SpringBoot中MongoDB聚合管道查询操作$facet$lookup$unwind$group

前言、官方文档、MongoTemplate中的概念前言

最近在做基于SpringBoot的MongoDB的聚合管道操作，JSON语句不难写，但是理清楚逻辑、顺序很麻烦，而且在Java（Springboot）上操作聚合管道，部分操作符的使用不清楚，加之网上可以参考的示例很零散，很多不够直观全面。

所以在翻阅了官方文档和一些个人分享的技术文章后，自己做了测试验证，汇总了这篇笔记，分享一下基于SpringBoot的MongoDB的聚合管道操作。

主要是聚焦于理解MongoDB Template提供的两种实现聚合管道的操作，重点基于$group，$lookup, $unwind, $facet, 这几个操作符，实际代码中也有涉及到$match, $count, $sortByCount的使用。

同时梳理一下MongoDB template中的几个定义（见“MongoTemplate中的概念”），希望有助于大家。

请大家配合标题导航食用，风味更佳~

禁止转载！！！！！！

官方文档

https://www.mongodb.com/docs/manual/reference/operator/aggregation/facet/

MongoTemplate中的概念

MongoTemplate：官方提供的操作MongoDB的对象。位于：org.springframework.data.mongodb.core。使用的时候，需要注入。
Query：用于创建查询条件的对象。位于：package org.springframework.data.mongodb.core.query。使用时一般需要传入如”Criteria”构建的查询条件。
Criteria: 构建具体查询条件的对象，和Query位于同个包下。

AggregationOperation：聚合管道的操作对象，这是适用于Aggregate Pipeline Stages的操作，比如$group/$lookup/$unwind/$sort…….使用的时候，需要先构建对应的聚合操作，比如$group（需要构建具体操作），可以创建多个，最后一并传入到Aggregation对象中，再交给template去执行管道聚合。位于：
Aggregation：Pipeline stage的集合，也就是上面AggregationOperation的集合，把上面的所有聚合操作存在一起，template调用aggregate方法的时候，传入该对象。
以上类位于packageorg.springframework.data.mongodb.core.aggregation;

Aggregates: Pipeline stage操作对象。和Aggregation有几乎一样的功能，但是会更加灵活，一般除了预先提供的操作符，还可以自己传入Bson操作对象去灵活实现。整体的使用难度，比Aggregation可能高一些。

Bson、BsonDocument、BsonField:Bson我理解就是灵活的表达式，查询条件、聚合操作符之类的构建定义，都可以由它接收，并最后传给template的aggregate方法去执行聚合操作。BsonDocument则是Bson的具体实现，用于灵活构建表达式的对象。关于这部分，具体可以往下看。BsonField也是构建灵活的聚合表达式的一个类，比如快速地定义{“count”:{$sum:1} ，作为聚合操作的一部分传入到具体的聚合阶段中。
以上类位于 packagecom.mongodb.client.model; Bson/BsonDocument则是另外的包中。org.bson中。感兴趣自行去源码中查找。

开发环境和参考文档

JDK1.8 + Maven

SpringBoot(Springboot-starter-parent): 2.7.5

Mongodb(spring-boot-starter-data-mongodb) 4.6.1

参考文档：

http://www.mydlq.club/article/85/#1maven-%E5%BC%95%E5%85%A5%E7%9B%B8%E5%85%B3%E4%BE%9D%E8%B5%96

https://learnku.com/articles/61052

参考了以上网友的分享案例。

代码和案例$count 和$match操作符官方定义

$count: Passes a document to the next stage that contains a count of the number of documents input to the stage.

https://www.mongodb.com/docs/manual/reference/operator/aggregation/count/

就是统计当前stage（聚合管道操作的阶段）存在的文档数量。

$match: Filters the documents to pass only the documents that match the specified condition(s) to the next pipeline stage.

过滤符合条件的数据到下个pipeline stage。

语法

{ $count:  // 这里的名称随便写，最后显示出来的结果就是 xxx : 总数 } 

// match语法

{ $match: { } } // 就是传入查询语句Json格式

在MongoDB中操作的官方示例

// 数据

{ “_id” : 1, “subject” : “History”, “score” : 88 }
{ “_id” : 2, “subject” : “History”, “score” : 92 }
{ “_id” : 3, “subject” : “History”, “score” : 97 }
{ “_id” : 4, “subject” : “History”, “score” : 71 }
{ “_id” : 5, “subject” : “History”, “score” : 79 }
{ “_id” : 6, “subject” : “History”, “score” : 83 }

// 执行

db.scores.aggregate(

// 先用match查找匹配的文档，然后直接用count统计当前match阶段存在的文档数量。

[{$match: {score: {$gt: 80}}},
{$count: “passing_scores” // 这里的passing_scores 也可以是其他任意名称
}]
)

// 返回结果
{ “passing_scores” : 4 }

MongoTemplate中实现的Java代码

数据参考上面官方示例。以下分别是Aggregation和Aggregates的实现。任意一种都可以。

    /**     * @Author zaoyu     */    @Autowired    private MongoTemplate mongoTemplate;    private String DEMO_COLLECTION = "demo";    /**     * 用Aggregates和Bson构建聚合操作对象，用预先生成的MongoCollection对象调用aggregate执行即可。     */    @Test    public void testCountWithAggregates(){        MongoCollection collection = mongoTemplate.getCollection(DEMO_COLLECTION);        // Aggregates提供各种操作符，返回一个Bson对象。这里用match，然后用Filters来实现过滤条件的构建，也是返回一个Bson对象。        Bson matchBson = Aggregates.match(Filters.gt("score", 80));        // 直接用Aggregates的count方法，如果不传自定义的名称，默认用“count”接收。        Bson countBson = Aggregates.count("myCount");        // 构建一个List, 并把每一个聚合操作Bson加进去，最后传入aggregate方法中执行。        List bsonList = new ArrayList();        bsonList.add(matchBson);        bsonList.add(countBson);        AggregateIterable resultList = collection.aggregate(bsonList);        for (Document document : resultList) {            System.out.println("result is :" + document);        }    }    /**     * 用Aggregation集合接收聚合操作，用MongoTemplate对象直接调用aggregate，传入聚合操作集合、表名、映射对象。     */    @Test    public void testCountWithAggregation(){        // 构建查询match条件：分数大于80        MatchOperation matchOperation = Aggregation.match(Criteria.where("score").gt(80));        // 构建count操作，用“myCount”名称接收        CountOperation countOperation = Aggregation.count().as("myCount");        // 传入多个aggregation（聚合操作），用Aggregation对象接收。        Aggregation aggregation = Aggregation.newAggregation(matchOperation, countOperation);        // 直接用mongoTemplate调用aggregate方法，传入aggregation集合，表名，还有用什么对象接收数据，这里我用Document接收，不再建类。        AggregationResults resultList = mongoTemplate.aggregate(aggregation, DEMO_COLLECTION, Document.class);        for (Document document : resultList) {            System.out.println("result is :" + document);        }    }

// 以上2个方法的输出结果一样，如下。

result is :Document{{myCount=4}}

$group 操作符官方定义

The$groupstage separates documents into groups according to a “group key”. The output is one document for each unique group key. A group key is often a field, or group of fields. The group key can also be the result of an expression. Use the_idfield in the$grouppipeline stage to set the group key.

大概意思就是把文档做分组。输出的格式是一条数据有一个唯一的分组键。这里可以简单类比mysql 的group by分组。

语法

{  $group:    {      _id: , // 用来分组的字段      : {  :  }, // 对某字段做处理 accumulator操作。       ...    } }

在MongoDB中操作的官方示例

// 插入数据db.sales.insertMany([  { "_id" : 1, "item" : "abc", "price" : NumberDecimal("10"), "quantity" : NumberInt("2"), "date" : ISODate("2014-03-01T08:00:00Z") },  { "_id" : 2, "item" : "jkl", "price" : NumberDecimal("20"), "quantity" : NumberInt("1"), "date" : ISODate("2014-03-01T09:00:00Z") },  { "_id" : 3, "item" : "xyz", "price" : NumberDecimal("5"), "quantity" : NumberInt( "10"), "date" : ISODate("2014-03-15T09:00:00Z") },  { "_id" : 4, "item" : "xyz", "price" : NumberDecimal("5"), "quantity" :  NumberInt("20") , "date" : ISODate("2014-04-04T11:21:39.736Z") },  { "_id" : 5, "item" : "abc", "price" : NumberDecimal("10"), "quantity" : NumberInt("10") , "date" : ISODate("2014-04-04T21:23:13.331Z") },  { "_id" : 6, "item" : "def", "price" : NumberDecimal("7.5"), "quantity": NumberInt("5" ) , "date" : ISODate("2015-06-04T05:08:13Z") },  { "_id" : 7, "item" : "def", "price" : NumberDecimal("7.5"), "quantity": NumberInt("10") , "date" : ISODate("2015-09-10T08:43:00Z") },  { "_id" : 8, "item" : "abc", "price" : NumberDecimal("10"), "quantity" : NumberInt("5" ) , "date" : ISODate("2016-02-06T20:20:13Z") },])

// 执行group，这里还加上了match 和 project和sort。一并使用展示。

db.getCollection("sales").aggregate(  // 第一个聚合管道：过滤出日期在2014-01-01到2015-01-01之间的数据[ {    $match : { "date": { $gte: new ISODate("2014-01-01"), $lt: new ISODate("2015-01-01") } }  }, // 第二个聚合管道：处理一下日期格式，方便等下做group。   {$project:      {quantity:1,          price:1,         myDate:{"$dateToString":{format: "%Y-%m-%d", date: "$date"}} }}, // 第三个聚合管道：分组统计，先按照日期分组，再统计每天的销售数量。         {$group:{_id:"$myDate",     perDayQuantity:{$sum:"$quantity"},    myCount: {$sum:1}   }}, // 第四个聚合管道：按照每日销售数量降序排序 {$sort:{"perDayQuantity":-1}}])

MongoTemplate中实现的Java代码

    /**
　　　* @Authorzaoyu     * Aggregation 实现match, group, sort。     */    @Test    public void testGroupAggregations(){        // 第一阶段，过滤查询日期介于14-1-1~15-1-1之间的数据，用Aggregation实现类MatchOperation接收。        MatchOperation match = Aggregation.match(Criteria                .where("date").gte(Instant.parse("2014-01-01T08:00:00.000Z"))                .andOperator(Criteria.where("date").lte(Instant.parse("2015-01-01T08:00:00.000Z"))));        // 第二阶段，处理一下日期格式，方便等下做group，用ProjectionOperation接收，也是Aggregation的实现类。        ProjectionOperation project = Aggregation.project("quantity", "price")                .andExpression("{\"$dateToString\":{format: \"%Y-%m-%d\", date: \"$date\"}}").as("myDate");        // 第三阶段，分组统计，先按照日期分组，再统计每天的销售数量。        GroupOperation group = Aggregation.group("myDate")                .sum("quantity").as("perDayQuantity")                // 这里是计算文档条数                .count().as("myCount");        // 第四阶段，排序。按照perDayQuantity字段升序展示。        SortOperation sort = Aggregation.sort(Sort.Direction.ASC, "perDayQuantity");        // 用newAggregation接收以上多个阶段的管道聚合指令，执行，得到结果。        Aggregation aggregations =Aggregation.newAggregation(match, project, group, sort);        AggregationResults resultList = mongoTemplate.aggregate(aggregations, SALES_COLLECTION, Document.class);        for (Document document : resultList) {            System.out.println("result is :" + document);        }    }
// 返回结果

result is :Document{{_id=2014-03-01, perDayQuantity=3, myCount=2}}
result is :Document{{_id=2014-03-15, perDayQuantity=10, myCount=1}}
result is :Document{{_id=2014-04-04, perDayQuantity=30, myCount=2}}

$unwind 操作符官方定义

Deconstructsanarrayfieldfromtheinputdocumentstooutputadocumentforeachelement.Eachoutputdocumentistheinputdocumentwiththevalueofthearrayfieldreplacedbytheelement.

大概意思就是把输入文档的数组字段按元素一个个拆分出来，并和原来的数据一并形成一条新文档输出。好比原来10条数据，其中每条数据都有长度为3的数组，那么拆出来（在元素不重复的情况下），会得到30条数据。

语法

{ $unwind:  }  和{  $unwind:    {      path: ,  // path是固定名称，沿用即可。  是数组字段，就是你要拆分的字段（值得是一个数组，不然没有意义）      includeArrayIndex: ,      preserveNullAndEmptyArrays: <boolean>    }}

在MongoDB中操作的官方示例

// 插入数据db.inventory.insertOne({ "_id" : 1, "item" : "ABC1", sizes: [ "S", "M", "L"] })// 执行unwind操作，这里是把 sizes字段的数组拆分出来 db.inventory.aggregate( [ { $unwind : "$sizes" } ] )// 执行结果  可以看到每条数据的sizes不再是list,而是具体的元素。{ "_id" : 1, "item" : "ABC1", "sizes" : "S" }{ "_id" : 1, "item" : "ABC1", "sizes" : "M" }{ "_id" : 1, "item" : "ABC1", "sizes" : "L" }

注意，如果要拆分的字段是一个空数组或者null，那么实际输出的数据，不会包含那条数据。如下示例。

// 插入多条数据，这里还放了三条特殊数据，一个是空集合，一个是null，一个是没有要拆分的字段 sizesdb.clothing.insertMany([  { "_id" : 1, "item" : "Shirt", "sizes": [ "S", "M", "L"] },  { "_id" : 2, "item" : "Shorts", "sizes" : [ ] },  { "_id" : 3, "item" : "Hat", "sizes": "M" },  { "_id" : 4, "item" : "Gloves" },  { "_id" : 5, "item" : "Scarf", "sizes" : null }])// 执行$unwinddb.clothing.aggregate( [ { $unwind: { path: "$sizes" } } ] )// 返回结果， 可以看到，sizes值为空数组和null的id=2以及id=5的数据都没有展示出来，同时没有该字段的id=4，也没有展示出来。{ _id: 1, item: 'Shirt', sizes: 'S' },{ _id: 1, item: 'Shirt', sizes: 'M' },{ _id: 1, item: 'Shirt', sizes: 'L' },{ _id: 3, item: 'Hat', sizes: 'M' }

MongoTemplate中实现的Java代码

以下分别是Aggregation和Aggregates的实现

    /**     * @Author zaoyu     * Aggregation 实现$unwind     */    @Test    public void testUnwindAggregations() {        String CLOTHING_COLLECTION = "clothing";        // 调用Aggregation中的unwind的聚合操作符        UnwindOperation unwind = Aggregation.unwind("sizes");        // 用newAggregation接收管道聚合指令，执行，得到结果。        Aggregation aggregations =Aggregation.newAggregation(unwind);        // mongoTemplate 直接调用aggregate方法，传入Aggregation对象，基于的表，映射类（这里简单化，我用Document）        AggregationResults resultList = mongoTemplate.aggregate(aggregations, CLOTHING_COLLECTION, Document.class);        for (Document document : resultList) {            System.out.println("result is :" + document);        }    }    /**     * @Author zaoyu     * Aggregates/Bson 实现$unwind     */    @Test    public void testUnwindAggregates(){        String CLOTHING_COLLECTION = "clothing";        // 调用Aggregates的unwind聚合操作符 注意，Aggregates这里需要传入$        Bson unwindBson = Aggregates.unwind("$sizes");        // 建一个List 把unwindBson传进去        List bsonList = new ArrayList();        bsonList.add(unwindBson);        // mongoTemplate先获得对应的collection对象，然后调用aggregate，传入List 获得结果        MongoCollection collection = mongoTemplate.getCollection(CLOTHING_COLLECTION);        AggregateIterable resultList = collection.aggregate(bsonList);        for (Document document : resultList) {            System.out.println("result is :" + document);        }    }

$lookup 操作符官方定义

Performsaleftouterjointoacollectioninthesamedatabasetofilterindocumentsfromthe”joined”collectionforprocessing.The$lookupstageaddsanewarrayfieldtoeachinputdocument.Thenewarrayfieldcontainsthematchingdocumentsfromthe”joined”collection.

The$lookupstagepassesthesereshapeddocumentstothenextstage.

StartinginMongoDB5.1,$lookupworksacrossshardedcollections.

其实可以简单理解类比Mysql的子查询。会把另外一张表匹配的数据，作为一个数组存入到当前数据中，需要自定义一个字段来接收显示。

类比如下的sql语句

SELECT *, <output array field>FROM collectionWHERE <output array field> IN (   SELECT *   FROM <collection to join>   WHERE <foreignField> = <collection.localField>);

【特别注意】如果当前DB是集群部署，那么在DB版本为5.1之前的情况，$lookup是不会生效的。如果你数据库是集群的，然后又要用$lookup，一定要检查版本是否大于等于5.1，否则是查不出来的。前阵子不知道这个，一直没有头绪为什么数据查不出来。

语法

{   $lookup:     {       from: ,  // 要联表查的表名       localField: , // 当前表的要和联表关联的字段       foreignField: , // 要被关联表的外键字段        as:  // 定义一个字段接收匹配关联的数据     }}

在MongoDB中操作的官方示例

// 插入表orders数据  db.orders.insertMany( [   { "_id" : 1, "item" : "almonds", "price" : 12, "quantity" : 2 },   { "_id" : 2, "item" : "pecans", "price" : 20, "quantity" : 1 },   { "_id" : 3  }] )// 插入表inventory数据db.inventory.insertMany( [   { "_id" : 1, "sku" : "almonds", "description": "product 1", "instock" : 120 },   { "_id" : 2, "sku" : "bread", "description": "product 2", "instock" : 80 },   { "_id" : 3, "sku" : "cashews", "description": "product 3", "instock" : 60 },   { "_id" : 4, "sku" : "pecans", "description": "product 4", "instock" : 70 },   { "_id" : 5, "sku": null, "description": "Incomplete" },   { "_id" : 6 }] )

执行代码

db.orders.aggregate( [ // db.orders 表示基于orders做聚合操作   {     $lookup:       {         from: "inventory",   // 联表inventory          localField: "item",  // 当前orders的字段         foreignField: "sku", // inventory中的sku字段，和orders的item关联          as: "inventory_docs" // 定义一个字段名接收 inventory中sku 和orders的item相同的数据，数组形式。        }  }] )

返回结果

{   "_id" : 1,   "item" : "almonds",   "price" : 12,   "quantity" : 2,   "inventory_docs" : [  // 这个inventory_docs字段就是自己命名的字段，存储着来自inventory的数据      { "_id" : 1, "sku" : "almonds", "description" : "product 1", "instock" : 120 }   ]}{   "_id" : 2,   "item" : "pecans",   "price" : 20,   "quantity" : 1,   "inventory_docs" : [      { "_id" : 4, "sku" : "pecans", "description" : "product 4", "instock" : 70 }   ]}{   "_id" : 3,   "inventory_docs" : [      { "_id" : 5, "sku" : null, "description" : "Incomplete" },      { "_id" : 6 }   ]}

MongoTemplate中实现的Java代码

以下分别是Aggregation和Aggregates的实现

    /**     * @Author zaoyu     * Aggregation 实现$lookup     */    @Test    public void testLookupAggregations(){        String INVENTORY_COLLECTION = "inventory";        String ORDERS_COLLECTION = "orders";        // Aggregation类，直接可以调用lookup方法，传入要关联的表、当前表和关联表关联的字段、要关联的表的字段、自定义名称接收关联匹配的数据        LookupOperation lookup = Aggregation.lookup(INVENTORY_COLLECTION, "item", "sku", "inventory_docs");        // 用newAggregation接收管道聚合指令，执行，得到结果。        Aggregation aggregations =Aggregation.newAggregation(lookup);        // mongoTemplate 直接调用aggregate方法，传入Aggregation对象，基于的表，映射类（这里简单化，我用Document）        AggregationResults resultList = mongoTemplate.aggregate(aggregations, ORDERS_COLLECTION, Document.class);        for (Document document : resultList) {            System.out.println("result is :" + document);        }    }    /**     * @Author zaoyu     * Aggregates/Bson 实现$lookup     */    @Test    public void testLookupAggregates(){        String INVENTORY_COLLECTION = "inventory";        String ORDERS_COLLECTION = "orders";        // 这里用Aggregates类直接调用lookup，传入的参数和上面的Aggregations的lookup是一样的，只不过这里返回的结果是一个Bson对象。        Bson lookupBson = Aggregates.lookup(INVENTORY_COLLECTION, "item", "sku", "inventory_docs");        // 建一个List 把lookupBson传进去        List bsonList = new ArrayList();        bsonList.add(lookupBson);        // mongoTemplate先获得对应的collection对象，然后调用aggregate，传入List 获得结果        MongoCollection collection = mongoTemplate.getCollection(ORDERS_COLLECTION);        AggregateIterable resultList = collection.aggregate(bsonList);        for (Document document : resultList) {            System.out.println("result is :" + document);        }    }

返回结果

result is :Document{{_id=1.0, item=almonds, price=12.0, quantity=2.0, inventory_docs=[Document{{_id=1.0, sku=almonds, description=product 1, instock=120.0}}]}}result is :Document{{_id=2.0, item=pecans, price=20.0, quantity=1.0, inventory_docs=[Document{{_id=4.0, sku=pecans, description=product 4, instock=70.0}}]}}result is :Document{{_id=3.0, inventory_docs=[Document{{_id=5.0, sku=null, description=Incomplete}}, Document{{_id=6.0}}]}}

$facet 操作符官方定义

Processesmultipleaggregationpipelineswithinasinglestageonthesamesetofinputdocuments.Eachsub-pipelinehasitsownfieldintheoutputdocumentwhereitsresultsarestoredasanarrayofdocuments.

Inputdocumentsarepassedtothe$facetstageonlyonce.$facetenablesvariousaggregationsonthesamesetofinputdocuments,withoutneedingtoretrievetheinputdocumentsmultipletimes.

简单来说，就是facet可以实现在facet管道操作完成多个stage管道操作。减少获取输入文档的次数。

我个人觉得有种场景很适合使用facet：分页查询文档数据的同时，把符合查询条件的总数也查询出来的场景下，如果使用$facet，同时获取分页数据和总数，不用做两次数据库查询（分别查询分页数据和总数）。

语法

{ $facet:   {      : [ , , ... ],  // 这里outputpufield 是自己定义的用来接收stage集合返回的文档数据。        : [ , , ... ],  // 可以基于上一个Facet继续做facet      ...   }}

在MongoDB中操作的官方示例

// 数据  插入artwork 表中 { "_id" : 1, "title" : "The Pillars of Society", "artist" : "Grosz", "year" : 1926,  "price" : NumberDecimal("199.99"),  "tags" : [ "painting", "satire", "Expressionism", "caricature" ] }{ "_id" : 2, "title" : "Melancholy III", "artist" : "Munch", "year" : 1902,  "price" : NumberDecimal("280.00"),  "tags" : [ "woodcut", "Expressionism" ] }{ "_id" : 3, "title" : "Dancer", "artist" : "Miro", "year" : 1925,  "price" : NumberDecimal("76.04"),  "tags" : [ "oil", "Surrealism", "painting" ] }{ "_id" : 4, "title" : "The Great Wave off Kanagawa", "artist" : "Hokusai",  "price" : NumberDecimal("167.30"),  "tags" : [ "woodblock", "ukiyo-e" ] }{ "_id" : 5, "title" : "The Persistence of Memory", "artist" : "Dali", "year" : 1931,  "price" : NumberDecimal("483.00"),  "tags" : [ "Surrealism", "painting", "oil" ] }{ "_id" : 6, "title" : "Composition VII", "artist" : "Kandinsky", "year" : 1913,  "price" : NumberDecimal("385.00"),  "tags" : [ "oil", "painting", "abstract" ] }{ "_id" : 7, "title" : "The Scream", "artist" : "Munch", "year" : 1893,  "tags" : [ "Expressionism", "painting", "oil" ] }{ "_id" : 8, "title" : "Blue Flower", "artist" : "O'Keefe", "year" : 1918,  "price" : NumberDecimal("118.42"),  "tags" : [ "abstract", "painting" ] }// 执行$facet聚合db.artwork.aggregate( [  {    $facet: {        // 第一个Facet操作，按照tag分类：先用unwind拆分tags字段的数组值，交给下一个聚合 $sortByCount, 按照tags的个数排序。      "categorizedByTags": [        { $unwind: "$tags" },        { $sortByCount: "$tags" }      ],    // 第二个Facet操作，按照price分类：先过滤数据（只处理存在price数据的文档），然后执行$bucket按照价格区间分组 0~150，151~200， 201~300， 301~400这样。      "categorizedByPrice": [        { $match: { price: { $exists: 1 } } },        {          $bucket: {            groupBy: "$price",            boundaries: [  0, 150, 200, 300, 400 ],            default: "Other",            output: {              "count": { $sum: 1 },              "titles": { $push: "$title" }            }          }        }      ],// 第三个Facet， 按照years分类。 分成4个区间。      "categorizedByYears(Auto)": [        {          $bucketAuto: {            groupBy: "$year",            buckets: 4          }        }      ]    }  }])

MongoTemplate中实现的Java代码

注：以下代码的实现，数据来源参考上边的官方示例的数据， artwork表。请自行插入数据。

1. 使用Aggregation对象实现

    /**     * @Author zaoyu     * Aggregation 实现$facet     */    @Test    public void testFacetAggregations(){        String ARTWORK_COLLECTION = "artwork";        // Facet中第一组分类（categorizedByTags）的两个聚合操作unwind 和 sortByCount        UnwindOperation unwindForByTags = Aggregation.unwind("$tags");        SortByCountOperation sortByCountForByTags = Aggregation.sortByCount("$tags");        // Facet中第二组分类（categorizedByPrice）的聚合操作match 和 match        MatchOperation matchForByPrice = Aggregation.match(Criteria.where("price").exists(true));        // 分别传入bucket分组的字段price，设置区间值，并设置桶内条数统计和值（这里用titles接收title的值）        BucketOperation bucketForByPrice = Aggregation.bucket("$price")                .withBoundaries(0, 150, 200, 300, 400)                .withDefaultBucket("Other")                .andOutput("count").sum(1).as("count")                .andOutput("$title").push().as("titles");        // Facet中第三组分类 （categorizedByYears(Auto)）的聚合操作，按年自动分成4个区间。        BucketAutoOperation bucketForByYears = Aggregation.bucketAuto("$year", 4);        // Aggregation调用facet方法，按照组别分类顺序，把每一组的聚合操作和输出的名称传进去。        FacetOperation facetOperation = Aggregation.facet(unwindForByTags, sortByCountForByTags).as("categorizedByTags")                .and(matchForByPrice, bucketForByPrice).as("categorizedByPrice")                .and(bucketForByYears).as("categorizedByYears(Auto)");        // 把facetOperation传入newAggregation得到Aggregation对象，调用mongoTemplate的Aggregate方法执行得到结果        Aggregation aggregation = Aggregation.newAggregation(facetOperation);        AggregationResults resultList = mongoTemplate.aggregate(aggregation, ARTWORK_COLLECTION, Document.class);        for (Document document : resultList) {            System.out.println("result is :" + document);        }    }

2. 使用Aggregates实现

  /**     * @Author zaoyu     * Aggregates 实现$facet     */    @Test    public void testFacetAggregates() {        String ARTWORK_COLLECTION = "artwork";        // Facet中第一组分类（categorizedByTags）的两个聚合操作unwind 和 sortByCount        Bson unwindBsonForByTags = Aggregates.unwind("$tags");        Bson sortByCountBsonForByTags = Aggregates.sortByCount("$tags");        // 新建Facet对象，传入第一组分类的接收名称，以及在第一组分类中要做的聚合操作。        Facet categorizedByTags = new Facet("categorizedByTags", unwindBsonForByTags, sortByCountBsonForByTags);        // Facet中第二组分类（categorizedByPrice）的聚合操作match 和 match        Bson matchBsonForPrice = Aggregates.match(Filters.exists("price"));        // 这里面要新建BsonField构建 {"count": { $sum: 1 }  和 "titles": { $push: "$title" }} 作为第二组分类中$Bucket聚合操作中output值        BsonField countOutput = new BsonField("count", new Document("$sum", 1));        BsonField titleOutput = new BsonField("titles", new Document("$push", "$price"));        // 上面2个操作传入到BucketOption对象，最后传到bucket操作        BucketOptions bucketOptions = new BucketOptions().defaultBucket("Other").output(countOutput).output(titleOutput);        Bson bucketBsonForByPrice = Aggregates.bucket("$price", Arrays.asList(0, 150, 200, 300, 400), bucketOptions);        Facet categorizedByPrice = new Facet("categorizedByPrice", matchBsonForPrice, bucketBsonForByPrice);        // Facet中第三组分类 （categorizedByYears(Auto)）的聚合操作，按年自动分成4个区间。        Bson bucketAutoBsonForByYears = Aggregates.bucketAuto("$year", 4);        Facet categorizedByYears = new Facet("categorizedByYears", bucketAutoBsonForByYears);        // 新建一个List把每组分类的Facet对象传进去。        List facetList = new ArrayList();        facetList.add(categorizedByTags);        facetList.add(categorizedByPrice);        facetList.add(categorizedByYears);        // 调用Aggregates的facet方法，传入List得到最终Bson对象，并添加到Bson集合中。        Bson facetBson = Aggregates.facet(facetList);        List bsonList = new ArrayList();        bsonList.add(facetBson);        // 调用方法执行得到结果        MongoCollection collection = mongoTemplate.getCollection(ARTWORK_COLLECTION);        AggregateIterable resultList = collection.aggregate(bsonList);        for (Document document : resultList) {            System.out.println("result is :" + document);        }    }

最终返回结果，二者一样。

result is :Document{{categorizedByTags=[Document{{_id=painting, count=6}}, Document{{_id=oil, count=4}}, Document{{_id=Expressionism, count=3}}, Document{{_id=Surrealism, count=2}}, Document{{_id=abstract, count=2}}, Document{{_id=woodblock, count=1}}, Document{{_id=ukiyo-e, count=1}}, Document{{_id=satire, count=1}}, Document{{_id=caricature, count=1}}, Document{{_id=woodcut, count=1}}], categorizedByPrice=[Document{{_id=0, titles=[76.04, 118.42]}}, Document{{_id=150, titles=[199.99, 167.30]}}, Document{{_id=200, titles=[280.00]}}, Document{{_id=300, titles=[385.00]}}, Document{{_id=Other, titles=[483.00]}}], categorizedByYears=[Document{{_id=Document{{min=null, max=1902.0}}, count=2}}, Document{{_id=Document{{min=1902.0, max=1918.0}}, count=2}}, Document{{_id=Document{{min=1918.0, max=1926.0}}, count=2}}, Document{{_id=Document{{min=1926.0, max=1931.0}}, count=2}}]}}

小结

整体来说，MonogoDB官方提供了很详细的资料，但是对于Java 层面的操作，或者说SpringBoot层面的操作，文档就比较简单。

个人感觉而言，Aggregations提供的方法比较直接，更适合不太熟悉Springboot上操作Mongo的同学来使用，而Aggregates会更加灵活，但是需要你知道Document, BsonField, Bson之间的转换和获取。

希望这篇文章能帮到大家，有错漏之处，欢迎指正。

SpringBoot中MongoDB聚合管道查询操作$facet$lookup$unwind$group

最新关注

热文推荐

【MySQL】查询操作（基础篇）

ZoKrates+Remix在线实现zkSNARK零知识证明

npm : 无法将“npm”项识别为 cmdlet、函数、脚本文件或可运行程序的名称。请检查名称的拼写，如果包括路径，请确保路径正确，然后再试一次。

一文读懂机器学习分类全流程

[20231204]快速删除大量文件测试(rsync版本)4.txt

vue3 解决各场景 loading过度，避免白屏尴尬！

SpringBoot中MongoDB聚合管道查询操作$facet$lookup$unwind$group

相关文章

最新关注

热文推荐