graphbrainz/lib/loaders.js

'use strict';

Object.defineProperty(exports, "__esModule", {
  value: true
});

var _toConsumableArray2 = require('babel-runtime/helpers/toConsumableArray');

var _toConsumableArray3 = _interopRequireDefault(_toConsumableArray2);

var _slicedToArray2 = require('babel-runtime/helpers/slicedToArray');

var _slicedToArray3 = _interopRequireDefault(_slicedToArray2);

var _promise = require('babel-runtime/core-js/promise');

var _promise2 = _interopRequireDefault(_promise);

exports.default = createLoaders;

var _dataloader = require('dataloader');

var _dataloader2 = _interopRequireDefault(_dataloader);

var _lruCache = require('lru-cache');

var _lruCache2 = _interopRequireDefault(_lruCache);

var _helpers = require('./types/helpers');

var _util = require('./util');

function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }

var debug = require('debug')('graphbrainz:loaders');

function createLoaders(client) {
  // All loaders share a single LRU cache that will remember 8192 responses,
  // each cached for 1 day.
  var cache = (0, _lruCache2.default)({
    max: parseInt(process.env.GRAPHBRAINZ_CACHE_SIZE || 8192, 10),
    maxAge: parseInt(process.env.GRAPHBRAINZ_CACHE_TTL || _util.ONE_DAY, 10),
    dispose: function dispose(key) {
      debug('Removed from cache. key=' + key);
    }
  });
  // Make the cache Map-like.
  cache.delete = cache.del;
  cache.clear = cache.reset;

  var lookup = new _dataloader2.default(function (keys) {
    return _promise2.default.all(keys.map(function (key) {
      var _key = (0, _slicedToArray3.default)(key, 3),
          entityType = _key[0],
          id = _key[1],
          _key$ = _key[2],
          params = _key$ === undefined ? {} : _key$;

      return client.lookup(entityType, id, params).then(function (entity) {
        if (entity) {
          // Store the entity type so we can determine what type of object this
          // is elsewhere in the code.
          entity._type = entityType;
        }
        return entity;
      });
    }));
  }, {
    cacheKeyFn: function cacheKeyFn(key) {
      return client.getLookupURL.apply(client, (0, _toConsumableArray3.default)(key));
    },
    cacheMap: cache
  });

  var browse = new _dataloader2.default(function (keys) {
    return _promise2.default.all(keys.map(function (key) {
      var _key2 = (0, _slicedToArray3.default)(key, 2),
          entityType = _key2[0],
          _key2$ = _key2[1],
          params = _key2$ === undefined ? {} : _key2$;

      return client.browse(entityType, params).then(function (list) {
        list[(0, _helpers.toPlural)(entityType)].forEach(function (entity) {
          // Store the entity type so we can determine what type of object this
          // is elsewhere in the code.
          entity._type = entityType;
        });
        return list;
      });
    }));
  }, {
    cacheKeyFn: function cacheKeyFn(key) {
      return client.getBrowseURL.apply(client, (0, _toConsumableArray3.default)(key));
    },
    cacheMap: cache
  });

  var search = new _dataloader2.default(function (keys) {
    return _promise2.default.all(keys.map(function (key) {
      var _key3 = (0, _slicedToArray3.default)(key, 3),
          entityType = _key3[0],
          query = _key3[1],
          _key3$ = _key3[2],
          params = _key3$ === undefined ? {} : _key3$;

      return client.search(entityType, query, params).then(function (list) {
        list[(0, _helpers.toPlural)(entityType)].forEach(function (entity) {
          // Store the entity type so we can determine what type of object this
          // is elsewhere in the code.
          entity._type = entityType;
        });
        return list;
      });
    }));
  }, {
    cacheKeyFn: function cacheKeyFn(key) {
      return client.getSearchURL.apply(client, (0, _toConsumableArray3.default)(key));
    },
    cacheMap: cache
  });

  return { lookup: lookup, browse: browse, search: search };
}