yapi/server_dist/utils/db.js

48 lines
1.3 KiB
JavaScript
Raw Normal View History

2017-07-03 16:16:05 +08:00
'use strict';
var _mongoose = require('mongoose');
var _mongoose2 = _interopRequireDefault(_mongoose);
var _yapi = require('../yapi.js');
var _yapi2 = _interopRequireDefault(_yapi);
2017-07-03 16:16:05 +08:00
var _mongooseAutoIncrement = require('mongoose-auto-increment');
var _mongooseAutoIncrement2 = _interopRequireDefault(_mongooseAutoIncrement);
2017-07-03 16:16:05 +08:00
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() {
2017-07-03 16:16:05 +08:00
_mongoose2.default.Promise = global.Promise;
var config = _yapi2.default.WEBCONFIG;
2017-07-03 16:16:05 +08:00
var db = _mongoose2.default.connect('mongodb://' + config.db.servername + ':' + config.db.port + '/' + config.db.DATABASE);
db.then(function (res) {
_yapi2.default.commons.log('mongodb load success...');
2017-07-03 16:16:05 +08:00
}, function (err) {
_yapi2.default.commons.log(err, 'Mongo connect error');
2017-07-03 16:16:05 +08:00
});
_mongooseAutoIncrement2.default.initialize(db);
2017-07-03 16:16:05 +08:00
return db;
}
_yapi2.default.db = model;
module.exports = {
model: model,
connect: connect
};