mongodb之利用findAndModify()和nodejs实现自定义自增id

mongodb有自己的ObjectId,为什么我们还要用自己的id呢?因为自定义的id可以实现自己的业务需求,比如我有一条记录,我需要更新它的一个字段,而且更新之后的数据要排在最前面。但是这个时候objectId是不变的,即使是使用毫秒级的时间戳,也可能同时插入多条数据,让你不知道更新的先后顺序。这个时候,你就需要自定义自增长的id.

项目使用的是nodejs,nodejs的io操作基本是异步的,逻辑运算等操作是同步的,所以我思考了大概两种实现自增长的方式,一种是设一个全局的id,只要有更新操作,就先将这个id先增加,再写入,如果有重启,初始化的时候就先从数据库把这个最大的id读出来,再作为重启后更新的id使用;第二种就是用findAndModify()的方式实现。

我使用的是第二种,主要利用了findAndModify()查找和更新是原子操作的特点。

storage部分代码:

'use strict';
var _ = require('lodash');

var mongodb = require('mongodb');


var Storage = function(){
	this.db = {};
}

Storage.prototype.connect = function(opts,cb){
	var self = this;
	var client = mongodb.MongoClient;
	client.connect(opts.uri,function(err,db){
		if(err){
			return cb(err);
		}
		self.db = db;
		return cb();
	})
}

Storage.prototype.updateNewIncoming = function(tx,cb){
	var self = this;

		 self.db.collection('counters').findAndModify( 
			{_id:'txid'},	
			[['_id','asc']],
			{$inc:{seq:1}},	
			{new:true},function(err,result){
			if(err){
				console.log(err);
				return cb(err);
			}
		
			console.log('result111'+JSON.stringify(result.value.seq));
			self.db.collection('tx_income').update({
				txid:tx.id,
				type:'newIncome',
			},{
				txid:tx.id,
				type:'newIncome',
				tx:tx,
				id:result.value.seq
			},{
				w:1,
				upsert:true,
			},cb)
		});
}

业务部分代码:

var Storage = require('./storage');
var config = require('./config');
var async = require('async');

var storage = new Storage();

var opts = {
    uri : config.mongodb.uri  // 自己设置config
};

async.series([
    function(next){
        storage.connect(opts,next);
    },
        function(next){
        storage.updateNewIncoming(tx,function(){ // tx自己设置
            console.log('next2');
            return next();
        })
    },    
],function(err){
// TODO(): 业务代码自己去写            
});

大家要注意的地方在findAndModify()里边,可能写法不一样,这个跟nodejs的驱动有关,我在命令行里边用这种方式:

db.counters.findAndModify({query:{'_id':'txid'},update:{$inc:{'seq':1}},new:true})

可以实现操作,但是在代码中不行,所以大家可以去试一下。

参考 https://docs.mongodb.com/manual/reference/method/db.collection.findAndModify/#db.collection.findAndModify

 

送你几颗比特币玩玩:

https://www.coincola.app/acts/red-packet?type=invitation&packet_id=gBbVudr7GwuTF3r71ihUq8vDiuiNdAgL&lang=zh-CN

 

posted @ 2017-03-05 11:03  三尺剑  阅读(5847)  评论(0编辑  收藏  举报