yapi/server_dist/utils/db.js

'use strict';

var _mongoose = require('mongoose');

var _mongoose2 = _interopRequireDefault(_mongoose);

var _yapi = require('../yapi.js');

var _yapi2 = _interopRequireDefault(_yapi);

var _mongooseAutoIncrement = require('mongoose-auto-increment');

var _mongooseAutoIncrement2 = _interopRequireDefault(_mongooseAutoIncrement);

function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }

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

    schema.set('autoIndex', false);

    return _yapi2.default.connect.model(model, schema, model);
}

function connect(callback) {
    _mongoose2.default.Promise = global.Promise;

    var config = _yapi2.default.WEBCONFIG;
    var options = {};

    if (config.db.user) {
        options.user = config.db.user;
        options.pass = config.db.pass;
    }

    var db = _mongoose2.default.connect('mongodb://' + config.db.servername + ':' + config.db.port + '/' + config.db.DATABASE, options);

    db.then(function () {
        _yapi2.default.commons.log('mongodb load success...');
        if (typeof callback === 'function') {
            callback.call(db);
        }
    }, function (err) {
        _yapi2.default.commons.log(err, 'Mongo connect error');
    });

    _mongooseAutoIncrement2.default.initialize(db);
    return db;
}

_yapi2.default.db = model;

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