Added "AddDataLoader<TService, TImplementation>" executor overload that takes a factory (#7787)

Author: danielreynolds1

src/HotChocolate/Core/src/Execution/DependencyInjection/RequestExecutorBuilderExtensions.DataLoader.cs

@@ -37,6 +37,18 @@ public static IRequestExecutorBuilder AddDataLoader<T>(
         builder.Services.TryAddScoped<T>(sp => sp.GetDataLoader<T>());
         return builder;
     }
+
+    public static IRequestExecutorBuilder AddDataLoader<TService, TImplementation>(
+        this IRequestExecutorBuilder builder,
+        Func<IServiceProvider, TImplementation> factory)
+        where TService : class, IDataLoader
+        where TImplementation : class, TService
+    {
+        builder.Services.AddSingleton(new DataLoaderRegistration(typeof(TService), typeof(TImplementation), sp => factory(sp)));
+        builder.Services.TryAddScoped<TImplementation>(sp => sp.GetDataLoader<TImplementation>());
+        builder.Services.TryAddScoped<TService>(sp => sp.GetDataLoader<TService>());
+        return builder;
+    }
 }
 
 file static class DataLoaderServiceProviderExtensions

src/HotChocolate/Core/test/Execution.Tests/Integration/DataLoader/DataLoaderTests.cs

@@ -472,6 +472,67 @@ await executor.ExecuteAsync(
         await snapshot.MatchMarkdownAsync();
     }
 
+    [Fact]
+    public async Task ClassDataLoader_Resolve_From_DependencyInjection_Using_Factory()
+    {
+        var snapshot = new Snapshot();
+
+        // arrange
+        var executor = await CreateExecutorAsync(
+            c => c
+                .AddQueryType<Query>()
+                .AddDataLoader<ITestDataLoader, TestDataLoader>(sp =>
+                    new TestDataLoader(
+                        sp.GetRequiredService<IBatchScheduler>(),
+                        new DataLoaderOptions()))
+                .ModifyRequestOptions(o => o.IncludeExceptionDetails = true)
+                .UseRequest(
+                    next => async context =>
+                    {
+                        await next(context);
+
+                        var dataLoader = (TestDataLoader)context.Services.GetRequiredService<ITestDataLoader>();
+
+                        context.Result = OperationResultBuilder
+                            .FromResult(((IOperationResult)context.Result!))
+                            .AddExtension("loads", dataLoader.Loads)
+                            .Build();
+                    })
+                .UseDefaultPipeline());
+
+        // act
+        snapshot.Add(
+            await executor.ExecuteAsync(
+                OperationRequestBuilder.New()
+                    .SetDocument(
+                        @"{
+                            a: dataLoaderWithInterface(key: ""a"")
+                            b: dataLoaderWithInterface(key: ""b"")
+                        }")
+                    .Build()));
+
+        snapshot.Add(
+            await executor.ExecuteAsync(
+                OperationRequestBuilder.New()
+                    .SetDocument(
+                        @"{
+                            a: dataLoaderWithInterface(key: ""a"")
+                        }")
+                    .Build()));
+
+        snapshot.Add(
+            await executor.ExecuteAsync(
+                OperationRequestBuilder.New()
+                    .SetDocument(
+                        @"{
+                            c: dataLoaderWithInterface(key: ""c"")
+                        }")
+                    .Build()));
+
+        // assert
+        await snapshot.MatchMarkdownAsync();
+    }
+
     [LocalFact]
     public async Task NestedDataLoader()
     {

src/HotChocolate/Core/test/Execution.Tests/Integration/DataLoader/__snapshots__/DataLoaderTests.ClassDataLoader_Resolve_From_DependencyInjection_Using_Factory.md

@@ -0,0 +1,55 @@
+# ClassDataLoader_Resolve_From_DependencyInjection_Using_Factory
+
+## Result 1
+
+```json
+{
+  "data": {
+    "a": "a",
+    "b": "b"
+  },
+  "extensions": {
+    "loads": [
+      [
+        "a",
+        "b"
+      ]
+    ]
+  }
+}
+```
+
+## Result 2
+
+```json
+{
+  "data": {
+    "a": "a"
+  },
+  "extensions": {
+    "loads": [
+      [
+        "a"
+      ]
+    ]
+  }
+}
+```
+
+## Result 3
+
+```json
+{
+  "data": {
+    "c": "c"
+  },
+  "extensions": {
+    "loads": [
+      [
+        "c"
+      ]
+    ]
+  }
+}
+```
+