Add OnDisconnecting service callback (#767)

Add OnDisconnecting service callback for mobile

client/internal/connect.go

@@ -164,6 +164,7 @@ func RunClient(ctx context.Context, config *Config, statusRecorder *peer.Status,
 		state.Set(StatusConnected)
 
 		<-engineCtx.Done()
+		statusRecorder.ClientTeardown()
 
 		backOff.Reset()
 

client/internal/peer/listener.go

@@ -5,6 +5,7 @@ type Listener interface {
 	OnConnected()
 	OnDisconnected()
 	OnConnecting()
+	OnDisconnecting()
 	OnAddressChanged(string, string)
 	OnPeersListChanged(int)
 }

client/internal/peer/notifier.go

@@ -8,6 +8,7 @@ const (
 	stateDisconnected = iota
 	stateConnected
 	stateConnecting
+	stateDisconnecting
 )
 
 type notifier struct {
@@ -57,8 +58,12 @@ func (n *notifier) updateServerStates(mgmState bool, signalState bool) {
 	}
 
 	n.currentServerState = newState
-	n.lastNotification = n.calculateState(newState, n.currentClientState)
 
+	if n.lastNotification == stateDisconnecting {
+		return
+	}
+
+	n.lastNotification = n.calculateState(newState, n.currentClientState)
 	go n.notifyAll(n.lastNotification)
 }
 
@@ -78,6 +83,14 @@ func (n *notifier) clientStop() {
 	go n.notifyAll(n.lastNotification)
 }
 
+func (n *notifier) clientTearDown() {
+	n.serverStateLock.Lock()
+	defer n.serverStateLock.Unlock()
+	n.currentClientState = false
+	n.lastNotification = stateDisconnecting
+	go n.notifyAll(n.lastNotification)
+}
+
 func (n *notifier) isServerStateChanged(newState bool) bool {
 	return n.currentServerState != newState
 }
@@ -99,6 +112,8 @@ func (n *notifier) notifyListener(l Listener, state int) {
 		l.OnConnected()
 	case stateConnecting:
 		l.OnConnecting()
+	case stateDisconnecting:
+		l.OnDisconnecting()
 	}
 }
 

client/internal/peer/status.go

@@ -288,6 +288,11 @@ func (d *Status) ClientStop() {
 	d.notifier.clientStop()
 }
 
+// ClientTeardown will notify all listeners about the service is under teardown
+func (d *Status) ClientTeardown() {
+	d.notifier.clientTearDown()
+}
+
 // AddConnectionListener add a listener to the notifier
 func (d *Status) AddConnectionListener(listener Listener) {
 	d.notifier.addListener(listener)