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

A GiraphQL plugin for attaching dataloader to object types

"use strict";
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
    if (k2 === undefined) k2 = k;
    Object.defineProperty(o, k2, { enumerable: true, get: function() { return m[k]; } });
}) : (function(o, m, k, k2) {
    if (k2 === undefined) k2 = k;
    o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
    Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
    o["default"] = v;
});
var __importStar = (this && this.__importStar) || function (mod) {
    if (mod && mod.__esModule) return mod;
    var result = {};
    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
    __setModuleDefault(result, mod);
    return result;
};
var __exportStar = (this && this.__exportStar) || function(m, exports) {
    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.GiraphQLDataloaderPlugin = void 0;
require("./global-types");
require("./field-builder");
require("./schema-builder");
const core_1 = __importStar(require("@giraphql/core"));
__exportStar(require("./refs"), exports);
__exportStar(require("./types"), exports);
__exportStar(require("./util"), exports);
const pluginName = 'dataloader';
class GiraphQLDataloaderPlugin extends core_1.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 ((0, core_1.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 ((0, core_1.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));
    }
}
exports.GiraphQLDataloaderPlugin = GiraphQLDataloaderPlugin;
core_1.default.registerPlugin(pluginName, GiraphQLDataloaderPlugin);
exports.default = pluginName;
//# sourceMappingURL=index.js.map