Mongodb中的查询操作比较丰富,支持逻辑操作,如:
{ $or: [ { <expression1> }, { <expression2> }, ... , { <expressionN> } ] }
在C++ driver中如何实现或查询呢?
源码包中的example中没有这样的例子,经过一番查找在bsonobjbuilder.h和jsobjtests.cpp找到相关代码
BSONObj actual() {
return OR( BSON( "a" << GT << 1 << LTE << "x"),
BSON( "b" << NE << 1 << NE << "f" << NE << 22.3),
BSON( "x" << "p" ) );
}
inline BSONObj OR(const BSONObj& a, const BSONObj& b)
{ return BSON( "$or" << BSON_ARRAY(a << b) ); }
inline BSONObj OR(const BSONObj& a, const BSONObj& b, const BSONObj& c)
{ return BSON( "$or" << BSON_ARRAY(a << b << c) ); }
inline BSONObj OR(const BSONObj& a, const BSONObj& b, const BSONObj& c, const BSONObj& d) { return BSON( "$or" << BSON_ARRAY(a << b << c << d) ); }
inline BSONObj OR(const BSONObj& a, const BSONObj& b, const BSONObj& c, const BSONObj& d, const BSONObj& e)
{ return BSON( "$or" << BSON_ARRAY(a << b << c << d << e) ); }
inline BSONObj OR(const BSONObj& a, const BSONObj& b, const BSONObj& c, const BSONObj& d, const BSONObj& e, const BSONObj& f)
{ return BSON( "$or" << BSON_ARRAY(a << b << c << d << e << f) ); }
可以看出有两种方法
- 使用OR函数
- 使用$or + BSON_ARRAY构造
- 先看第一种方法:使用OR函数构造查询语句
mongo::BSONObj query_obj;
query_obj = mongo::OR(BSON("n"<<max_v), BSON("a"<<"test1"), BSON("a"<<"test2"));
auto_ptr<mongo::DBClientCursor> cursor = conn.query("test.c1", mongo::Query(query_obj));
查询时调用mongo::Query(BSONobj &obj)方法
- 再看第二种方法
auto_ptr<mongo::DBClientCursor> cursor = conn.query("test.c1", QUERY("$or" << BSON_ARRAY(BSON("n" << max_v) << BSON("a" << "test1") << BSON("a" << "test2"))));
查询时调用QUERY(X)宏
#define QUERY(x) ::mongo::Query( BSON(x) )
两种方法从代码来看,其实是一样的,只是外观不同,可以任意使用。
问题:如何动态的构造一个OR操作呢?
如:在一个数组里,有多个查询参数,在运行中可变,因此不能用写死的方法,调用宏在这种情况下不合适。既然宏能构造数组,那么宏所调用的函数必定支持动态数组的创建,因为对于C++driver来说,引用中创建的数组都是动态的!
方法如下:
vector<string> ids;
ids.push_back("1001");
ids.push_back("1002");
ids.push_back("1003");
mongo::BSONArrayBuilder ab;
for (vector<string>::iterator it = ids.begin(); it != ids.end(); ++it) {
tmp = BSON("mid" << *it);
ab << tmp;
}
mongo::BSONArray or_array;
or_array = ab.arr();
auto_ptr<mongo::DBClientCursor> cursor = conn.query("test.c1", QUERY("$or" << or_array));
while (cursor->more()) {
mongo::BSONObj p = cursor->next();
cout << p.getStringField("a") << endl;
}
这里用到了BSONArrayBuilder类,它用于构造BSONArray类,构造完成后,调用arr()方法得到BSONArray对象。
BSONArrayBuilder与BSONObjBuilder都支持直接将想要的数据添加到Builder中
BSONArrayBuilder ab;
ab << "item1";
ab << "item2";
ab << "item3";
BSONArray myArray = ab.arr();
# {[ "item1", "item2", "item3"] }
BSONObjBuilder ob;
ob << "key" << "item1";
BSONObj query = ob.obj();
# {"key": "item1"}
mongo::BSONObj obj1 = BSON("$or" << or_array);
cout << obj1.jsonString() << endl;
运行结果为:
{ "$or" : [ { "mid" : "1001" }, { "mid" : "1002" }, { "mid" : "1003" } ] }
此时数组的构造问题解决了,还有一个问题,“有时候,需要在逻辑查询之后,再进行一次赛选,如何将逻辑操作与其它的查询条件结合呢?
根据BSON()的定义
#define BSON(x) (( ::mongo::BSONObjBuilder(64) << x ).obj())
可知,经过BSON宏处理过后得到的是一个BSONOjb,BSONObj是缺少BSON构造功能的,只能通过BSONObjBuilder构造,需要用到如下两个函数
template<class T>
inline BSONObjBuilder& BSONObjBuilderValueStream::operator<<( T value ) {
_builder->append(_fieldName, value);
_fieldName = StringData();
return *_builder;
}
inline BSONObjBuilder& BSONObjBuilderValueStream::operator<<( const BSONElement& e ) {
_builder->appendAs( e , _fieldName );
_fieldName = StringData();
return *_builder;
}
注意到BSONObjBuilder的operation<<的参数是BSONElement 与 class T,经过测试,不直接支持BSONObj与BSONArray,即连个BSONObj不能直接相连
BSONObj obj1;
BSONObj obj2;
obj1 = BSON("id" << "2");
obj2 = BSON("name" << "joe");
obj1 << obj2;
这是不允许的,上面只是一个简单的示例,用以说明情况。
mongo::BSONObjBuilder objbuilder;
objbuilder << "$or" << or_array << "price" << BSON( "$gt" << 1000);
cout << objbuilder.obj().jsonString() << endl;
其运行结果为类似的:
{ "$or" : [ { "mid" : "1001" }, { "mid" : "1002" }, { "mid" : "1003" } ], "price" : { "$gt" : 100 } }
按照规则可以组成我们需要的查询条件。