mongoDB，ID自增实现

最新推荐文章于 2024-08-09 17:12:01 发布

早餐核桃牛奶

最新推荐文章于 2024-08-09 17:12:01 发布

阅读量1.6k

点赞数

文章标签： mongodb javascript 数据库

本文链接：https://blog.youkuaiyun.com/m0_46262108/article/details/129115063

版权

counter.js

const mongoose = require('mongoose');
const Schema = mongoose.Schema;

const counterSchema = new Schema({
  _id: { type: String, required: true },
  sequence_value: { type: Number, default: 1 },
});

counterSchema.statics.getNextSequenceValue = async function (sequenceName) {
  const sequenceDocument = await this.findOneAndUpdate(
    { _id: sequenceName },
    { $inc: { sequence_value: 1 } },
    { new: true, upsert: true }
  );
  return sequenceDocument.sequence_value;
};

module.exports = mongoose.model("Counter", counterSchema);

$inc 操作符用于自增计数器集合中的 sequence_value 属性

new 选项设置为 true，返回更新后的计数器文档
如果计数器文档不存在，则使用 upsert 选项创建一个新的文档。

然后，然后返回序列号

role.js

const mongoose = require('mongoose');
const Schema = mongoose.Schema;

const Counter = require("./counter");

const RoleSchema = new Schema({
  rid: {
    type: Number,
    unique: true, // 唯一索引，用于确保角色名称的唯一性
  },
  name: {
    type: String,
    required: true,
    unique: true, // 唯一索引，用于确保角色名称的唯一性
  },
  description: {
    type: String,
    default: "",
  },
});

RoleSchema.pre("save", async function (next) {
  const role = this;
  if (!role.isNew || role.rid) {
    return next();
  }
  try {
    if (!role.rid) {
      role.rid = await Counter.getNextSequenceValue("rid"); // rid是你需要自增的属性
    }
  } catch (err) {
    next(err);
  }
});

module.exports = mongoose.model("Role", RoleSchema);