[置顶] MongoDB自增主键id生成器的设计和实现

mongodb自增主键id生成器的设计和实现

 

文章地址: http://blog.csdn.net/5iasp/article/details/8794396

作者： javaboy2012
Email：[email protected]
qq:    1046011462

 

场景： 假设有如下两个系统：

  A 系统和  B系统

采用不同的编程语言共同操作mongodb的一个表 如  news表

相关字段有

news_id

name

title

desc

有个要求：

news_id的生成字段为整数int

该id能够通过最后一位数字区别news的类别 比如 末尾为1 表示 国内新闻 末尾为 2 表示国际新闻

下面是设计思路

初始化数据库:

db.ids.save({name:"china_news", id:1});
db.ids.save({name:"international_news", id:2});　

取国内新闻id: 执行如下：

db.ids.findAndModify({update:{$inc:{'id':10}}, query:{"name":"china_news"}, new:true});

取国际新闻id: 执行如下：

db.ids.findAndModify({update:{$inc:{'id':10}}, query:{"name":"international_news"}, new:true});

国内新闻的id: 格式如下

1 11 21 31 41 51 61 xx1

国际新闻id:格式如下

1 22 22 32 42 52 62 xx2

 

java代码实现：

主要依靠：findAndModify(queryObj,update); 方法来实现自增，

package com.test;

import java.net.UnknownHostException;

import com.mongodb.BasicDBObject;
import com.mongodb.DB;
import com.mongodb.DBCollection;
import com.mongodb.DBObject;
import com.mongodb.Mongo;
import com.mongodb.MongoException;

public class Test {
 
 public static String MONGODB_DATABASE_NAME = "test"; // mongodb数据库名称
 public static String MONGODB_SERVER_IP = "192.168.0.1"; // mongodb数据库服务器ip地址
 public static int MONGODB_SERVER_PORT = 17017; // mongodb数据库服务器端口
 public static String MONGODB_TABLE_NAME_IDS = "ids"; // ids表

 /**
  * @param args
  */
 public static void main(String[] args) {
  
  for(int i=0;i<10;i++)
  {
   System.out.println(getID());
   
  }
  
 }
 
 public static String getID()
 {
  String ret="";
  Mongo mongo = null;
  try {
   mongo = new Mongo(MONGODB_SERVER_IP, MONGODB_SERVER_PORT);
   DB db = mongo.getDB(MONGODB_DATABASE_NAME);
   db.slaveOk();
   DBCollection topic = db.getCollection(MONGODB_TABLE_NAME_IDS);
   BasicDBObject queryObj = new BasicDBObject();
   queryObj.put("name", "china_news");
   BasicDBObject update=new BasicDBObject("$inc", new BasicDBObject("id", 10));
   DBObject d=topic.findAndModify(queryObj,update);
   //System.out.println(d.get("id"));
   ret=d.get("id").toString();
   
  } catch (UnknownHostException e) {
   e.printStackTrace();
  } catch (MongoException e) {
   e.printStackTrace();
  } finally {
   mongo.close();

  }
  return ret;
  
 }

}

代码输出：

1
11
21
31
41
51
61
71
81
91