@giraphql/plugin-dataloader/esm/index.js

A GiraphQL plugin for attaching dataloader to object types

import './global-types.js';
import './field-builder.js';
import './schema-builder.js';
import SchemaBuilder, { BasePlugin, isThenable, } from '@giraphql/core';
export * from './refs/index.js';
export * from './types.js';
export * from './util.js';
const pluginName = "dataloader";
export class GiraphQLDataloaderPlugin extends BasePlugin {
    wrapResolve(resolver, fieldConfig) {
        var _a, _b;
        const isList = fieldConfig.type.kind === "List";
        const type = fieldConfig.type.kind === "List" ? fieldConfig.type.type : fieldConfig.type;
        const options = this.buildCache.getTypeConfig(type.ref)
            .giraphqlOptions;
        const getDataloader = (_a = options.extensions) === null || _a === void 0 ? void 0 : _a.getDataloader;
        if (!getDataloader) {
            return resolver;
        }
        const cacheResolved = (_b = options.extensions) === null || _b === void 0 ? void 0 : _b.cacheResolved;
        function loadIfID(idOrResult, loader) {
            if (idOrResult == null) {
                return idOrResult;
            }
            if (isThenable(idOrResult)) {
                return idOrResult.then((result) => loadIfID(result, loader));
            }
            switch (typeof idOrResult) {
                case "number":
                case "bigint":
                case "string":
                    return loader.load(idOrResult);
                default:
                    if (cacheResolved) {
                        const key = cacheResolved(idOrResult);
                        loader.prime(key, idOrResult);
                    }
                    return idOrResult;
            }
        }
        if (isList) {
            return (parent, args, context, info) => {
                const loader = getDataloader(context);
                const promiseOrResults = resolver(parent, args, context, info);
                if (isThenable(promiseOrResults)) {
                    return promiseOrResults.then((results) => results === null || results === void 0 ? void 0 : results.map((item) => loadIfID(item, loader)));
                }
                return promiseOrResults === null || promiseOrResults === void 0 ? void 0 : promiseOrResults.map((item) => loadIfID(item, loader));
            };
        }
        return (parent, args, context, info) => loadIfID(resolver(parent, args, context, info), getDataloader(context));
    }
}
SchemaBuilder.registerPlugin(pluginName, GiraphQLDataloaderPlugin);
export default pluginName;
//# sourceMappingURL=index.js.map