54 lines
1.3 KiB
Go
54 lines
1.3 KiB
Go
|
package discovery
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/ipfs/go-cid"
|
||
|
|
"github.com/ipfs/go-datastore"
|
||
|
|
"github.com/ipfs/go-datastore/namespace"
|
||
|
|
dshelp "github.com/ipfs/go-ipfs-ds-help"
|
||
|
|
cbor "github.com/ipfs/go-ipld-cbor"
|
||
|
|
logging "github.com/ipfs/go-log"
|
||
|
|
|
||
|
|
"github.com/filecoin-project/go-lotus/node/modules/dtypes"
|
||
|
|
)
|
||
|
|
|
||
|
|
var log = logging.Logger("ret-discovery")
|
||
|
|
|
||
|
|
type Local struct {
|
||
|
|
ds datastore.Datastore
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewLocal(ds dtypes.MetadataDS) *Local {
|
||
|
|
return &Local{ds: namespace.Wrap(ds, datastore.NewKey("/deals/local"))}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (l *Local) AddPeer(cid cid.Cid, peer RetrievalPeer) error {
|
||
|
|
// TODO: allow multiple peers here
|
||
|
|
// (implement an util for tracking map[thing][]otherThing, use in sectorBlockstore too)
|
||
|
|
|
||
|
|
log.Warn("Tracking multiple retrieval peers not implemented")
|
||
|
|
|
||
|
|
entry, err := cbor.DumpObject(peer)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
return l.ds.Put(dshelp.CidToDsKey(cid), entry)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (l *Local) GetPeers(data cid.Cid) ([]RetrievalPeer, error) {
|
||
|
|
entry, err := l.ds.Get(dshelp.CidToDsKey(data))
|
||
|
|
if err == datastore.ErrNotFound {
|
||
|
|
return []RetrievalPeer{}, nil
|
||
|
|
}
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
var peer RetrievalPeer
|
||
|
|
if err := cbor.DecodeInto(entry, &peer); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return []RetrievalPeer{peer}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
var _ PeerResolver = &Local{}
|