4ff467794f
refs: https://github.com/TryGhost/DevOps/issues/11 This is a pretty huge commit, but the relevant points are: * Each importer no longer needs to be passed a set of data, it just gets the data it needs * Each importer specifies its dependencies, so that the order of import can be determined at runtime using a topological sort * The main data generator function can just tell each importer to import the data it has This makes working on the data generator much easier. Some other benefits are: * Batched importing, massively speeding up the whole process * `--tables` to set the exact tables you want to import, and specify the quantity of each
40 lines
1.2 KiB
JavaScript
40 lines
1.2 KiB
JavaScript
const TableImporter = require('./TableImporter');
|
|
const {faker} = require('@faker-js/faker');
|
|
const {luck} = require('../utils/random');
|
|
|
|
class MembersLabelsImporter extends TableImporter {
|
|
static table = 'members_labels';
|
|
static dependencies = ['labels', 'members'];
|
|
|
|
constructor(knex, transaction, {labels}) {
|
|
super(MembersLabelsImporter.table, knex, transaction);
|
|
this.labels = labels;
|
|
}
|
|
|
|
async import(quantity) {
|
|
const members = await this.transaction.select('id').from('members');
|
|
this.labels = await this.transaction.select('id').from('labels');
|
|
|
|
await this.importForEach(members, quantity ? quantity / members.length : 1);
|
|
}
|
|
|
|
generate() {
|
|
if (luck(90)) {
|
|
// 90% of members don't have labels
|
|
return;
|
|
}
|
|
// TODO: Ensure we don't generate the same member label twice
|
|
return {
|
|
id: faker.database.mongodbObjectId(),
|
|
member_id: this.model.id,
|
|
label_id: this.labels[faker.datatype.number({
|
|
min: 0,
|
|
max: this.labels.length - 1
|
|
})].id,
|
|
sort_order: 0
|
|
};
|
|
}
|
|
}
|
|
|
|
module.exports = MembersLabelsImporter;
|