package storage import ( "encoding/binary" "path/filepath" "time" "github.com/anacrolix/missinggo/expect" "github.com/boltdb/bolt" "github.com/anacrolix/torrent/metainfo" ) const ( // Chosen to match the usual chunk size in a torrent client. This way, // most chunk writes are to exactly one full item in bolt DB. chunkSize = 1 << 14 ) type boltDBClient struct { db *bolt.DB } type boltDBTorrent struct { cl *boltDBClient ih metainfo.Hash } func NewBoltDB(filePath string) ClientImpl { db, err := bolt.Open(filepath.Join(filePath, "bolt.db"), 0600, &bolt.Options{ Timeout: time.Second, }) expect.Nil(err) db.NoSync = true return &boltDBClient{db} } func (me *boltDBClient) Close() error { return me.db.Close() } func (me *boltDBClient) OpenTorrent(info *metainfo.Info, infoHash metainfo.Hash) (TorrentImpl, error) { return &boltDBTorrent{me, infoHash}, nil } func (me *boltDBTorrent) Piece(p metainfo.Piece) PieceImpl { ret := &boltDBPiece{ p: p, db: me.cl.db, ih: me.ih, } copy(ret.key[:], me.ih[:]) binary.BigEndian.PutUint32(ret.key[20:], uint32(p.Index())) return ret } func (boltDBTorrent) Close() error { return nil }