From e480ecb09388042fe13485678ba8b3d8d6ddb383 Mon Sep 17 00:00:00 2001 From: nikk gitanes Date: Sun, 27 Aug 2023 02:38:43 +0300 Subject: [PATCH] add torrent storage capacity func --- server/torr/storage/torrstor/storage.go | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) 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 }