Spring MongoDB 聚合 DBRef

问题描述

我有 spring mongoDB 收集用户，数据如下：

    [{
  "_id": {
    "$oid": "5fd81fb277b245717d34c25e"
  },"userName": "banana","email": "banana","userRole": {
    "$ref": "roles","$id": {
      "$oid": "5fdc248b1846a46d71a71374"
    }
  }
},{
  "_id": {
    "$oid": "5fd81fd777b245717d34c25f"
  },"$id": {
      "$oid": "5fd6e784f03b1f5e76699ebb"
    }
  }
},{
  "_id": {
    "$oid": "5fdc24691846a46d71a71373"
  },{
  "_id": {
    "$oid": "5fe19f2f136f696e0bf83167"
  },"$id": {
      "$oid": "5fdc248b1846a46d71a71374"
    }
  }
}]

我尝试使用 Spring MongoDB 对其应用聚合

GroupOperation groupOperation = Aggregation.group("userRole.$id")
                .addToSet("email").as("email")
                .addToSet("firstName").as("firstName");

        Aggregation aggregation = Aggregation.newAggregation(groupOperation);

        AggregationResults<Map> result = mongoOperations
                .aggregate(aggregation,"user",Map.class);
        return result.getMappedResults();

但它不起作用。我的问题是 Spring MongoDB Aggregation 是否支持 DBref ？

解决方法

Spring data mongo 确实支持聚合，您可以在此处阅读更多相关信息 - https://www.baeldung.com/spring-data-mongodb-projections-aggregations

根据您提供的集合示例，我没有看到任何名为 firstName 的字段，您是否可能指的是 userName 并且基于您的聚合我提供的是否与此类似：

db.collection.aggregate({
  "$group": {
    "_id": {
      "userRole": "$userRole.$id"
    },"email": {
      "$addToSet": "$email"
    },"username": {
      "$addToSet": "$userName"
    }
  }
})

如果是这样，那么您可以直接将此管道添加到您的存储库方法中。像这样

public interface UserRepository extends MongoRepository<User,String> {
    
     @Aggregation(pipeline = {
        "{ \"$group\": { \"_id\": { \"userRole\": \"$userRole.$id\" },\"email\": { \"$addToSet\": \"$email\" },\"username\": { \"$addToSet\": \"$userName\" } } }"
     })
     AggregationResults<Result> findAndGroup();
}

*注意 - 这里的 Result 是具有等同于返回文档的字段的类

然后

@Autowired
private UserRepository userRepository;

userRepository.findAndGroup().getMappedResults();