SpringData MongoDb，如何计算查询的不同部分？

Question

我正在SpringBootAPI中使用mongoDb进行分页搜索。

对于客户搜索路径，我正在构建一个查询，根据用户输入使用一系列条件。然后我进行计数以显示结果总数（以及计算出的关联页面数）

  Long total = mongoTemplate.count(query, MyEntity.class);

然后我做分页查询只返回当前页面结果

query.with(PageRequest.of(pagination.getPage(), pagination.getPageSize()));
query.with(Sort.by(Sort.Direction.DESC, "creationDate"));
List<MyEntity> listResults = mongoTemplate.find(query, MyEntity.class);

这一切工作得很好。现在在我的总结果上，我经常有相同用户的多个结果，我想显示分页列表中的那些，但我也想显示一个新的计数器，其中包含搜索中的总不同用户。

我看到了发现。不同的参数

mongoTemplate.findDistinct(query, "userId", OnboardingItineraryEntity.class, String.class);

但是我不想检索一个巨大的列表并对其进行计数。有没有一种方法可以轻松做到：

    mongoTemplate.countDistinct(query, "userId", OnboardingItineraryEntity.class, String.class);

因为我有大量的条件，所以我觉得必须从头开始重建聚合对象很难过？

另外一个问题是，有时userId将为null，是否有一种简单的方法在一个查询中计算null的不同（非null）数？或者我需要做一个查询，当我在userId为null的情况下添加一个额外的条件时，对其进行计数，然后对all进行不同的计数，并在我的代码中手动地将它们相加（减1）。

Answer 1

MongoDB聚合通过几种方式解决了这个问题。

db.myEntity.aggregate([
  {$match:...}, //add here MatchOperation
  {
    "$group": {
      "_id": {
        "$type": "$userId"
      },
      "count": {
        "$sum": 1
      }
    }
  }
])

MongoPlayground

---Ouput---

[
  {
    "_id": "null", //null values
    "count": 2
  },
  {
    "_id": "missing", // if userId doesn't exists at all
    "count": 1
  },
  {
    "_id": "string", //not null values
    "count": 4
  }
]

db.myEntity.aggregate([
  {$match:...}, //add here MatchOperation
  {
    "$group": {
      "_id": "",
      "null": {
        $sum: {
          $cond: [
            {
              $ne: [{ "$type": "$userId"}, "string"]
            },
            1,
            0
          ]
        }
      },
      "nonNull": {
        "$sum": {
          $cond: [
            {
              $eq: [{ "$type": "$userId" }, "string"]
            },
            1,
            0
          ]
        }
      }
    }
  }
])

MongoPlayground

---输出---

[
  {
    "_id": "",
    "nonNull": 4,
    "null": 3
  }
]

db.myEntity.aggregate([
  {$match:...}, //add here MatchOperation
  {
    $facet: {
      "null": [
        {
          $match: {
            $or: [
              {
                userId: {
                  $exists: false
                }
              },
              {
                userId: null
              }
            ]
          }
        },
        {
          $count: "count"
        }
      ],
      "nonNull": [
        {
          $match: {
            $and: [
              {
                userId: {
                  $exists: true
                }
              },
              {
                userId: {
                  $ne: null
                }
              }
            ]
          }
        },
        {
          $count: "count"
        }
      ]
    }
  },
  {
    $project: {
      "null": {
        $ifNull: [
          {
            $arrayElemAt: [
              "$null.count",
              0
            ]
          },
          0
        ]
      },
      "nonNull": {
        $ifNull: [
          {
            $arrayElemAt: [
              "$nonNull.count",
              0
            ]
          },
          0
        ]
      }
    }
  }
])

MongoPlayground

注意：尝试这些解决方案中的任何一种，如果您在创建MongoDB聚合时遇到任何问题，请告诉我。