枫叶飞翔 发表于 2015-11-11 09:16:47

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

  mongodb自增主键id生成器的设计和实现
  
  文章地址: http://blog.iyunv.com/5iasp/article/details/8794396
  
作者: javaboy2012
Email:yanek@163.com
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&#43;&#43;)
{
   System.out.println(getID());
   
}

}

public static String getID()
{
String ret=&quot;&quot;;
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(&quot;name&quot;, &quot;china_news&quot;);
   BasicDBObject update=new BasicDBObject(&quot;$inc&quot;, new BasicDBObject(&quot;id&quot;, 10));
   DBObject d=topic.findAndModify(queryObj,update);
   //System.out.println(d.get(&quot;id&quot;));
   ret=d.get(&quot;id&quot;).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

         版权声明:本文为博主原创文章,未经博主允许不得转载。
页: [1]
查看完整版本: MongoDB自增主键id生成器的设计和实现