mirror of
https://github.com/louislam/uptime-kuma.git
synced 2024-09-07 09:51:57 +00:00
140 lines
3.8 KiB
JavaScript
140 lines
3.8 KiB
JavaScript
const { log } = require("../src/util");
|
|
const childProcess = require("child_process");
|
|
const fs = require("fs");
|
|
|
|
/**
|
|
* It is only used inside the docker container
|
|
*/
|
|
class EmbeddedMariaDB {
|
|
|
|
static instance = null;
|
|
|
|
exec = "mariadbd";
|
|
|
|
mariadbDataDir = "/app/data/mariadb";
|
|
|
|
runDir = "/app/data/run/mariadb";
|
|
|
|
socketPath = this.runDir + "/mysqld.sock";
|
|
|
|
childProcess = null;
|
|
running = false;
|
|
|
|
started = false;
|
|
|
|
/**
|
|
*
|
|
* @returns {EmbeddedMariaDB}
|
|
*/
|
|
static getInstance() {
|
|
if (!EmbeddedMariaDB.instance) {
|
|
EmbeddedMariaDB.instance = new EmbeddedMariaDB();
|
|
}
|
|
return EmbeddedMariaDB.instance;
|
|
}
|
|
|
|
static hasInstance() {
|
|
return !!EmbeddedMariaDB.instance;
|
|
}
|
|
|
|
/**
|
|
*
|
|
*/
|
|
start() {
|
|
if (this.childProcess) {
|
|
log.info("mariadb", "Already started");
|
|
return;
|
|
}
|
|
|
|
if (!fs.existsSync(this.mariadbDataDir)) {
|
|
log.info("mariadb", `Embedded MariaDB: ${this.mariadbDataDir} is not found, create one now.`);
|
|
fs.mkdirSync(this.mariadbDataDir, {
|
|
recursive: true,
|
|
});
|
|
|
|
let result = childProcess.spawnSync("mysql_install_db", [
|
|
"--user=node",
|
|
"--ldata=" + this.mariadbDataDir,
|
|
]);
|
|
|
|
if (result.status !== 0) {
|
|
let error = result.stderr.toString("utf-8");
|
|
log.error("mariadb", error);
|
|
return;
|
|
} else {
|
|
log.info("mariadb", "Embedded MariaDB: mysql_install_db done:" + result.stdout.toString("utf-8"));
|
|
}
|
|
}
|
|
|
|
if (!fs.existsSync(this.runDir)) {
|
|
log.info("mariadb", `Embedded MariaDB: ${this.runDir} is not found, create one now.`);
|
|
fs.mkdirSync(this.runDir, {
|
|
recursive: true,
|
|
});
|
|
}
|
|
|
|
this.running = true;
|
|
log.info("mariadb", "Starting Embedded MariaDB");
|
|
this.childProcess = childProcess.spawn(this.exec, [
|
|
"--user=node",
|
|
"--datadir=" + this.mariadbDataDir,
|
|
`--socket=${this.socketPath}`,
|
|
`--pid-file=${this.runDir}/mysqld.pid`,
|
|
]);
|
|
|
|
this.childProcess.on("close", (code) => {
|
|
this.running = false;
|
|
this.childProcess = null;
|
|
this.started = false;
|
|
log.info("mariadb", "Stopped Embedded MariaDB: " + code);
|
|
|
|
if (code !== 0) {
|
|
log.info("mariadb", "Try to restart Embedded MariaDB as it is not stopped by user");
|
|
this.start();
|
|
}
|
|
});
|
|
|
|
this.childProcess.on("error", (err) => {
|
|
if (err.code === "ENOENT") {
|
|
log.error("mariadb", `Embedded MariaDB: ${this.exec} is not found`);
|
|
} else {
|
|
log.error("mariadb", err);
|
|
}
|
|
});
|
|
|
|
let handler = (data) => {
|
|
log.debug("mariadb", data.toString("utf-8"));
|
|
if (data.toString("utf-8").includes("ready for connections")) {
|
|
log.info("mariadb", "Embedded MariaDB is ready for connections");
|
|
this.started = true;
|
|
}
|
|
};
|
|
|
|
this.childProcess.stdout.on("data", handler);
|
|
this.childProcess.stderr.on("data", handler);
|
|
|
|
return new Promise((resolve) => {
|
|
let interval = setInterval(() => {
|
|
if (this.started) {
|
|
clearInterval(interval);
|
|
resolve();
|
|
} else {
|
|
log.info("mariadb", "Waiting for Embedded MariaDB to start...");
|
|
}
|
|
}, 1000);
|
|
});
|
|
}
|
|
|
|
stop() {
|
|
if (this.childProcess) {
|
|
this.childProcess.kill("SIGINT");
|
|
this.childProcess = null;
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
module.exports = {
|
|
EmbeddedMariaDB,
|
|
};
|