[client] Fix controller re-connection (#2758)

Rethink the peer reconnection implementation

relay/client/manager.go

@@ -65,6 +65,7 @@ type Manager struct {
 	relayClientsMutex sync.RWMutex
 
 	onDisconnectedListeners map[string]*list.List
+	onReconnectedListenerFn func()
 	listenerLock            sync.Mutex
 }
 
@@ -101,6 +102,7 @@ func (m *Manager) Serve() error {
 	m.relayClient = client
 
 	m.reconnectGuard = NewGuard(m.ctx, m.relayClient)
+	m.relayClient.SetOnConnectedListener(m.onServerConnected)
 	m.relayClient.SetOnDisconnectListener(func() {
 		m.onServerDisconnected(client.connectionURL)
 	})
@@ -138,6 +140,18 @@ func (m *Manager) OpenConn(serverAddress, peerKey string) (net.Conn, error) {
 	return netConn, err
 }
 
+// Ready returns true if the home Relay client is connected to the relay server.
+func (m *Manager) Ready() bool {
+	if m.relayClient == nil {
+		return false
+	}
+	return m.relayClient.Ready()
+}
+
+func (m *Manager) SetOnReconnectedListener(f func()) {
+	m.onReconnectedListenerFn = f
+}
+
 // AddCloseListener adds a listener to the given server instance address. The listener will be called if the connection
 // closed.
 func (m *Manager) AddCloseListener(serverAddress string, onClosedListener OnServerCloseListener) error {
@@ -240,6 +254,13 @@ func (m *Manager) openConnVia(serverAddress, peerKey string) (net.Conn, error) {
 	return conn, nil
 }
 
+func (m *Manager) onServerConnected() {
+	if m.onReconnectedListenerFn == nil {
+		return
+	}
+	go m.onReconnectedListenerFn()
+}
+
 func (m *Manager) onServerDisconnected(serverAddress string) {
 	if serverAddress == m.relayClient.connectionURL {
 		go m.reconnectGuard.OnDisconnected()