node, cmd/devp2p/internal/v5test: use slices.Contains

This commit is contained in:
Aaron Chen 2024-04-05 14:38:34 +08:00 committed by Felix Lange
parent cc348a601e
commit 4e28c73924
4 changed files with 6 additions and 22 deletions

View file

@ -19,6 +19,7 @@ package v5test
import ( import (
"bytes" "bytes"
"net" "net"
"slices"
"sync" "sync"
"time" "time"
@ -266,7 +267,7 @@ func (s *Suite) TestFindnodeResults(t *utesting.T) {
n := bn.conn.localNode.Node() n := bn.conn.localNode.Node()
expect[n.ID()] = n expect[n.ID()] = n
d := uint(enode.LogDist(n.ID(), s.Dest.ID())) d := uint(enode.LogDist(n.ID(), s.Dest.ID()))
if !containsUint(dists, d) { if !slices.Contains(dists, d) {
dists = append(dists, d) dists = append(dists, d)
} }
} }

View file

@ -252,12 +252,3 @@ func checkRecords(records []*enr.Record) ([]*enode.Node, error) {
} }
return nodes, nil return nodes, nil
} }
func containsUint(ints []uint, x uint) bool {
for i := range ints {
if ints[i] == x {
return true
}
}
return false
}

View file

@ -25,6 +25,7 @@ import (
"os" "os"
"path/filepath" "path/filepath"
"reflect" "reflect"
"slices"
"strings" "strings"
"sync" "sync"
@ -278,16 +279,6 @@ func (n *Node) openEndpoints() error {
return err return err
} }
// containsLifecycle checks if 'lfs' contains 'l'.
func containsLifecycle(lfs []Lifecycle, l Lifecycle) bool {
for _, obj := range lfs {
if obj == l {
return true
}
}
return false
}
// stopServices terminates running services, RPC and p2p networking. // stopServices terminates running services, RPC and p2p networking.
// It is the inverse of Start. // It is the inverse of Start.
func (n *Node) stopServices(running []Lifecycle) error { func (n *Node) stopServices(running []Lifecycle) error {
@ -571,7 +562,7 @@ func (n *Node) RegisterLifecycle(lifecycle Lifecycle) {
if n.state != initializingState { if n.state != initializingState {
panic("can't register lifecycle on running/stopped node") panic("can't register lifecycle on running/stopped node")
} }
if containsLifecycle(n.lifecycles, lifecycle) { if slices.Contains(n.lifecycles, lifecycle) {
panic(fmt.Sprintf("attempt to register lifecycle %T more than once", lifecycle)) panic(fmt.Sprintf("attempt to register lifecycle %T more than once", lifecycle))
} }
n.lifecycles = append(n.lifecycles, lifecycle) n.lifecycles = append(n.lifecycles, lifecycle)

View file

@ -23,6 +23,7 @@ import (
"net" "net"
"net/http" "net/http"
"reflect" "reflect"
"slices"
"strings" "strings"
"testing" "testing"
@ -116,7 +117,7 @@ func TestLifecycleRegistry_Successful(t *testing.T) {
noop := NewNoop() noop := NewNoop()
stack.RegisterLifecycle(noop) stack.RegisterLifecycle(noop)
if !containsLifecycle(stack.lifecycles, noop) { if !slices.Contains(stack.lifecycles, noop) {
t.Fatalf("lifecycle was not properly registered on the node, %v", err) t.Fatalf("lifecycle was not properly registered on the node, %v", err)
} }
} }