30 lines
702 B
Go
30 lines
702 B
Go
package storage
|
|
|
|
import (
|
|
"github.com/anacrolix/log"
|
|
|
|
"github.com/anacrolix/torrent/metainfo"
|
|
)
|
|
|
|
type PieceCompletionGetSetter interface {
|
|
Get(metainfo.PieceKey) (Completion, error)
|
|
Set(_ metainfo.PieceKey, complete bool) error
|
|
}
|
|
|
|
// Implementations track the completion of pieces. It must be concurrent-safe.
|
|
type PieceCompletion interface {
|
|
PieceCompletionGetSetter
|
|
Close() error
|
|
}
|
|
|
|
// use sqlite by default
|
|
// if sqlite connection fails, use a map
|
|
func pieceCompletionForDir(dir string) (ret PieceCompletion) {
|
|
ret, err := NewDefaultPieceCompletionForDir(dir)
|
|
if err != nil {
|
|
log.Printf("couldn't open piece completion db in %q: %s", dir, err)
|
|
ret = NewMapPieceCompletion()
|
|
}
|
|
return
|
|
}
|