refactoring

Signed-off-by: Erick Wendel <erick.workspace@gmail.com>

Author: ErickWendel

src/background-task.js

@@ -8,7 +8,7 @@ process.on('message', (items) => {
     for (const item of items) {
         db.students.insert(item)
             .then(() => {
-                process.send({ status: 'done' });
+                process.send('item-done');
             })
             .catch((error) => {
                 console.error(error);

src/cluster.js

@@ -25,7 +25,10 @@ function initializeCluster({ backgroundTaskFile, clusterSize, onMessage }) {
             process.exit(1)
         })
 
-        child.on('message', onMessage)
+        child.on('message', (message) => {
+            if (message !== 'item-done') return
+            onMessage(message)
+        })
 
         processes.set(child.pid, child)
     }
@@ -39,21 +42,9 @@ function initializeCluster({ backgroundTaskFile, clusterSize, onMessage }) {
 
 }
 
-export function initialize({ backgroundTaskFile, clusterSize, amountToBeProcessed, onDone, onMessage }) {
-    let totalProcessed = 0
-    const _onMessage = (message) => {
-        onMessage(message)
-
-        // ++totalProcessed
-        // if (totalProcessed === amountToBeProcessed) {
-        //     // console.log(`all ${amountToBeProcessed} processed! Exiting...`)
-        //     onDone(totalProcessed)
-        //     // killAll()
-        // }
-    }
-
+export function initialize({ backgroundTaskFile, clusterSize, onMessage }) {
 
-    const { getProcess, killAll } = initializeCluster({ backgroundTaskFile, clusterSize, onMessage: _onMessage })
+    const { getProcess, killAll } = initializeCluster({ backgroundTaskFile, clusterSize, onMessage })
     // console.log(`starting with ${clusterSize} processes`)
 
     function sendToChild(message) {

src/index.js

@@ -40,18 +40,16 @@ const cp = initialize(
         async onMessage(message) {
             progress.increment()
 
-            ++totalProcessed
-
-            if (totalProcessed === total) {
-                // console.log(`all ${amountToBeProcessed} processed! Exiting...`)
-                progress.stop()
-                cp.killAll()
-
-                const insertedOnSQLite = await postgresDB.students.count()
-                console.log(`total on MongoDB ${total} and total on PostGres ${insertedOnSQLite}`)
-                console.log(`are the same? ${total === insertedOnSQLite ? 'yes' : 'no'}`)
-                process.exit()
-            }
+            if (++totalProcessed !== total) return
+            // console.log(`all ${amountToBeProcessed} processed! Exiting...`)
+            progress.stop()
+            cp.killAll()
+
+            const insertedOnSQLite = await postgresDB.students.count()
+            console.log(`total on MongoDB ${total} and total on PostGres ${insertedOnSQLite}`)
+            console.log(`are the same? ${total === insertedOnSQLite ? 'yes' : 'no'}`)
+            process.exit()
+
         }
     }
 )

src/seed.js

@@ -2,7 +2,7 @@ import { faker } from '@faker-js/faker';
 import { getMongoConnection, getPostgresConnection } from './db.js'
 
 async function seedMongoDB(amount) {
-    const { students, client } = getMongoConnection()
+    const { students, client } = await getMongoConnection()
     console.log('deleting all students')
     await students.deleteMany({})
     let person = []
@@ -31,5 +31,5 @@ async function seedPostegres() {
     console.log('table students created with success')
     await db.client.end()
 }
-// await seedMongoDB(1_000_000)
+await seedMongoDB(1_000_000)
 await seedPostegres()