Ghost/ghost/data-generator/lib/importers/MembersFeedbackImporter.js
Sam Lord 4ff467794f Entirely rewrote data generator to simplify codebase
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
2023-08-04 13:36:09 +01:00

46 lines
1.7 KiB
JavaScript

const TableImporter = require('./TableImporter');
const {faker} = require('@faker-js/faker');
const {luck} = require('../utils/random');
const dateToDatabaseString = require('../utils/database-date');
class MembersFeedbackImporter extends TableImporter {
static table = 'members_feedback';
static dependencies = ['emails', 'email_recipients'];
constructor(knex, transaction, {emails}) {
super(MembersFeedbackImporter.table, knex, transaction);
this.emails = emails;
}
async import(quantity) {
const emailRecipients = await this.transaction.select('id', 'opened_at', 'email_id', 'member_id').from('email_recipients');
this.emails = await this.transaction.select('id', 'post_id').from('emails');
await this.importForEach(emailRecipients, quantity ? quantity / emailRecipients.length : 1);
}
generate() {
// ~10% of people who opened the email will leave feedback
if (!this.model.opened_at || luck(90)) {
return null;
}
const openedAt = new Date(this.model.opened_at);
const laterOn = new Date(this.model.opened_at);
laterOn.setMinutes(laterOn.getMinutes() + 60);
const feedbackTime = new Date(openedAt.valueOf() + (Math.random() * (laterOn.valueOf() - openedAt.valueOf())));
const postId = this.emails.find(email => email.id === this.model.email_id).post_id;
return {
id: faker.database.mongodbObjectId(),
score: luck(70) ? 1 : 0,
member_id: this.model.member_id,
post_id: postId,
created_at: dateToDatabaseString(feedbackTime),
updated_at: dateToDatabaseString(feedbackTime)
};
}
}
module.exports = MembersFeedbackImporter;