mongodb字段值自增长实现代码

mongodb 没有像 sql 一样有自动增长的功能, mongodb 的 _id 是系统自动生成的12字节唯一标识。但在某些情况下,我们可能需要实现 objectid 自动增长功能。由于 mongodb 没有实现这个功能,我们可以通过编程的方式来实现,以下我们将在 counters 集合中实现_id字段自动增长。

1.创建计数器集合

期望_id字段从1,2,3,4到n,启动一个自动递增的整数序列,如:

{
  "_id":1,
  "title": "标题",
  "content": "内容1",
  "type": "类型"
}

为此,创建 counters 集合,序列字段值可以实现自动长:

db.createcollection("counters")

初始化集合,以objid作为主键,sequence_value 字段是序列通过自动增长后的一个值:

db.counters.insert({_id:"objid",sequence_value:0})

2.查询序列号

查询返回更新后的序列号

db.counters.findandmodify({
  query: {_id: "objid" },
  update: {$inc:{sequence_value:1}},
  new: true
}).sequence_value;

操作符解释:

$inc可以对文档的某个值为数字型(只能为满足要求的数字)的键进行增减的操作;

db.collection.findandmodify({
  query: <document>, //定义关于哪些记录需要修改的选择标准
  sort: <document>, //确定选择标准检索多个文档时应修改的文档
  new: <boolean>, //表示将显示修改后的文档
  fields: <document>, //指定要返回的字段集
  upsert: <boolean> //如果选择标准无法检索文档,则创建一个新文档
  remove: <boolean> //为true,query指定的文档将从数据库中删除
)}

3.测试

创建测试集合sms:

db.createcollection("sms")

在sms集合中新增文档,实现_id自增长: 

db.sms.insert({
  _id: db.counters.findandmodify({query:{_id: "objid" },update: {$inc:{sequence_value:1}},"new":true}).sequence_value,
  title: "标题1",
  content: "短信1",
  type: "1"
})

查询sms集合:

db.sms.find({}).sort({_id:1})

4.java实现

java实现以上功能,数据库驱动版本不同运行效果有差异,仅供参考:

private mongodatabase conn;
 
static{
  this.conn = getdatabase(databasename);
}
 
/**
 * 连接数据库
 * @param databasename 数据库名称
 * @return       数据库连接对象
 */
private static mongodatabase getdatabase(databasename){
  mongodatabase mongodatabase = null;
  try{  
    // 连接到 mongodb 服务
    mongoclient mongoclient = new mongoclient( "localhost" , 27017 );
  
    // 连接到数据库
    mongodatabase mongodatabase = mongoclient.getdatabase(databasename); 
    system.out.println("connect to database successfully");
    
  }catch(exception e){
    system.err.println( e.getclass().getname() + ": " + e.getmessage() );
  }
  return mongodatabase;
}
 
/**
 * 获取最新序列号
 * @return 序列号
 */
private static int getnextsequencevalue(){
  dbcollection collection = conn.getcollection("counters");
  dbobject query = new basicdbobject("_id", new basicdbobject("$eq", "objid"));
  dbobject newdocument =new basicdbobject();
  newdocument.put("$inc", new basicdbobject().append("sequence_value", 1));
  newdocument.put("new": true);
  dbobject ret = collection.findandmodify(query, newdocument);
  if (ret == null){
    return 0;
  }else{
    return (integer)ret.get("sequence_value");
  }
}
 
/**
 * 新增集合文档
 */
public static void addsms(){
  int id = getnextsequencevalue();
  if(id != 0){
    dbcollection collection = conn.getcollection("sms");
    list<document> documents = new arraylist<document>();
    for(int i = 0; i < 20; i++){
      int id = getnextsequencevalue();
      document document = new document("_id", id).
      append("title", "标题" + i).
      append("content", "短信" + i). 
      append("type", 1);
      documents.add(document);
    }
    collection.insertmany(documents); 
    system.out.println("文档插入成功"); 
 
  }
}
 
/**
 * 查询集合
 */
public static void findsms(){
  dbcollection collection = conn.getcollection("sms");
  finditerable<document> finditerable = collection.find(); 
  mongocursor<document> mongocursor = finditerable.iterator(); 
  while(mongocursor.hasnext()){ 
    system.out.println(mongocursor.next()); 
  }
}

5.总结

有了字段自增长功能,可以实现订单流水号、编码的流水号等功能,可以实现同mysql自增字段、oracle序列的相同效果。

到此这篇关于mongodb字段值自增长实现的文章就介绍到这了,更多相关mongodb字段值自增长内容请搜索www.887551.com以前的文章或继续浏览下面的相关文章希望大家以后多多支持www.887551.com!

(0)
上一篇 2022年3月21日
下一篇 2022年3月21日

相关推荐