mongoDB 自增ID C++实现方法

mongoDB 自增ID C++实现方法

标签： mongodbc++stringquery数据库脚本

2012-05-01 20:06  3876人阅读  评论(0)  收藏  举报

  分类：

C++（2） 

版权声明：本文为博主原创文章，未经博主允许不得转载。

目录(?)[+]

1、实现原理

假设要为CollectionB实现自增ID,，需要引入另外一个计算“_id”的CollectionA。

CollectionA中存放一条记录:{'_id':'CollectionB',  'currentIdValue':1}， 其中currentIdValue表示CollectionB的当前最大id值+1，每次往CollectionB里插入数据前，先到CollectionA查询currentIdValue 值并把这个值+1

从上面的原理，需要一个查询并加1的原子操作，mongodb提供了这样的接口，而且是线程安全的，即是findAndModify。

2、Mongo命令示例

mongodb的脚本如下：
db.runCommand(findAndModify:'CollectionA',query:{_id:'CollectionB'}, update:{$inc:{'currentIdValue':1}}, new:true};
查询与更新，查询'CollectionA'表_id为'CollectionB'的内容，并将'currentIdValue'属性值+1，返回该内容

3、C++示例

MongoDB.h文件

[cpp]  view plain  copy

1. using namespace mongo;  
2.   
3. // 8小时  
4. #define UTC_TIME_ZONE 28800000  
5.   
6. class MongoDB  
7. {  
8. public:  
9.     // 数据库名称  
10.     static string mNameDB;  
11.   
12.     static BOOL connectDB(string conText,string dbName);  
13.     // 数据库链接  
14.     static DBClientConnection mMongoConn;  
15.   
16.     static int getMid(string db);  
17. };  

MongoDB.cpp文件

[cpp]  view plain  copy

1. #include "MongoDB.h"  
2.   
3. // 静态MONGO链接声明  
4. DBClientConnection MongoDB::mMongoConn;  
5. string MongoDB::mNameDB;  
6.   
7. BOOL MongoDB::connectDB(string conText,string dbName)  
8. {  
9.     MongoDB::mNameDB = dbName;  
10.     string strErr;  
11.     // 链接数据库  
12.     if(!MongoDB::mMongoConn.connect(conText,strErr))  
13.     {  
14.         /*TRACE("DB connect ERROR %s\r\n",strErr);*/  
15.         return FALSE;  
16.     }  
17.     //　登陆数据库  
18.     if(!MongoDB::mMongoConn.auth(dbName,"admin","aaa111+++",strErr))  
19.     {  
20.         //TRACE("DB auth ERROR %s\r\n",strErr);  
21.         return FALSE;  
22.     }  
23.     return TRUE;  
24. }  

 

[cpp]  view plain  copy

1. // 返回自增ID　(表不存在自动补表)  
2. int MongoDB::getMid(string db)  
3. {  
4.     string err;  
5.     BSONObj b;  
6.   
7.     BSONObjBuilder bj;  
8.     //　当前集合Collection  
9.     bj.append("findAndModify","Collection");  
10.     //　查询串  
11.     bj.append("query",BSON("_id"<<db));  
12.     // 更新串  
13.     bj.append("update",BSON("$inc"<<BSON("currentIdValue"<<1)));  
14.       
15.     MongoDB::mMongoConn.runCommand(MongoDB::mNameDB,bj.obj(),b);  
16.     err = MongoDB::mMongoConn.getLastError();  
17.     if(!err.empty())  
18.         return 0;  
19.     // 获取返回内容  
20.     BSONElement bel = b.getField("value");  
21.     if(bel.isNull())   
22.     {  
23.         int tmp = 1;  
24.         // 数据库不存在记录,创建自增ID  
25.         MongoDB::mMongoConn.insert(mNameDB + ".Collection", BSON("_id"<<db<<"currentIdValue"<<tmp));      
26.         err = MongoDB::mMongoConn.getLastError();  
27.         if(err.empty())  
28.         {  
29.             return 1;  
30.         }  
31.         return 0;  
32.     }     
33.     // 当EL是结束  
34.     if(bel.eoo()) return 0;  
35.       
36.     // 检查类型是INT或DOUBLE  
37.     if(bel.type() == mongo::NumberDouble)  
38.         return (int)bel["currentIdValue"].Double();  
39.     else  
40.         return bel["currentIdValue"].Int();  
41. }