114 lines
2.6 KiB
Go
114 lines
2.6 KiB
Go
package torrent
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"reflect"
|
|
"sync/atomic"
|
|
|
|
pp "github.com/anacrolix/torrent/peer_protocol"
|
|
)
|
|
|
|
// Various connection-level metrics. At the Torrent level these are
|
|
// aggregates. Chunks are messages with data payloads. Data is actual torrent
|
|
// content without any overhead. Useful is something we needed locally.
|
|
// Unwanted is something we didn't ask for (but may still be useful). Written
|
|
// is things sent to the peer, and Read is stuff received from them.
|
|
type ConnStats struct {
|
|
// Total bytes on the wire. Includes handshakes and encryption.
|
|
BytesWritten Count
|
|
BytesWrittenData Count
|
|
|
|
BytesRead Count
|
|
BytesReadData Count
|
|
BytesReadUsefulData Count
|
|
|
|
ChunksWritten Count
|
|
|
|
ChunksRead Count
|
|
ChunksReadUseful Count
|
|
ChunksReadUnwanted Count
|
|
|
|
// Number of pieces data was written to, that subsequently passed verification.
|
|
PiecesDirtiedGood Count
|
|
// Number of pieces data was written to, that subsequently failed
|
|
// verification. Note that a connection may not have been the sole dirtier
|
|
// of a piece.
|
|
PiecesDirtiedBad Count
|
|
}
|
|
|
|
func (me *ConnStats) Copy() (ret ConnStats) {
|
|
for i := 0; i < reflect.TypeOf(ConnStats{}).NumField(); i++ {
|
|
n := reflect.ValueOf(me).Elem().Field(i).Addr().Interface().(*Count).Int64()
|
|
reflect.ValueOf(&ret).Elem().Field(i).Addr().Interface().(*Count).Add(n)
|
|
}
|
|
return
|
|
}
|
|
|
|
type Count struct {
|
|
n int64
|
|
}
|
|
|
|
var _ fmt.Stringer = (*Count)(nil)
|
|
|
|
func (me *Count) Add(n int64) {
|
|
atomic.AddInt64(&me.n, n)
|
|
}
|
|
|
|
func (me *Count) Int64() int64 {
|
|
return atomic.LoadInt64(&me.n)
|
|
}
|
|
|
|
func (me *Count) String() string {
|
|
return fmt.Sprintf("%v", me.Int64())
|
|
}
|
|
|
|
func (cs *ConnStats) wroteMsg(msg *pp.Message) {
|
|
// TODO: Track messages and not just chunks.
|
|
switch msg.Type {
|
|
case pp.Piece:
|
|
cs.ChunksWritten.Add(1)
|
|
cs.BytesWrittenData.Add(int64(len(msg.Piece)))
|
|
}
|
|
}
|
|
|
|
func (cs *ConnStats) readMsg(msg *pp.Message) {
|
|
switch msg.Type {
|
|
case pp.Piece:
|
|
cs.ChunksRead.Add(1)
|
|
cs.BytesReadData.Add(int64(len(msg.Piece)))
|
|
}
|
|
}
|
|
|
|
func (cs *ConnStats) incrementPiecesDirtiedGood() {
|
|
cs.PiecesDirtiedGood.Add(1)
|
|
}
|
|
|
|
func (cs *ConnStats) incrementPiecesDirtiedBad() {
|
|
cs.PiecesDirtiedBad.Add(1)
|
|
}
|
|
|
|
func add(n int64, f func(*ConnStats) *Count) func(*ConnStats) {
|
|
return func(cs *ConnStats) {
|
|
p := f(cs)
|
|
p.Add(n)
|
|
}
|
|
}
|
|
|
|
type connStatsReadWriter struct {
|
|
rw io.ReadWriter
|
|
c *connection
|
|
}
|
|
|
|
func (me connStatsReadWriter) Write(b []byte) (n int, err error) {
|
|
n, err = me.rw.Write(b)
|
|
me.c.wroteBytes(int64(n))
|
|
return
|
|
}
|
|
|
|
func (me connStatsReadWriter) Read(b []byte) (n int, err error) {
|
|
n, err = me.rw.Read(b)
|
|
me.c.readBytes(int64(n))
|
|
return
|
|
}
|