95 lines
2.1 KiB
Go
95 lines
2.1 KiB
Go
package torrent
|
|
|
|
import (
|
|
"io"
|
|
"sync"
|
|
|
|
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 int64
|
|
BytesWrittenData int64
|
|
|
|
BytesRead int64
|
|
BytesReadData int64
|
|
BytesReadUsefulData int64
|
|
|
|
ChunksWritten int64
|
|
|
|
ChunksRead int64
|
|
ChunksReadUseful int64
|
|
ChunksReadUnwanted int64
|
|
|
|
// Number of pieces data was written to, that subsequently passed verification.
|
|
PiecesDirtiedGood int64
|
|
// 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 int64
|
|
}
|
|
|
|
func (cs *ConnStats) wroteMsg(msg *pp.Message) {
|
|
// TODO: Track messages and not just chunks.
|
|
switch msg.Type {
|
|
case pp.Piece:
|
|
cs.ChunksWritten++
|
|
cs.BytesWrittenData += int64(len(msg.Piece))
|
|
}
|
|
}
|
|
|
|
func (cs *ConnStats) readMsg(msg *pp.Message) {
|
|
switch msg.Type {
|
|
case pp.Piece:
|
|
cs.ChunksRead++
|
|
cs.BytesReadData += int64(len(msg.Piece))
|
|
}
|
|
}
|
|
|
|
func (cs *ConnStats) incrementPiecesDirtiedGood() {
|
|
cs.PiecesDirtiedGood++
|
|
}
|
|
|
|
func (cs *ConnStats) incrementPiecesDirtiedBad() {
|
|
cs.PiecesDirtiedBad++
|
|
}
|
|
|
|
func add(n int64, f func(*ConnStats) *int64) func(*ConnStats) {
|
|
return func(cs *ConnStats) {
|
|
p := f(cs)
|
|
*p += n
|
|
}
|
|
}
|
|
|
|
type connStatsReadWriter struct {
|
|
rw io.ReadWriter
|
|
l sync.Locker
|
|
c *connection
|
|
}
|
|
|
|
func (me connStatsReadWriter) Write(b []byte) (n int, err error) {
|
|
n, err = me.rw.Write(b)
|
|
go func() {
|
|
me.l.Lock()
|
|
me.c.wroteBytes(int64(n))
|
|
me.l.Unlock()
|
|
}()
|
|
return
|
|
}
|
|
|
|
func (me connStatsReadWriter) Read(b []byte) (n int, err error) {
|
|
n, err = me.rw.Read(b)
|
|
go func() {
|
|
me.l.Lock()
|
|
me.c.readBytes(int64(n))
|
|
me.l.Unlock()
|
|
}()
|
|
return
|
|
}
|