FedP2P/conn_stats.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
}