Support custom DHT servers

[btrtrc.git] / torrent.go

diff --git a/torrent.go b/torrent.go
index 474acfd1d1d334ace0d89287fccaa8a4d3bf22fc..4cd6cd4b9e3821323624e6544a84e90c047fa34e 100644 (file)
--- a/torrent.go
+++ b/torrent.go
@@ -1359,12 +1359,12 @@ func (t *Torrent) consumeDhtAnnouncePeers(pvs <-chan dht.PeersValues) {
        }
 }
 
-func (t *Torrent) announceToDht(impliedPort bool, s *dht.Server) error {
+func (t *Torrent) announceToDht(impliedPort bool, s DhtServer) error {
        ps, err := s.Announce(t.infoHash, t.cl.incomingPeerPort(), impliedPort)
        if err != nil {
                return err
        }
-       go t.consumeDhtAnnouncePeers(ps.Peers)
+       go t.consumeDhtAnnouncePeers(ps.Peers())
        select {
        case <-t.closed.LockedChan(t.cl.locker()):
        case <-time.After(5 * time.Minute):
@@ -1373,7 +1373,7 @@ func (t *Torrent) announceToDht(impliedPort bool, s *dht.Server) error {
        return nil
 }
 
-func (t *Torrent) dhtAnnouncer(s *dht.Server) {
+func (t *Torrent) dhtAnnouncer(s DhtServer) {
        cl := t.cl
        for {
                select {