我的应用程序有几个用户，每个用户都有文档。每个文档都需要有一个序列号，可能看起来像这样：2013-1、2013-2（年份和序列号），或者只是一个简单的数字：1、2、3...
目前，我在创建 Mongoose 文档时从用户设置中分配序列号。根据该序列号和用户设置的数字格式，我正在生成最终的文档编号。
我意识到，当同时创建 2 个文档时，它们将获得完全相同的编号，因为我在保存文档后在设置中增加了序列号。但是我在创建（尚未保存）文档时分配了序列号，因此两个文档的序列号将完全相同。
我显然需要一种方法来处理这个序列号在保存时自动递增......
我如何确保这个数字是唯一的并且自动递增/生成？
@emre 和 @WiredPraire 为我指明了正确的方向，但我想为我的问题提供一个完全兼容 Mongoose 的答案。我最终得到了以下解决方案：
var Settings = new Schema({
nextSeqNumber: { type: Number, default: 1 }
});
var Document = new Schema({
_userId: { type: Schema.Types.ObjectId, ref: "User" },
number: { type: String }
});
// Create a compound unique index over _userId and document number
Document.index({ "_userId": 1, "number": 1 }, { unique: true });
// I make sure this is the last pre-save middleware (just in case)
Document.pre('save', function(next) {
var doc = this;
// You have to know the settings_id, for me, I store it in memory: app.current.settings.id
Settings.findByIdAndUpdate( settings_id, { $inc: { nextSeqNumber: 1 } }, function (err, settings) {
if (err) next(err);
doc.number = settings.nextSeqNumber - 1; // substract 1 because I need the 'current' sequence number, not the next
next();
});
});
请注意，使用此方法无法要求模式中的数字路径，也没有任何意义，因为它是自动添加的。
您可以通过以下方式实现：
这是一个使用 todo 应用程序的工作和测试示例。
var mongoose = require('mongoose');
mongoose.connect('mongodb://localhost/todoApp');
// Create a sequence
function sequenceGenerator(name){
var SequenceSchema, Sequence;
SequenceSchema = new mongoose.Schema({
nextSeqNumber: { type: Number, default: 1 }
});
Sequence = mongoose.model(name + 'Seq', SequenceSchema);
return {
next: function(callback){
Sequence.find(function(err, data){
if(err){ throw(err); }
if(data.length < 1){
// create if doesn't exist create and return first
Sequence.create({}, function(err, seq){
if(err) { throw(err); }
callback(seq.nextSeqNumber);
});
} else {
// update sequence and return next
Sequence.findByIdAndUpdate(data[0]._id, { $inc: { nextSeqNumber: 1 } }, function(err, seq){
if(err) { throw(err); }
callback(seq.nextSeqNumber);
});
}
});
}
};
}
// sequence instance
var sequence = sequenceGenerator('todo');
var TodoSchema = new mongoose.Schema({
name: String,
completed: Boolean,
priority: Number,
note: { type: String, default: '' },
updated_at: { type: Date, default: Date.now }
});
TodoSchema.pre('save', function(next){
var doc = this;
// get the next sequence
sequence.next(function(nextSeq){
doc.priority = nextSeq;
next();
});
});
var Todo = mongoose.model('Todo', TodoSchema);
您可以在节点控制台中对其进行测试，如下所示
function cb(err, data){ console.log(err, data); }
Todo.create({name: 'hola'}, cb);
Todo.find(cb);
随着每个新创建的对象，您将看到优先级增加。干杯!
此代码取自MongoDB手册，它实际上描述了使 _id 字段自动递增。但是，它可以应用于任何领域。您想要的是在插入文档后检查插入的值是否存在于数据库中。如果它已经插入，重新增加值然后尝试再次插入。通过这种方式，您可以检测重复值并重新增加它们。
while (1) {
var cursor = targetCollection.find( {}, { f: 1 } ).sort( { f: -1 } ).limit(1);
var seq = cursor.hasNext() ? cursor.next().f + 1 : 1;
doc.f = seq;
targetCollection.insert(doc);
var err = db.getLastErrorObj();
if( err && err.code ) {
if( err.code == 11000 /* dup key */ )
continue;
else
print( "unexpected error inserting data: " + tojson( err ) );
}
break;
}
在此示例中， f 是文档中要自动递增的字段。要完成这项工作，您需要使您的字段 UNIQUE 可以使用索引来完成。
db.myCollection.ensureIndex( { "f": 1 }, { unique: true } )
您可以mongoose-auto-increment按如下方式使用包：
var mongoose = require('mongoose');
var autoIncrement = require('mongoose-auto-increment');
/* connect to your database here */
/* define your DocumentSchema here */
autoIncrement.initialize(mongoose.connection);
DocumentSchema.plugin(autoIncrement.plugin, 'Document');
var Document = mongoose.model('Document', DocumentSchema);
您只需要初始化autoIncrement一次。