DataLoaderX

DataLoaderX is an expansion upon DataLoader, designed to abstract some details away, and add common functionality.

This library provides 2 classes to work with: ObjectDataLoader and ArrayDataLoader.

• ObjectDataLoader is for use cases where you wish to load a single entity for each provided ID (e.g. load user by ID)
• ArrayDataLoader is for use cases where you wish to load multiple entities for each provided ID (e.g. load all books by author ID)

Both classes require a context to be passed to them, which is any object which will persist for the duration of the request that is currently being processed. This allows dataloader instances to be cached for the duration of the request, as recommended by the base dataloader library.

ObjectDataLoader

const idLoader = new ObjectDataLoader(
  async (ids: number[]) => getUsersByIds(ids),
  result => result.id
);

await idLoader.getDataLoader(context).load(1); // returns the user with an ID of 1, or null if not found

ArrayDataLoader

const authorIdLoader = new ArrayDataLoader(
  async (ids: number[]) => getBooksByAuthorIds(ids),
  result => result.authorId
);

await authorIdLoader.getDataLoader(context).load(2); // returns an array of all books with an authorId of 2

Hooks

There are 2 hooks available; one runs before the loading occurs, and one runs after the loading has finished. This allows for implementing functionality such as timing how long the loading takes, implementing tracing, etc.

Additionally, a value can optionally be passed from the before-load hook to the after-load hook.

const idLoader = new ObjectDataLoader(
  async (ids: number[]) => getUsersByIds(ids),
  result => result.id,
  {
    hooks: {
      beforeLoad: () => {
        const start = Date.now();
        return start;
      },
      afterLoad: (start: number) => {
        const end = Date.now();
        const time = end - start;
        console.log(`Loading users took ${time}ms`);
      }
    }
  }
);