Returning to offline state when a transport failure occurs

Author: LPLafontaineB

Assets/Scripts/ConnectionManagement/ConnectionManager.cs

@@ -96,6 +96,7 @@ void Start()
             NetworkManager.OnClientDisconnectCallback += OnClientDisconnectCallback;
             NetworkManager.OnServerStarted += OnServerStarted;
             NetworkManager.ConnectionApprovalCallback += ApprovalCheck;
+            NetworkManager.OnTransportFailure += OnTransportFailure;
         }
 
         void OnDestroy()
@@ -104,7 +105,7 @@ void OnDestroy()
             NetworkManager.OnClientDisconnectCallback -= OnClientDisconnectCallback;
             NetworkManager.OnServerStarted -= OnServerStarted;
             NetworkManager.ConnectionApprovalCallback -= ApprovalCheck;
-
+            NetworkManager.OnTransportFailure -= OnTransportFailure;
         }
 
         internal void ChangeState(ConnectionState nextState)
@@ -139,6 +140,11 @@ void ApprovalCheck(NetworkManager.ConnectionApprovalRequest request, NetworkMana
             m_CurrentState.ApprovalCheck(request, response);
         }
 
+        void OnTransportFailure()
+        {
+            m_CurrentState.OnTransportFailure();
+        }
+
         public void StartClientLobby(string playerName)
         {
             m_CurrentState.StartClientLobby(playerName);

Assets/Scripts/ConnectionManagement/ConnectionState/ConnectionState.cs

@@ -39,5 +39,10 @@ public virtual void OnUserRequestedShutdown() { }
         public virtual void OnDisconnectReasonReceived(ConnectStatus disconnectReason) { }
 
         public virtual void ApprovalCheck(NetworkManager.ConnectionApprovalRequest request, NetworkManager.ConnectionApprovalResponse response) { }
+
+        public virtual void OnTransportFailure()
+        {
+            m_ConnectionManager.ChangeState(m_ConnectionManager.m_Offline);
+        }
     }
 }