123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206 |
- /**
- * mongodb Helper
- *
- * @author caiaolin
- * @date 2017/5/22.
- */
- import mongoose from "mongoose";
- class MongooseHelper {
- /**
- * mongoose连接对象
- *
- * @var {object}
- */
- connect = null;
- /**
- * mongoose数据模型
- *
- * @var {object}
- */
- model = null;
- /**
- * 查找单一数据
- *
- * @param {object} conditions
- * @param {object} fields
- * @return {Promise}
- */
- findOne(conditions, fields = null) {
- let self = this;
- return new Promise(function (resolve, reject) {
- self.model.findOne(conditions, fields, function (error, data) {
- if (error) {
- reject(null);
- } else {
- resolve(data);
- }
- });
- });
- }
- /**
- * 查找数据
- *
- * @param {object} conditions
- * @param {object} fields
- * @param {object} option
- * @return {Promise}
- */
- find(conditions, fields = null, option = null) {
- let self = this;
- return new Promise(function (resolve, reject) {
- self.model.find(conditions, fields, option, function (error, data) {
- if (error) {
- reject(error);
- } else {
- resolve(data);
- }
- });
- });
- }
- /**
- * 关联查找数据
- *
- * @param {object} conditions
- * @param {object} fields
- * @param {String|Object} populate
- * @return {Promise}
- */
- findWithPopulate(conditions, fields = null, populate = '') {
- let self = this;
- return new Promise(function (resolve, reject) {
- self.model.find(conditions, fields).populate(populate).exec(function(error, data) {
- if (error) {
- reject(error);
- } else {
- resolve(data);
- }
- });
- });
- }
- /**
- * 查找且更新(原子操作)
- *
- * @param {object} update
- * @param {object} condition
- * @param {object} options
- * @return {Promise}
- */
- findAndModify(condition, update, options) {
- let self = this;
- return new Promise(function (resolve, reject) {
- self.model.findOneAndUpdate(condition, update, options, function(error, data) {
- if (error) {
- reject(error);
- } else {
- resolve(data);
- }
- });
- });
- }
- /**
- * 新增操作
- *
- * @param {object} data
- * @return {Promise}
- */
- create(data) {
- let self = this;
- return new Promise(function (resolve, reject) {
- self.model.create(data, function(error, data) {
- if (error) {
- reject(error);
- } else {
- resolve(data);
- }
- });
- });
- }
- /**
- * 统计数据数量
- *
- * @param {Object} condition
- * @return {Promise}
- */
- count(condition) {
- let self = this;
- return new Promise(function(resolve, reject) {
- self.model.count(condition, function(error, data) {
- if (error) {
- reject(error);
- } else {
- resolve(data);
- }
- });
- });
- }
- /**
- * 删除数据
- *
- * @param {Object} condition
- * @return {Promise}
- */
- delete(condition) {
- let self = this;
- return new Promise(function(resolve, reject) {
- condition = self._convertId(condition);
- self.model.remove(condition, function(error, data) {
- if (error) {
- reject(error);
- } else {
- resolve(data);
- }
- });
- });
- }
- /**
- * 更新数据
- *
- * @param {Object} condition
- * @param {Object} updateData
- * @return {Promise}
- */
- update(condition, updateData) {
- let self = this;
- return new Promise(function(resolve, reject) {
- condition = self._convertId(condition);
- self.model.update(condition, {$set: updateData}, function(error, data) {
- if (error) {
- reject(error);
- } else {
- resolve(data);
- }
- });
- });
- }
- /**
- * id转换为objectId
- *
- * @param {object} condition
- * @return {object}
- */
- _convertId(condition) {
- // 对于ID的处理
- if (condition === null || condition._id === undefined) {
- return condition;
- }
- let result = mongoose.Types.ObjectId(condition._id);
- condition._id = result;
- return condition;
- }
- }
- export default MongooseHelper;
|