Merge remote-tracking branch 'upstream/master'

This commit is contained in:
nikk gitanes
2021-06-17 12:20:52 +03:00
3 changed files with 51 additions and 16 deletions

View File

@@ -2,6 +2,10 @@ package settings
import ( import (
"encoding/json" "encoding/json"
"io"
"io/fs"
"path/filepath"
"strings"
"server/log" "server/log"
) )
@@ -62,6 +66,23 @@ func SetBTSets(sets *BTSets) {
if sets.TorrentsSavePath == "" { if sets.TorrentsSavePath == "" {
sets.UseDisk = false sets.UseDisk = false
} else if sets.UseDisk {
BTsets = sets
go filepath.WalkDir(sets.TorrentsSavePath, func(path string, d fs.DirEntry, err error) error {
if err != nil {
return err
}
if d.IsDir() && strings.ToLower(d.Name()) == ".tsc" {
BTsets.TorrentsSavePath = path
log.TLogln("Find directory \"" + BTsets.TorrentsSavePath + "\", use as cache dir")
return io.EOF
}
if d.IsDir() && strings.HasPrefix(d.Name(), ".") {
return filepath.SkipDir
}
return nil
})
} }
BTsets = sets BTsets = sets

View File

@@ -107,21 +107,37 @@ func GetTorrent(hashHex string) *Torrent {
func SetTorrent(hashHex, title, poster, data string) *Torrent { func SetTorrent(hashHex, title, poster, data string) *Torrent {
hash := metainfo.NewHashFromHex(hashHex) hash := metainfo.NewHashFromHex(hashHex)
tor := bts.GetTorrent(hash) torr := bts.GetTorrent(hash)
if tor != nil { torrDb := GetTorrentDB(hash)
tor.Title = title
tor.Poster = poster if title == "" && torr == nil && torrDb != nil {
tor.Data = data torr = GetTorrent(hashHex)
torr.GotInfo()
if torr.Torrent != nil && torr.Torrent.Info() != nil {
title = torr.Info().Name
}
} }
tor = GetTorrentDB(hash) if torr != nil {
if tor != nil { if title == "" && torr.Torrent != nil && torr.Torrent.Info() != nil {
tor.Title = title title = torr.Info().Name
tor.Poster = poster }
tor.Data = data torr.Title = title
AddTorrentDB(tor) torr.Poster = poster
torr.Data = data
}
if torrDb != nil {
torrDb.Title = title
torrDb.Poster = poster
torrDb.Data = data
AddTorrentDB(torrDb)
}
if torr != nil {
return torr
} else {
return torrDb
} }
return tor
} }
func RemTorrent(hashHex string) { func RemTorrent(hashHex string) {

View File

@@ -194,12 +194,10 @@ func (c *Cache) cleanPieces() {
c.removePiece(p) c.removePiece(p)
rems-- rems--
if rems <= 0 { if rems <= 0 {
break utils.FreeOSMemGC()
return
} }
} }
if rems <= 0 {
utils.FreeOSMemGC()
}
} }
} }