使用条件查询从 Spring 数据 Mongo 中获取结果时更新文档中的字段之一

Update one of the field in documents while fetching results using criteria query from Spring Data Mongo

我正在使用条件查询从 MongoDB 中获取文档。我在这里的要求是我想通过使用 Spring 数据 Mongo 中的条件查询子文档来更新父文档值的字段。父文档是评论,嵌套文档是回复。我可以使用以下代码获取评论列表以及子文档,

Data Set:
{
"_id" : ObjectId("5cb726937a7148376094d393"),
"_class" : "vzi.cpei.Comments",
"text" : "first comment on money control",
"replies" : [        
    {
        "_id" : "3cfef1cd-e0da-4883-86a4-17b223639087",
        "text" : "extract the traces",
        "status" : true
    },
    {
        "_id" : "3cfef1cd-e0da-4883-86a4-17b153690087",
        "text" : "replied deiberate",
        "status" : false
    },
    {
        "_id" : "3cfef1cd-e0da-4883-86a4-17b153139087",
        "text" : "Bgm",
        "status" : true
    }],
}
Response DTO:
      
      public class CommentsDTO{
      private String id;
      private String text;
      private List<Replies> replies;
      private Integer totalReplies;
    }

使用 Spring 数据 mongo 条件查询,Spring 中编写的代码,

    Query query = new Query();
    Criteria criteria =Criteria.where("_id").is(new ObjectId("5efe3d1f8a2ef008249f72d9"));
    query.addCriteria(criteria);
    List<Comments> comments = mongoOps.find(query,"Comments", 
  CommentsDTO.class);
    return comments;

在结果中,我想用状态为 true 的回复总数更新字段 repliesCount,因此预期输出应该是,

{
"_id" : ObjectId("5cb726937a7148376094d393"),
"_class" : "vzi.cpei.Comments",
"text" : "first comment on money control",
"totalReplies" : 2
"replies" : [        
    {
        "_id" : "3cfef1cd-e0da-4883-86a4-17b223639087",
        "text" : "extract the traces",
        "status" : true
    },
    {
        "_id" : "3cfef1cd-e0da-4883-86a4-17b153690087",
        "text" : "replied deiberate",
        "status" : false
    },
    {
        "_id" : "3cfef1cd-e0da-4883-86a4-17b153139087",
        "text" : "Bgm",
        "status" : true
    }],
}

我完全不知道在抓取的时候在哪里做这个操作。

您需要执行MongoDB aggregation

pipeline

的解释
  1. 我们应用 $match 阶段来过滤结果(类似于 .find(query)
  2. 我们应用 $project 来转换文档结构,并包含根据 replies
  3. 计算的 totalReplies 字段

Shell

db.Comments.aggregate([
  {
    $match: {
      "_id": ObjectId("5cb726937a7148376094d393")
    }
  },
  {
    $project: {
      _class: 1,
      replies: 1,
      totalReplies: {
        $size: {
          "$filter": {
            input: "$replies.status",
            as: "status",
            cond: "$$status"
          }
        }
      }
    }
  }
])

MongoPlayground

Spring-Mongo

import org.springframework.data.mongodb.core.aggregation.Aggregation;
import org.springframework.data.mongodb.core.aggregation.AggregationResults;
import org.springframework.data.mongodb.core.aggregation.ArrayOperators;
import static org.springframework.data.mongodb.core.aggregation.Aggregation.*;
...

Aggregation agg = Aggregation.newAggregation(
        match(Criteria.where("_id").is(new ObjectId("5cb726937a7148376094d393"))),
        project("_id", "_class", "replies").and(ArrayOperators.Size.lengthOfArray(
                ArrayOperators.Filter.filter("replies.status").as("filter").by("$$filter")
                )).as("totalReplies"));

//System.out.println(agg);
return mongoOps.aggregate(agg, mongoOps.getCollectionName(CommentsDTO.class), CommentsDTO.class);

编辑: 旧版本 Spring-boot 1.4.2

project("_id", "_class", "replies")
        .and(AggregationFunctionExpressions.SIZE
            .of(new BasicDBObject("$filter",
                new BasicDBObject("input", "$replies.status")
                    .append("as", "status")
                    .append("cond", "$$status"))))
        .as("totalReplies")