status-im/go-libp2p-pubsub
2
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

added mutex to timecache

Browse Source
This commit is contained in:
Aarsh Shah 2020-02-05 14:13:21 +05:30
parent 3f48f468ac
commit 677ca1025b
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
blacklist.go
View File

@ -1,6 +1,7 @@
package pubsub package pubsub
import ( import (
"sync"
"time" "time"
"github.com/libp2p/go-libp2p-core/peer" "github.com/libp2p/go-libp2p-core/peer"
@ -32,19 +33,25 @@ func (b MapBlacklist) Contains(p peer.ID) bool {
// TimeCachedBlacklist is a blacklist implementation using a time cache // TimeCachedBlacklist is a blacklist implementation using a time cache
type TimeCachedBlacklist struct { type TimeCachedBlacklist struct {
sync.RWMutex
tc *timecache.TimeCache tc *timecache.TimeCache
} }
// NewTimeCachedBlacklist creates a new TimeCachedBlacklist with the given expiry duration // NewTimeCachedBlacklist creates a new TimeCachedBlacklist with the given expiry duration
func NewTimeCachedBlacklist(expiry time.Duration) (Blacklist, error) { func NewTimeCachedBlacklist(expiry time.Duration) (Blacklist, error) {
b := &TimeCachedBlacklist{timecache.NewTimeCache(expiry)} b := &TimeCachedBlacklist{tc: timecache.NewTimeCache(expiry)}
return b, nil return b, nil
} }
func (b TimeCachedBlacklist) Add(p peer.ID) { func (b *TimeCachedBlacklist) Add(p peer.ID) {
b.Lock()
b.tc.Add(p.String()) b.tc.Add(p.String())
b.Unlock()
} }
func (b TimeCachedBlacklist) Contains(p peer.ID) bool { func (b *TimeCachedBlacklist) Contains(p peer.ID) bool {
b.RLock()
defer b.RUnlock()
return b.tc.Has(p.String()) return b.tc.Has(p.String())
} }