35 lines
606 B
Go
35 lines
606 B
Go
|
package storage
|
||
|
|
|
||
|
|
import (
|
||
|
|
"sync"
|
||
|
|
|
||
|
|
"github.com/anacrolix/torrent/metainfo"
|
||
|
|
)
|
||
|
|
|
||
|
|
type mapPieceCompletion struct {
|
||
|
|
// TODO: Generics
|
||
|
|
m sync.Map
|
||
|
|
}
|
||
|
|
|
||
|
|
var _ PieceCompletion = (*mapPieceCompletion)(nil)
|
||
|
|
|
||
|
|
func NewMapPieceCompletion() PieceCompletion {
|
||
|
|
return &mapPieceCompletion{}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (*mapPieceCompletion) Close() error { return nil }
|
||
|
|
|
||
|
|
func (me *mapPieceCompletion) Get(pk metainfo.PieceKey) (c Completion, err error) {
|
||
|
|
v, ok := me.m.Load(pk)
|
||
|
|
if ok {
|
||
|
|
c.Complete = v.(bool)
|
||
|
|
}
|
||
|
|
c.Ok = ok
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func (me *mapPieceCompletion) Set(pk metainfo.PieceKey, b bool) error {
|
||
|
|
me.m.Store(pk, b)
|
||
|
|
return nil
|
||
|
|
}
|