yapi/server/utils/db.js

import mongoose from 'mongoose'
import yapi from '../yapi.js'
import autoIncrement from 'mongoose-auto-increment'

function model(model, schema){
    if(schema instanceof mongoose.Schema === false){
        schema = new mongoose.Schema(schema);
    }

    
    schema.set('autoIndex', false);
    return yapi.connect.model(model, schema, model)
}

function connect(){
    mongoose.Promise = global.Promise;
    let config = yapi.WEBCONFIG;

    let db = mongoose.connect(`mongodb://${config.db.servername}:${config.db.port}/${config.db.DATABASE}`);

    db.then(function (res) {
        yapi.commons.log('mongodb load success...')
    }, function (err) {
        yapi.commons.log(err, 'Mongo connect error');
    })

    autoIncrement.initialize(db);

    checkDatabase();
    return db;
}

function checkDatabase(){
    let exist = yapi.commons.fileExist(yapi.path.join(yapi.WEBROOT_RUNTIME, 'init.lock'))
    if(!exist){
        yapi.commons.log('lock is not exist')
    }
}

yapi.db = model;


module.exports =   {
    model: model,
    connect: connect
};