webRtc/Assets/PongStreamSender.cs

using System;
using Unity.RenderStreaming.Signaling;
using UnityEngine;
using Unity.WebRTC;

namespace Unity.RenderStreaming.Samples
{
    /// <summary>
    /// 心跳接收通道 - 继承 DataChannelBase（对应博客中的 ReceiveChannel）
    /// 接收服务端发送的 Ping，回复 Pong
    /// </summary>
    class HeartbeatReceiveChannel : DataChannelBase
    {
        // 消息类型（必须与服务端保持一致）
        private const int MSG_TYPE_PING = 1;
        private const int MSG_TYPE_PONG = 2;
        
        // 统计信息
        public long LastRtt { get; private set; }
        public float AverageRtt { get; private set; }
        public int ReceivedPingCount { get; private set; }
        
        // 事件
        public event Action<long> OnPingReceived;  // 参数：RTT
        public event Action OnTimeoutWarning;       // 超时警告
        
        private System.Collections.Generic.Queue<long> _latencyHistory = new System.Collections.Generic.Queue<long>(10);
        
        [Serializable]
        private class HeartbeatMessage
        {
            public int type;        // 1=ping, 2=pong
            public long timestamp;  // 服务端发送时间戳
            public int seq;         // 序列号
        }
        

        /// <summary>
        /// 重写 OnMessage - 当收到数据时自动调用（对应博客中的 ReceiveChannel 实现）
        /// </summary>
        protected override void OnMessage(byte[] bytes)
        {
            try
            {
                string json = System.Text.Encoding.UTF8.GetString(bytes);
                var msg = JsonUtility.FromJson<HeartbeatMessage>(json);
                
                if (msg.type == MSG_TYPE_PING)
                {
                    HandlePing(msg);
                }
            }
            catch (Exception e)
            {
                Debug.LogError($"[心跳接收通道] 消息解析失败: {e.Message}");
            }
        }
        
        /// <summary>
        /// 处理 Ping 消息并回复 Pong
        /// </summary>
        private void HandlePing(HeartbeatMessage ping)
        {
            long now = DateTimeOffset.UtcNow.ToUnixTimeMilliseconds();
            long latency = now - ping.timestamp;
            
            ReceivedPingCount++;
            LastRtt = latency;
            
            // 更新平均延迟
            _latencyHistory.Enqueue(latency);
            if (_latencyHistory.Count > 10)
                _latencyHistory.Dequeue();
            
            long sum = 0;
            foreach (var lat in _latencyHistory)
                sum += lat;
            AverageRtt = sum / (float)_latencyHistory.Count;
            
            Debug.Log($"[心跳接收通道] 收到 Ping #{ping.seq}, 延迟: {latency}ms, 平均: {AverageRtt:F1}ms");
            
            // 立即回复 Pong（关键：回传原始时间戳，方便服务端计算 RTT）
            SendPong(ping.timestamp, ping.seq);
            
            // 触发事件
            OnPingReceived?.Invoke(latency);
        }
        
        /// <summary>
        /// 发送 Pong 响应
        /// </summary>
        private void SendPong(long originalTimestamp, int seq)
        {
            if (Channel == null || Channel.ReadyState != RTCDataChannelState.Open)
            {
                Debug.LogWarning("[心跳接收通道] 通道未打开，无法发送 Pong");
                return;
            }
            
            var pong = new HeartbeatMessage
            {
                type = MSG_TYPE_PONG,
                timestamp = originalTimestamp,  // 回传服务端的时间戳
                seq = seq
            };
            
            string json = JsonUtility.ToJson(pong);
            byte[] bytes = System.Text.Encoding.UTF8.GetBytes(json);
            
            Channel.Send(bytes);
            Debug.Log($"[心跳接收通道] 发送 Pong #{seq}");
        }
        
        /// <summary>
        /// 获取当前连接状态
        /// </summary>
        public bool IsConnected => Channel != null && Channel.ReadyState == RTCDataChannelState.Open;
    }
}