From: YenForYang <YenForYang@users.noreply.github.com>
Date: Sat, 4 Sep 2021 02:24:35 +0000 (-0500)
Subject: Simplify dhtPort()
X-Git-Tag: v1.32.0~100
X-Git-Url: http://www.git.stargrave.org/?a=commitdiff_plain;h=f86355ac0bb46b817f83c68a831d6e9c4aefa72a;p=btrtrc.git

Simplify dhtPort()

A similar case of `haveDhtServer`
---

diff --git a/client.go b/client.go
index c5ab3373..a0465059 100644
--- a/client.go
+++ b/client.go
@@ -991,10 +991,10 @@ func (cl *Client) sendInitialMessages(conn *PeerConn, torrent *Torrent) {
 }
 
 func (cl *Client) dhtPort() (ret uint16) {
-	cl.eachDhtServer(func(s DhtServer) {
-		ret = uint16(missinggo.AddrPort(s.Addr()))
-	})
-	return
+	if len(cl.dhtServers) == 0 {
+		return
+	}
+	return uint16(missinggo.AddrPort(cl.dhtServers[len(cl.dhtServers)-1].Addr()))
 }
 
 func (cl *Client) haveDhtServer() bool {