fix mongo connection while backup

This commit is contained in:
cawa 2023-06-06 15:53:03 +03:00
parent abee37a52c
commit 85fc6b6a86

View File

@ -11,7 +11,7 @@ const getDirectories = async source =>
.filter(dirent => dirent.isDirectory()) .filter(dirent => dirent.isDirectory())
.map(dirent => dirent.name); .map(dirent => dirent.name);
async function performBackup(databaseName) { async function performBackup(databaseName, uri) {
try { try {
const timestamp = moment().format("DD.MM.YYYY_HHmmss"); const timestamp = moment().format("DD.MM.YYYY_HHmmss");
const { backupFolder } = await prompt([ const { backupFolder } = await prompt([
@ -24,37 +24,64 @@ async function performBackup(databaseName) {
]); ]);
const folderName = "dumps/" + backupFolder; const folderName = "dumps/" + backupFolder;
const backupCommand = `mongodump --db ${databaseName} --out ${folderName}`;
console.log(`Creating backup of database ${databaseName}...`); let backupCommand = spawn("mongodump", [
await exec(backupCommand); "--authenticationDatabase=admin",
"--uri",
uri,
"--db",
databaseName,
"--out",
folderName,
]);
console.log(`Backup created successfully in folder ${folderName}.`); backupCommand.stdout.on("data", function (data) {
console.log(data.toString());
});
backupCommand.stderr.on("data", function (data) {
console.log(data.toString());
});
const waitForEnd = new Promise((resolve, reject) => {
backupCommand.on("exit", function (code) {
console.log("child process exited with code " + code.toString());
console.log(`Search you backup in folder ${folderName}.`);
resolve();
});
});
await waitForEnd;
return;
} catch (error) { } catch (error) {
console.error("An error occurred during the backup process:", error); console.error("An error occurred during the backup process:", error);
return;
} }
} }
function performRestore(restoreFolder, uri) { async function performRestore(restoreFolder, uri) {
try { try {
console.log(`Restoring database from folder ${restoreFolder}...`); console.log(`Restoring database from folder ${restoreFolder}...`);
let ls = spawn("mongorestore", ["--uri", uri, "--drop", restoreFolder]); let restoreCommand = spawn("mongorestore", ["--uri", uri, "--drop", restoreFolder]);
ls.stdout.on("data", function (data) { restoreCommand.stdout.on("data", function (data) {
console.log(data.toString()); console.log(data.toString());
}); });
ls.stderr.on("data", function (data) { restoreCommand.stderr.on("data", function (data) {
console.log(data.toString()); console.log(data.toString());
}); });
ls.on("exit", function (code) { const waitForEnd = new Promise((resolve, reject) => {
restoreCommand.on("exit", function (code) {
console.log("child process exited with code " + code.toString()); console.log("child process exited with code " + code.toString());
resolve();
}); });
});
console.log("Database restored successfully."); await waitForEnd;
return;
} catch (error) { } catch (error) {
console.error("An error occurred during the restore process:", error); console.error("An error occurred during the restore process:", error);
return;
} }
} }
@ -161,6 +188,7 @@ async function main() {
await client.connect(); await client.connect();
console.log("Connected to MongoDB successfully."); console.log("Connected to MongoDB successfully.");
while (true) {
const { action } = await prompt([ const { action } = await prompt([
{ {
type: "list", type: "list",
@ -184,7 +212,7 @@ async function main() {
}, },
]); ]);
await performBackup(databaseName); await performBackup(databaseName, uri);
} else if (action === "Restore") { } else if (action === "Restore") {
try { try {
const dirs = await getDirectories("./dumps"); const dirs = await getDirectories("./dumps");
@ -230,6 +258,8 @@ async function main() {
await client.close(); await client.close();
console.log("Disconnected from MongoDB."); console.log("Disconnected from MongoDB.");
} }
console.log("\nPress Ctrl+C to exit\n");
}
} catch (error) { } catch (error) {
console.error("An error occurred:", error); console.error("An error occurred:", error);
} }