sovonegr/TorrServerJellyfin
1
0
Fork 0
mirror of https://github.com/Ernous/TorrServerJellyfin.git synced 2025-12-19 21:46:11 +05:00
Code Issues Packages Projects Releases Wiki Activity

add torrent storage capacity func

Browse Source
This commit is contained in:
nikk gitanes
2023-08-27 02:38:43 +03:00
parent 3463464902
commit e480ecb093
1 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
server/torr/storage/torrstor/storage.go
View File

@@ -6,7 +6,7 @@ import (
"server/torr/storage" "server/torr/storage"
"github.com/anacrolix/torrent/metainfo" "github.com/anacrolix/torrent/metainfo"
storage2 "github.com/anacrolix/torrent/storage" ts "github.com/anacrolix/torrent/storage"
) )
type Storage struct { type Storage struct {
@@ -24,16 +24,20 @@ func NewStorage(capacity int64) *Storage {
return stor 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() s.mu.Lock()
defer s.mu.Unlock() defer s.mu.Unlock()
ch := NewCache(s.capacity, s) ch := NewCache(s.capacity, s)
ch.Init(info, infoHash) ch.Init(info, infoHash)
s.caches[infoHash] = ch s.caches[infoHash] = ch
// return ch, nil // return ch, nil
return storage2.TorrentImpl{ return ts.TorrentImpl{
Piece: ch.Piece, Piece: ch.Piece,
Close: ch.Close, Close: ch.Close,
Capacity: &capFunc,
}, nil }, nil
} }