forked from anacrolix/torrent
/
dht.go
51 lines (40 loc) · 1.05 KB
/
dht.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
package torrent
import (
"io"
"net"
"github.com/anacrolix/dht/v2"
"github.com/anacrolix/dht/v2/krpc"
)
type DhtServer interface {
Stats() interface{}
ID() [20]byte
Addr() net.Addr
AddNode(ni krpc.NodeInfo) error
Ping(addr *net.UDPAddr)
Announce(hash [20]byte, port int, impliedPort bool) (DhtAnnounce, error)
WriteStatus(io.Writer)
}
type DhtAnnounce interface {
Close()
Peers() <-chan dht.PeersValues
}
type anacrolixDhtServerWrapper struct {
*dht.Server
}
func (me anacrolixDhtServerWrapper) Stats() interface{} {
return me.Server.Stats()
}
type anacrolixDhtAnnounceWrapper struct {
*dht.Announce
}
func (me anacrolixDhtAnnounceWrapper) Peers() <-chan dht.PeersValues {
return me.Announce.Peers
}
func (me anacrolixDhtServerWrapper) Announce(hash [20]byte, port int, impliedPort bool) (DhtAnnounce, error) {
ann, err := me.Server.Announce(hash, port, impliedPort)
return anacrolixDhtAnnounceWrapper{ann}, err
}
func (me anacrolixDhtServerWrapper) Ping(addr *net.UDPAddr) {
me.Server.Ping(addr, nil)
}
var _ DhtServer = anacrolixDhtServerWrapper{}