forked from MidLevel/MLAPI.WebSockets
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathWebSocketClientFactory.cs
More file actions
108 lines (95 loc) · 3.84 KB
/
WebSocketClientFactory.cs
File metadata and controls
108 lines (95 loc) · 3.84 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
#if JSLIB
using System;
using System.Collections.Generic;
using System.Runtime.InteropServices;
using AOT;
#endif
namespace MLAPI.WebSockets
{
public static class WebSocketClientFactory
{
#if JSLIB
private static readonly byte[] _buffer = new byte[4096];
private static readonly Dictionary<int, JsWebSocketClient> sockets = new Dictionary<int, JsWebSocketClient>();
internal delegate void OnOpenCallback(int instanceId);
internal delegate void OnMessageCallback(int instanceId, IntPtr messagePointer, int messageSize);
internal delegate void OnErrorCallback(int instanceId, IntPtr errorPointer);
internal delegate void OnCloseCallback(int instanceId, int closeCode);
[DllImport("__Internal")]
internal static extern int CreateWebSocketInstance(string url);
[DllImport("__Internal")]
internal static extern void DestroyWebSocketInstance(int instanceId);
[DllImport("__Internal")]
internal static extern void SetOnOpenDelegate(OnOpenCallback callback);
[DllImport("__Internal")]
internal static extern void SetOnMessageDelegate(OnMessageCallback callback);
[DllImport("__Internal")]
internal static extern void SetOnErrorDelegate(OnErrorCallback callback);
[DllImport("__Internal")]
internal static extern void SetOnCloseDelegate(OnCloseCallback callback);
internal static void HandleInstanceDestroy(int instanceId)
{
sockets.Remove(instanceId);
DestroyWebSocketInstance(instanceId);
}
[MonoPInvokeCallback(typeof(OnOpenCallback))]
internal static void DelegateOnOpenEvent(int instanceId)
{
if (sockets.ContainsKey(instanceId))
{
sockets[instanceId].OnOpen();
}
}
[MonoPInvokeCallback(typeof(OnMessageCallback))]
internal static void DelegateOnMessageEvent(int instanceId, IntPtr payloadPointer, int size)
{
if (sockets.ContainsKey(instanceId))
{
Marshal.Copy(payloadPointer, _buffer, 0, size);
sockets[instanceId].OnPayload(new ArraySegment<byte>(_buffer, 0, size));
}
}
[MonoPInvokeCallback(typeof(OnErrorCallback))]
internal static void DelegateOnErrorEvent(int instanceId, IntPtr errorPointer)
{
if (sockets.ContainsKey(instanceId))
{
string errorMessage = Marshal.PtrToStringAuto(errorPointer);
sockets[instanceId].OnError(errorMessage);
}
}
[MonoPInvokeCallback(typeof(OnCloseCallback))]
internal static void DelegateOnCloseEvent(int instanceId, int disconnectCode)
{
if (sockets.ContainsKey(instanceId))
{
DisconnectCode code = (DisconnectCode)disconnectCode;
if (!Enum.IsDefined(typeof(DisconnectCode), code))
{
code = DisconnectCode.Unknown;
}
sockets[instanceId].OnClose(code);
}
}
#endif
public static IWebSocketClient Create(string url)
{
#if JSLIB
// Set the js delegates
SetOnOpenDelegate(DelegateOnOpenEvent);
SetOnMessageDelegate(DelegateOnMessageEvent);
SetOnErrorDelegate(DelegateOnErrorEvent);
SetOnCloseDelegate(DelegateOnCloseEvent);
// Create js websocket instance
int instanceId = CreateWebSocketInstance(url);
// Construct a wrapper for the JSLIB
JsWebSocketClient webSocket = new JsWebSocketClient(instanceId);
sockets.Add(instanceId, webSocket);
return webSocket;
#else
// Return a native websocket wrapper
return new NativeWebSocketClient(url);
#endif
}
}
}