diff --git a/libp2p/network/network_interface.py b/libp2p/network/network_interface.py index 156a71b..c759a41 100644 --- a/libp2p/network/network_interface.py +++ b/libp2p/network/network_interface.py @@ -1,7 +1,7 @@ from abc import ABC, abstractmethod from typing import TYPE_CHECKING, Dict, Sequence -from async_service import Service +from async_service import ServiceAPI from multiaddr import Multiaddr from libp2p.network.connection.net_connection_interface import INetConn @@ -73,5 +73,5 @@ class INetwork(ABC): pass -class INetworkService(INetwork, Service): +class INetworkService(INetwork, ServiceAPI): ... diff --git a/libp2p/network/swarm.py b/libp2p/network/swarm.py index 45d85b1..e03d65b 100644 --- a/libp2p/network/swarm.py +++ b/libp2p/network/swarm.py @@ -1,6 +1,7 @@ import logging from typing import Dict, List, Optional +from async_service import Service from multiaddr import Multiaddr from libp2p.io.abc import ReadWriteCloser @@ -30,7 +31,7 @@ from .stream.net_stream_interface import INetStream logger = logging.getLogger("libp2p.network.swarm") -class Swarm(INetworkService): +class Swarm(Service, INetworkService): self_id: ID peerstore: IPeerStore