Egg.js 中基于 Mongoose 的分页

背景

分页是 Web 开发中常见的功能,基本上所有 Web 框架都有分页的实现,或内置于框架之内,或使用插件的形式。
最近在使用 Egg.js 做后端开发,由于数据库用的是 MongoDB,所以自然选用官方的 egg-mongoose 作为 ODM Plugin,在开发过程中恰好就遇到一个分页问题。

Koa 中的实现

为了便于理解,首先先讲讲 Koa 中的实现。
使用 Koa 框架时直接基于 Mongoose 本身的 Plugin 机制实现了一个(大家也可以选择三方的实现:mongoose-paginate-v2
分页 Plugin 使用起来也很方便,在 Mongoose 初始化时加载该 Plugin 即可:

const mongoose = require('mongoose')
const config = require('config')
const paginate = require('../middlewares/paginate')

// 启用分页插件
mongoose.plugin(paginate)

mongoose.connect(config.get('db'), { useNewUrlParser: true })
mongoose.Promise = global.Promise
const db = mongoose.connection

这样全局的 Model 中都可以使用此 Plugin,即:将 paginate 扩展成 Mongoose 模型的一个方法,如:

const apps = await AppModel.paginate({}, {page})

Egg 中的问题

在 Egg 中,由于使用了 egg-mongoose 插件,所以初始化过程已经被插件本身处理过了,那么如果想让插件生效,可以直接在对应的 Model 中增加:

module.exports = app => {
    const mongoose = app.mongoose
    // 启用分页插件
    mongoose.plugin(paginate)
    const Schema = mongoose.Schema

    const schema = new Schema({
    })

    return mongoose.model('App', schema);
}

这么逻辑上是没有问题的,也能够全局生效,但是当有多个 Model时,加载 Plugin 的逻辑到底放在哪个 Model 中合适呢?
貌似放在哪个都不合适,毕竟这是个全局的设置,万一加载 Plugin 的 Model 废弃了,如果忘记将此逻辑迁移的话,就会造成一个 Bug,时间一长排查起来就比较麻烦。

那么,要是放在所有的 Model 中呢,这样不就解决问题了吗?
其实这样也不妥,太过冗余,维护起来也不方便。

所以我们还是有必要找一个全局的地方放置,尽可能保证在 App 初始化阶段就能生效,这样只要服务处于运行状态,该插件就必然生效,不会存在 paginate is not a function 的风险

解决问题

仔细看了一下 Egg 的文档,发现 Egg 支持启动时对各种生命周期的自定义(真心赞),应该可以解决我们的问题,于是在项目目录下创建 app.js 文件:

class AppBootHook {
    constructor(app) {
        this.app = app;
    }

    configWillLoad() {
        // Ready to call configDidLoad,
        // Config, plugin files are referred,
        // this is the last chance to modify the config.
    }

    configDidLoad() {
        // Config, plugin files have been loaded.
    }

    async didLoad() {
        // All files have loaded, start plugin here.
    }

    async willReady() {
        // All plugins have started, can do some thing before app ready
    }

    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.
    }
}

module.exports = AppBootHook

看着这么多生命周期,新的问题再此出现:Mongoose 的 Plugin 应该放在哪个生命周期中呢?
一个个试太不科学了,要想解决此问题,势必需要了解 egg-mongoose 的初始化过程,所以直接看源码
源码很简单,大概扫了一眼就找到如下行:

app.beforeStart(() => {
  loadModelToApp(app);
});

嗯,原来是在 beforeStart 方法中将 Model 们加载到 App 中的。
所以只要在 beforeStart 之前加载 Plugin 即可全局生效,但是上面的生命周期类貌似没有 beforeStart 方法,倒是有一个 beforeClose 方法。

仔细找了一圈文档,终于在文档最下面找到一行字:

如果你的 Egg 框架的生命周期函数是旧版本的,建议你升级到类方法模式;详情请查看升级你的生命周期事件函数

难道是新旧版本问题?带着问题继续看文档,哈,果然是因为 Egg 升级导致的:

// app.js 或 agent.js 文件:
class AppBootHook {

  constructor(app) {
    this.app = app;
  }

  async didLoad() {
    // 请将你的插件项目中 app.beforeStart 中的代码置于此处。
  }

  async willReady() {
    // 请将你的应用项目中 app.beforeStart 中的代码置于此处。
  }
}

module.exports = AppBootHook;

后面的事就简单了,为了在此之前让分页插件生效,需要将对应的逻辑放在 didLoad 之前的生命周期中,即:configDidLoad,加上相关代码

const paginate = require('./app/middleware/paginate')
class AppBootHook {
    constructor(app) {
        this.app = app;
    }

    configWillLoad() {
        // Ready to call configDidLoad,
        // Config, plugin files are referred,
        // this is the last chance to modify the config.
    }

    configDidLoad() {
        // Config, plugin files have been loaded.
        const mongoose = this.app.mongoose
        mongoose.plugin(paginate)
    }
    ...
}

module.exports = AppBootHook

大功告成

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 158,736评论 4 362
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 67,167评论 1 291
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 108,442评论 0 243
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 43,902评论 0 204
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,302评论 3 287
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,573评论 1 216
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,847评论 2 312
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,562评论 0 197
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,260评论 1 241
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,531评论 2 245
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,021评论 1 258
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,367评论 2 253
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 33,016评论 3 235
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,068评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,827评论 0 194
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,610评论 2 274
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,514评论 2 269

推荐阅读更多精彩内容