Merge pull request #164 from graphql-dotnet/websocket-connection-status

Websocket connection status

Author: rose-a

src/GraphQL.Client.Abstractions.Websocket/GraphQLWebsocketConnectionState.cs

@@ -0,0 +1,7 @@
+namespace GraphQL.Client.Abstractions.Websocket {
+	public enum GraphQLWebsocketConnectionState {
+		Disconnected,
+		Connecting,
+		Connected
+	}
+}

src/GraphQL.Client/GraphQLHttpClient.cs

@@ -1,6 +1,5 @@
 using System;
 using System.Collections.Concurrent;
-using System.Linq;
 using System.Net.Http;
 using System.Text;
 using System.Threading;
@@ -33,6 +32,11 @@ public class GraphQLHttpClient : IGraphQLClient {
 		/// </summary>
 		public IObservable<Exception> WebSocketReceiveErrors => graphQlHttpWebSocket.ReceiveErrors;
 
+		/// <summary>
+		/// the websocket connection state
+		/// </summary>
+		public IObservable<GraphQLWebsocketConnectionState> WebsocketConnectionState =>
+			graphQlHttpWebSocket.ConnectionState;
 
 		#region Constructors
 
@@ -47,15 +51,15 @@ public GraphQLHttpClient(Action<GraphQLHttpClientOptions> configure) : this(conf
 		public GraphQLHttpClient(GraphQLHttpClientOptions options, HttpClient httpClient) {
 			Options = options;
 			this.HttpClient = httpClient ?? throw new ArgumentNullException(nameof(httpClient));
-			this.graphQlHttpWebSocket = new GraphQLHttpWebSocket(GetWebSocketUri(), Options);
+			this.graphQlHttpWebSocket = new GraphQLHttpWebSocket(GetWebSocketUri(), this);
 			Options.JsonSerializer = JsonSerializer.EnsureAssigned();
 		}
 
 		public GraphQLHttpClient(GraphQLHttpClientOptions options, HttpClient httpClient, IGraphQLWebsocketJsonSerializer serializer) {
 			Options = options ?? throw new ArgumentNullException(nameof(options));
 			Options.JsonSerializer = serializer ?? throw new ArgumentNullException(nameof(serializer));
 			this.HttpClient = httpClient ?? throw new ArgumentNullException(nameof(httpClient));
-			this.graphQlHttpWebSocket = new GraphQLHttpWebSocket(GetWebSocketUri(), Options);
+			this.graphQlHttpWebSocket = new GraphQLHttpWebSocket(GetWebSocketUri(), this);
 		}
 
 		#endregion

src/GraphQL.Client/GraphQLHttpClientOptions.cs

@@ -49,5 +49,10 @@ public class GraphQLHttpClientOptions {
 		/// Request preprocessing function. Can be used i.e. to inject authorization info into a GraphQL request payload.
 		/// </summary>
 		public Func<GraphQLRequest, GraphQLHttpClient, Task<GraphQLRequest>> PreprocessRequest { get; set; } = (request, client) => Task.FromResult(request);
+
+		/// <summary>
+		/// This callback is called after successfully establishing a websocket connection but before any regular request is made. 
+		/// </summary>
+		public Func<GraphQLHttpClient, Task> OnWebsocketConnected { get; set; } = client => Task.CompletedTask;
 	}
 }