mongodb自增主键id生成器的设计和实现
文章地址: http://blog.csdn.net/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++)
{
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