protected static ResolverContext[] CreateInitialBatch(
IExecutionContext executionContext,
IDictionary <string, object> result)
{
ImmutableStack <object> source = ImmutableStack <object> .Empty
.Push(executionContext.Operation.RootValue);
IReadOnlyCollection <FieldSelection> fieldSelections =
executionContext.CollectFields(
executionContext.Operation.RootType,
executionContext.Operation.Definition.SelectionSet,
null);
int i = 0;
ResolverContext[] batch =
ArrayPool <ResolverContext> .Shared.Rent(
fieldSelections.Count);
foreach (FieldSelection fieldSelection in fieldSelections)
{
batch[i++] = ResolverContext.Rent(
executionContext,
fieldSelection,
source,
result);
}
return(batch);
}
private async Task <IExecutionResult> ExecuteInternalAsync(
IExecutionContext executionContext)
{
object rootValue = executionContext.Operation.RootValue;
FieldSelection fieldSelection = executionContext.CollectFields(
executionContext.Schema.SubscriptionType,
executionContext.Operation.Definition.SelectionSet,
null)
.Single();
ImmutableStack <object> source = ImmutableStack.Create(rootValue);
var subscribeContext = ResolverContext.Rent(
executionContext,
fieldSelection,
source,
new FieldData(1));
SubscribeResolverDelegate subscribeResolver =
fieldSelection.Field.SubscribeResolver
?? DefaultSubscribeResolverAsync;
try
{
IAsyncEnumerable <object> sourceStream =
await subscribeResolver(subscribeContext)
.ConfigureAwait(false);
return(new SubscriptionResult(
sourceStream,
message =>
{
IExecutionContext cloned = executionContext.Clone();
cloned.ContextData[WellKnownContextData.EventMessage] = message;
return cloned;
},
ExecuteSubscriptionQueryAsync,
executionContext.ServiceScope,
executionContext.RequestAborted));
}
finally
{
ResolverContext.Return(subscribeContext);
}
}