From 0f6aeb23d7a73b1692725332e8e46a83f83cefc0 Mon Sep 17 00:00:00 2001 From: vyzo Date: Wed, 29 Jul 2020 20:09:06 +0300 Subject: [PATCH] disallow duplicate score inspectors --- score.go | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/score.go b/score.go index deb4945..fda88f5 100644 --- a/score.go +++ b/score.go @@ -152,6 +152,10 @@ func WithPeerScoreInspect(inspect interface{}, period time.Duration) Option { return fmt.Errorf("peer scoring is not enabled") } + if gs.score.inspect != nil || gs.score.inspectEx != nil { + return fmt.Errorf("duplicate peer score inspector") + } + switch i := inspect.(type) { case PeerScoreInspectFn: gs.score.inspect = i @@ -352,7 +356,8 @@ func (ps *peerScore) background(ctx context.Context) { func (ps *peerScore) inspectScores() { if ps.inspect != nil { ps.inspectScoresSimple() - } else if ps.inspectEx != nil { + } + if ps.inspectEx != nil { ps.inspectScoresExtended() } }