mongoose实现mongodb的自增字段

最近帮朋友弄一个小项目,数据库最初选定了mongodb,后来在需求文档中发现对玩家的id有自增的需求。这放在mysql中是天生支持的,可是在mongodb中为了分布式部署不出现id冲突,id是由机器名、时间等共同组成的。所以这个需求我们假设是mongodb单机部署,将来也不会分片的前提下才能按照本文的方法去做。

大致的思想是创建一个公共的counter集合,它记录其他集合的id数值,其他集合需要自增id前,先原子增加counter中对应的字段,然后用返回的增加后的值赋予自增id。用mongoose实现的话,可以hook住save之前的操作,让自增透明化,关键代码如下:

const CounterSchema = dbHelper.mongoose.Schema({
    _id: { type: String, required: true },
    seq: { type: Number, default: 0 }
});

const Counter = dbHelper.mongoose.model('counter', CounterSchema);

playerSchema.pre('save', function (next) {
    let doc = this;
    Counter.findByIdAndUpdate({ _id: 'playerId' }, { $inc: { seq: 1 } }, { new: true, upsert: true }, function (error, counter) {
        if (error)
            return next(error);
        doc.id = counter.seq;
        next();
    });
});

 

posted @ 2018-04-21 07:40 doublerain 阅读(...) 评论(...) 编辑 收藏