Loader

Egg 在 Koa 的基础上进行增强最重要的就是基于一定的约定,根据功能差异将代码放到不同的目录下管理,对整体团队的开发成本提升有着明显的效果。Loader 实现了这套约定,并抽象了很多底层 API 可以进一步扩展。

# 应用、框架和插件

Egg 是一个底层框架,应用可以直接使用,但 Egg 本身的插件比较少,应用需要自己配置插件增加各种特性,比如 MySQL。

// 应用配置
// package.json
{
"dependencies": {
"egg": "^2.0.0",
"egg-mysql": "^3.0.0"
}
}

// config/plugin.js
module.exports = {
mysql: {
enable: true,
package: 'egg-mysql',
},
}

当应用达到一定数量,我们会发现大部分应用的配置都是类似的,这时可以基于 Egg 扩展出一个框架,应用的配置就会简化很多。

// 框架配置
// package.json
{
"name": "framework1",
"version": "1.0.0",
"dependencies": {
"egg-mysql": "^3.0.0",
"egg-view-nunjucks": "^2.0.0"
}
}

// config/plugin.js
module.exports = {
mysql: {
enable: false,
package: 'egg-mysql',
},
view: {
enable: false,
package: 'egg-view-nunjucks',
}
}

// 应用配置
// package.json
{
"dependencies": {
"framework1": "^1.0.0",
}
}

// config/plugin.js
module.exports = {
// 开启插件
mysql: true,
view: true,
}

从上面的使用场景可以看到应用、插件和框架三者之间的关系。

+-----------------------------------+--------+
| app1, app2, app3, app4 | |
+-----+--------------+--------------+ |
| | | framework3 | |
+ | framework1 +--------------+ plugin |
| | | framework2 | |
+ +--------------+--------------+ |
| Egg | |
+-----------------------------------+--------|
| Koa |
+-----------------------------------+--------+

# loadUnit

Egg 将应用、框架和插件都称为加载单元(loadUnit),因为在代码结构上几乎没有什么差异,下面是目录结构

loadUnit
├── package.json
├── app.js
├── agent.js
├── app
│ ├── extend
│ | ├── helper.js
│ | ├── request.js
│ | ├── response.js
│ | ├── context.js
│ | ├── application.js
│ | └── agent.js
│ ├── service
│ ├── middleware
│ └── router.js
└── config
├── config.default.js
├── config.prod.js
├── config.test.js
├── config.local.js
└── config.unittest.js

不过还存在着一些差异

文件 应用 框架 插件
package.json ✔︎ ✔︎ ✔︎
config/plugin.{env}.js ✔︎ ✔︎
config/config.{env}.js ✔︎ ✔︎ ✔︎
app/extend/application.js ✔︎ ✔︎ ✔︎
app/extend/request.js ✔︎ ✔︎ ✔︎
app/extend/response.js ✔︎ ✔︎ ✔︎
app/extend/context.js ✔︎ ✔︎ ✔︎
app/extend/helper.js ✔︎ ✔︎ ✔︎
agent.js ✔︎ ✔︎ ✔︎
app.js ✔︎ ✔︎ ✔︎
app/service ✔︎ ✔︎ ✔︎
app/middleware ✔︎ ✔︎ ✔︎
app/controller ✔︎
app/router.js ✔︎

文件按表格内的顺序自上而下加载

在加载过程中,Egg 会遍历所有的 loadUnit 加载上述的文件(应用、框架、插件各有不同),加载时有一定的优先级

比如有这样一个应用配置了如下依赖

app
| ├── plugin2 (依赖 plugin3)
| └── plugin3
└── framework1
| └── plugin1
└── egg

最终的加载顺序为

=> plugin1
=> plugin3
=> plugin2
=> egg
=> framework1
=> app

plugin1 为 framework1 依赖的插件,配置合并后 object key 的顺序会优先于 plugin2/plugin3。因为 plugin2 和 plugin3 的依赖关系,所以交换了位置。framework1 继承了 egg,顺序会晚于 egg。应用最后加载。

请查看 Loader.getLoadUnits 方法

# 文件顺序

上面已经列出了默认会加载的文件,Egg 会按如下文件顺序加载,每个文件或目录再根据 loadUnit 的顺序去加载(应用、框架、插件各有不同)。

注意

# 生命周期

Egg提供了应用启动(beforeStart), 启动完成(ready), 关闭(beforeClose)这三个生命周期方法。

   init master process

init agent worker process

loader.load | beforeStart

await agent worker ready

call ready callback

init app worker processes

loader.load | beforeStart

await app workers ready

call ready callback

send egg-ready to master,
agent,app workers

# beforeStart

beforeStart 方法在 loading 过程中调用, 所有的方法并行执行。 一般用来执行一些异步方法, 例如检查连接状态等, 比如 egg-mysql 就用 beforeStart 来检查与 mysql 的连接状态。所有的 beforeStart 任务结束后, 状态将会进入 ready 。不建议执行一些耗时较长的方法, 可能会导致应用启动超时。

# ready

ready 方法注册的任务在 load 结束并且所有的 beforeStart 方法执行结束后顺序执行, HTTP server 监听也是在这个时候开始, 此时代表所有的插件已经加载完毕并且准备工作已经完成, 一般用来执行一些启动的后置任务。

# beforeClose

beforeClose 注册方法在 app/agent 实例的 close 方法被调用后, 按注册的逆序执行。一般用于资源的释放操作, 例如 egg 用来关闭 logger , 删除监听方法等。

这个方法不建议在生产环境使用, 可能遇到未执行完就结束进程的问题。

e.g.:

// app.js
console.time('app before start 200ms');
console.time('app before start 100ms');

app.beforeStart(async () => {
await sleep(200);
console.timeEnd('app before start 200ms');
});

app.beforeStart(async () => {
await sleep(100);
console.timeEnd('app before start 100ms');
});

app.on('server', () => {
console.log('server is ready');
});

app.ready(() => {
console.log('app ready');
cp.execSync(`kill ${process.ppid}`);
console.time('app before close 100ms');
console.time('app before close 200ms');
});

app.beforeClose(async () => {
await sleep(200);
console.timeEnd('app before close 200ms');
});

app.beforeClose(async () => {
await sleep(100);
console.timeEnd('app before close 100ms');
});

// agent.js
console.time('agent before start 200ms');
console.time('agent before start 100ms');

agent.beforeStart(async () => {
await sleep(200);
console.timeEnd('agent before start 200ms');
});

agent.beforeStart(async () => {
await sleep(100);
console.timeEnd('agent before start 100ms');
});

agent.ready(() => {
console.log('agent ready');
console.time('agent before close 200ms');
console.time('agent before close 100ms');
});

agent.beforeClose(async () => {
await sleep(200);
console.timeEnd('agent before close 200ms');
});

agent.beforeClose(async () => {
await sleep(100);
console.timeEnd('agent before close 100ms');
});

print:

agent before start 100ms: 131.096ms
agent before start 200ms: 224.396ms // 并行执行

agent ready

app before start 100ms: 147.546ms
app before start 200ms: 245.405ms // 并行执行

app ready

// 开流量
server is ready

agent before close 100ms: 866.218ms
app before close 100ms: 108.007ms // LIFO, 后注册先执行
app before close 200ms: 310.549ms // 串行执行
agent before close 200ms: 1070.865ms

可以使用 egg-development 来查看加载过程。

# 文件加载规则

框架在加载文件时会进行转换,因为文件命名风格和 API 风格存在差异。我们推荐文件使用下划线,而 API 使用驼峰。比如 app/service/user_info.js 会转换成 app.service.userInfo

框架也支持连字符和驼峰的方式

Loader 还提供了 caseStyle 强制指定首字母大小写,比如加载 model 时 API 首字母大写,app/model/user.js => app.model.User,就可以指定 caseStyle: 'upper'

# 扩展 Loader

Loader 是一个基类,并根据文件加载的规则提供了一些内置的方法,但基本本身并不会去调用,而是由继承类调用。

Egg 基于 Loader 实现了 AppWorkerLoaderAgentWorkerLoader,上层框架基于这两个类来扩展,Loader 的扩展只能在框架进行

// 自定义 AppWorkerLoader
// lib/framework.js
const path = require('path');
const egg = require('egg');
const EGG_PATH = Symbol.for('egg#eggPath');

class YadanAppWorkerLoader extends egg.AppWorkerLoader {
constructor(opt) {
super(opt);
// 自定义初始化
}

loadConfig() {
super.loadConfig();
// 对 config 进行处理
}

load() {
super.load();
// 自定义加载其他目录
// 或对已加载的文件进行处理
}
}

class Application extends egg.Application {
get [EGG_PATH]() {
return path.dirname(__dirname);
}
// 覆盖 Egg 的 Loader,启动时使用这个 Loader
get [EGG_LOADER]() {
return YadanAppWorkerLoader;
}
}

module.exports = Object.assign(egg, {
Application,
// 自定义的 Loader 也需要 export,上层框架需要基于这个扩展
AppWorkerLoader: YadanAppWorkerLoader,
});

通过 Loader 提供的这些 API,可以很方便的定制团队的自定义加载,如 this.model.xxapp/extend/filter.js 等等。

以上只是说明 Loader 的写法,具体可以查看框架开发

# Loader API

Loader 还提供一些底层的 API,在扩展时可以简化代码,全部 API 请查看

# loadFile

用于加载一个文件,比如加载 app.js 就是使用这个方法。

// app/xx.js
module.exports = app => {
console.log(app.config);
};

// app.js
// 以 app/xx.js 为例,我们可以在 app.js 加载这个文件
const path = require('path');
module.exports = app => {
app.loader.loadFile(path.join(app.config.baseDir, 'app/xx.js'));
};

如果文件 export 一个函数会被调用,并将 app 作为参数,否则直接使用这个值。

# loadToApp

用于加载一个目录下的文件到 app,比如 app/controller/home.js 会加载到 app.controller.home

// app.js
// 以下只是示例,加载 controller 请用 loadController
module.exports = app => {
const directory = path.join(app.config.baseDir, 'app/controller');
app.loader.loadToApp(directory, 'controller');
};

一共有三个参数 loadToApp(directory, property, LoaderOptions)

  1. directory 可以为 String 或 Array,Loader 会从这些目录加载文件
  2. property 为 app 的属性
  3. LoaderOptions 为一些配置

# loadToContext

与 loadToApp 有一点差异,loadToContext 是加载到 ctx 上而非 app,而且是懒加载。加载时会将文件都放到一个临时对象上,在调用 ctx API 时才实例化对象。

比如 service 的加载就是使用这种模式

// 以下为示例,请使用 loadService
// app/service/user.js
const Service = require('egg').Service;
class UserService extends Service {

}
module.exports = UserService;

// app.js
// 获取所有的 loadUnit
const servicePaths = app.loader.getLoadUnits().map(unit => path.join(unit.path, 'app/service'));

app.loader.loadToContext(servicePaths, 'service', {
// service 需要继承 app.Service,所以要拿到 app 参数
// 设置 call 在加载时会调用函数返回 UserService
call: true,
// 将文件加载到 app.serviceClasses
fieldClass: 'serviceClasses',
});

文件加载后 app.serviceClasses.user 就是 UserService,当调用 ctx.service.user 时会实例化 UserService, 所以这个类只有每次请求中首次访问时才会实例化,实例化后会被缓存,同一个请求多次调用也只会实例化一次。

# LoaderOptions

# ignore [String]

ignore 可以忽略一些文件,支持 glob,默认为空

app.loader.loadToApp(directory, 'controller', {
// 忽略 app/controller/util 下的文件
ignore: 'util/**',
});

# initializer [Function]

对每个文件 export 出来的值进行处理,默认为空

// app/model/user.js
module.exports = class User {
constructor(app, path) {}
}

// 从 app/model 目录加载,加载时可做一些初始化处理
const directory = path.join(app.config.baseDir, 'app/model');
app.loader.loadToApp(directory, 'model', {
initializer(model, opt) {
// 第一个参数为 export 的对象
// 第二个参数为一个对象,只包含当前文件的路径
return new model(app, opt.path);
},
});

# caseStyle [String]

文件的转换规则,可选为 camelupperlower,默认为 camel

三者都会将文件名转换成驼峰,但是对于首字母的处理有所不同。

在加载不同文件时配置不同

文件 配置
app/controller lower
app/middleware lower
app/service lower

# override [Boolean]

遇到已经存在的文件时是直接覆盖还是抛出异常,默认为 false

比如同时加载应用和插件的 app/service/user.js 文件,如果为 true 应用会覆盖插件的,否则加载应用的文件时会报错。

在加载不同文件时配置不同

文件 配置
app/controller true
app/middleware false
app/service false

# call [Boolean]

当 export 的对象为函数时则调用,并获取返回值,默认为 true

在加载不同文件时配置不同

文件 配置
app/controller true
app/middleware false
app/service true