最近在检索MongoDB的数据时需要用到分组操作,由于没有现成的说明文档可参考,只能是在代码中不断调试、摸索前进;目前已现实了Java对MongoDB的分组操作,并统计各个分组的数量。现通过示例详细解析,步骤如下流程所示:
(1)测试环境条件准备:MongoDB的版本为2.0.1,如下图所示
(2)测试数据源准备:MongoDB的数据源示例如下图所示
(3)测试示例过程:
- 最初测试过程
1、示例代码
public void test(String guid) {
String groupKey = "sex"; // 按性别来分组
GroupBy groupBy = new GroupBy(groupKey);
final String COUNT = "count"; // 统计数量值的key,可以任意字符串,例如total
DBObject initial = new BasicDBObject();
initial.put(COUNT, 0); // 初始值为0
// reduce必须要定义,否则调用MongoDB的group方法时会抛出异常
String reduce = "function (document, output) { "
+ " output." + COUNT + "++; " // 统计数量值的key自增
+ "}";
// 可以不用定义finalize变量
String finalize = "function (output) {"
+ "return output;"
+ "}";
groupBy = groupBy.reduceFunction(reduce); // 必须要设置,不能为null,该函数的含义是在分组操作时定义一个操作文档的聚合函数
groupBy = groupBy.collation(null); // 设置为null吧
groupBy = groupBy.finalizeFunction(finalize); // 可以设置为null,该函数的含义是在group函数返回最终值之前,定义一个运行每个分组的结果集的函数
groupBy = groupBy.initialDocument(initial.toString()); // 初始化分组统计数量
String criteriaKey = "guid"; // 过滤的条件
String criteriaValue = guid; // 过滤值
Criteria criteria = Criteria.where(criteriaKey).is(criteriaValue);
String collectionName = "test_mongodb"; // MongoDB中数据集合名称
GroupByResults<Map> mapGroupByResults = mongoTemplate.group(criteria, collectionName, groupBy, Map.class);
logger.info(mapGroupByResults.toString());
}
变量groupBy的属性值如下所示:
从上图中可以看出代码设置的值是有具体的含义的,可以明白这些值在group()函数中作用。
2、示例结果
从上图中我们可以看出分组结果已出来了,根据guid为"2"的条件下按性别进行分组:男的有2人,女的也有2人。而guid为"2"的记录总共有count=4.0,即4人。
需要注意的地方是,count=4.0的count属性是GroupByResult自带的,跟代码中设置的final String COUNT="count"没有关联的,在下面的测试结果中会发现这种差别。
- 修改测试过程
1、代码稍作修改
public void test(String guid) {
String groupKey = "sex";
GroupBy groupBy = new GroupBy(groupKey);
final String COUNT = "total"; // 修改为total
DBObject initial = new BasicDBObject();
initial.put(COUNT, 0);
String reduce = "function (document, output) { "
+ " output." + COUNT + "++; "
+ "}";
// String finalize = "function (output) {"
// + "return output;"
// + "}";
groupBy = groupBy.reduceFunction(reduce);
groupBy = groupBy.collation(null);
// groupBy = groupBy.finalizeFunction(finalize);
groupBy = groupBy.finalizeFunction(null);
groupBy = groupBy.initialDocument(initial.toString());
String criteriaKey = "guid";
String criteriaValue = guid;
Criteria criteria = Criteria.where(criteriaKey).is(criteriaValue);
String collectionName = "test_mongodb";
GroupByResults<Map> mapGroupByResults = mongoTemplate.group(criteria, collectionName, groupBy, Map.class);
logger.info(mapGroupByResults.toString());
}
变量groupBy的属性值如下所示:
从上图中可以看到细微的差别是:initial的value值由"{"count":0}”变为"{"total":0}",这是因为在代码中修改了COUNT的赋值;此外,finalize也被赋值为null了,说明它对结果不会有影响。
2、测试结果
测试结果跟第一种的测试结果是类似的,只不过是通过关键字total来代表统计数量值了。
------20191209闪
阅读是一种修养,分享是一种美德。