如何缩短MongoDB ObjectId并在Mongoose模式

时间:2016-05-17 19:22:11

标签: javascript node.js mongodb mongoose objectid

我正在寻找一种方法来缩短MongoDB数据库生成的ObjectID,我知道一些现有的库,例如           

然而,我的问题是我希望创建一个新的' shortID'对于集合中存在的每个文档,mongoose模式中的字段,并且此新字段应该是唯一的,理想情况下是附加到文档的ObjectId的截断版本。 像这样:

var subjectSchema = new Schema({

    shortID: {required: true, unique: true}, 
    //must be unique to each document and is same everytime server restarts

    Sex: {type: String, enum: enumSex},
    Diagnosis: String,
    Projects:[projectPerSubjectSchema]
});   

'短片' https://www.npmjs.com/package/shortid库运行良好并且生成唯一ID,但是,每次服务器重新启动时它会生成不同的id,这不是我想要的。

我试过的另一个图书馆&short; mongo-id' https://www.npmjs.com/package/short-mongo-id能够将ObjectId转换为唯一ID字符串的截断版本,但是,我不确定如何在创建模式时使用它。我试过了:

ID: {type: String, 'default': shortid((this.ObjectId).valueOf()), unique: true}

尝试使用this.ObjectId获取文档的ObjectId,使用valueOf()对其进行字符串化,但终端显示:

TypeError: Cannot read property 'valueOf' of undefined 

所有代码都在Node.JS中完成,我对NodeJS和MongoDB都是新手,所以如果我在上面提供的代码中出现一些重大错误,请纠正我。提前谢谢!

1 个答案:

答案 0 :(得分:1)

shortid模块应该可以满足您的需要,您只需将shortId保存为普通的String字段,并确保在保存之前生成新代码,如下所示(在mongoose模式定义内):

const mongoose = require('mongoose')
const shortid = require('shortid')
const Schema = mongoose.Schema
const schema = new Schema({
  _shortId: {
    type: String,
    unique: true
  },
  ... // other fields...
})
// This is for checking if the document already have a _shortId, 
// so you don't replace it with a new code
schema.pre('save', function (next) {
  let doc = this

  if (!this._shortId) { 
    addShortId(doc, next)
  } 
})

function addShortId (doc, next) {
  let newShortId = shortid.generate()
  doc.constructor.findOne({_shortId: newShortId}).then((docRes) => {
    if (docRes) {
      addShortId(doc, next)
    } else {
      doc._shortId = newCode
      next()
    }
  }, (err) => {
    next(err)
  })
}

如果要为所有现有文档插入_showId,只需forEach并再次保存模型,这就是我正在做的事情(使用https://github.com/JMPerez/promise-throttle,导致它& #39;一个庞大的集合,一次调用save()会降低服务器的速度):

XXXXX.find({ '_shortId': { '$exists': false } }).then((data) => {
  var promiseThrottle = new PromiseThrottle({
    requestsPerSecond: 50,
    promiseImplementation: Promise
  })
  var i = 1
  data.forEach(element => {
  var myFunction = function (j) {
    return element.save()
  }
  promiseThrottle.add(myFunction.bind(this, i++)).then(ss => {
    console.log('done ' + ss._shortId)
  })
})

部分代码改编自:https://github.com/dylang/shortid/issues/65