test2

[bytom/bytom.git] / p2p / switch.go

diff --git a/p2p/switch.go b/p2p/switch.go
index d125f3b..27dec0d 100644 (file)
--- a/p2p/switch.go
+++ b/p2p/switch.go
@@ -407,7 +407,7 @@ func (sw *Switch) addPeerWithConnection(conn net.Conn) error {
 func (sw *Switch) AddBannedPeer(peer *Peer) error {
        sw.mtx.Lock()
        defer sw.mtx.Unlock()
-       key := peer.RemoteAddr
+       key := peer.NodeInfo.RemoteAddrHost()
        sw.bannedPeer[key] = time.Now().Add(defaultBanDuration)
        datajson, err := json.Marshal(sw.bannedPeer)
        if err != nil {