| | 1 | | using System; |
| | 2 | | using NLog; |
| | 3 | | using ReFlex.Core.Networking.Interfaces; |
| | 4 | | using ReFlex.Core.Networking.Util; |
| | 5 | | using WebSocketSharp; |
| | 6 | | using Logger = NLog.Logger; |
| | 7 | |
|
| | 8 | | namespace ReFlex.Core.Networking.Components |
| | 9 | | { |
| | 10 | | public class WebSocketClient : IClient, IDisposable |
| | 11 | | { |
| 0 | 12 | | private static readonly Logger Logger = LogManager.GetCurrentClassLogger(); |
| | 13 | |
|
| | 14 | | private readonly WebSocket _client; |
| 0 | 15 | | private readonly string _address = ""; |
| | 16 | | private readonly string _endpoint; |
| | 17 | | private readonly int _port; |
| | 18 | |
|
| 0 | 19 | | private readonly Guid _clientId = Guid.NewGuid(); |
| | 20 | |
|
| | 21 | | public event EventHandler<NetworkingDataMessage> NewDataReceived; |
| 0 | 22 | | public bool IsInitialized { get; } |
| 0 | 23 | | public bool IsConnected { get; private set; } |
| | 24 | |
|
| 0 | 25 | | public NetworkInterface Type => NetworkInterface.Websockets; |
| | 26 | |
|
| 0 | 27 | | public string Address => $"{_address}:{_port}{_endpoint}"; |
| | 28 | |
|
| 0 | 29 | | public string Id => _clientId.ToString(); |
| | 30 | |
|
| 0 | 31 | | public WebSocketClient(string address, int port, string endpoint) |
| 0 | 32 | | { |
| 0 | 33 | | _address = address; |
| 0 | 34 | | _port = port; |
| 0 | 35 | | _endpoint = endpoint; |
| | 36 | |
|
| 0 | 37 | | _client = new WebSocket($"{_address}:{_port}{_endpoint}"); |
| 0 | 38 | | IsInitialized = true; |
| 0 | 39 | | } |
| | 40 | |
|
| | 41 | | /// <summary> |
| | 42 | | /// |
| | 43 | | /// </summary> |
| | 44 | | /// <param name="address">c</param> |
| | 45 | | /// <param name="port"></param> |
| | 46 | | /// <returns></returns> |
| | 47 | | public bool Connect() |
| 0 | 48 | | { |
| 0 | 49 | | _client.Connect(); |
| 0 | 50 | | _client.OnOpen += WebSocketOpened; |
| 0 | 51 | | _client.OnMessage += WebSocketMessageReceived; |
| 0 | 52 | | _client.OnClose += WebSocketClosed; |
| 0 | 53 | | _client.OnError += HandleWebSocketError; |
| | 54 | |
|
| 0 | 55 | | IsConnected = _client.ReadyState == WebSocketState.Open; |
| | 56 | |
|
| 0 | 57 | | return IsConnected; |
| 0 | 58 | | } |
| | 59 | |
|
| | 60 | | private void WebSocketClosed(object sender, CloseEventArgs e) |
| 0 | 61 | | { |
| 0 | 62 | | Logger.Debug( |
| 0 | 63 | | $"WebSocket on '{_address}' closed with code {e.Code}. {Environment.NewLine}Reason: {e.Reason}, was clea |
| | 64 | |
|
| 0 | 65 | | IsConnected = false; |
| 0 | 66 | | } |
| | 67 | |
|
| | 68 | |
|
| | 69 | | private void HandleWebSocketError(object sender, ErrorEventArgs e) |
| 0 | 70 | | { |
| 0 | 71 | | Logger.Error(e?.Exception, |
| 0 | 72 | | $"{e?.Exception?.GetType()} on WebSocket on '{_address}': {e?.Message}."); |
| | 73 | |
|
| 0 | 74 | | IsConnected = false; |
| 0 | 75 | | } |
| | 76 | |
|
| | 77 | | private void WebSocketOpened(object sender, EventArgs e) |
| 0 | 78 | | { |
| 0 | 79 | | Logger.Debug( |
| 0 | 80 | | $"WebSocket on '{_address}' opened. ClientId: {_clientId}."); |
| 0 | 81 | | } |
| | 82 | |
|
| | 83 | | private void WebSocketMessageReceived(object sender, MessageEventArgs e) |
| 0 | 84 | | { |
| 0 | 85 | | Logger.Trace( |
| 0 | 86 | | $"WebSocket on '{_address}' received data: {Environment.NewLine}{e.Data}{Environment.NewLine}ClientId: { |
| | 87 | |
|
| 0 | 88 | | var message = new NetworkingDataMessage(e.Data, _clientId); |
| 0 | 89 | | OnNewDataReceived(this, message); |
| 0 | 90 | | } |
| | 91 | |
|
| | 92 | | public void Disconnect() |
| 0 | 93 | | { |
| 0 | 94 | | Disconnect($"Client with Id {_clientId} disconnected."); |
| | 95 | |
|
| 0 | 96 | | IsConnected = false; |
| 0 | 97 | | } |
| | 98 | |
|
| | 99 | | private void Disconnect(string message) |
| 0 | 100 | | { |
| 0 | 101 | | _client.Close(CloseStatusCode.Normal, message); |
| 0 | 102 | | _client.OnOpen -= WebSocketOpened; |
| 0 | 103 | | _client.OnMessage -= WebSocketMessageReceived; |
| 0 | 104 | | _client.OnClose -= WebSocketClosed; |
| 0 | 105 | | _client.OnError -= HandleWebSocketError; |
| 0 | 106 | | } |
| | 107 | |
|
| | 108 | | public void OnNewDataReceived(object sender, NetworkingDataMessage message) |
| 0 | 109 | | { |
| 0 | 110 | | NewDataReceived?.Invoke(sender, message); |
| 0 | 111 | | } |
| | 112 | |
|
| | 113 | | public void Send(NetworkingDataMessage message) |
| 0 | 114 | | { |
| 0 | 115 | | _client.Send(message.Message); |
| 0 | 116 | | } |
| | 117 | |
|
| | 118 | | public void Dispose() |
| 0 | 119 | | { |
| 0 | 120 | | Disconnect($"App for client {_clientId} is closing."); |
| 0 | 121 | | ((IDisposable)_client)?.Dispose(); |
| 0 | 122 | | } |
| | 123 | | } |
| | 124 | | } |