aboutsummaryrefslogtreecommitdiffstats
path: root/swarm/network/simulation/service.go
diff options
context:
space:
mode:
Diffstat (limited to 'swarm/network/simulation/service.go')
-rw-r--r--swarm/network/simulation/service.go8
1 files changed, 4 insertions, 4 deletions
diff --git a/swarm/network/simulation/service.go b/swarm/network/simulation/service.go
index 02e7ad0cc..819602e9e 100644
--- a/swarm/network/simulation/service.go
+++ b/swarm/network/simulation/service.go
@@ -18,13 +18,13 @@ package simulation
import (
"github.com/ethereum/go-ethereum/node"
- "github.com/ethereum/go-ethereum/p2p/discover"
+ "github.com/ethereum/go-ethereum/p2p/enode"
"github.com/ethereum/go-ethereum/p2p/simulations/adapters"
)
// Service returns a single Service by name on a particular node
// with provided id.
-func (s *Simulation) Service(name string, id discover.NodeID) node.Service {
+func (s *Simulation) Service(name string, id enode.ID) node.Service {
simNode, ok := s.Net.GetNode(id).Node.(*adapters.SimNode)
if !ok {
return nil
@@ -48,9 +48,9 @@ func (s *Simulation) RandomService(name string) node.Service {
// Services returns all services with a provided name
// from nodes that are up.
-func (s *Simulation) Services(name string) (services map[discover.NodeID]node.Service) {
+func (s *Simulation) Services(name string) (services map[enode.ID]node.Service) {
nodes := s.Net.GetNodes()
- services = make(map[discover.NodeID]node.Service)
+ services = make(map[enode.ID]node.Service)
for _, node := range nodes {
if !node.Up {
continue