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

Update ttlmanager to use ds.Batching

Browse Source
This commit is contained in:
Cole Brown 2018-06-26 17:22:04 -04:00
parent 78cd2c5abd
commit 9eeea343d5
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
addr_manager_ds.go
View File

@ -231,7 +231,7 @@ type ttlmanager struct {
ctx context.Context
cancel context.CancelFunc
ticker *time.Ticker
ds ds.Datastore
ds ds.Batching
}
func newTTLManager(parent context.Context, d ds.Datastore, tick time.Duration) *ttlmanager {
@ -265,14 +265,23 @@ func (mgr *ttlmanager) tick() {
defer mgr.RUnlock()
now := time.Now()
batch, err := mgr.ds.Batch()
if err != nil {
log.Error(err)
return
}
for key, entry := range mgr.entries {
if entry.ExpiresAt.Before(now) {
if err := mgr.ds.Delete(key); err != nil {
if err := batch.Delete(key); err != nil {
log.Error(err)
}
delete(mgr.entries, key)
}
}
err := batch.Commit()
if err != nil {
log.Error(err)
}
}
func (mgr *ttlmanager) setTTLs(keys []ds.Key, ttl time.Duration) {