mongodb 自定义函数—自增长id

最新推荐文章于 2023-05-11 17:00:00 发布

需于沙

最新推荐文章于 2023-05-11 17:00:00 发布

阅读量7.7k

点赞数 2

分类专栏： DBA 文章标签： mongodb 结构自定义函数

本文链接：https://blog.youkuaiyun.com/baiyvwuxia/article/details/47274243

版权

DBA 专栏收录该内容

1 篇文章

订阅专栏

本文介绍了如何在MongoDB中创建自定义函数以生成自增ID。通过建立特殊表结构和使用findAndModify函数，解决MongoDB中无内置自增ID的问题。同时，讨论了函数保存和导入的方法，包括保存到system.js集合和使用db.loadServerScripts()或db.eval()进行引用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

最近在做mongodb的备份工作，需要做一些自定义函数以方便备份的相关运作。在mongodb中，可以把一些自定义函数放入数据库中以方便今后的使用。
下面以自增id函数为例：
在进行数据设计时，不可避免的需要用到自增id，但是mongodb中无法像mysql中那样，提供专门的自增id索引。因此需要人为对id进行自增操作。在官网的mongodb手册上是这么实现的：

首先需要建立一个表结构，用于存储你的id信息：

db.counters.insert(
   {
      _id: "userid",
      seq: 0
   }
)

然后定义一个函数，每次调用实现id++功能

 function getNextSequence(name) {
   var ret = db.counters.findAndModify(
          {
            query: { _id: name },
            update: { $inc: { seq: 1 } },
            new: true
          }
   );

   return ret.seq;
}

上面的函数，主要用到了findAndModify这个系统函数，此系统函数其实就是实现了update的quary语句。
3. 接下来就可以在其他数据库中进行应用了

b.users.insert(
   {
     _id: getNextSequence("userid"),
     name: "Sarah C."
   }
)
db.users.insert(
   {
     _id: getNextSequence("userid"),
     name: "Bob D."
   }
)

我们来查看一下结果：

db.users.find()；
{_id : 1, name : "Sarah C."}
{_id : 2, name : "Bob D."}

看来是成功了，但是，按照上面的方法，有一个问题。就是每次退出mongo重进之后，我的getNextSequence都无法正常使用了，会出现下面所示的错误：

ReferenceError: getNextSequence is not defined

引起这个的原因是，本实例中定义的函数没有保存，在mongo实例退出后会自动清零。所以需要对我们辛辛苦苦定义的函数进行保存，要不然岂不是白干了。
那好我们就保存在一个表中，这样总不至于把函数搞丢了吧。保存的方式也有两种，一种是插入，如下所示。

db.system.js.insert(
{_id:"getNextSequence",value:function getNextSequence(name) {
   var ret = db.counters.findAndModify(
          {
            query: { _id: name },
            update: { $inc: { seq: 1 } },
            new: true
          }
   );
   return ret.seq;
}
});

还有一种保存方法，在官网上有提到，就是用db.collection.save(）函数进行保存，其实都是一样的。

db.system.js.save({
     _id : "myAddFunction" ,
     value : function (name){ 
    var ret = db.counters.findAndModify(
          {
            query: { _id: name },
            update: { $inc: { seq: 1 } },
            new: true
          }
   );
   return ret.seq;
   }
});

可能有人会问，为什么是system.js这个表，而不是其他的，因为这个表是专门用来保存js函数的。如果你定义完函数退出mongo实例，会发现还是无法使用函数getNextSequence,这是因为虽然你定义了函数，但是没有把函数引入进来，也就是说每次重新进入mongo的实例后需要把相应的js函数重新导入一遍。通过，db.loadServerScripts();把system.js中的函数，引入到mongo实例。这也是保存到system.js中原因，mongodb数据库提供了很好的库函数进行函数引用。
还有一种引入方式如下：
db.eval(‘getNextSequence(“userid”)’);
可能大家感觉上面的方式很麻烦，但是不同的方式其应用场景不太相同。大家在进行编程时慢慢体会吧。举个bash中完整应用的例子

#/home/test/mongodb/bin/mongo --port 20017 <<EOF
use mgdb
db.loadServerScripts()
db.mgtest.insert({
id:getNextSequence("userid"),
name:"daxia"
})
exit
EOF