【m】插件上传
This commit is contained in:
341
Packages/com.unity.renderstreaming@3.1.0-exp.9/Tests/Runtime/Signaling/MockSignaling.cs
vendored
Normal file
341
Packages/com.unity.renderstreaming@3.1.0-exp.9/Tests/Runtime/Signaling/MockSignaling.cs
vendored
Normal file
@@ -0,0 +1,341 @@
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
using System.Threading;
|
||||
using System.Threading.Tasks;
|
||||
using Unity.RenderStreaming.Signaling;
|
||||
using Unity.WebRTC;
|
||||
using UnityEngine;
|
||||
|
||||
namespace Unity.RenderStreaming.RuntimeTest.Signaling
|
||||
{
|
||||
internal class MockSignaling : ISignaling
|
||||
{
|
||||
private string participantId ;
|
||||
interface IMockSignalingManager
|
||||
{
|
||||
Task Add(MockSignaling signaling);
|
||||
Task Remove(MockSignaling signaling);
|
||||
Task OpenConnection(MockSignaling signaling, string connectionId);
|
||||
Task CloseConnection(MockSignaling signaling, string connectionId);
|
||||
Task Offer(MockSignaling owner, DescData data);
|
||||
Task Answer(MockSignaling owner, DescData data);
|
||||
Task Candidate(MockSignaling owner, CandidateData data);
|
||||
}
|
||||
|
||||
class MockPublicSignalingManager : IMockSignalingManager
|
||||
{
|
||||
private Dictionary<MockSignaling, HashSet<string>> signalingToConnectionLookup = new Dictionary<MockSignaling, HashSet<string>>();
|
||||
private Dictionary<string, HashSet<MockSignaling>> connectionToSignalingLookup = new Dictionary<string, HashSet<MockSignaling>>();
|
||||
private const int MillisecondsDelay = 10;
|
||||
|
||||
public async Task Add(MockSignaling signaling)
|
||||
{
|
||||
await Task.Delay(MillisecondsDelay);
|
||||
signalingToConnectionLookup[signaling] = new HashSet<string>();
|
||||
|
||||
signaling.OnStart?.Invoke(signaling);
|
||||
}
|
||||
|
||||
public async Task Remove(MockSignaling signaling)
|
||||
{
|
||||
await Task.Delay(MillisecondsDelay);
|
||||
|
||||
if (signalingToConnectionLookup.ContainsKey(signaling))
|
||||
{
|
||||
foreach (var connectionId in signalingToConnectionLookup[signaling])
|
||||
{
|
||||
foreach (var signalingToDisconnect in connectionToSignalingLookup[connectionId])
|
||||
{
|
||||
signalingToDisconnect.OnDestroyConnection?.Invoke(signaling, connectionId);
|
||||
}
|
||||
|
||||
connectionToSignalingLookup.Remove(connectionId);
|
||||
}
|
||||
|
||||
signalingToConnectionLookup.Remove(signaling);
|
||||
}
|
||||
}
|
||||
|
||||
public async Task OpenConnection(MockSignaling signaling, string connectionId)
|
||||
{
|
||||
await Task.Delay(MillisecondsDelay);
|
||||
addToLookups(signaling, connectionId);
|
||||
|
||||
signaling.OnCreateConnection?.Invoke(signaling, connectionId, true);
|
||||
}
|
||||
|
||||
public async Task CloseConnection(MockSignaling signaling, string connectionId)
|
||||
{
|
||||
await Task.Delay(MillisecondsDelay);
|
||||
|
||||
if (connectionToSignalingLookup.ContainsKey(connectionId))
|
||||
{
|
||||
foreach (var signalingToDisconnect in connectionToSignalingLookup[connectionId])
|
||||
{
|
||||
signalingToDisconnect.OnDestroyConnection?.Invoke(signalingToDisconnect, connectionId);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public async Task Offer(MockSignaling owner, DescData data)
|
||||
{
|
||||
await Task.Delay(MillisecondsDelay);
|
||||
data.polite = false;
|
||||
foreach (var signaling in signalingToConnectionLookup.Keys.Where(e => e != owner))
|
||||
{
|
||||
signaling.OnOffer?.Invoke(signaling, data);
|
||||
}
|
||||
}
|
||||
|
||||
public async Task Answer(MockSignaling owner, DescData data)
|
||||
{
|
||||
await Task.Delay(MillisecondsDelay);
|
||||
foreach (var signaling in signalingToConnectionLookup.Keys.Where(e => e != owner))
|
||||
{
|
||||
addToLookups(owner, data.connectionId);
|
||||
signaling.OnAnswer?.Invoke(signaling, data);
|
||||
}
|
||||
}
|
||||
|
||||
public async Task Candidate(MockSignaling owner, CandidateData data)
|
||||
{
|
||||
await Task.Delay(MillisecondsDelay);
|
||||
foreach (var signaling in signalingToConnectionLookup.Keys.Where(e => e != owner))
|
||||
{
|
||||
signaling.OnIceCandidate?.Invoke(signaling, data);
|
||||
}
|
||||
}
|
||||
|
||||
private void addToLookups(MockSignaling signaling, string connectionId)
|
||||
{
|
||||
if (!connectionToSignalingLookup.TryGetValue(connectionId, out var signalingSet))
|
||||
{
|
||||
signalingSet = new HashSet<MockSignaling>();
|
||||
connectionToSignalingLookup[connectionId] = signalingSet;
|
||||
}
|
||||
signalingSet.Add(signaling);
|
||||
|
||||
if (!signalingToConnectionLookup.TryGetValue(signaling, out var connectionSet))
|
||||
{
|
||||
connectionSet = new HashSet<string>();
|
||||
signalingToConnectionLookup[signaling] = connectionSet;
|
||||
}
|
||||
connectionSet.Add(connectionId);
|
||||
}
|
||||
}
|
||||
|
||||
class MockPrivateSignalingManager : IMockSignalingManager
|
||||
{
|
||||
private Dictionary<string, List<MockSignaling>> connectionIds = new Dictionary<string, List<MockSignaling>>();
|
||||
private const int MillisecondsDelay = 10;
|
||||
public async Task Add(MockSignaling signaling)
|
||||
{
|
||||
await Task.Delay(MillisecondsDelay);
|
||||
signaling.OnStart?.Invoke(signaling);
|
||||
}
|
||||
|
||||
public async Task Remove(MockSignaling signaling)
|
||||
{
|
||||
await Task.Delay(MillisecondsDelay);
|
||||
}
|
||||
|
||||
public async Task OpenConnection(MockSignaling signaling, string connectionId)
|
||||
{
|
||||
await Task.Delay(MillisecondsDelay);
|
||||
bool peerExists = connectionIds.TryGetValue(connectionId, out var list);
|
||||
if (!peerExists)
|
||||
{
|
||||
list = new List<MockSignaling>();
|
||||
connectionIds.Add(connectionId, list);
|
||||
}
|
||||
|
||||
list.Add(signaling);
|
||||
|
||||
signaling.OnCreateConnection?.Invoke(signaling, connectionId, peerExists);
|
||||
}
|
||||
|
||||
public async Task CloseConnection(MockSignaling signaling, string connectionId)
|
||||
{
|
||||
await Task.Delay(MillisecondsDelay);
|
||||
bool peerExists = connectionIds.TryGetValue(connectionId, out var list);
|
||||
if (!peerExists || !list.Contains(signaling))
|
||||
{
|
||||
Debug.LogError($"{connectionId} This connection id is not used.");
|
||||
}
|
||||
|
||||
foreach (var element in list)
|
||||
{
|
||||
element.OnDestroyConnection?.Invoke(element, connectionId);
|
||||
}
|
||||
|
||||
list.Remove(signaling);
|
||||
if (list.Count == 0)
|
||||
{
|
||||
connectionIds.Remove(connectionId);
|
||||
}
|
||||
}
|
||||
|
||||
List<MockSignaling> FindList(MockSignaling owner, string connectionId)
|
||||
{
|
||||
if (!connectionIds.TryGetValue(connectionId, out var list))
|
||||
{
|
||||
return null;
|
||||
}
|
||||
|
||||
list = list.Where(e => e != owner).ToList();
|
||||
if (list.Count == 0)
|
||||
{
|
||||
return null;
|
||||
}
|
||||
|
||||
return list;
|
||||
}
|
||||
|
||||
public async Task Offer(MockSignaling owner, DescData data)
|
||||
{
|
||||
await Task.Delay(MillisecondsDelay);
|
||||
var list = FindList(owner, data.connectionId);
|
||||
if (list == null)
|
||||
{
|
||||
Debug.LogWarning($"{data.connectionId} This connection id is not ready other session.");
|
||||
return;
|
||||
}
|
||||
|
||||
data.polite = true;
|
||||
foreach (var signaling in list.Where(x => x != owner))
|
||||
{
|
||||
signaling.OnOffer?.Invoke(signaling, data);
|
||||
}
|
||||
}
|
||||
|
||||
public async Task Answer(MockSignaling owner, DescData data)
|
||||
{
|
||||
await Task.Delay(MillisecondsDelay);
|
||||
var list = FindList(owner, data.connectionId);
|
||||
if (list == null)
|
||||
{
|
||||
Debug.LogWarning($"{data.connectionId} This connection id is not ready other session.");
|
||||
return;
|
||||
}
|
||||
|
||||
foreach (var signaling in list.Where(x => x != owner))
|
||||
{
|
||||
signaling.OnAnswer?.Invoke(signaling, data);
|
||||
}
|
||||
}
|
||||
|
||||
public async Task Candidate(MockSignaling owner, CandidateData data)
|
||||
{
|
||||
await Task.Delay(MillisecondsDelay);
|
||||
var list = FindList(owner, data.connectionId);
|
||||
if (list == null)
|
||||
{
|
||||
Debug.LogWarning($"{data.connectionId} This connection id is not ready other session.");
|
||||
return;
|
||||
}
|
||||
|
||||
foreach (var signaling in list.Where(e => e != owner))
|
||||
{
|
||||
signaling.OnIceCandidate?.Invoke(signaling, data);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private static IMockSignalingManager manager = null;
|
||||
|
||||
public static void Reset(bool enablePrivateMode)
|
||||
{
|
||||
if (enablePrivateMode)
|
||||
{
|
||||
manager = new MockPrivateSignalingManager();
|
||||
}
|
||||
else
|
||||
{
|
||||
manager = new MockPublicSignalingManager();
|
||||
}
|
||||
}
|
||||
|
||||
public string Url { get { return string.Empty; } }
|
||||
|
||||
public float Interval { get { return 0.1f; } }
|
||||
|
||||
static MockSignaling()
|
||||
{
|
||||
manager = new MockPublicSignalingManager();
|
||||
}
|
||||
|
||||
public MockSignaling(SignalingSettings settings = null, SynchronizationContext context = null)
|
||||
{
|
||||
}
|
||||
|
||||
public void Start()
|
||||
{
|
||||
manager.Add(this);
|
||||
}
|
||||
|
||||
public void Stop()
|
||||
{
|
||||
manager.Remove(this);
|
||||
}
|
||||
|
||||
public event OnStartHandler OnStart;
|
||||
public event OnConnectHandler OnCreateConnection;
|
||||
public event OnDisconnectHandler OnDestroyConnection;
|
||||
public event OnOfferHandler OnOffer;
|
||||
public event OnAnswerHandler OnAnswer;
|
||||
public event OnIceCandidateHandler OnIceCandidate;
|
||||
|
||||
public void OpenConnection(string connectionId)
|
||||
{
|
||||
if (string.IsNullOrEmpty(connectionId))
|
||||
throw new ArgumentException("connectionId is null or empty.");
|
||||
manager.OpenConnection(this, connectionId);
|
||||
}
|
||||
|
||||
public void CloseConnection(string connectionId)
|
||||
{
|
||||
if (string.IsNullOrEmpty(connectionId))
|
||||
throw new ArgumentException("connectionId is null or empty.");
|
||||
manager.CloseConnection(this, connectionId);
|
||||
}
|
||||
|
||||
public void SendOffer(string connectionId, RTCSessionDescription offer)
|
||||
{
|
||||
if (string.IsNullOrEmpty(connectionId))
|
||||
throw new ArgumentException("connectionId is null or empty.");
|
||||
DescData data = new DescData
|
||||
{
|
||||
connectionId = connectionId,
|
||||
sdp = offer.sdp
|
||||
};
|
||||
manager.Offer(this, data);
|
||||
}
|
||||
|
||||
public void SendAnswer(string connectionId, RTCSessionDescription answer)
|
||||
{
|
||||
if (string.IsNullOrEmpty(connectionId))
|
||||
throw new ArgumentException("connectionId is null or empty.");
|
||||
DescData data = new DescData
|
||||
{
|
||||
connectionId = connectionId,
|
||||
sdp = answer.sdp
|
||||
};
|
||||
manager.Answer(this, data);
|
||||
}
|
||||
|
||||
public void SendCandidate(string connectionId, RTCIceCandidate candidate)
|
||||
{
|
||||
if (string.IsNullOrEmpty(connectionId))
|
||||
throw new ArgumentException("connectionId is null or empty.");
|
||||
CandidateData data = new CandidateData
|
||||
{
|
||||
connectionId = connectionId,
|
||||
candidate = candidate.Candidate,
|
||||
sdpMid = candidate.SdpMid,
|
||||
sdpMLineIndex = candidate.SdpMLineIndex.GetValueOrDefault()
|
||||
};
|
||||
manager.Candidate(this, data);
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user