mirror of
https://github.com/TryGhost/Ghost.git
synced 2024-12-29 22:01:49 +03:00
08250d44b4
refs: https://github.com/TryGhost/Toolbox/issues/453 This makes it so that a JSON bundle can be imported as well as the data generation script
39 lines
1.1 KiB
JavaScript
39 lines
1.1 KiB
JavaScript
const {faker} = require('@faker-js/faker');
|
|
|
|
class JsonImporter {
|
|
constructor(knex) {
|
|
this.knex = knex;
|
|
}
|
|
|
|
/**
|
|
* @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<Array<Object.<string, any>>>} Set of rows returned from database
|
|
*/
|
|
async import({
|
|
name,
|
|
data,
|
|
rows = []
|
|
}) {
|
|
for (const obj of data) {
|
|
if (!('id' in obj)) {
|
|
obj.id = faker.database.mongodbObjectId();
|
|
}
|
|
}
|
|
if (rows.findIndex(row => row === 'id') === -1) {
|
|
rows.unshift('id');
|
|
}
|
|
await this.knex.batchInsert(name, data, 500);
|
|
return await this.knex.select(...rows).whereIn('id', data.map(obj => obj.id)).from(name);
|
|
}
|
|
}
|
|
|
|
module.exports = JsonImporter;
|