113 lines
3.0 KiB
TypeScript
113 lines
3.0 KiB
TypeScript
import 'reflect-metadata'
|
||
import * as mongoose from 'mongoose';
|
||
import { Application, IBoot } from 'egg';
|
||
import { loadGmDb, loadSubDb } from '@db/index';
|
||
import { connectRedis } from '@pubUtils/redis';
|
||
|
||
export default class FooBoot implements IBoot {
|
||
private readonly app: Application;
|
||
|
||
constructor(app: Application) {
|
||
this.app = app;
|
||
}
|
||
|
||
async configWillLoad() {
|
||
// Ready to call configDidLoad,`
|
||
// Config, plugin files are referred,`
|
||
// this is the last chance to modify the config.
|
||
await this.connectDB(this.app)
|
||
await this.connectGMDB(this.app);
|
||
await this.connectSubDB(this.app);
|
||
await this.connectRedis(this.app);
|
||
|
||
this.app.config.realEnv = this.app.config.env;
|
||
if(this.app.config.env == 'local') {
|
||
this.app.config.realEnv = 'development';
|
||
}
|
||
|
||
// 如果gm使用的就是本机代理，host不转发到target而只把path替换
|
||
if(this.app.config.httpProxy && this.app.config.httpProxy[`/api/${this.app.config.realEnv}/`]) {
|
||
this.app.config.httpProxy[`/api/${this.app.config.realEnv}/`].changeOrigin = false;
|
||
}
|
||
}
|
||
|
||
configDidLoad() {
|
||
// Config, plugin files have loaded.
|
||
}
|
||
|
||
async didLoad() {
|
||
// All files have loaded, start plugin here.
|
||
}
|
||
|
||
async willReady() {
|
||
// All plugins have started, can do some thing before app ready.
|
||
// await this.customLoadModel();
|
||
}
|
||
|
||
async didReady() {
|
||
// Worker is ready, can do some things
|
||
// don't need to block the app boot.
|
||
}
|
||
|
||
async serverDidReady() {
|
||
// Server is listening.
|
||
}
|
||
|
||
async beforeClose() {
|
||
// Do some thing before app close.
|
||
}
|
||
|
||
//#region 手动挂载model，测试需要ctx.model
|
||
|
||
public async connectDB(app: Application) {
|
||
const { url, options } = app.config.mongoose
|
||
try {
|
||
if (url) {
|
||
const connection = await mongoose.connect(url, options)
|
||
console.log('******connectDB suc', url, options)
|
||
app.context.connection = connection
|
||
}
|
||
} catch(e) {
|
||
console.log(e)
|
||
}
|
||
}
|
||
|
||
public async connectGMDB(app: Application) {
|
||
const { url, options } = app.config.gmmongoose
|
||
try {
|
||
if (url) {
|
||
const connection = await mongoose.createConnection(url, options)
|
||
app.context.connectionGM = connection;
|
||
loadGmDb(connection);
|
||
console.log('******connectGMDB suc', url, options)
|
||
}
|
||
} catch(e) {
|
||
console.log(e)
|
||
}
|
||
}
|
||
|
||
public async connectSubDB(app: Application) {
|
||
const { url, options } = app.config.submongoose||app.config.mongoose
|
||
try {
|
||
if (url) {
|
||
const connection = await mongoose.createConnection(url, options)
|
||
app.context.connectionGM = connection;
|
||
loadSubDb(connection);
|
||
console.log('******connectSubDB suc', url, options)
|
||
}
|
||
} catch(e) {
|
||
console.log(e)
|
||
}
|
||
}
|
||
|
||
public async connectRedis(app: Application) {
|
||
const { url, pw } = app.config.redis
|
||
if (url) {
|
||
const redisClient = connectRedis(url, pw);
|
||
app.context.redisClient = redisClient;
|
||
}
|
||
}
|
||
//#endregion
|
||
}
|
||
|
||
module.exports = FooBoot; |