2021-10-09 08:33:47 -07:00
|
|
|
const { parentPort, workerData } = require("worker_threads");
|
|
|
|
const Database = require("../database");
|
|
|
|
const path = require("path");
|
|
|
|
|
2022-04-16 03:10:51 -07:00
|
|
|
/**
|
|
|
|
* Send message to parent process
|
|
|
|
* @param {any} any The message to log
|
|
|
|
*/
|
2021-10-09 08:33:47 -07:00
|
|
|
const log = function (any) {
|
|
|
|
if (parentPort) {
|
|
|
|
parentPort.postMessage(any);
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
2022-04-16 03:10:51 -07:00
|
|
|
/**
|
|
|
|
* Exit the worker process
|
|
|
|
* @param {number} error The status code to exit
|
|
|
|
*/
|
2021-10-09 08:33:47 -07:00
|
|
|
const exit = function (error) {
|
|
|
|
if (error && error != 0) {
|
|
|
|
process.exit(error);
|
|
|
|
} else {
|
|
|
|
if (parentPort) {
|
|
|
|
parentPort.postMessage("done");
|
|
|
|
} else {
|
|
|
|
process.exit(0);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
2022-04-16 03:10:51 -07:00
|
|
|
/** Connects to the database */
|
2021-10-09 08:33:47 -07:00
|
|
|
const connectDb = async function () {
|
|
|
|
const dbPath = path.join(
|
|
|
|
process.env.DATA_DIR || workerData["data-dir"] || "./data/"
|
|
|
|
);
|
|
|
|
|
|
|
|
Database.init({
|
|
|
|
"data-dir": dbPath,
|
|
|
|
});
|
|
|
|
|
|
|
|
await Database.connect();
|
|
|
|
};
|
|
|
|
|
|
|
|
module.exports = {
|
|
|
|
log,
|
|
|
|
exit,
|
|
|
|
connectDb,
|
|
|
|
};
|