如何缩短MongoDB ObjectId并在Mongoose模式中使用它

Question

我正在寻找一种缩短MongoDB数据库生成的ObjectID的方法，我知道一些现有的库，如'shortid'shortmongo-id'

然而，我的问题是，我希望在mongoose架构中为集合中存在的每个文档创建一个新的"shortID"字段，并且这个新字段应该是唯一的，最好是附加到文档的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}

尝试使用此获取文档的ObjectId。ObjectId，用valueOf（）字符串化它，但终端显示：

TypeError: Cannot read property 'valueOf' of undefined 

所有的代码都是在Node.JS中完成的，我对NodeJS和MongoDB都很陌生，所以如果我在上面提供的代码中犯了一些重大错误，请纠正我。提前感谢！

Answer 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，因为这是一个巨大的集合，一次调用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