mirror of
https://github.com/libp2p/go-libp2p-peerstore.git
synced 2024-12-26 23:30:32 +08:00
82 lines
2.3 KiB
Go
82 lines
2.3 KiB
Go
package pstoreds
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/gob"
|
|
|
|
pool "github.com/libp2p/go-buffer-pool"
|
|
"github.com/libp2p/go-libp2p-core/peer"
|
|
pstore "github.com/libp2p/go-libp2p-core/peerstore"
|
|
|
|
ds "github.com/ipfs/go-datastore"
|
|
"github.com/ipfs/go-datastore/query"
|
|
"github.com/multiformats/go-base32"
|
|
)
|
|
|
|
// Metadata is stored under the following db key pattern:
|
|
// /peers/metadata/<b32 peer id no padding>/<key>
|
|
var pmBase = ds.NewKey("/peers/metadata")
|
|
|
|
type dsPeerMetadata struct {
|
|
ds ds.Datastore
|
|
}
|
|
|
|
var _ pstore.PeerMetadata = (*dsPeerMetadata)(nil)
|
|
|
|
func init() {
|
|
// Gob registers basic types by default.
|
|
//
|
|
// Register complex types used by the peerstore itself.
|
|
gob.Register(make(map[string]struct{}))
|
|
}
|
|
|
|
// NewPeerMetadata creates a metadata store backed by a persistent db. It uses gob for serialisation.
|
|
//
|
|
// See `init()` to learn which types are registered by default. Modules wishing to store
|
|
// values of other types will need to `gob.Register()` them explicitly, or else callers
|
|
// will receive runtime errors.
|
|
func NewPeerMetadata(_ context.Context, store ds.Datastore, _ Options) (*dsPeerMetadata, error) {
|
|
return &dsPeerMetadata{store}, nil
|
|
}
|
|
|
|
func (pm *dsPeerMetadata) Get(p peer.ID, key string) (interface{}, error) {
|
|
k := pmBase.ChildString(base32.RawStdEncoding.EncodeToString([]byte(p))).ChildString(key)
|
|
value, err := pm.ds.Get(context.TODO(), k)
|
|
if err != nil {
|
|
if err == ds.ErrNotFound {
|
|
err = pstore.ErrNotFound
|
|
}
|
|
return nil, err
|
|
}
|
|
|
|
var res interface{}
|
|
if err := gob.NewDecoder(bytes.NewReader(value)).Decode(&res); err != nil {
|
|
return nil, err
|
|
}
|
|
return res, nil
|
|
}
|
|
|
|
func (pm *dsPeerMetadata) Put(p peer.ID, key string, val interface{}) error {
|
|
k := pmBase.ChildString(base32.RawStdEncoding.EncodeToString([]byte(p))).ChildString(key)
|
|
var buf pool.Buffer
|
|
if err := gob.NewEncoder(&buf).Encode(&val); err != nil {
|
|
return err
|
|
}
|
|
return pm.ds.Put(context.TODO(), k, buf.Bytes())
|
|
}
|
|
|
|
func (pm *dsPeerMetadata) RemovePeer(p peer.ID) {
|
|
result, err := pm.ds.Query(context.TODO(), query.Query{
|
|
Prefix: pmBase.ChildString(base32.RawStdEncoding.EncodeToString([]byte(p))).String(),
|
|
KeysOnly: true,
|
|
})
|
|
if err != nil {
|
|
log.Warnw("querying datastore when removing peer failed", "peer", p, "error", err)
|
|
return
|
|
}
|
|
for entry := range result.Next() {
|
|
pm.ds.Delete(context.TODO(), ds.NewKey(entry.Key))
|
|
}
|
|
}
|