OperationContext.Pooling.cs

using System;
using System.Collections.Generic;
using System.Runtime.CompilerServices;
using System.Threading;
using HotChocolate.Execution.DependencyInjection;
using HotChocolate.Execution.Instrumentation;
using HotChocolate.Execution.Processing.Tasks;
using HotChocolate.Fetching;
using HotChocolate.Types;
using HotChocolate.Utilities;
using Microsoft.Extensions.DependencyInjection;
using static HotChocolate.Execution.ThrowHelper;

namespace HotChocolate.Execution.Processing;

internal sealed partial class OperationContext
{
    private readonly IFactory<ResolverTask> _resolverTaskFactory;
    private readonly WorkScheduler _workScheduler;
    private readonly DeferredWorkScheduler _deferredWorkScheduler;
    private readonly ResultBuilder _resultBuilder;
    private readonly PooledPathFactory _pathFactory;
    private ISchema _schema = default!;
    private IErrorHandler _errorHandler = default!;
    private IActivator _activator = default!;
    private IExecutionDiagnosticEvents _diagnosticEvents = default!;
    private IDictionary<string, object?> _contextData = default!;
    private CancellationToken _requestAborted;
    private IOperation _operation = default!;
    private IVariableValueCollection _variables = default!;
    private IServiceProvider _services = default!;
    private Func<object?> _resolveQueryRootValue = default!;
    private IBatchDispatcher _batchDispatcher = default!;
    private InputParser _inputParser = default!;
    private object? _rootValue;
    private bool _isInitialized;

    public OperationContext(
        IFactory<ResolverTask> resolverTaskFactory,
        PooledPathFactory pathFactory,
        ResultBuilder resultBuilder,
        ITypeConverter typeConverter)
    {
        _resolverTaskFactory = resolverTaskFactory;
        _pathFactory = pathFactory;
        _workScheduler = new(this);
        _deferredWorkScheduler = new();
        _resultBuilder = resultBuilder;
        Converter = typeConverter;
    }

    public bool IsInitialized => _isInitialized;

    public void Initialize(
        IRequestContext requestContext,
        IServiceProvider scopedServices,
        IBatchDispatcher batchDispatcher,
        IOperation operation,
        IVariableValueCollection variables,
        object? rootValue,
        Func<object?> resolveQueryRootValue)
    {
        _schema = requestContext.Schema;
        _errorHandler = requestContext.ErrorHandler;
        _activator = requestContext.Activator;
        _diagnosticEvents = requestContext.DiagnosticEvents;
        _contextData = requestContext.ContextData;
        _requestAborted = requestContext.RequestAborted;
        _operation = operation;
        _variables = variables;
        _services = scopedServices;
        _inputParser = scopedServices.GetRequiredService<InputParser>();
        _rootValue = rootValue;
        _resolveQueryRootValue = resolveQueryRootValue;
        _batchDispatcher = batchDispatcher;
        _isInitialized = true;

        IncludeFlags = _operation.CreateIncludeFlags(variables);
        _workScheduler.Initialize(batchDispatcher);
        _deferredWorkScheduler.Initialize(this);
        _resultBuilder.Initialize(_operation, _errorHandler, _diagnosticEvents);
    }

    public void InitializeFrom(OperationContext context)
    {
        _schema = context._schema;
        _errorHandler = context._errorHandler;
        _activator = context._activator;
        _diagnosticEvents = context._diagnosticEvents;
        _contextData = context.ContextData;
        _requestAborted = context._requestAborted;
        _operation = context._operation;
        _variables = context._variables;
        _services = context._services;
        _inputParser = context._inputParser;
        _rootValue = context._rootValue;
        _resolveQueryRootValue = context._resolveQueryRootValue;
        _batchDispatcher = context._batchDispatcher;
        _isInitialized = true;

        IncludeFlags = _operation.CreateIncludeFlags(_variables);
        _workScheduler.Initialize(_batchDispatcher);
        _deferredWorkScheduler.InitializeFrom(this, context._deferredWorkScheduler);
        _resultBuilder.Initialize(_operation, _errorHandler, _diagnosticEvents);
    }

    public void Clean()
    {
        if (_isInitialized)
        {
            _pathFactory.Clear();
            _workScheduler.Clear();
            _resultBuilder.Clear();
            _deferredWorkScheduler.Clear();
            _schema = default!;
            _errorHandler = default!;
            _activator = default!;
            _diagnosticEvents = default!;
            _contextData = default!;
            _operation = default!;
            _variables = default!;
            _services = default!;
            _rootValue = null;
            _resolveQueryRootValue = default!;
            _batchDispatcher = default!;
            _isInitialized = false;
        }
    }

    [MethodImpl(MethodImplOptions.AggressiveInlining)]
    private void AssertInitialized()
    {
        if (!_isInitialized)
        {
            throw Object_Not_Initialized();
        }
    }
}