一、BasicDBObject
整個(gè)聚合查詢(xún)是統(tǒng)計(jì)用戶(hù)的各種狀態(tài)下的用戶(hù)數(shù)量為場(chǎng)景：
1.篩選條件：
date為查詢(xún)?nèi)掌冢?/p>
|
1
2
3
|
BasicDBObject Query = new BasicDBObject(); Query.put("time",new BasicDBObject("$gte", date + " 00:00:00") .append("$lte", date + " 23:59:59")); |
如果有多個(gè)條件：直接加Query.put("status", 0);
如果有OR篩選：
|
1
2
3
4
|
BasicDBList values = new BasicDBList();values.add(new BasicDBObject("status", new BasicDBObject("$exists",false)));values.add(new BasicDBObject("status", 0));Query.put("$or", values); |
其中 new BasicDBObject("$exists",false)) 就是“status”不存在（mongodb文檔可不等于實(shí)體的字段）
2.拼接match
|
1
|
DBObject Match = new BasicDBObject("$match", Query); |
將上一步的Query放入macth篩選中
3.拼接分組條件
此步相當(dāng)于SQL的Group By
|
1
2
|
BasicDBObject GroupBy = new BasicDBObject();// 分組條件GroupBy.put("userId", "$userId");//根據(jù)用戶(hù)ID來(lái)分組 |
同第一步的篩選一樣，如果有多個(gè)分組可直接添加在后面：GroupBy.put("status", "$status");
4.分組后查詢(xún)
首先把分組拼接上：
|
1
|
DBObject Select = new BasicDBObject("_id", GroupBy); |
如果我們要對(duì)分組后的數(shù)據(jù)進(jìn)行操作：（這里是用戶(hù)的統(tǒng)計(jì)數(shù)量）
|
1
|
Select.put("count", new BasicDBObject("$sum", 1)); |
5.拼接$group
|
1
|
DBObject Group = new BasicDBObject("$group", Select); |
至此。我們拼接，mongodb的語(yǔ)句到此結(jié)束
下面我們開(kāi)始查詢(xún)操作：
|
1
2
3
4
|
import com.mongodb.AggregationOutput;import org.springframework.data.mongodb.core.MongoTemplate;AggregationOutput Output = mongoTemplate.getCollection("這里是mongodb的數(shù)據(jù)表名") .aggregate(Match, Group); |
讀取查詢(xún)結(jié)果：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
|
Iterable<DBObject> map = Output.results();//遍歷mapfor (DBObject dbObject : map) { //這里讀取到的map為{"_id":{"userId":1,"status"0}}... Map<String, Object> resultMap = (Map<String, Object>) dbObject.get("_id"); //這里截取掉.0 Integer userId = Integer.parseInt(CommUtil.toString(ausgMap.get("userId")).replace(".0","")); //在這里我們就可以對(duì)我們的數(shù)據(jù)進(jìn)行操作了； Integer count = Integer.parseInt(CommUtil.toString(dbObject.get("count"))); } |
二、DBObject
不多說(shuō)，如果大家覺(jué)得第一種方法特別長(zhǎng)篇大論，那么下面我給大家?guī)?lái)一種代碼更簡(jiǎn)潔的聚合查詢(xún)：
直接上代碼吧。
1.注意導(dǎo)入的包:
|
1
|
import com.mongodb.DBObject; import com.mongodb.util.JSON; |
2.$macth:
(beginDay-查詢(xún)開(kāi)始日期；endDay-查詢(xún)結(jié)束日期)
|
1
2
|
DBObject match = (DBObject) JSON.parse("{$match:{'status':'1','time': {$gte:'"+beginDay+"',$lte:'"+endDay+"'}}}"); |
3.$group:
（以u(píng)serId分組，統(tǒng)計(jì)用戶(hù)數(shù)量，累加amout字段值，取amout字段最大值）
|
1
2
|
DBObject group = (DBObject) JSON.parse("{$group:{_id:'$userId', count:{$sum:1},total:{$sum:'$amount'},maxSigle:{$max:'$amount'}}}"); |
條件拼接完，開(kāi)始查詢(xún)：
|
1
2
3
|
List<DBObject> list = new ArrayList<DBObject>();list.add(match); list.add(group);AggregationOutput Output = mongoTemplate.getCollection("表名").aggregate(list); |
遍歷結(jié)果就不用在說(shuō)了。上面已經(jīng)詳細(xì)描述了
第二種方法是不是比一種方式簡(jiǎn)單多了。但是比較考驗(yàn)自身對(duì)mongodb語(yǔ)句熟悉性。
【擴(kuò)展】
接著上面的方法進(jìn)行擴(kuò)展
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
|
//status 為0或者為1 ; type 不為11；time在beginDay ~ endDay時(shí)間段內(nèi)DBObject match = (DBObject) JSON.parse("{$match:{'status':{$in:['0','1']}, type:{$ne:11},'time':{$gte:'"+beginDay+"',$lte:'"+endDay+"'}}}");//以用戶(hù)Id分組，統(tǒng)計(jì)查詢(xún)次數(shù)，取最后一次time的時(shí)間DBObject group = (DBObject) JSON.parse("{$group:{_id:'$userId', count:{$sum:1},lastTime:{$max:'$time'} }}");//在上一步統(tǒng)計(jì)出的結(jié)果中篩選次數(shù)大于100的數(shù)據(jù)DBObject groupMatch = (DBObject) JSON.parse("{$match:{count:{$gte:100}}}");//$project----查詢(xún)結(jié)果中需要顯示哪些字段，顯示則設(shè)置為1。如下需要不顯示_id 域（字段），則需如下指定：//db.集合名.aggregate( [ { $project : { _id: 0, count: 1 , lastSuccTime: 1 } } ] )DBObject project = (DBObject) JSON.parse("{$project:{_id:1,count:1,lastSuccTime:1}}");DBObject sort = (DBObject) JSON.parse("{$sort:{'count':-1}}");//排序List<DBObject> list = new ArrayList<DBObject>();list.add(match);list.add(group);list.add(groupMatch);list.add(project);list.add(sort);AggregationOutput catchOutPut = mongoTemplate.getCollection("表名") .aggregate(list);//查詢(xún)結(jié)果 |
三、BasicDBObject+腳本語(yǔ)句
|
1
2
3
4
5
6
7
8
9
10
|
BasicDBObject groupIndex = new BasicDBObject();BasicDBObject distinctQuery = new BasicDBObject();distinctQuery.put("userId", new BasicDBObject("$in", userIds));//UserIds數(shù)組BasicDBObject initIndex = new BasicDBObject();initIndex.put("count", 0);//給count賦初值// 腳本（doc代表數(shù)據(jù)庫(kù)的數(shù)據(jù).prev代表查詢(xún)結(jié)果//prev.count這里的count就是上一步initIndex的count）String reduce = "function(doc, prev) {if(doc.status==0){prev.count+= 1;};}";List<Map<String, Object>> basicDBList = (List<Map<String, Object>>) mongoTemplate. getCollection("Collection").group(groupIndex, distinctQuery, initIndex, reduce, null); |
最后循環(huán)遍歷List即可得到結(jié)果【PS：group這種聚合方式有個(gè)限制條件--->超過(guò)20000數(shù)據(jù)就會(huì)報(bào)錯(cuò)】
以上就是我在普通業(yè)務(wù)處理中用到mongodb聚合處理。如果有更好的方法或是有需要改進(jìn)的地方，歡迎大家給我留言~
原文鏈接：https://juejin.im/post/5ab8a2306fb9a028d9372666
