You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
34 lines
712 B
34 lines
712 B
1 year ago
|
package blockedpeers
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/harmony-one/harmony/internal/utils/lrucache"
|
||
|
|
libp2p_peer "github.com/libp2p/go-libp2p/core/peer"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Manager struct {
|
||
|
|
internal *lrucache.Cache[libp2p_peer.ID, time.Time]
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewManager(size int) *Manager {
|
||
|
|
return &Manager{
|
||
|
|
internal: lrucache.NewCache[libp2p_peer.ID, time.Time](size),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *Manager) IsBanned(key libp2p_peer.ID, now time.Time) bool {
|
||
|
|
future, ok := m.internal.Get(key)
|
||
|
|
if ok {
|
||
|
|
return future.After(now) // future > now
|
||
|
|
}
|
||
|
|
return ok
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *Manager) Ban(key libp2p_peer.ID, future time.Time) {
|
||
|
|
m.internal.Set(key, future)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *Manager) Contains(key libp2p_peer.ID) bool {
|
||
|
|
return m.internal.Contains(key)
|
||
|
|
}
|