import yapi from '../yapi.js'; import mongoose from 'mongoose'; import autoIncrement from 'mongoose-auto-increment'; /** * 所有的model都需要继承baseModel, 且需要 getSchema和getName方法,不然会报错 */ class baseModel{ constructor(){ this.schema = new mongoose.Schema(this.getSchema()); this.name = this.getName(); if(this.isNeedAutoIncrement() === true){ this.schema.plugin(autoIncrement.plugin, { model: this.name, field: this.getPrimaryKey(), startAt: 101, incrementBy: yapi.commons.rand(1, 10) }); } this.model = yapi.db(this.name, this.schema); } isNeedAutoIncrement(){ return true; } /** * 可通过覆盖此方法生成其他自增字段 */ getPrimaryKey(){ return '_id'; } /** * 获取collection的schema结构 */ getSchema(){ yapi.commons.log('Model Class need getSchema function', 'error'); } getName(){ yapi.commons.log('Model Class need name', 'error'); } } module.exports = baseModel;