1. 首页
  2. 开发

MongoDB的Group查询

  • Group

为了方便我还是把我的表结构贴上来:

和数据库一样group常常用于统计。MongoDB的group还有很多限制,如:返回结果集不能超过16M, group操作不会处理超过10000个唯一键,好像还不能利用索引[不很确定]。

Group****大约需要一下几个参数。

  1. key:用来分组文档的字段。和keyf两者必须有一个
  2. keyf:可以接受一个javascript函数。用来动态的确定分组文档的字段。和key两者必须有一个
  3. initial:reduce中使用变量的初始化
  4. reduce:执行的reduce函数。函数需要返回值。
  5. cond:执行过滤的条件。
  6. finallize:在reduce执行完成,结果集返回之前对结果集最终执行的函数。可选的

下面我用Java对他们做一些测试。

我们以 age 年龄统计集合中存在的用户。 Spring Schema 和上次的一样。有了 MongoTemplate 对象我们可以做所有事的。以 age 统计用户测试代码如:

@Test
    public void testGroupBy() throws Exception {
        String reduce = "function(doc, aggr){" +
                "            aggr.count += 1;" +
                "        }";
        Query query = Query.query(Criteria.where("age").exists(true));
        DBObject result = mongoTemplate.getCollection("person").group(new BasicDBObject("age", 1), 
                query.getQueryObject(), 
                new BasicDBObject("count", 0),
                reduce);
         
        Map map = result.toMap();
        System.out.println(map);
        for (Map.Entry o : map.entrySet()) {
            System.out.println(o.getKey() + "  " + o.getValue());
        }
    }

key 为 new BasicDBObject(“age”, 1)

cond 为: Criteria.where(“age”).exists(true) 。即用户中存在 age 字段的。

initial 为: new BasicDBObject(“count”, 0) ,即初始化 reduce 中人的个数为 count 为 0 。假如我们想在查询的时候给每个年龄的人增加 10 个假用户。我们只需要传入 BasicDBObject(“count”, 10).

reduce 为: reduce 的 javascript 函数

上面的执行输出如:

2  [age:23.0, count:1.0]
1  [age:25.0, count:1.0]
0  [age:24.0, count:1.0]

前面的是一个序号,是Mongo的java-driver加上去的。我们可以看到结果在后面。

不过你可能都觉得 reduce 这段代码用 Java 写的太繁琐了,要是和 Python 一样支持多行字符串多好啊。 我也烦。下面的例子我用 Groovy 写,不过我尽量写的贴近 Java 。

同样的 reduce ,用 Groovy 只需这样:

def reduce = """
        function(doc, aggr){
             aggr.count += 1;
        }
               """;

用 age 统计用户这是基本的需求了。下面我来几个高级点的。

我的表结构中用户的朋友 [myFriends] 是一个数组类型的, mongo 提供的查询中对数组查询时数组长度 $size 只能用来判断,却不能用来输出 [ 至少我没找到 ] 。那么我们用 group 操作来统计一下每个人有几个朋友。测试代码如:

@Test
    void testFriendGroupUserFriendCount() throws Exception {
        def reduce = """
                     function(doc, aggr){
                     aggr.manId = doc.manId;
                         doc.myFriends.forEach(function(z){
                              aggr.count += 1;
                         })
                      }
                      """;
  
        Query query = Query.query(Criteria.where("myFriends").exists(true));
        DBObject result = mongoTemplate.getCollection("person").group(
                new BasicDBObject("manId", 1),
                query.getQueryObject(),
                new BasicDBObject("count", 0),
                reduce);
  
        Map map = result.toMap();
        for (Map.Entry o : map.entrySet()) {
            System.out.println(o.getKey() + " ==> " + o.getValue());
        }
    }

执行结果如:

2 ==> [manId:345678, count:2.0]
1 ==> [manId:234567, count:2.0]
0 ==> [manId:123456, count:4.0]

上面的 reduce 中遍历文档的数组用了 forEach ,我记得好像 Javascript 中不能这么做吧? 我对 js 不熟,希望牛人解答下。

  上面的例子一直都没用到finallize,下面的测试我希望能用上。

我们统计每个人最喜欢水果是哪个?每个人都有 n 个自己的喜欢的水果, fruits.boost 是每个水果的权重。那么我们找出最喜欢的那个?

测试代码:

@Test
    void testGroupByFruitFinallize() throws Exception {
        def reduce = """
                function(doc, out) {
                    out.name = doc.manName;
                    for(i in doc.fruits) {
                       if(doc.fruits[i] in out.fruits) {
                           out.fruits[doc.fruits[i].fruitId]++;
                      } else {
                           out.fruits[doc.fruits[i].fruitId] = 1;
                       }
                    }
                }
                """;
  
        def finallizer = """
                function(out) {
                    var mostPopular = 0;
                    for(i in out.fruits) {
                        if(out.fruits[i] > mostPopular) {
                            out.fruitId = i;
                           mostPopular = out.fruits[i];
                        }
                    }
                   delete out.fruits;
                    return out;
                }
                """;
  
        Query query = new BasicQuery("{}");
        long time = System.currentTimeMillis();
        DBObject result = mongoTemplate.getCollection("person").group(new BasicDBObject("fruits", true),
                query.getQueryObject(),
                new BasicDBObject("fruits",  new BasicDBObject()),
                reduce,
                finallizer);
        System.out.println("use time: " + (System.currentTimeMillis() - time));
  
        Map map = result.toMap();
        for (Map.Entry o : map.entrySet()) {
            System.out.println(o.getKey() + "  " + o.getValue());
        }
    }

它的测试输出为: 2 [name:ZhenZi, fruitId:www] 1 [name:YangYan, fruitId:www] 0 [name:ZhenQin, fruitId:aaa]

OK ,他完成了。

原文地址 http://my.oschina.net/zhzhenqin/blog/99846

收藏

暂无评论

登录后可以进行评论。没有账号?马上注册