dht.go | 6 ++++++ diff --git a/dht.go b/dht.go index da79aee472b853e153175ef2b752a68a867fc8f2..b69c28845196458ec22d2aadc372f4ee5e7467da 100644 --- a/dht.go +++ b/dht.go @@ -6,6 +6,7 @@ "net" "github.com/anacrolix/dht/v2" "github.com/anacrolix/dht/v2/krpc" + peer_store "github.com/anacrolix/dht/v2/peer-store" ) type DhtServer interface { @@ -16,6 +17,11 @@ AddNode(ni krpc.NodeInfo) error Ping(addr *net.UDPAddr) Announce(hash [20]byte, port int, impliedPort bool) (DhtAnnounce, error) WriteStatus(io.Writer) +} + +// Optional interface for DhtServer's that can expose their peer store (if any). +type PeerStorer interface { + PeerStore() peer_store.Interface } type DhtAnnounce interface {