C++ 操作mongodb 的代码汇总

mongodb 是UTF8格式, 所以存储中文的时候, 需要用下面这个函数将中文转换为 UTF8 格式

bool MB_To_UTF8( const std::string& strIn, std::string& strOut ){    strOut.clear();    int nInputLen = MultiByteToWideChar( CP_ACP, 0, strIn.c_str(), strIn.length(), NULL, 0 );    wchar_t wszStr[ 100 ] = { 0 };    char szResult[ 100 ] = { 0 };    int nWstrLen = MultiByteToWideChar( CP_ACP, 0, strIn.c_str(), strIn.length(), wszStr, nInputLen );    if( nWstrLen != nInputLen )    {        return false;    }    int nUTF8_Len = WideCharToMultiByte( CP_UTF8, 0, wszStr, nInputLen, NULL, 0, NULL, NULL );    if( nUTF8_Len <= 0 )    {        return false;    }    nWstrLen = WideCharToMultiByte( CP_UTF8, 0, wszStr, nInputLen, szResult, nUTF8_Len, NULL, NULL );    if( nWstrLen != nUTF8_Len )    {        return false;    }    strOut = szResult;    return true;}

定义数据库集合名称和 mongodb 实例

std::string strCollection = "test.foo";mongo::DBClientConnection oDB;

假设要把学生信息保存到集合 "test.foo" 中, 学生信息包括学号, 名字, 入学年月日，用下面这个结构体来描述学生信息

typedef struct tagStudengInfo{    int nStudentID;    char szName[ 50 ];    // 入学年月日份    int nDmission_year;    int nDmission_mon;    int nDmission_mday;}ST_STUDENT_INFO;

保存到数据库中的代码如下

void InputStudentInfo( const ST_STUDENT_INFO& stStudentInfo ){mongo::BSONObjBuilder oBOJCharacter;oBOJCharacter.append( "student_id", stStudentInfo.nStudentID ); // 学号std::string strName;MB_To_UTF8( stStudentInfo.szName, strName );oBOJCharacter.append( "name", strName ); // 名字struct tm tmDmissionTime;memset( (void*)&tmDmissionTime, 0, sizeof( tmDmissionTime ) );tmDmissionTime.tm_year = stStudentInfo.nDmission_year - 1900;tmDmissionTime.tm_mon = stStudentInfo.nDmission_mon - 1;tmDmissionTime.tm_mday = stStudentInfo.nDmission_mday;time_t t_DmissionTime = mktime( &tmDmissionTime );mongo::Date_t stDate( t_DmissionTime * 1000 );oBOJCharacter.append( "admission_time", stDate ); // 入学年月日try{oDB.insert( strCollection, oBOJCharacter.obj() );std::string strDBError = oDB.getLastError();if( !strDBError.empty() ){std::cout << "mongodb error, err = " << strDBError.c_str() << std::endl;}// 对 "student_id" 字段建立唯一索引mongo::IndexSpec oIds;oIds.addKey( "student_id" );oIds.unique( true );oDB.createIndex( strCollection, oIds );}catch ( const mongo::DBException &e ){std::cout << "InputStudentInfo caught " << e.what() << std::endl;}}

然后执行下面的代码

void main(){mongo::client::initialize();try{oDB.connect("localhost");}catch( const mongo::DBException &e ){std::cout << "caught " << e.what() << std::endl;return;}// 入学年份都是 2005 年 11 月 1 日ST_STUDENT_INFO stStudentInfo_1, stStudentInfo_2;stStudentInfo_1.nStudentID = 1;strcpy(stStudentInfo_1.szName, "我的老板是笨蛋" );stStudentInfo_1.nDmission_year = 2005;stStudentInfo_1.nDmission_mon = 11;stStudentInfo_1.nDmission_mday = 1;stStudentInfo_2.nStudentID = 2;strcpy( stStudentInfo_2.szName, "money" );stStudentInfo_2.nDmission_year = 2005;stStudentInfo_2.nDmission_mon = 11;stStudentInfo_2.nDmission_mday = 1;InputStudentInfo( stStudentInfo_1 );InputStudentInfo( stStudentInfo_2 );mongo::client::shutdown();}

使用 mongoVUE 查看数据库集合中的信息如下

接下来, 为集合中的每个文档增加性别和年龄字段

mongo::BSONObjBuilder oBOJ;oBOJ.append( "sex", 0 ); // 性别, 0 表示男, 1 表示女oBOJ.append( "age", 18 ); // 年龄oDB.update( strCollection, mongo::BSONObj(), BSON( "$set" << oBOJ.obj() ), false, true );

假设 "我的老板是笨蛋", 性别是男, 年龄40, 需要修改 年龄值, 即 age 字段为 40
 

// 因为 student_id 是唯一索引, 所以按照 student_id 进行查询并更新 age 字段// "我的老板是笨蛋" 的学号 "student_id" 值是 1oDB.update( strCollection, mongo::Query( BSON( "student_id" << 1 ) ), BSON( "$set" << BSON( "age" << 40 ) ) );

假设 "money", 性别是女, 年龄18, 需要修改性别, 即 sex 字段

// 按照 student_id 进行查询并更新 sex 字段// "money" 的学号 "student_id" 值是 2oDB.update( strCollection, mongo::Query( BSON( "student_id" << 2 ) ), BSON( "$set" << BSON( "sex" << 1 ) ) );

执行完上述几行代码后, 数据库集合中的信息如下

每个文档多了 sex 和 age 字段

为所有文档增加成绩信息, 保存在一个数组 "transcripts" 中   

mongo::BSONObjBuilder oBOJ_1;mongo::BSONArrayBuilder arr_o;for( int i = 0; i < 2; i++ ){    // 数组中的每个元素是一个子文档      mongo::BSONObjBuilder o;    o.append( "course_name", "name" ); // 表示课程名称, 字段类型是 String, 默认值是 "name"      o.append( "score", 0 );  // 表示分数, 字段类型 Int32 默认值是 0      arr_o.append( o.obj() );}oBOJ_1.appendArray( "transcripts", arr_o.arr() ); //  oDB.update( strCollection, mongo::BSONObj(), BSON( "$set" << oBOJ_1.obj() ), false, true );

执行完上述几行代码后, 数据库集合中的信息如下

假设数组 "transcripts" 的第一个子文档表示语文成绩, 第二个子文档表示数学成绩, 需要把课程名称 "course_name" 字段的值分别修改为 "chinese" 和 “math”
这样可以让更直观的表达出信息

mongo::BSONObj obj_chinese = BSON( "$set" << BSON( "transcripts.0.course_name" << "chinese" ) ); // 课程名称是 "chinese" 表示 语文成绩mongo::BSONObj obj_math = BSON( "$set" << BSON( "transcripts.1.course_name" << "math" ) ); // 课程名称是 "math" 表示 数学成绩oDB.update( strCollection, mongo::BSONObj(), obj_chinese, false, true );  oDB.update( strCollection, mongo::BSONObj(), obj_math, false, true );  

更新 "我的老板是笨蛋" 语文成绩 100 分, 数学 59 分; "money" 语文 80 分, 数学 100 分

oDB.update( strCollection, mongo::Query( BSON( "student_id" << 1 ) ), BSON( "$set" << BSON( "transcripts.0.score" << 100 << "transcripts.1.score" << 59 ) ) );  oDB.update( strCollection, mongo::Query( BSON( "student_id" << 2 ) ), BSON( "$set" << BSON( "transcripts.0.score" << 80 << "transcripts.1.score" << 100 ) ) );  

执行完上述几行代码后, 数据库集合中的信息如下

对 transcripts 数组扩容，增加英语成绩的信息

// 数组扩容, 增加英语成绩mongo::BSONObj o = BSON( "$push" << BSON( "transcripts" << BSON( "course_name" << "English" << "score" << 80 ) ) );oDB.update( strCollection, mongo::Query(), o, false, true );

// 增加表示年级的字段

// 增加表示年级的字段mongo::BSONObjBuilder oBOJ_2;oBOJ_2.append( "grade", 1 );oDB.update( strCollection, mongo::Query(), BSON( "$set" << oBOJ_2.obj() ), false, true );