libp2p/go-libp2p-peerstore
1
0
Fork 0
mirror of https://github.com/libp2p/go-libp2p-peerstore.git synced 2024-12-27 23:40:16 +08:00
Code Issues Projects Releases Wiki Activity

adopt new ds.Write interface.

Browse Source
This commit is contained in:
Raúl Kripalani 2019-02-05 16:53:23 +00:00
parent d88042c422
commit 20a0dc649d
1 changed files with 3 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
pstoreds/addr_book.go
View File

@ -36,11 +36,6 @@ var (
addrBookBase = ds.NewKey("/peers/addrs")
)
type dsWriter interface {
Put(key ds.Key, value []byte) error
Delete(key ds.Key) error
}
// addrsRecord decorates the AddrBookRecord with locks and metadata.
type addrsRecord struct {
sync.RWMutex
@ -50,17 +45,17 @@ type addrsRecord struct {
// flush writes the record to the datastore by calling ds.Put, unless the record is
// marked for deletion, in which case we call ds.Delete.
func (r *addrsRecord) flush(ds dsWriter) (err error) {
func (r *addrsRecord) flush(write ds.Write) (err error) {
key := addrBookBase.ChildString(b32.RawStdEncoding.EncodeToString([]byte(r.Id.ID)))
if len(r.Addrs) == 0 {
return ds.Delete(key)
return write.Delete(key)
}
data, err := r.Marshal()
if err != nil {
return err
}
if err = ds.Put(key, data); err != nil {
if err = write.Put(key, data); err != nil {
return err
}
// write succeeded; record is no longer dirty.