Skip to main content

Implementing DataLoader For Query Execution Using Hot Chocolate GraphQL In .Net

In this article, we are going to understand and implement the DataLoader concept in Hot Chocolate GraphQL.

Here we will use an existing Hot Chocolate sample application that was created in the introduction articles on Hot chocolate GraphQL. I recommend reading those articles:
Part-1 Introduction On Hot Chocolate GraphQL

Issues Need To Fix:

Before implementing the DataLoader into our sample we must aware of the issues we are facing. In GrphQL we have an option like Fragments Query. Fragments query is used to display the comparison results very effective way.
Fragments Query:
query{
  Gadget1:GetByBrand(brand:"samsung"){
    Id
    ProductName
    Brand
  }
   Gadget2:GetByBrand(brand:"red mi"){
    Id
    ProductName
    Brand
  }
}
Here we request a GraphQL endpoint to serve the different gadgets for comparison this type of query called Fragments Query.

Now this kind of Fragments Query produces issues like:
  • We will face a threading issue with the database context on using the async database communication calls.
  • In a normal process, this Fragments Query will run the query resolver method(logic that contains database communication) count is like based on n-number of comparison queries. This means if you observe the above sample query we requesting 2 comparison results so the revolver method will execute 2 times(which means 2 database calls). This might leads to performance issues.

Async Calls Database Context Issue:

Let's expose ourselves to the kind of issue we will get using the asynchronous calls.

In Part-2  written a resolver method as synchronous like below.
SchemaCore/Query.cs:(Existing Logic)
public List<Gadgets> GetByBrand(string brand, [Service] MyWorldDbContext context)
{
	return context.Gadgets.Where(_ => _.Brand.ToLower() == brand.ToLower()).ToList();
}
The results look as below.

Now let's update our resolver to an asynchronous approach.
SchemaCore/Query.cs:(Existing Logic)
public async Task<List<Gadgets>> GetByBrand(string brand, [Service] MyWorldDbContext context)
{
  return await context.Gadgets
  .Where(_ => _.Brand.ToLower() == brand.ToLower()).ToListAsync();
}
Let's check the results for the asynchronous calls.
You can see we are getting errors in the result set. But the actual error is like this:-

An exception occurred while iterating over the results of a query for contexttype'HC.GraphQL.Sample.Data.Context.MyWorldDbContext'.
System.InvalidOperationException: A second operation was started in this context before a previous operation completed. This is usually caused by different threads concurrently using the same instance of DbContext. For more information on how to avoid threading issues with DbContext.

This error because in the fragment query we are requested 2 comparison results, since our resolver is an asynchronous resolver so 2 requests are executed parallel and both are trying to access the database context at the same time.

Use Database Context Pool To Resolve The Threading Issue:

In .net DbContext pool keeps by default 128 Dbcontext alive in the pool. So that these instances will be served to requests without destroying them immediately. So if our fragment request will use separate DBContext for each thread request.

Existing DbContext registration looks like as below:
Startup.cs:(Existing Code)
services.AddDbContext<MyWorldDbContext>(options => {
  options.UseSqlServer(Configuration.GetConnectionString("MyWorldDbConnection"));
});
Now let's update our code to register the DbContext pool as below:
Startup.cs:(Updated Code)
services.AddPooledDbContextFactory<MyWorldDbContext>(options => {
  options.UseSqlServer(Configuration.GetConnectionString("MyWorldDbConnection"));
});
Now let's implement our logic to fetch the DbContext instance from the pool of DbContext. So let's create an extension method of the IObjectFieldDescriptor that will generate DbContext as below.
Extensions/ObjectFieldDescriptorExtensions:
using HotChocolate.Types;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.DependencyInjection;

namespace HC.GraphQL.Sample.Extensions
{
  public static class ObjectFieldDescriptorExtensions
  {
   public static IObjectFieldDescriptor UseDbContext<TDbContext>(
	this IObjectFieldDescriptor descriptor
   ) where TDbContext: DbContext
   {
	return descriptor.UseScopedService<TDbContext>(
		create: s => s.GetRequiredService<IDbContextFactory<TDbContext>>().CreateDbContext(),
		disposeAsync: (s,c) => c.DisposeAsync()
	);
   }
  }
}
  • (Line: 14) Fetching the context from the pool.
  • (Line: 15) Releases the allocated resource to DbContext.
Now let's create an Attribute which we are going to decorate on resolver so that they receive scoped DbContext from the pool.
Attributes/UseMyWordDbContextAttribute.cs:
using System.Reflection;
using HC.GraphQL.Sample.Data.Context;
using HotChocolate.Types;
using HotChocolate.Types.Descriptors;

namespace HC.GraphQL.Sample.Extensions
{
  public class UseMyWorldDbContextAttribute : ObjectFieldDescriptorAttribute
  {
   public override void OnConfigure(IDescriptorContext context, 
   IObjectFieldDescriptor descriptor, MemberInfo member)
   {
		descriptor.UseDbContext<MyWorldDbContext>();
   }
  }
}
  • (Line: 8) Inherits from the 'HotChocolate.Types.ObjectFieldDescriptorAttribute'.
  • (Line: 13) Above create generic extension method registered here with our database context.
Now update our resolver method to decorate the attribute on it as below.
Schemas/Query.cs:
[UseMyWorldDbContext]
public List<Gadgets> GetByBrand(string brand, [ScopedService] MyWorldDbContext context)
{
  return context.Gadgets.Where(_ => _.Brand.ToLower() == brand.ToLower()).ToList();
}
  • (Line: 1)Decorated our 'UseMyWorldDbContext' attribute
  • (Line: 2) Used the 'ScopedService' attribute to inject the Dbcontext into our resolver method.
Now let's run our fragments query on our asynchronous resolver which will now work perfectly without any threading issue.

Fragment Query Multiple Database Calls Issue:

As mentioned that fragment query will run the resolver method multiple times based on the number of comparison queries inside of it. For the above sample query or screenshots, we used only 2 comparison queries inside of the fragment query so this will invoke the resolver method 2 times which might be performance issues. 

From the above 2 images, we can observe resolver executed 2 times.

DataLoader To Fix Multiple Database Calls Issue:

DataLoader technique in GraphQL to fetch the fragments of data with a single execution of the resolver method which gives us performance benifits.

Let's implement a DataLoader for our sample.
DataLoader/GadgetsByBrandDataLoader.cs:
using System.Collections.Generic;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
using HC.GraphQL.Sample.Data.Context;
using HC.GraphQL.Sample.Data.Entities;
using HotChocolate.DataLoader;
using HotChocolate.Fetching;
using Microsoft.EntityFrameworkCore;

namespace HC.GraphQL.Sample.DataLoader
{
  public class GadgetsByBrandDataLoader : BatchDataLoader<string, List<Gadgets>>
  {
	private readonly IDbContextFactory<MyWorldDbContext> _dbContextFactory;
	public GadgetsByBrandDataLoader(IDbContextFactory<MyWorldDbContext> dbContextFactory,
	BatchScheduler batchScheduler) : base(batchScheduler)
	{
	  _dbContextFactory = dbContextFactory;
	}
	protected override async Task<IReadOnlyDictionary<string, List<Gadgets>>> LoadBatchAsync(IReadOnlyList<string> keys, CancellationToken cancellationToken)
	{
	 await using (MyWorldDbContext dbContext = _dbContextFactory.CreateDbContext())
	 {
	  var gadgets = await dbContext.Gadgets
	  .Where(g => keys.Select(_ => _.ToLower()).ToList().Contains(g.Brand)).ToListAsync(cancellationToken);
	  var result = gadgets.GroupBy( _ => _.Brand.ToLower())
	 .Select(_ =>new {
		key = _.Key.ToLower(),
		gadgets = _.ToList()
	 }).ToDictionary(_ => _.key, _ => _.gadgets);
	  return result;
	 };
	}
  }
}
  • (Line: 13) To create a HotChocolate GraphQL Dataloader we need to inherit base class DataLoader that is 'HotChocolate.DataLoader.BatchDataLoader<TKey,TValue>'. Here 'TKey' is the type of our query parameter value and 'TValue' is the output type returned from the delivery.
  • (Line: 16&17) Injected 'HotChocolate.Fetching.BatchScheduler' and then its value passed to the base constructor. Also injected 'Microsoft.EntityFrameworkCore.IDbContextFactory'.
  • (Line: 21) Implemented the 'LoadBatchAsync' method whose return type is key-value pair dictionary.
  • This method has input parameters is like collection string as the variable name 'keys', all these collections of string nothing but our query parameters from each comparison query this helps to fetch the data from the database with a single attempt.
  • (Line: 25-31) Fetching results based on the input parameter collection 'keys' and return the result as dictionary data.
Now register our DataLoader in Startup.cs file.
Startup.cs:
services.AddGraphQLServer()
.AddQueryType<QueryObjectType>()
.AddMutationType<MutationObjectType>()
.AddDataLoader<GadgetsByBrandDataLoader>();
Now let's implement our resolver method where we are going to use our DataLoader to output the results.
SchemaCore/Query.cs:
public async Task<List<Gadgets>> GetByBrandLoader(
  string brand,
  GadgetsByBrandDataLoader dataLoader,
  CancellationToken cancellationToken)
{
   var result = await dataLoader.LoadAsync(brand,cancellationToken);
   return result;
}
  • Here for our resolver first parameter is our query parameter and the second is our Dataloader passed as a parameter this will be passed by the framework automatically.
Now let's create a GraphQL field for our new resolver method.
ObjectTypes/QueryObjectType.cs:
descriptor.Field(g => g.GetByBrandLoader(default, default, default)).Type<ListType<GadgetsObjectType>>()
Name("GetByBrandLoader");
Now let's test our fragments query that will return the result with a single time execution of the resolver method with the help of DataLoader.
That's all about the DataLoader in the HotChocolate GraphQL.

Support Me!
Buy Me A Coffee PayPal Me

Wrapping Up:

Hopefully, I think this article delivered some useful information about DataLoader in GraphQL. I love to have your feedback, suggestions, and better techniques in the comment section below.

Refer:

Follow Me:

Comments

  1. Nice article, but I don't see the difference between the existing and the updated DbContext registration. Am I missing something?

    ReplyDelete
  2. The updated DbContext registration should be this (I think, the author copied the code but forgot to change the method name):

    services.AddPooledDbContextFactory(options => {
    options.UseSqlServer(Configuration.GetConnectionString("MyWorldDbConnection"));
    });

    The article itself is awesome, thanks for it.

    ReplyDelete
  3. good stuff! thanks but the code was going off the screenand i had to scroll alot to the right, just an fyi

    ReplyDelete

Post a Comment

Popular posts from this blog

Angular 14 Reactive Forms Example

In this article, we will explore the Angular(14) reactive forms with an example. Reactive Forms: Angular reactive forms support model-driven techniques to handle the form's input values. The reactive forms state is immutable, any form filed change creates a new state for the form. Reactive forms are built around observable streams, where form inputs and values are provided as streams of input values, which can be accessed synchronously. Some key notations that involve in reactive forms are like: FormControl - each input element in the form is 'FormControl'. The 'FormControl' tracks the value and validation status of form fields. FormGroup - Track the value and validate the state of the group of 'FormControl'. FormBuilder - Angular service which can be used to create the 'FormGroup' or FormControl instance quickly. Form Array - That can hold infinite form control, this helps to create dynamic forms. Create An Angular(14) Application: Let'

.NET 7 Web API CRUD Using Entity Framework Core

In this article, we are going to implement a sample .NET 7 Web API CRUD using the Entity Framework Core. Web API: Web API is a framework for building HTTP services that can be accessed from any client like browser, mobile devices, and desktop apps. In simple terminology API(Application Programming Interface) means an interface module that contains programming functions that can be requested via HTTP calls either to fetch or update data for their respective clients. Some of the Key Characteristics of API: Supports HTTP verbs like 'GET', 'POST', 'PUT', 'DELETE', etc. Supports default responses like 'XML' and 'JSON'. Also can define custom responses. Supports self-hosting or individual hosting, so that all different kinds of apps can consume it. Authentication and Authorization are easy to implement. The ideal platform to build the REST full services. Install The SQL Server And SQL Management Studio: Let's install the SQL server on our l

ReactJS(v18) JWT Authentication Using HTTP Only Cookie

In this article, we will implement the ReactJS application authentication using the HTTP-only cookie. HTTP Only Cookie: In a SPA(Single Page Application) Authentication JWT token either can be stored in browser 'LocalStorage' or in 'Cookie'. Storing the JWT token inside of the cookie then the cookie should be HTTP Only. The HTTP-ONly cookie nature is that it will be only accessible by the server application. Client apps like javascript-based apps can't access the HTTP-Only cookie. So if we use the authentication with HTTP-only JWT cookie then we no need to implement the custom logic like adding authorization header or storing token data, etc at our client application. Because once the user authenticated cookie will be automatically sent to the server by the browser on every API call. Authentication API: To authenticate our client application with JWT HTTP-only cookie, I developed a NetJS(which is a node) Mock API. Check the GitHub link and read the document on G

.NET6 Web API CRUD Operation With Entity Framework Core

In this article, we are going to do a small demo on AspNetCore 6 Web API CRUD operations. What Is Web API: Web API is a framework for building HTTP services that can be accessed from any client like browser, mobile devices, desktop apps. In simple terminology API(Application Programming Interface) means an interface module that contains a programming function that can be requested via HTTP calls to save or fetch the data for their respective clients. Some of the key characteristics of API: Supports HTTP verbs like 'GET', 'POST', 'PUT', 'DELETE', etc. Supports default responses like 'XML' and 'JSON'. Also can define custom responses. Supports self-hosting or individual hosting, so that all different kinds of apps can consume it. Authentication and Authorization are easy to implement. The ideal platform to build REST full services. Create A .NET6 Web API Application: Let's create a .Net6 Web API sample application to accomplish our

Angular 14 State Management CRUD Example With NgRx(14)

In this article, we are going to implement the Angular(14) state management CRUD example with NgRx(14) NgRx Store For State Management: In an angular application to share consistent data between multiple components, we use NgRx state management. Using NgRx state helps to avoid unwanted API calls, easy to maintain consistent data, etc. The main building blocks for the NgRx store are: Actions - NgRx actions represents event to trigger the reducers to save the data into the stores. Reducer - Reducer's pure function, which is used to create a new state on data change. Store - The store is the model or entity that holds the data. Selector - Selector to fetch the slices of data from the store to angular components. Effects - Effects deals with external network calls like API. The effect gets executed based the action performed Ngrx State Management flow: The angular component needs data for binding.  So angular component calls an action that is responsible for invoking the API call.  Aft

Angular 14 Crud Example

In this article, we will implement CRUD operation in the Angular 14 application. Angular: Angular is a framework that can be used to build a single-page application. Angular applications are built with components that make our code simple and clean. Angular components compose of 3 files like TypeScript File(*.ts), Html File(*.html), CSS File(*.cs) Components typescript file and HTML file support 2-way binding which means data flow is bi-directional Component typescript file listens for all HTML events from the HTML file. Create Angular(14) Application: Let's create an Angular(14) application to begin our sample. Make sure to install the Angular CLI tool into our local machine because it provides easy CLI commands to play with the angular application. Command To Install Angular CLI npm install -g @angular/cli Run the below command to create the angular application. Command To Create Angular Application ng new name_of_your_app Note: While creating the app, you will see a noti

Unit Testing Asp.NetCore Web API Using xUnit[.NET6]

In this article, we are going to write test cases to an Asp.NetCore Web API(.NET6) application using the xUnit. xUnit For .NET: The xUnit for .Net is a free, open-source, community-focused unit testing tool for .NET applications. By default .Net also provides a xUnit project template to implement test cases. Unit test cases build upon the 'AAA' formula that means 'Arrange', 'Act' and 'Assert' Arrange - Declaring variables, objects, instantiating mocks, etc. Act - Calling or invoking the method that needs to be tested. Assert - The assert ensures that code behaves as expected means yielding expected output. Create An API And Unit Test Projects: Let's create a .Net6 Web API and xUnit sample applications to accomplish our demo. We can use either Visual Studio 2022 or Visual Studio Code(using .NET CLI commands) to create any.Net6 application. For this demo, I'm using the 'Visual Studio Code'(using the .NET CLI command) editor. Create a fo

Part-1 Angular JWT Authentication Using HTTP Only Cookie[Angular V13]

In this article, we are going to implement a sample angular application authentication using HTTP only cookie that contains a JWT token. HTTP Only JWT Cookie: In a SPA(Single Page Application) Authentication JWT token either can be stored in browser 'LocalStorage' or in 'Cookie'. Storing JWT token inside of the cookie then the cookie should be HTTP Only. The HTTP-Only cookie nature is that it will be only accessible by the server application. Client apps like javascript-based apps can't access the HTTP-Only cookie. So if we use authentication with HTTP only JWT cookie then we no need to implement custom logic like adding authorization header or storing token data, etc at our client application. Because once the user authenticated cookie will be automatically sent to the server by the browser on every API call. Authentication API: To implement JWT cookie authentication we need to set up an API. For that, I had created a mock authentication API(Using the NestJS Se

ReactJS(v18) Authentication With JWT AccessToken And Refresh Token

In this article, we are going to do ReactJS(v18) application authentication using the JWT Access Token and Refresh Token. JSON Web Token(JWT): JSON Web Token is a digitally signed and secured token for user validation. The JWT is constructed with 3 important parts: Header Payload Signature Create ReactJS Application: Let's create a ReactJS application to accomplish our demo. npx create-react-app name-of-your-app Configure React Bootstrap Library: Let's install the React Bootstrap library npm install react-bootstrap bootstrap Now add the bootstrap CSS reference in 'index.js'. src/index.js: import 'bootstrap/dist/css/bootstrap.min.css' Create A React Component 'Layout': Let's add a React component like 'Layout' in 'components/shared' folders(new folders). src/components/shared/Layout.js: import Navbar from "react-bootstrap/Navbar"; import { Container } from "react-bootstrap"; import Nav from "react-boot

A Small Guide On NestJS Queues

NestJS Application Queues helps to deal with application scaling and performance challenges. When To Use Queues?: API request that mostly involves in time taking operations like CPU bound operation, doing them synchronously which will result in thread blocking. So to avoid these issues, it is an appropriate way to make the CPU-bound operation separate background job.  In nestjs one of the best solutions for these kinds of tasks is to implement the Queues. For queueing mechanism in the nestjs application most recommended library is '@nestjs/bull'(Bull is nodejs queue library). The 'Bull' depends on Redis cache for data storage like a job. So in this queueing technique, we will create services like 'Producer' and 'Consumer'. The 'Producer' is used to push our jobs into the Redis stores. The consumer will read those jobs(eg: CPU Bound Operations) and process them. So by using this queues technique user requests processed very fastly because actually