在mongoose中自动增加id



如何在猫鼬中自动增加id ?我希望我的id以1 2 3 4开头,而不是mongodb为你创建的奇怪id数字?

这是我的schema:

var PortfolioSchema = mongoose.Schema({
    url: String,
    createTime: { type: Date, default: Date.now },
    updateTime: { type: Date, default: Date.now },
    user: {type: Schema.Types.ObjectId, ref: 'User'}
});

使用mongoose-auto-increment:https://github.com/codetunnel/mongoose-auto-increment

var mongoose = require('mongoose');
var autoIncrement = require('mongoose-auto-increment');
var connection = ....;
autoIncrement.initialize(connection);
var PortfolioSchema = new mongoose.Schema({
    url: String,
    createTime: { type: Date, default: Date.now },
    updateTime: { type: Date, default: Date.now },
    user: {type: Schema.Types.ObjectId, ref: 'User'}
});
//Auto-increment
PortfolioSchema.plugin(autoIncrement.plugin, { model: 'Portfolio' });
module.exports = mongoose.model('Portfolio', PortfolioSchema);

或者,如果您喜欢使用额外的字段而不是覆盖_id,只需添加字段并在自动增量初始化中列出它:

var PortfolioSchema = new mongoose.Schema({
    portfolioId: {type: Number, required: true},
    url: String,
    createTime: { type: Date, default: Date.now },
    updateTime: { type: Date, default: Date.now },
    user: {type: Schema.Types.ObjectId, ref: 'User'}
});
//Auto-increment
PortfolioSchema.plugin(autoIncrement.plugin, { model: 'Portfolio', field: 'portfolioId' });

如果你想在_id中有一个递增的数值,那么基本的过程是你需要一些东西从某个地方的存储返回该值。一种方法是使用MongoDB本身来存储保存每个集合的_id值的计数器的数据,这在手册本身中创建和自动递增序列字段中进行了描述。

然后,当您创建每个新项时,您使用实现的函数来获取该"计数器"值,并将其用作文档中的_id

当覆盖这里的默认行为时,mongoose要求您明确指定_id和它的类型,例如_id: Number,并且您告诉它不再自动尝试提供ObjectId类型,{ "_id": false }作为模式上的选项。

下面是一个实际的例子:

var async = require('async'),
    mongoose = require('mongoose'),
    Schema = mongoose.Schema;
mongoose.connect('mongodb://localhost/test');
var counterSchema = new Schema({
  "_id": String,
  "counter": { "type": Number, "default": 1 }
},{ "_id": false });
counterSchema.statics.getNewId = function(key,callback) {
  return this.findByIdAndUpdate(key,
    { "$inc": { "counter": 1 } },
    { "upsert": true, "new": true },
    callback
  );
};
var sampleSchema = new Schema({
  "_id": Number,
  "name": String
},{ "_id": false });
var Counter = mongoose.model( 'Counter', counterSchema ),
    ModelA = mongoose.model( 'ModelA', sampleSchema ),
    ModelB = mongoose.model( 'ModelB', sampleSchema );

async.series(
  [
    function(callback) {
      async.each([Counter,ModelA,ModelB],function(model,callback) {
        model.remove({},callback);
      },callback);
    },
    function(callback) {
      async.eachSeries(
        [
          { "model": "ModelA", "name": "bill" },
          { "model": "ModelB", "name": "apple" },
          { "model": "ModelA", "name": "ted" },
          { "model": "ModelB", "name": "oranage" }
        ],
        function(item,callback) {
          async.waterfall(
            [
              function(callback) {
                Counter.getNewId(item.model,callback);
              },
              function(counter,callback) {
                mongoose.model(item.model).findByIdAndUpdate(
                  counter.counter,
                  { "$set": { "name": item.name } },
                  { "upsert": true, "new": true },
                  function(err,doc) {
                    console.log(doc);
                    callback(err);
                  }
                );
              }
            ],
            callback
          );
        },
        callback
      );
    },
    function(callback) {
      Counter.find().exec(function(err,result) {
        console.log(result);
        callback(err);
      });
    }
  ],
  function(err) {
    if (err) throw err;
    mongoose.disconnect();
  }
);

为了方便起见,它在模型上实现了一个静态方法,如.getNewId(),它只是描述性地包装了.findByIdAndUpdate()中使用的主函数。这是手册页部分提到的.findAndModify()的一种形式。

这样做的目的是,它将在Counter模型集合中查找一个特定的"键"(实际上还是_id),并执行一个操作来"增加"该键的计数器值并返回修改后的文档。"upsert"选项也有助于这一点,因为如果所请求的"key"还没有文档存在,那么它将被创建,否则将通过$inc增加值,并且总是这样,默认值将为1。

下面的例子显示了两个计数器被独立维护:

{ _id: 1, name: 'bill', __v: 0 }
{ _id: 1, name: 'apple', __v: 0 }
{ _id: 2, name: 'ted', __v: 0 }
{ _id: 2, name: 'oranage', __v: 0 }
[ { _id: 'ModelA', __v: 0, counter: 2 },
  { _id: 'ModelB', __v: 0, counter: 2 } ]

首先列出创建时的每个文档,然后显示"counters"集合的结束状态,该集合保存了请求的每个键的最后使用值。

还要注意,这些"奇怪的数字"有一个特定的目的,即总是保证是唯一的,并且总是按顺序递增。注意,它们这样做不需要再次访问数据库,以便安全地存储和使用一个递增的数字。

最新更新