@nestjs/typeorm/dist/entities-metadata.storage.js

Nest - modern, fast, powerful node.js web framework (@typeorm)

export class EntitiesMetadataStorage {
    static storage = new Map();
    static addEntitiesByDataSource(dataSource, entities) {
        const dataSourceToken = typeof dataSource === 'string'
            ? dataSource
            : dataSource.name;
        if (!dataSourceToken) {
            return;
        }
        let collection = this.storage.get(dataSourceToken);
        if (!collection) {
            collection = [];
            this.storage.set(dataSourceToken, collection);
        }
        entities.forEach((entity) => {
            if (collection.includes(entity)) {
                return;
            }
            collection.push(entity);
        });
    }
    static getEntitiesByDataSource(dataSource) {
        const dataSourceToken = typeof dataSource === 'string'
            ? dataSource
            : dataSource.name;
        if (!dataSourceToken) {
            return [];
        }
        return this.storage.get(dataSourceToken) || [];
    }
}