[개발일지] Heartbeat Lobby

qweasfjbv·2025년 6월 4일

개발일지

목록 보기

6/8

개요

Unity에서 Host에 연결된 Client는 Host가 연결끊길 경우에 Client들에게 RPC를 통해 알려줄 수 있습니다.
하지만, 예기치 못한 상황에 강제종료를 하게되면 제대로 RPC가 호출되지 않고 Client들은 호스트가 없는 상태에서 게임이 멈추게 됩니다.
따라서 Host가 끊길 때 RPC를 보내는 것이 아니라, 계속 보내다가 RPC를 안보내면 Host의 연결이 끊긴 것으로 판단해야 하며 이를 Heartbeat Lobby 라 합니다.

이번에는 해당 Heartbeat Lobby 를 구현하고, 하는 김에 Ping 까지 측정할 수 있도록 해보겠습니다.

구현

		public float pingInterval = 2.0f;
		public float timeoutThreshold = 10.0f;

		private float lastPongTime;		// 가장 최근 받은 pong응답 시간
		private float pingTimer;		// ping 보내기 까지 남은 시간
		private float pingSentTime;		// 가장 최근 ping 보낸 시간
		private bool isDisconnected = true;

		public float LastPingMs { get; private set; } = -1;

우선 변수 선언입니다.

2초마다 핑을 보내고, 10초간 응답이 없으면 끊겼다고 판단하겠습니다.

		private void Update()
		{
			if (!isHeartbeating) return;
			if (!IsClient || IsHost) return;

			pingTimer -= Time.deltaTime;

			if (pingTimer <= 0f)
			{
				if (NetworkManager.Singleton.IsConnectedClient)
				{
					SendPingServerRpc();
					pingSentTime = Time.time;
				}
				pingTimer = pingInterval;
			}

			if (!isDisconnected && (Time.time - lastPongTime) > timeoutThreshold)
			{
				isDisconnected = true;
				Debug.LogWarning("[HeartbeatChecker] : Host Disconnected");
                
                // TODO - Disconnected
			}
		}

		public void StartHeartbeat()
		{
			if (IsClient && !IsHost)
			{
				lastPongTime = Time.time;
				pingTimer = pingInterval;
				isHeartbeating = true;
			}
		}

		public void EndHeartbeat()
		{
			isHeartbeating = false;
		}

우선 Host는 항상 핑이 0ms 이기 때문에 제외해주도록 합니다.
우선, SendPingServerRPC 함수를 통해 서버와 핑을 주고받아 가장 최근 응답받은 pong 시간을 기록합니다.
그 후에 최근에 받은 pong 응답 시간과 현재 시간의 차이를 통해 연결이 끊겼는지 판단하였습니다.

		[ServerRpc(RequireOwnership = false)]
		private void SendPingServerRpc(ServerRpcParams rpcParams = default)
		{
			if (!NetworkManager.Singleton.IsHost)
				return;

			ReceivePongClientRpc(rpcParams.Receive.SenderClientId);
		}
		[ClientRpc]
		private void ReceivePongClientRpc(ulong clientId)
		{
			if (clientId != NetworkManager.Singleton.LocalClientId)
				return;

			lastPongTime = Time.time;
			if (isDisconnected)
			{
				Debug.Log("[HeartbeatChecker] 서버 응답 복구됨!");
				isDisconnected = false;
			}

			LastPingMs = (Time.time - pingSentTime) * 1000.0f;
			Debug.Log($"[HeartbeatChecker] Ping: {LastPingMs:F0} ms");
		}