修改数据库加载

This commit is contained in:
whyour
2021-11-08 22:47:46 +08:00
parent 04045fd410
commit 8559438611
9 changed files with 71 additions and 60 deletions
+3 -14
View File
@@ -10,24 +10,17 @@ import { getFileContentByName } from '../config/util';
import PQueue from 'p-queue';
import { promises, existsSync } from 'fs';
import { promisify } from 'util';
import { dbs } from '../loaders/db';
@Service()
export default class CronService {
private cronDb = new DataStore({ filename: config.cronDbFile });
private cronDb = dbs.cronDb;
private queue = new PQueue({
concurrency: parseInt(process.env.MaxConcurrentNum as string) || 5,
});
constructor(@Inject('logger') private logger: winston.Logger) {
this.cronDb.loadDatabase((err) => {
if (err) throw err;
});
}
public getDb(): DataStore {
return this.cronDb;
}
constructor(@Inject('logger') private logger: winston.Logger) {}
private isSixCron(cron: Crontab) {
const { schedule } = cron;
@@ -427,10 +420,6 @@ export default class CronService {
this.cronDb.update({}, { $set: { saved: true } }, { multi: true });
}
private reload_db() {
this.cronDb.loadDatabase();
}
public import_crontab() {
exec('crontab -l', (error, stdout, stderr) => {
var lines = stdout.split('\n');
+4 -10
View File
@@ -12,22 +12,16 @@ import {
import _ from 'lodash';
import { spawn } from 'child_process';
import SockService from './sock';
import { dbs } from '../loaders/db';
@Service()
export default class DependenceService {
private dependenceDb = new DataStore({ filename: config.dependenceDbFile });
private dependenceDb = dbs.dependenceDb;
constructor(
@Inject('logger') private logger: winston.Logger,
private sockService: SockService,
) {
this.dependenceDb.loadDatabase((err) => {
if (err) throw err;
});
}
public getDb(): DataStore {
return this.dependenceDb;
}
) {}
public async create(payloads: Dependence[]): Promise<Dependence[]> {
const tabs = payloads.map((x) => {
+3 -9
View File
@@ -6,19 +6,13 @@ import * as fs from 'fs';
import DataStore from 'nedb';
import { Env, EnvStatus, initEnvPosition } from '../data/env';
import _ from 'lodash';
import { dbs } from '../loaders/db';
@Service()
export default class EnvService {
private envDb = new DataStore({ filename: config.envDbFile });
constructor(@Inject('logger') private logger: winston.Logger) {
this.envDb.loadDatabase((err) => {
if (err) throw err;
});
}
private envDb = dbs.envDb;
public getDb(): DataStore {
return this.envDb;
}
constructor(@Inject('logger') private logger: winston.Logger) {}
public async create(payloads: Env[]): Promise<Env[]> {
const envs = await this.envs();
+3 -9
View File
@@ -5,19 +5,13 @@ import config from '../config';
import DataStore from 'nedb';
import { App } from '../data/open';
import { v4 as uuidV4 } from 'uuid';
import { dbs } from '../loaders/db';
@Service()
export default class OpenService {
private appDb = new DataStore({ filename: config.appDbFile });
constructor(@Inject('logger') private logger: winston.Logger) {
this.appDb.loadDatabase((err) => {
if (err) throw err;
});
}
private appDb = dbs.appDb;
public getDb(): DataStore {
return this.appDb;
}
constructor(@Inject('logger') private logger: winston.Logger) {}
public async findTokenByValue(token: string): Promise<App> {
return new Promise((resolve) => {
+3 -6
View File
@@ -15,22 +15,19 @@ import ScheduleService from './schedule';
import { spawn } from 'child_process';
import SockService from './sock';
import got from 'got';
import { dbs } from '../loaders/db';
@Service()
export default class UserService {
@Inject((type) => NotificationService)
private notificationService!: NotificationService;
private authDb = new DataStore({ filename: config.authDbFile });
private authDb = dbs.authDb;
constructor(
@Inject('logger') private logger: winston.Logger,
private scheduleService: ScheduleService,
private sockService: SockService,
) {
this.authDb.loadDatabase((err) => {
if (err) throw err;
});
}
) {}
public async login(
payloads: {