tstor/src/host/storage/storage.go

58 lines
1.8 KiB
Go
Raw Normal View History

2023-12-31 22:54:55 +00:00
package storage
import (
"fmt"
"os"
"path/filepath"
"git.kmsign.ru/royalcat/tstor/src/config"
"github.com/anacrolix/torrent/storage"
)
2024-01-01 18:17:32 +00:00
func SetupStorage(cfg config.TorrentClient) (storage.ClientImplCloser, storage.PieceCompletion, error) {
2023-12-31 22:54:55 +00:00
pcp := filepath.Join(cfg.DataFolder, "piece-completion")
if err := os.MkdirAll(pcp, 0744); err != nil {
return nil, nil, fmt.Errorf("error creating piece completion folder: %w", err)
}
pc, err := storage.NewBoltPieceCompletion(pcp)
if err != nil {
return nil, nil, fmt.Errorf("error creating servers piece completion: %w", err)
}
// pc, err := NewBadgerPieceCompletion(pcp)
// if err != nil {
// return nil, nil, fmt.Errorf("error creating servers piece completion: %w", err)
// }
// TODO implement cache/storage switching
// cacheDir := filepath.Join(tcfg.DataFolder, "cache")
// if err := os.MkdirAll(cacheDir, 0744); err != nil {
// return nil, nil, fmt.Errorf("error creating piece completion folder: %w", err)
// }
// fc, err := filecache.NewCache(cacheDir)
// if err != nil {
// return nil, nil, fmt.Errorf("error creating cache: %w", err)
// }
// log.Info().Msg(fmt.Sprintf("setting cache size to %d MB", 1024))
// fc.SetCapacity(1024 * 1024 * 1024)
// rp := storage.NewResourcePieces(fc.AsResourceProvider())
// st := &stc{rp}
2024-01-01 18:17:32 +00:00
// filesDir := filepath.Join(cfg.DataFolder, "files")
// if err := os.MkdirAll(filesDir, 0744); err != nil {
// return nil, nil, fmt.Errorf("error creating piece completion folder: %w", err)
// }
//st := NewFileStorage(filesDir, pc)
piecesDir := filepath.Join(cfg.DataFolder, "pieces")
if err := os.MkdirAll(piecesDir, 0744); err != nil {
2023-12-31 22:54:55 +00:00
return nil, nil, fmt.Errorf("error creating piece completion folder: %w", err)
}
2024-01-01 18:17:32 +00:00
st := storage.NewMMapWithCompletion(piecesDir, pc)
2023-12-31 22:54:55 +00:00
return st, pc, nil
}