FedP2P/storage/piece-completion.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
}