Weishen/Hydro
2
0 Fork 0
mirror of https://gitcode.com/gh_mirrors/hy/Hydro.git synced 2026-10-17 09:36:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

core: fix backup & restore

Browse Source
This commit is contained in:
undefined committed 2021-12-02 21:47:17 +08:00
1 parent a0734a1030
commit 2ef9daf37a
2 files changed
+16 -10
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

packages/hydrooj/bin/hydrooj.js
+15 -9
View File
@@ -23,6 +23,12 @@ const argv = require('cac')().parse();
const child = require('child_process');
const esbuild = require('esbuild');
const exec = (...args) => {
const res = child.spawnSync(...args);
if (res.error) throw res.error;
return res;
};
if (!process.env.NODE_APP_INSTANCE) process.env.NODE_APP_INSTANCE = '0';
const major = +process.version.split('.')[0].split('v')[1];
const minor = +process.version.split('.')[1];
@@ -64,7 +70,7 @@ require.extensions['.ts'] = require.extensions['.tsx'] = function loader(module,
function buildUrl(opts) {
let mongourl = `${opts.protocol || 'mongodb'}://`;
if (opts.username) mongourl += `${opts.username}:${opts.password}@`;
mongourl += `${opts.host}:${opts.port}/${opts.name}`;
mongourl += `${opts.host}:${opts.port}/${opts.db}`;
if (opts.url) mongourl = opts.url;
return mongourl;
}
@@ -85,15 +91,15 @@ if (argv.args[0] === 'backup') {
const dbConfig = fs.readFileSync(path.resolve(hydroPath, 'config.json'), 'utf-8');
const url = buildUrl(JSON.parse(dbConfig));
const dir = `${os.tmpdir()}/${Math.random().toString(36).substring(2)}`;
child.spawnSync('mongodump', [url, `--out=${dir}/dump`], { stdio: 'inherit' });
exec('mongodump', [url, `--out=${dir}/dump`], { stdio: 'inherit' });
const env = `${os.homedir()}/.hydro/env`;
if (fs.existsSync(env)) fs.copySync(env, `${dir}/env`);
const target = `${process.cwd()}/backup-${new Date().toISOString()}.zip`;
child.spawnSync('zip', ['-r', target, 'dump'], { cwd: dir, stdio: 'inherit' });
exec('zip', ['-r', target, 'dump'], { cwd: dir, stdio: 'inherit' });
if (!argv.options.dbOnly) {
child.spawnSync('zip', ['-r', target, 'file'], { cwd: '/data', stdio: 'inherit' });
exec('zip', ['-r', target, 'file'], { cwd: '/data', stdio: 'inherit' });
}
child.spawnSync('rm', ['-rf', dir]);
exec('rm', ['-rf', dir]);
console.log(`Database backup saved at ${target}`);
return;
}
@@ -106,11 +112,11 @@ if (argv.args[0] === 'restore') {
console.error('Cannot find file');
return;
}
child.spawnSync('unzip', [argv.args[1], '-d', dir], { stdio: 'inherit' });
child.spawnSync('mongorestore', [url, `${dir}/dump`, '--drop'], { stdio: 'inherit' });
exec('unzip', [argv.args[1], '-d', dir], { stdio: 'inherit' });
exec('mongorestore', [`--uri=${url}`, `--dir=${dir}/dump/${JSON.parse(dbConfig).db}`, '--drop'], { stdio: 'inherit' });
if (fs.existsSync(`${dir}/file`)) {
child.spawnSync('rm', ['-rf', '/data/file/**'], { stdio: 'inherit' });
child.spawnSync('mv', ['-f', `${dir}/file/**`, '/data/file'], { stdio: 'inherit' });
exec('rm', ['-rf', '/data/file/*'], { stdio: 'inherit' });
exec('mv', ['-f', `${dir}/file/*`, '/data/file'], { stdio: 'inherit' });
}
if (fs.existsSync(`${dir}/env`)) {
fs.copySync(`${dir}/env`, `${os.homedir()}/.hydro/env`, { overwrite: true });