TryGhost/Ghost

View on GitHub
ghost/data-generator/lib/utils/JsonImporter.js

Summary

Maintainability
A
0 mins
Test Coverage
const TableImporter = require('../importers/TableImporter');

class JsonImporter extends TableImporter {
    constructor(knex, transaction) {
        super();
        this.knex = knex;
        this.transaction = transaction;
    }

    /**
     * @typedef {Object} JsonImportOptions
     * @property {string} name Name of the table to import
     * @property {Object} data Models without ids to be imported
     * @property {Array<string>} [rows] Set of rows to be returned
     */

    /**
     * Import a dataset to the database
     * @param {JsonImportOptions} options
     * @returns {Promise}
     */
    async import({
        name,
        data,
        rows = []
    }) {
        for (const obj of data) {
            if (!('id' in obj)) {
                obj.id = this.fastFakeObjectId();
            }
        }
        if (rows.findIndex(row => row === 'id') === -1) {
            rows.unshift('id');
        }
        await this.knex.batchInsert(name, data, 500).transacting(this.transaction);
    }
}

module.exports = JsonImporter;