diff --git a/server/torr/storage/torrstor/cache.go b/server/torr/storage/torrstor/cache.go index 3074b31..efe861d 100644 --- a/server/torr/storage/torrstor/cache.go +++ b/server/torr/storage/torrstor/cache.go @@ -83,7 +83,7 @@ func (c *Cache) Piece(m metainfo.Piece) storage.PieceImpl { if val, ok := c.pieces[m.Index()]; ok { return val } - return nil + return &PieceFake{} } func (c *Cache) Close() error { diff --git a/server/torr/storage/torrstor/piecefake.go b/server/torr/storage/torrstor/piecefake.go new file mode 100644 index 0000000..6434bb2 --- /dev/null +++ b/server/torr/storage/torrstor/piecefake.go @@ -0,0 +1,33 @@ +package torrstor + +import ( + "errors" + "github.com/anacrolix/torrent/storage" +) + +type PieceFake struct{} + +func (PieceFake) ReadAt(p []byte, off int64) (n int, err error) { + err = errors.New("fake") + return +} + +func (PieceFake) WriteAt(p []byte, off int64) (n int, err error) { + err = errors.New("fake") + return +} + +func (PieceFake) MarkComplete() error { + return errors.New("fake") +} + +func (PieceFake) MarkNotComplete() error { + return errors.New("fake") +} + +func (PieceFake) Completion() storage.Completion { + return storage.Completion{ + Complete: false, + Ok: true, + } +} diff --git a/server/torr/storage/torrstor/storage.go b/server/torr/storage/torrstor/storage.go index 9fcf069..1c92d0b 100644 --- a/server/torr/storage/torrstor/storage.go +++ b/server/torr/storage/torrstor/storage.go @@ -6,7 +6,7 @@ import ( "server/torr/storage" "github.com/anacrolix/torrent/metainfo" - storage2 "github.com/anacrolix/torrent/storage" + ts "github.com/anacrolix/torrent/storage" ) type Storage struct { @@ -24,16 +24,20 @@ func NewStorage(capacity int64) *Storage { return stor } -func (s *Storage) OpenTorrent(info *metainfo.Info, infoHash metainfo.Hash) (storage2.TorrentImpl, error) { +func (s *Storage) OpenTorrent(info *metainfo.Info, infoHash metainfo.Hash) (ts.TorrentImpl, error) { + capFunc := func() (int64, bool) { + return s.capacity, true + } s.mu.Lock() defer s.mu.Unlock() ch := NewCache(s.capacity, s) ch.Init(info, infoHash) s.caches[infoHash] = ch // return ch, nil - return storage2.TorrentImpl{ - Piece: ch.Piece, - Close: ch.Close, + return ts.TorrentImpl{ + Piece: ch.Piece, + Close: ch.Close, + Capacity: &capFunc, }, nil }