Add holepunching stats and tests

[btrtrc.git] / storage / piece-resource.go

diff --git a/storage/piece-resource.go b/storage/piece-resource.go
index d56280bc5a2654806e6e6e4455d3bb584bda6ab9..ec3848df2111186259ce1cb76c9f6749aa8623f9 100644 (file)
--- a/storage/piece-resource.go
+++ b/storage/piece-resource.go
@@ -26,6 +26,7 @@ type ResourcePiecesOpts struct {
        // Sized puts require being able to stream from a statement executed on another connection.
        // Without them, we buffer the entire read and then put that.
        NoSizedPuts bool
+       Capacity    *int64
 }
 
 func NewResourcePieces(p PieceProvider) ClientImpl {
@@ -49,10 +50,11 @@ func (piecePerResourceTorrentImpl) Close() error {
 }
 
 func (s piecePerResource) OpenTorrent(info *metainfo.Info, infoHash metainfo.Hash) (TorrentImpl, error) {
-       return piecePerResourceTorrentImpl{
+       t := piecePerResourceTorrentImpl{
                s,
                make([]sync.RWMutex, info.NumPieces()),
-       }, nil
+       }
+       return TorrentImpl{Piece: t.Piece, Close: t.Close}, nil
 }
 
 func (s piecePerResourceTorrentImpl) Piece(p metainfo.Piece) PieceImpl {