strapi/packages/strapi-database/lib/connector-registry.js

'use strict';
/**
 * Database connector registry
 */

const _ = require('lodash');
const requireConnector = require('./require-connector');

const createConnectorRegistry = ({ defaultConnection, connections }) => {
  const _connectors = new Map();

  return {
    /**
     * Load connector modules
     */
    load() {
      for (const connection of Object.values(connections)) {
        const { connector } = connection;
        if (!_connectors.has(connector)) {
          _connectors.set(connector, requireConnector(connector)(strapi));
        }
      }
    },

    /**
     * Initialize connectors
     */
    async initialize() {
      for (const connector of _connectors.values()) {
        await connector.initialize();
      }
    },

    getAll() {
      return Array.from(_connectors.values());
    },

    get(key) {
      return _connectors.get(key);
    },

    set(key, val) {
      _connectors.set(key, val);
      return this;
    },

    get default() {
      const defaultConnector = connections[defaultConnection].connector;
      return _connectors.get(defaultConnector);
    },

    getByConnection(connection) {
      if (!_.has(connections, connection)) {
        throw new Error('Trying to access a connector for an unknown connection');
      }

      const connectorKey = connections[connection].connector;
      return _connectors.get(connectorKey);
    },
  };
};

module.exports = createConnectorRegistry;