graphbrainz/src/extensions/fanart-tv/loader.js

import DataLoader from 'dataloader'
import LRUCache from 'lru-cache'

const debug = require('debug')('graphbrainz:extensions/fanart-tv')

export default function createLoader (options) {
  const { client } = options
  const cache = LRUCache({
    max: options.cacheSize,
    maxAge: options.cacheTTL,
    dispose (key) {
      debug(`Removed from cache. key=${key}`)
    }
  })
  // Make the cache Map-like.
  cache.delete = cache.del
  cache.clear = cache.reset

  const loader = new DataLoader(keys => {
    return Promise.all(keys.map(key => {
      const [ entityType, id ] = key
      return client.musicEntity(entityType, id)
        .catch(err => {
          if (err.statusCode === 404) {
            // 404s are OK, just return empty data.
            return {
              artistbackground: [],
              artistthumb: [],
              musiclogo: [],
              hdmusiclogo: [],
              musicbanner: [],
              musiclabel: [],
              albums: {}
            }
          }
          throw err
        }).then(body => {
          if (entityType === 'artist') {
            const releaseGroupIDs = Object.keys(body.albums)
            debug(`Priming album cache with ${releaseGroupIDs.length} album(s).`)
            releaseGroupIDs.forEach(key => loader.prime(['release-group', key], body))
          }
          return body
        })
    }))
  }, {
    cacheKeyFn: ([ entityType, id ]) => `${entityType}/${id}`,
    cacheMap: cache
  })
  return loader
}