fix: restoring lobby heartbeat to connected and hosting states (#703)

Author: fernando-cortez

Assets/Scripts/ConnectionManagement/ConnectionState/ClientConnectedState.cs

@@ -1,4 +1,7 @@
 using System;
+using Unity.Multiplayer.Samples.BossRoom.Shared.Net.UnityServices.Lobbies;
+using UnityEngine;
+using VContainer;
 
 namespace Unity.Multiplayer.Samples.BossRoom
 {
@@ -8,7 +11,16 @@ namespace Unity.Multiplayer.Samples.BossRoom
     /// </summary>
     class ClientConnectedState : ConnectionState
     {
-        public override void Enter() { }
+        [Inject]
+        protected LobbyServiceFacade m_LobbyServiceFacade;
+
+        public override void Enter()
+        {
+            if (m_LobbyServiceFacade.CurrentUnityLobby != null)
+            {
+                m_LobbyServiceFacade.BeginTracking();
+            }
+        }
 
         public override void Exit() { }
 

Assets/Scripts/ConnectionManagement/ConnectionState/HostingState.cs

@@ -34,6 +34,11 @@ public override void Enter()
             //The "BossRoom" server always advances to CharSelect immediately on start. Different games
             //may do this differently.
             SceneLoaderWrapper.Instance.LoadScene("CharSelect", useNetworkSceneManager: true);
+
+            if (m_LobbyServiceFacade.CurrentUnityLobby != null)
+            {
+                m_LobbyServiceFacade.BeginTracking();
+            }
         }
 
         public override void Exit()