MongoDB的使用学习之（七）MongoDB的聚合查询（两种方式）附项目源码
先来张在路上……
铛铛铛……项目源码下载地址：http://files.cnblogs.com/ontheroad_lee/MongoDBDemo.rar
此项目是用Maven创建的，没有使用Maven的，自己百度、谷歌去；直接用Junit测试就行，先执行里面的save方法，添加10000条测试数据提供各种聚合查询等。
少废话，上干货……
一、MongoDB数据库的配置（mongodb.xml）
以下是我自己的配置，红色字体请改为自己本机的东东，你说不懂设置端口，不会创建数据库名称，不会配置用户名密码，那有请查阅本系列的第4节（MongoDB的使用学习之（四）权限设置--用户名、密码、端口==），你说懒得设置，那就@#￥%……&*（）！
<!-- Default bean name is 'mongo' --> <!-- 定义mongo对象，对应的是mongodb官方jar包中的Mongo，replica-set设置集群副本的ip地址和端口 --> <mongo:mongo id="mongo" host="localhost" port="47017" /> <mongo:db-factory id="mongoDbFactory" dbname="mongoTest" mongo-ref="mongo" username="root1" password="root1" /> <!-- 自动扫描以下包的类 --> <!-- 映射转换器，扫描back-package目录下的文件，根据注释，把它们作为mongodb的一个collection的映射 --> <mongo:mapping-converter base-package="com.ontheroad.**.po" /> <bean id="mappingContext" class="org.springframework.data.mongodb.core.mapping.MongoMappingContext" /> <!-- 配置mongodb映射类型 --> <bean id="mappingMongoConverter" class="org.springframework.data.mongodb.core.convert.MappingMongoConverter"> <constructor-arg name="mongoDbFactory" ref="mongoDbFactory" /> <constructor-arg name="mappingContext" ref="mappingContext" /> <property name="typeMapper" ref="defaultMongoTypeMapper" /> </bean> <!-- 默认Mongodb类型映射 --> <bean id="defaultMongoTypeMapper" class="org.springframework.data.mongodb.core.convert.DefaultMongoTypeMapper"> <constructor-arg name="typeKey"> <null /><!-- 这里设置为空,可以把 spring data mongodb 多余保存的_class字段去掉 --> </constructor-arg> </bean> <!-- 操作mongodb --> <!-- mongodb的主要操作对象mongoTemplate(Spring Data提供)，所有对mongodb的增删改查的操作都是通过它完成 --> <bean id="mongoTemplate" class="org.springframework.data.mongodb.core.MongoTemplate"> <constructor-arg name="mongoDbFactory" ref="mongoDbFactory" /> <constructor-arg name="mongoConverter" ref="mappingMongoConverter" /> </bean>
二、各种聚合查询方法
以下只是展示一些常用的，聚合查询方法，无奈个人功力尚浅，没啥高深的东西，待日后，有时间有精力有实力，再整理些高级一点的
1、添加测试数据
@Test public void save() { News n = null; for (int i = 0; i < 10000; i++) { n = new News(); n.setTitle("title_" + i); n.setUrl("url_" + i); //2014-01-01到2014-01-01之间的随机时间 Date randomDate=DateUtil.randomDate("2014-01-01","2014-05-11"); //MongoDB里如果时间类型存的是Date，那么会差8个小时的时区，因为MongoDB使用的格林威治时间，中国所处的是+8区，so…… //比如我保存的是2014-05-01 00:00:00，那么保存到MongoDB里则是2014-05-01 08:00:00，所以为了统一方面，那就保存字符串类型，底下保存的long类型 n.setPublishTimeStr(DateUtil.formatDateTimeByDate(randomDate)); //long类型在查询速度中肯定会比较快 n.setPublishTime(randomDate.getTime()); n.setPublishDate(randomDate); n.setPublishMedia("publishMedia_" + i); String[] areaArr = {"1024", "102401", "102402", "102403", "102404", "102405", "102406", "102407", "102408" , "10240101", "10240102", "10240201", "10240202", "10240301", "10240302", "10240401", "10240402" , "10240501", "10240502", "10240601", "10240602", "10240701", "10240702", "10240801", "10240802"}; int areaNum=(int)(Math.random() * areaArr.length);//产生0-strs.length的整数随机数 String area = areaArr[areaNum]; n.setArea(area); String[] ckeyArr = {"A101", "A102", "A201", "A202", "A203" , "B101", "B102", "B103", "C201", "C202", "C203", "22", "23", "24", "25", "26"}; int ckeyNum=(int)(Math.random() * ckeyArr.length);//产生0-strs.length的整数随机数 List<String> list = new ArrayList<String>(); for (int j = 0; j < ckeyNum; j ++) { int ckeyNum1=(int)(Math.random() * ckeyArr.length);//产生0-strs.length的整数随机数 list.add(ckeyArr[ckeyNum1]); } n.setClassKey(list); Integer[] evalArr = {1, 0}; int evalNum=(int)(Math.random() * evalArr.length);//产生0-strs.length的整数随机数 n.setEvaluate(evalArr[evalNum]); Integer[] mproArr = {1, 2, 100}; int mproNum=(int)(Math.random() * mproArr.length);//产生0-strs.length的整数随机数 n.setMediaProperty(mproArr[mproNum]); Integer[] mtypeArr = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11}; int mtypeNum=(int)(Math.random() * mtypeArr.length);//产生0-strs.length的整数随机数 n.setMediaType(mtypeArr[mtypeNum]); Integer[] levelArr = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12}; int levelNum=(int)(Math.random() * levelArr.length);//产生0-strs.length的整数随机数 n.setLevel(levelArr[levelNum]); newsService.save(n); } System.out.println("OK"); }
2、简单的分组查询--使用Mongo本身提供的AggregationOutput进行分组查询
/** * * 功能：使用Mongo本身提供的AggregationOutput进行分组查询 * 参数： * 创建人：OnTheRoad_Lee * 修改人：OnTheRoad_Lee * 最后修改时间：2014-5-26 */ public void testGroup1 () { //按照eval字段进行分组，注意$eval必须是存在mongodb里面的字段，不能写$evaluate(此字段是News类中定义的，和存入mongodb中的有区别) //{$group:{_id:{'AAA':'$BBB'},CCC:{$sum:1}}}固定格式：把要分组的字段放在_id:{}里面，BBB是mongodb里面的某个字段，AAA是BBB的重命名，CCC是$sum:1的重命名 //此查询语句== select eval as eval, count(*) as docsNum from news group by eval having docsNum>=85 order by docsNum desc //具体的mongodb和sql的对照可以参考：http://docs.mongodb.org/manual/reference/sql-aggregation-comparison/ String groupStr = "{$group:{_id:{'eval':'$eval'},docsNum:{$sum:1}}}"; DBObject group = (DBObject) JSON.parse(groupStr); String matchStr = "{$match:{docsNum:{$gte:85}}}"; DBObject match = (DBObject) JSON.parse(matchStr); String sortStr = "{$sort:{_id.docsNum:-1}}"; DBObject sort = (DBObject) JSON.parse(sortStr); AggregationOutput output = mongoTemplate.getCollection("news").aggregate(group, match, sort); System.out.println(output.getCommand()); //转换为执行原生的mongodb查询语句 //{ "aggregate" : "news" , "pipeline" : [ { "$group" : { "_id" : { "eval" : "$eval"} , "docsNum" : { "$sum" : 1}}} , { "$match" : { "docsNum" : { "$gte" : 85}}} , { "$sort" : { "_id.docsNum" : -1}}]} System.out.println(output.getCommandResult()); //查询结果 //{ "serverUsed" : "localhost/127.0.0.1:47017" , "result" : [ { "_id" : { "evaluate" : 1} , "docsNum" : 9955} , { "_id" : { "evaluate" : 0} , "docsNum" : 10047}] , "ok" : 1.0} //也可以把查询结果封装到NewsNumDTO，这样以一个dto对象返回前台操作就更容易了 NewsNumDTO dto = new NewsNumDTO(); for( Iterator< DBObject > it = output.results().iterator(); it.hasNext(); ){ BasicDBObject dbo = ( BasicDBObject ) it.next(); BasicDBObject keyValus = (BasicDBObject)dbo.get("_id"); int eval = keyValus.getInt("eval"); long docsNum = ((Integer)dbo.get("docsNum")).longValue(); if(eval == 1){ dto.setPositiveNum(docsNum); }else { dto.setNegativeNum(docsNum); } } }
3、获取和testGroup1方法同样结果的另一种写法，Spring Data MongoDB隆重登场，语法更加简洁易懂
/** * * 功能：获取和testGroup1方法同样结果的另一种写法，Spring Data MongoDB隆重登场，语法更加简洁易懂 * 参数： * 创建人：OnTheRoad_Lee * 修改人：OnTheRoad_Lee * 最后修改时间：2014-5-26 */ public void testAggregation1() { TypedAggregation<News> agg = Aggregation.newAggregation( News.class, project("evaluate") ,group("evaluate").count().as("totalNum") ,match(Criteria.where("totalNum").gte(85)) ,sort(Sort.Direction.DESC, "totalNum") ); AggregationResults<BasicDBObject> result = mongoTemplate.aggregate(agg, BasicDBObject.class); System.out.println(agg.toString()); //执行语句差不多 //{ "aggregate" : "__collection__" , "pipeline" : [ { "$project" : { "evaluate" : 1}} , { "$group" : { "_id" : "$evaluate" , "totalNum" : { "$sum" : 1}}} , { "$match" : { "totalNum" : { "$gte" : 85}}} , { "$sort" : { "totalNum" : -1}}]} System.out.println(result.getMappedResults()); //查询结果简洁明了 //[{ "_id" : 0 , "totalNum" : 10047}, { "_id" : 1 , "totalNum" : 9955}] //使用此方法，如果封装好了某一个类，类里面的属性和结果集的属性一一对应，那么，Spring是可以直接把结果集给封装进去的 //就是AggregationResults<BasicDBObject> result = mongoTemplate.aggregate(agg, BasicDBObject);中的BasicDBObject改为自己封装的类 //但是感觉这样做有点不灵活，其实吧，应该是自己现在火候还不到，还看不到他的灵活性，好处在哪里；等火候旺了再说呗 //所以，就用这个万能的BasicDBObject类来封装返回结果 List<BasicDBObject> resultList = result.getMappedResults(); NewsNumDTO dto = new NewsNumDTO(); for(BasicDBObject dbo : resultList){ int eval = dbo.getInt("_id"); long num = dbo.getLong("totalNum"); if(eval == 1){ dto.setPositiveNum(num); }else { dto.setNegativeNum(num); } } System.out.println(dto.getPositiveNum()); }
4、previousOperation的简单使用--为分组的字段（_id）建立别名
/** * * 功能：previousOperation的简单使用--为分组的字段（_id）建立别名 * 参数： * 创建人：OnTheRoad_Lee * 修改人：OnTheRoad_Lee * 最后修改时间：2014-5-26 */ public void testAggregation2() { TypedAggregation<News> agg = Aggregation.newAggregation( News.class, // match(Criteria.where("mediaType").is(100)) project("evaluate") ,group("evaluate").count().as("totalNum") ,project("evaluate", "totalNum") .and("eval").previousOperation() //为分组的字段（_id）建立别名 ); AggregationResults<BasicDBObject> result = mongoTemplate.aggregate(agg, BasicDBObject.class); System.out.println(agg.toString()); // { "aggregate" : "__collection__" , "pipeline" : [ { "$project" : { "evaluate" : 1}} , { "$group" : { "_id" : "$evaluate" , "totalNum" : { "$sum" : 1}}} , { "$project" : { "evaluate" : "$_id.evaluate" , "totalNum" : 1 , "_id" : 0 , "eval" : "$_id"}}]} System.out.println(result.getMappedResults()); // [{ "totalNum" : 10047 , "eval" : 0}, { "totalNum" : 9955 , "eval" : 1}] }
5、unwind()的使用，通过Spring Data MongoDB
/** * * 功能：unwind()的使用，通过Spring Data MongoDB * unwind()就是$unwind这个命令的转换， * $unwind - 可以将一个包含数组的文档切分成多个, 比如你的文档有 中有个数组字段 A, A中有10个元素, 那么 * 经过 $unwind处理后会产生10个文档，这些文档只有 字段 A不同 * 详见：http://my.oschina.net/GivingOnenessDestiny/blog/88006 * 参数： * 创建人：OnTheRoad_Lee * 修改人：OnTheRoad_Lee * 最后修改时间：2014-5-26 */ public void testAggregation3() { TypedAggregation<News> agg = Aggregation.newAggregation( News.class, unwind("classKey") ,project("evaluate", "classKey") // 这里说明一点就是如果group>=2个字段，那么结果集的分组字段就没有_id了，取而代之的是具体的字段名（和testAggregation()最对比） ,group("evaluate", "classKey").count().as("totalNum") ,sort(Sort.Direction.DESC, "totalNum") ); AggregationResults<BasicDBObject> result = mongoTemplate.aggregate(agg, BasicDBObject.class); System.out.println(agg.toString()); // { "aggregate" : "__collection__" , "pipeline" : [ { "$unwind" : "$classKey"} , { "$project" : { "evaluate" : 1 , "classKey" : 1}} , { "$group" : { "_id" : { "evaluate" : "$evaluate" , "classKey" : "$classKey"} , "totalNum" : { "$sum" : 1}}} , { "$sort" : { "totalNum" : -1}}]} System.out.println(result.getMappedResults()); // 结果就是酱紫，一目了然，怎么操作，就交给你自己了 // [{ "evaluate" : 0 , "classKey" : "A201" , "totalNum" : 4857}, { "evaluate" : 0 , "classKey" : "23" , "totalNum" : 4857}, { "evaluate" : 0 , "classKey" : "A101" , "totalNum" : 4842}, { "evaluate" : 0 , "classKey" : "24" , "totalNum" : 4806}, { "evaluate" : 1 , "classKey" : "A101" , "totalNum" : 4787}, { "evaluate" : 0 , "classKey" : "C201" , "totalNum" : 4787}, { "evaluate" : 0 , "classKey" : "A102" , "totalNum" : 4783},……] }
6、Spring Data MongoDB，按照时间（字符串）分组
/** * * 功能：Spring Data MongoDB，按照时间（字符串）分组 * 参数： * 创建人：OnTheRoad_Lee * 修改人：OnTheRoad_Lee * 最后修改时间：2014-5-26 */ public void testAggregation4() { TypedAggregation<News> agg = Aggregation.newAggregation( News.class, //project().andExpression()里面是一个表达式 // 详见api：http://docs.spring.io/spring-data/data-mongodb/docs/current/reference/htmlsingle/#mongo.aggregation // 搜索 .andExpression 定位到具体的方法模块 project("evaluate") .andExpression("substr(publishTimeStr,0,10)").as("publishDate") ,group("evaluate", "publishDate").count().as("totalNum") ,sort(Sort.Direction.DESC, "totalNum") ); AggregationResults<BasicDBObject> result = mongoTemplate.aggregate(agg, BasicDBObject.class); System.out.println(agg.toString()); // { "aggregate" : "__collection__" , "pipeline" : [ { "$project" : { "evaluate" : 1 , "publishDate" : { "$substr" : [ "$publishTimeStr" , 0 , 10]}}} , { "$group" : { "_id" : { "evaluate" : "$evaluate" , "publishDate" : "$publishDate"} , "totalNum" : { "$sum" : 1}}} , { "$sort" : { "totalNum" : -1}}]} System.out.println(result.getMappedResults()); // [{ "evaluate" : 0 , "publishDate" : "2014-03-09" , "totalNum" : 101}, { "evaluate" : 1 , "publishDate" : "2014-02-14" , "totalNum" : 100}, { "evaluate" : 1 , "publishDate" : "2014-02-11" , "totalNum" : 99}, { "evaluate" : 0 , "publishDate" : "2014-03-17" , "totalNum" : 98}, { "evaluate" : 1 , "publishDate" : "2014-03-26" , "totalNum" : 98}, ……] // 这个查询结果貌似不是我们想要的效果，理想，一目了然的效果应该是，以日期为单位，日期底下正面多少，负面多少： // [ // { "publishDate" : "2014-03-09" , "evalInfo" : [{"evaluate" : 0 , "totalNum" : 101}, {"evaluate" : 1 , "totalNum" : 44}]} // { "publishDate" : "2014-03-12" , "evalInfo" : [{"evaluate" : 0 , "totalNum" : 11}, {"evaluate" : 1 , "totalNum" : 32}]}, // …… // ] // 无奈本人功力尚浅，查了N多资料，各种论坛，中文的，英文的都查了，就是找不到Spring Data MongoDB 分组方法 // ，所以就引出了testAggregation5 }
7、使用原生态mongodb语法，按照时间（字符串）分组，整合查询结果，使用$push命令
/** * * 功能：使用原生态mongodb语法，按照时间（字符串）分组，整合查询结果，使用$push命令 * 参数： * 创建人：OnTheRoad_Lee * 修改人：OnTheRoad_Lee * 最后修改时间：2014-5-26 */ public void testAggregation5() { /* Group操作*/ String groupStr = "{$group:{_id:{'evaluate':'$eval','ptimes':{$substr:['$ptimes',0,10]}},totalNum:{$sum:1}}}"; DBObject group = (DBObject) JSON.parse(groupStr); /* Reshape Group Result*/ DBObject projectFields = new BasicDBObject(); projectFields.put("ptimes", "$_id.ptimes"); projectFields.put("evalInfo", new BasicDBObject("evaluate","$_id.evaluate").append("totalNum", "$totalNum")); DBObject project = new BasicDBObject("$project", projectFields); /* 将结果push到一起*/ DBObject groupAgainFields = new BasicDBObject("_id", "$ptimes"); groupAgainFields.put("evalInfo", new BasicDBObject("$push", "$evalInfo")); DBObject reshapeGroup = new BasicDBObject("$group", groupAgainFields); /* 查看Group结果 */ AggregationOutput output = mongoTemplate.getCollection("news").aggregate(group, project, reshapeGroup); System.out.println(output.getCommand()); // { "aggregate" : "news" , "pipeline" : [ { "$group" : { "_id" : { "evaluate" : "$eval" , "ptimes" : { "$substr" : [ "$ptimes" , 0 , 10]}} , "totalNum" : { "$sum" : 1}}} , { "$project" : { "ptimes" : "$_id.ptimes" , "evalInfo" : { "evaluate" : "$_id.evaluate" , "totalNum" : "$totalNum"}}} , { "$group" : { "_id" : "$ptimes" , "evalInfo" : { "$push" : "$evalInfo"}}}]} System.out.println(output.getCommandResult()); // { "serverUsed" : "localhost/127.0.0.1:47017" , "result" : [ // { "_id" : "2014-04-24" , "evalInfo" : [ { "evaluate" : 1 , "totalNum" : 67} , { "evaluate" : 0 , "totalNum" : 76}]} // , { "_id" : "2014-02-05" , "evalInfo" : [ { "evaluate" : 1 , "totalNum" : 70} , { "evaluate" : 0 , "totalNum" : 84}]} // , { "_id" : "2014-03-21" , "evalInfo" : [ { "evaluate" : 0 , "totalNum" : 82} , { "evaluate" : 1 , "totalNum" : 89}]}}……] // , "ok" : 1.0} }
来源：oschina
链接：https://my.oschina.net/u/1757911/blog/2250428