| #region License |
| |
| /* |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance |
| * with the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, |
| * software distributed under the License is distributed on an |
| * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| * KIND, either express or implied. See the License for the |
| * specific language governing permissions and limitations |
| * under the License. |
| */ |
| |
| #endregion |
| |
| using System; |
| using System.IO; |
| using System.Net.WebSockets; |
| using System.Threading; |
| using System.Threading.Tasks; |
| |
| namespace Gremlin.Net.Driver |
| { |
| internal class WebSocketConnection : IDisposable |
| { |
| private const int ReceiveBufferSize = 1024; |
| private const WebSocketMessageType MessageType = WebSocketMessageType.Binary; |
| private readonly ClientWebSocket _client; |
| |
| public WebSocketConnection(Action<ClientWebSocketOptions> webSocketConfiguration) |
| { |
| _client = new ClientWebSocket(); |
| webSocketConfiguration?.Invoke(_client.Options); |
| } |
| |
| public async Task ConnectAsync(Uri uri) |
| { |
| await _client.ConnectAsync(uri, CancellationToken.None).ConfigureAwait(false); |
| } |
| |
| public async Task CloseAsync() |
| { |
| if (CloseAlreadyInitiated) return; |
| |
| try |
| { |
| await |
| _client.CloseAsync(WebSocketCloseStatus.NormalClosure, string.Empty, CancellationToken.None) |
| .ConfigureAwait(false); |
| } |
| catch (Exception) |
| { |
| // Swallow exceptions silently as there is nothing to do when closing fails |
| } |
| } |
| |
| private bool CloseAlreadyInitiated => _client.State == WebSocketState.Closed || |
| _client.State == WebSocketState.Aborted || |
| _client.State == WebSocketState.CloseSent; |
| |
| public async Task SendMessageAsync(byte[] message) |
| { |
| await |
| _client.SendAsync(new ArraySegment<byte>(message), MessageType, true, CancellationToken.None) |
| .ConfigureAwait(false); |
| } |
| |
| public async Task<byte[]> ReceiveMessageAsync() |
| { |
| using (var ms = new MemoryStream()) |
| { |
| WebSocketReceiveResult received; |
| var buffer = new byte[ReceiveBufferSize]; |
| |
| do |
| { |
| var receiveBuffer = new ArraySegment<byte>(buffer); |
| received = await _client.ReceiveAsync(receiveBuffer, CancellationToken.None).ConfigureAwait(false); |
| ms.Write(receiveBuffer.Array, receiveBuffer.Offset, received.Count); |
| } while (!received.EndOfMessage); |
| |
| return ms.ToArray(); |
| } |
| } |
| |
| public bool IsOpen => _client.State == WebSocketState.Open; |
| |
| #region IDisposable Support |
| |
| private bool _disposed; |
| |
| public void Dispose() |
| { |
| Dispose(true); |
| GC.SuppressFinalize(this); |
| } |
| |
| protected virtual void Dispose(bool disposing) |
| { |
| if (!_disposed) |
| { |
| if (disposing) |
| _client?.Dispose(); |
| _disposed = true; |
| } |
| } |
| |
| #endregion |
| } |
| } |