StuQ《Node.js微信开发》

mongoose模型扩展扩展

为什么要扩展mongoose模型?

我们对业务进行分层处理

service(多模型操作) -> dao(单一模型操作) -> model(模型定义)

所以我们在dao层需要很多单一模型的数据库操作方法的封装,如果业务非常复杂,比如一个超级查询方法,然后又有各种具体业务定义方法,难道我们一个一个的都写在dao层么?

事实上dao只做暴露给service的方法,而具有一定业务约定的方法是可以放到model层上做的,如果我们dao层的很多底层方法可以下沉到model,这样就可以让它们的职责更加清晰。所以,我们里有mongoose的模型扩展来重新定义一下分层

service(多模型操作) -> dao(单一模型操作) -> model(模型定义 + 扩展方法)

mongoose提供的模型模块有2种

  • statics:类上扩展
  • methods:对象上扩展

根据业务需求来确定,是通过类方法调用,还是通过实体对象上的方法来调用,然后我们根据对应业务,写mongoose相应的扩展即可。下面会分别讲解这2种扩展的差别:

方式1 statics:类上扩展

比如我们有这样一个需求,根据用户名查找用户,调用的时候是放到类上好,还是放在实例上好呢?

UserSchema.statics.find_by_username = function(username, cb) {
  return this.findOne({
    username: username
  }, cb);
};

调用

User.find_by_username(openid, cb)

方式2 methods:对象上扩展

UserSchema.methods.is_exist = function(cb) {
  var query;
  query = {
    username: this.username,
    password: this.password
  };
  return this.model('UserModel').findOne(query, cb);
};

调用

var user = new User({});
user.is_exist(cb)

测试代码

具体测试代码见db/extend/test.js

import test from 'ava';

// 1、引入`mongoose connect`
require('./connect');

// 2、引入`User` Model
const User = require('./user');

// 3、定义`user` Entity
const user = new User({
  username: 'i5ting',
  password: '0123456789',
  openid  : 'xxxxxx1'
});

const user2 = new User({
  username: 'i5ting for is_exist',
  password: '0123456789',
  openid  : 'xxxxxx2'
});

test('#find_by_username()', t => {
  user.save((err, u) => {
    t.ifError(err)

    User.find_by_username('i5ting', (err, user) => {
      t.not(err)
      t.not(user)
      t.is(user.username, 'i5ting');
      t.is(user.password, '0123456789');
    })
  });
});

test('#is_exist()', t => {
  user2.save((err, u) => {
    t.ifError(err)
    // console.log(user)
    var is_exist = u.is_exist();
    t.true(is_exist)
  });
});


test.after(t => {
  User.remove({}, (err, result) => {

  })
})

执行测试

$ cd db
$ npm run extend

> koa-db@1.0.0 extend /Users/sang/workspace/17koa/book-source/db
> ava -v extend/test.js


数据库连接成功
  ✔ #find_by_username()

  1 test passed