单个数据库文件大于500兆时,裁剪文件,避免服务无法启动

This commit is contained in:
whyour 2021-11-11 22:41:12 +08:00
parent 50dd235e39
commit 9936097086

View File

@ -1,6 +1,7 @@
import DataStore from 'nedb'; import DataStore from 'nedb';
import config from '../config'; import config from '../config';
import Logger from './logger'; import Logger from './logger';
import fs from 'fs';
interface Dbs { interface Dbs {
cronDb: DataStore; cronDb: DataStore;
@ -12,7 +13,28 @@ interface Dbs {
const db: Dbs = {} as any; const db: Dbs = {} as any;
export default () => { async function truncateDb() {
return new Promise((resolve) => {
const files = [
config.cronDbFile,
config.dependenceDbFile,
config.envDbFile,
config.appDbFile,
config.authDbFile,
];
for (const file of files) {
if (fs.statSync(file).size >= 1024 * 1024 * 500) {
fs.truncateSync(file, 1024 * 1024 * 500);
}
}
resolve(null);
});
}
export default async () => {
try {
await truncateDb();
db.cronDb = new DataStore({ filename: config.cronDbFile }); db.cronDb = new DataStore({ filename: config.cronDbFile });
db.dependenceDb = new DataStore({ filename: config.dependenceDbFile }); db.dependenceDb = new DataStore({ filename: config.dependenceDbFile });
db.envDb = new DataStore({ filename: config.envDbFile }); db.envDb = new DataStore({ filename: config.envDbFile });
@ -43,6 +65,10 @@ export default () => {
}); });
Logger.info('✌️ DB loaded'); Logger.info('✌️ DB loaded');
} catch (error) {
Logger.info('✌️ DB load failed');
Logger.info(error);
}
}; };
export const dbs: Dbs = db; export const dbs: Dbs = db;