1 // modernc.org/sqlite depends on modernc.org/libc which doesn't work for JS (and probably wasm but I
2 // think JS is the stronger signal).
3 //go:build cgo && !nosqlite
4 // +build cgo,!nosqlite
14 "crawshaw.io/sqlite/sqlitex"
15 "github.com/anacrolix/torrent/metainfo"
18 // sqlite is always the default when available.
19 func NewDefaultPieceCompletionForDir(dir string) (PieceCompletion, error) {
20 return NewSqlitePieceCompletion(dir)
23 type sqlitePieceCompletion struct {
29 var _ PieceCompletion = (*sqlitePieceCompletion)(nil)
31 func NewSqlitePieceCompletion(dir string) (ret *sqlitePieceCompletion, err error) {
32 p := filepath.Join(dir, ".torrent.db")
33 db, err := sqlite.OpenConn(p, 0)
37 err = sqlitex.ExecScript(db, `create table if not exists piece_completion(infohash, "index", complete, unique(infohash, "index"))`)
42 ret = &sqlitePieceCompletion{db: db}
46 func (me *sqlitePieceCompletion) Get(pk metainfo.PieceKey) (c Completion, err error) {
50 me.db, `select complete from piece_completion where infohash=? and "index"=?`,
51 func(stmt *sqlite.Stmt) error {
52 c.Complete = stmt.ColumnInt(0) != 0
56 pk.InfoHash.HexString(), pk.Index)
60 func (me *sqlitePieceCompletion) Set(pk metainfo.PieceKey, b bool) error {
64 return errors.New("closed")
68 `insert or replace into piece_completion(infohash, "index", complete) values(?, ?, ?)`,
70 pk.InfoHash.HexString(), pk.Index, b)
73 func (me *sqlitePieceCompletion) Close() (err error) {