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

use a Pool for address record objects.

Browse Source
This commit is contained in:
Raúl Kripalani 2018-11-15 18:41:25 +00:00
parent 390e8c3f44
commit ef8e8ff359
1 changed files with 25 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
pstoreds/addr_book.go
View File

@ -17,8 +17,8 @@ import (
pb "github.com/libp2p/go-libp2p-peerstore/pb"
pstoremem "github.com/libp2p/go-libp2p-peerstore/pstoremem"
lru "github.com/hashicorp/golang-lru"
ma "github.com/multiformats/go-multiaddr"
lru "github.com/raulk/golang-lru"
b32 "github.com/whyrusleeping/base32"
)
@ -34,6 +34,14 @@ var (
// Peer addresses are stored under the following db key pattern:
// /peers/addr/<b32 peer id no padding>/<hash of maddr>
addrBookBase = ds.NewKey("/peers/addrs")
arPool = &sync.Pool{
New: func() interface{} {
return &addrsRecord{
AddrBookRecord: &pb.AddrBookRecord{},
dirty: false,
}
},
}
)
// addrsRecord decorates the AddrBookRecord with locks and metadata.
@ -43,6 +51,11 @@ type addrsRecord struct {
dirty bool
}
func (r *addrsRecord) Reset() {
r.AddrBookRecord.Reset()
r.dirty = false
}
// FlushInTxn writes the record to the datastore by calling ds.Put, unless the record is
// marked for deletion, in which case the deletion is executed via ds.Delete.
func (r *addrsRecord) FlushInTxn(txn ds.Txn) (err error) {
@ -155,7 +168,11 @@ var _ pstore.AddrBook = (*dsAddrBook)(nil)
func NewAddrBook(ctx context.Context, store ds.TxnDatastore, opts Options) (ab *dsAddrBook, err error) {
var cache cache = new(noopCache)
if opts.CacheSize > 0 {
if cache, err = lru.NewARC(int(opts.CacheSize)); err != nil {
evictCallback := func(key interface{}, value interface{}) {
value.(*addrsRecord).Reset()
arPool.Put(value)
}
if cache, err = lru.NewARCWithEvict(int(opts.CacheSize), evictCallback); err != nil {
return nil, err
}
}
@ -221,7 +238,7 @@ func (ab *dsAddrBook) loadRecord(id peer.ID, cache bool, update bool) (pr *addrs
}
if err == nil {
pr = &addrsRecord{AddrBookRecord: &pb.AddrBookRecord{}}
pr = arPool.Get().(*addrsRecord)
if err = pr.Unmarshal(data); err != nil {
return nil, err
}
@ -229,7 +246,8 @@ func (ab *dsAddrBook) loadRecord(id peer.ID, cache bool, update bool) (pr *addrs
ab.asyncFlush(pr)
}
} else {
pr = &addrsRecord{AddrBookRecord: &pb.AddrBookRecord{Id: &pb.ProtoPeerID{ID: id}}}
pr = arPool.Get().(*addrsRecord)
pr.Id = &pb.ProtoPeerID{ID: id}
}
if cache {
@ -275,7 +293,9 @@ var purgeQuery = query.Query{Prefix: addrBookBase.String()}
// purgeCycle runs a GC cycle
func (ab *dsAddrBook) purgeCycle() {
var id peer.ID
record := &addrsRecord{AddrBookRecord: &pb.AddrBookRecord{}}
record := arPool.Get().(*addrsRecord)
defer arPool.Put(record)
txn, err := ab.ds.NewTransaction(false)
if err != nil {
log.Warningf("failed while purging entries: %v\n", err)