Add overrides for subscribe/unsubscribe

Author: wravery

include/graphqlservice/GraphQLService.h

@@ -879,7 +879,10 @@ class Request : public std::enable_shared_from_this<Request>
 	GRAPHQLSERVICE_EXPORT std::future<response::Value> resolve(std::launch launch, const std::shared_ptr<RequestState>& state, peg::ast& query, const std::string& operationName, response::Value&& variables) const;
 
 	GRAPHQLSERVICE_EXPORT SubscriptionKey subscribe(SubscriptionParams&& params, SubscriptionCallback&& callback);
+	GRAPHQLSERVICE_EXPORT SubscriptionKey subscribe(std::launch launch, SubscriptionParams&& params, SubscriptionCallback&& callback);
+
 	GRAPHQLSERVICE_EXPORT void unsubscribe(SubscriptionKey key);
+	GRAPHQLSERVICE_EXPORT void unsubscribe(std::launch launch, SubscriptionKey key);
 
 	GRAPHQLSERVICE_EXPORT void deliver(const SubscriptionName& name, const std::shared_ptr<Object>& subscriptionObject) const;
 	GRAPHQLSERVICE_EXPORT void deliver(const SubscriptionName& name, const SubscriptionArguments& arguments, const std::shared_ptr<Object>& subscriptionObject) const;

src/GraphQLService.cpp

@@ -2104,6 +2104,39 @@ SubscriptionKey Request::subscribe(SubscriptionParams&& params, SubscriptionCall
 	return key;
 }
 
+SubscriptionKey Request::subscribe(std::launch launch, SubscriptionParams&& params, SubscriptionCallback&& callback)
+{
+	const auto key = subscribe(std::move(params), std::move(callback));
+	const auto itrOperation = _operations.find(std::string { strSubscription });
+	const auto itrSubscription = _subscriptions.find(key);
+
+	if (itrOperation != _operations.cend()
+		&& itrSubscription != _subscriptions.cend())
+	{
+		const auto& operation = itrOperation->second;
+		const auto& registration = itrSubscription->second;
+		response::Value emptyFragmentDirectives(response::Type::Map);
+		const SelectionSetParams selectionSetParams {
+			ResolverContext::NotifySubscribe,
+			registration->data->state,
+			registration->data->directives,
+			emptyFragmentDirectives,
+			emptyFragmentDirectives,
+			emptyFragmentDirectives,
+			{},
+			launch,
+		};
+
+		std::async(launch,
+			[](std::future<response::Value> document)
+		{
+			return document.get();
+		}, operation->resolve(selectionSetParams, registration->selection, registration->data->fragments, registration->data->variables)).get();
+	}
+
+	return key;
+}
+
 void Request::unsubscribe(SubscriptionKey key)
 {
 	auto itrSubscription = _subscriptions.find(key);
@@ -2133,6 +2166,38 @@ void Request::unsubscribe(SubscriptionKey key)
 	}
 }
 
+void Request::unsubscribe(std::launch launch, SubscriptionKey key)
+{
+	const auto itrOperation = _operations.find(std::string { strSubscription });
+	const auto itrSubscription = _subscriptions.find(key);
+
+	if (itrOperation != _operations.cend()
+		&& itrSubscription != _subscriptions.cend())
+	{
+		const auto& operation = itrOperation->second;
+		const auto& registration = itrSubscription->second;
+		response::Value emptyFragmentDirectives(response::Type::Map);
+		const SelectionSetParams selectionSetParams {
+			ResolverContext::NotifyUnsubscribe,
+			registration->data->state,
+			registration->data->directives,
+			emptyFragmentDirectives,
+			emptyFragmentDirectives,
+			emptyFragmentDirectives,
+			{},
+			launch,
+		};
+
+		std::async(launch,
+			[](std::future<response::Value> document)
+		{
+			return document.get();
+		}, operation->resolve(selectionSetParams, registration->selection, registration->data->fragments, registration->data->variables)).get();
+	}
+
+	unsubscribe(key);
+}
+
 void Request::deliver(const SubscriptionName& name, const std::shared_ptr<Object>& subscriptionObject) const
 {
 	deliver(std::launch::deferred, name, subscriptionObject);