Commit aa270de0 authored by Steven Allen's avatar Steven Allen

chore: slightly simplify backoff logic

parent 6711d98c
...@@ -131,15 +131,9 @@ func (db *DialBackoff) background(ctx context.Context) { ...@@ -131,15 +131,9 @@ func (db *DialBackoff) background(ctx context.Context) {
func (db *DialBackoff) Backoff(p peer.ID, addr ma.Multiaddr) (backoff bool) { func (db *DialBackoff) Backoff(p peer.ID, addr ma.Multiaddr) (backoff bool) {
db.lock.Lock() db.lock.Lock()
defer db.lock.Unlock() defer db.lock.Unlock()
bp, found := db.entries[p]
if found {
ap, found := bp[string(addr.Bytes())]
if found && time.Now().Before(ap.until) {
return true
}
}
return false ap, found := db.entries[p][string(addr.Bytes())]
return found && time.Now().Before(ap.until)
} }
// BackoffBase is the base amount of time to backoff (default: 5s). // BackoffBase is the base amount of time to backoff (default: 5s).
...@@ -167,12 +161,8 @@ func (db *DialBackoff) AddBackoff(p peer.ID, addr ma.Multiaddr) { ...@@ -167,12 +161,8 @@ func (db *DialBackoff) AddBackoff(p peer.ID, addr ma.Multiaddr) {
defer db.lock.Unlock() defer db.lock.Unlock()
bp, ok := db.entries[p] bp, ok := db.entries[p]
if !ok { if !ok {
db.entries[p] = make(map[string]*backoffAddr) bp = make(map[string]*backoffAddr, 1)
db.entries[p][saddr] = &backoffAddr{ db.entries[p] = bp
tries: 1,
until: time.Now().Add(BackoffBase),
}
return
} }
ba, ok := bp[saddr] ba, ok := bp[saddr]
if !ok { if !ok {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment