chore: make seed more dynamic

Author: alextanhongpin

scripts/mysql/index.ts

@@ -10,8 +10,8 @@ const connection = await mysql.createConnection({
 
 connection.connect();
 
-const service = createService(connection);
+const service = createService(connection, "accounts");
 await service.migrate();
-console.log(await service.seed(2000));
+console.log(await service.seed(1_000_000));
 
 connection.end();

scripts/mysql/service.ts

@@ -1,33 +1,57 @@
 import { faker } from "@faker-js/faker";
 
-class Service {
-  constructor(db) {
-    this.db = db;
-  }
+const USERS = "users";
+const ACCOUNTS = "accounts";
 
-  async count() {
-    const [rows] = await this.db.query("select 1 + 1 as solution");
-    return rows;
-  }
+class Service {
+  MAX_BATCH_SIZE = 1000;
 
-  async migrate() {
-    return this.db.execute(
-      `create table if not exists users (
+  sql = {
+    [USERS]: `
+      create table if not exists users (
         id int auto_increment,
         name varchar(80) not null,
         email varchar(80) not null,
         primary key (id),
         unique(email)
-      )`
-    );
+      )
+    `,
+    [ACCOUNTS]: `
+      create table if not exists accounts (
+        id int auto_increment,
+        name varchar(80) not null,
+        email varchar(80) not null,
+        created_at timestamp not null default now(),
+        updated_at timestamp not null default now(),
+        primary key (id)
+      )
+    `,
+  };
+
+  constructor(db, ns) {
+    if (!this.sql[ns]) {
+      throw new Error(`unknown namespace: ${ns}`);
+    }
+    this.db = db;
+    this.ns = ns;
+    this.seeder = {
+      [USERS]: (n) => this.seedUsers(n),
+      [ACCOUNTS]: (n) => this.seedAccounts(n),
+    };
+  }
+
+  async migrate() {
+    const sql = this.sql[this.ns];
+    return this.db.execute(sql);
   }
 
   async seed(n = 1_000) {
-    const sizes = batch(n, 1000);
-    return Promise.all(sizes.flatMap((n) => this._seed(n)));
+    const fn = this.seeder[this.ns];
+    const sizes = batch(n, this.MAX_BATCH_SIZE);
+    return Promise.all(sizes.flatMap((n) => fn(n)));
   }
 
-  async _seed(n) {
+  async seedUsers(n) {
     const names = Array.from({ length: n }).map(() => [
       faker.person.fullName(),
       faker.internet.email(),
@@ -39,6 +63,19 @@ class Service {
       [names]
     );
   }
+
+  async seedAccounts(n) {
+    const names = Array.from({ length: n }).map(() => [
+      faker.person.fullName(),
+      faker.internet.email(),
+    ]);
+    return this.db.query(
+      `
+      insert into accounts(name, email) values ?
+    `,
+      [names]
+    );
+  }
 }
 function batch(n, size) {
   const result = [];
@@ -49,6 +86,6 @@ function batch(n, size) {
   return result;
 }
 
-export default function createService(conn) {
-  return new Service(conn);
+export default function createService(conn, ns) {
+  return new Service(conn, ns);
 }