GraphQL Data Loader Simplified

batchLoadData.ts

import DataLoader = require('dataloader');

/**
 * Speciallized function that can utilize batching the data basing on 
 * DataLoader library
 * 
 * @param context - resolver context object that will be used to apply new loader
 * @param name - name of the object we want to load
 * @param id - id of the object we want to load 
 */
const batchLoadData = (name: string, id: string, context: any) => {
    const keyName = `${name}DataLoader`;
    if (!context[keyName]) {
        context[keyName] = new DataLoader<string, any>((keys: string[]) => {
            return this.db.batchRead(name, keys);
        });
    }
    return context[keyName.load(id);  
}

// Can be used in resolver as follows
batchLoadData('test', 3, context)