Added unregistration on stop

youdianwuliao · Mar 15, 2018 · bc22544 · bc22544
1 parent 2fcd950
commit bc22544
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/lib/peer.go b/lib/peer.go
@@ -280,7 +280,7 @@ func (np *NetworkPeer) stateWaitingToConnect(ptpc *PeerToPeer) error {
 	recheck := time.Now()
 	recheckTimeout := time.Duration(5000 * time.Millisecond)
 	for {
-		if np.RemoteState == PeerStateWaitingToConnect || np.RemoteState == PeerStateConnecting {
+		if np.RemoteState == PeerStateWaitingToConnect || np.RemoteState == PeerStateConnecting || np.RemoteState == PeerStateConnected {
 			Log(Debug, "Peer [%s] have joined required state: %s", np.ID, StringifyState(np.RemoteState))
 			np.SetState(PeerStateConnecting, ptpc)
 			break

diff --git a/stop.go b/stop.go
@@ -80,6 +80,7 @@ func (p *Daemon) Stop(args *DaemonArgs, resp *Response) error {
 				}
 			}
 			usedIPs = usedIPs[:k]
+			bootstrap.unregisterInstance(args.Hash)
 			return nil
 		}
 	} else if args.Dev != "" {