使用Spring Data MongoDB进行复合ID的聚合查询

Question

我在使用聚合框架读取MongoDB文档时遇到了问题：我的结果中总是得到null的ID。这只会发生在具有复合ID的文档中。我尝试了各种版本的spring-data-mongodb（1.10.12、2.0.7），但结果相同。

实体定义类

    @Document(collection="entities")
    public class MyEntity {
    static class CompositeKey implements Serializable {
        private String stringKey;

        private Integer intKey;

        public CompositeKey(String stringKey, Integer intKey) {
            this.stringKey = stringKey;
            this.intKey = intKey;
        }

        public Integer getIntKey() {
            return intKey;
        }

        public String getStringKey() {
            return stringKey;
        }

        public String toString() {
            return "{" + stringKey + " - " + intKey + "}";
        }
    }

    @Id
    private CompositeKey id;

    private String param;

    public MyEntity() {}

    public MyEntity(String stringKey, Integer intKey) {
        id = new CompositeKey(stringKey, intKey);
    }

    public CompositeKey getId(){
        return id;
    }

    public void setId(CompositeKey id) {
        this.id = id;
    }

    public String getParam() {
        return param;
    }

    public void setParam(String param) {
        this.param = param;
    }
}

测试代码

public static void main(String[] args) {        
    MongoClient client = new MongoClient("127.0.0.1");
    SimpleMongoDbFactory factory = new SimpleMongoDbFactory(client, "aggTest");
    MongoTemplate mongoTemplate = new MongoTemplate(factory);

    MyEntity entity = new MyEntity();
    entity.setId(new MyEntity.CompositeKey("one", 1));
    entity.setParam("param1");
    mongoTemplate.save(entity);

    entity = new MyEntity();
    entity.setId(new MyEntity.CompositeKey("two", 2));
    entity.setParam("param2");
    mongoTemplate.save(entity);

    Criteria crit = Criteria.where("param").ne("param3");
    List<AggregationOperation> aggOpList = new ArrayList<AggregationOperation>();
    aggOpList.add(Aggregation.match(crit));

    System.out.println("Documents fetched with find: ");        
    for (MyEntity aggResult : mongoTemplate.find(new Query(crit), MyEntity.class).toArray(new MyEntity[0]))
        System.out.println(aggResult.getId() + " - " + aggResult.getParam());

    System.out.println("\nDocuments fetched with aggregate: ");        
    TypedAggregation<MyEntity> aggregation = new TypedAggregation<>(MyEntity.class, aggOpList);
    AggregationResults<MyEntity> aggregate = mongoTemplate.aggregate(aggregation, MyEntity.class);      
    for (MyEntity aggResult : aggregate.getMappedResults())
        System.out.println(aggResult.getId() + " - " + aggResult.getParam());       
}

输出

Documents fetched with find: 
{one - 1} - param1
{two - 2} - param2

Documents fetched with aggregate: 
null - param1
null - param2

在调试以下方法MappingMongoConverter.read(final MongoPersistentEntity entity, final Document bson, final ObjectPath path)时，我发现在第一种情况（find方法）中，documentAccessor变量的内容如下：

Document{{_id=Document{{stringKey=one, intKey=1}}, param=param1, _class=MyEntity}}

而在第二种情况（聚合查询）中，它看起来像这样：

Document{{stringKey=one, intKey=1, param=param1, _class=MyEntity}}

某种方式使文档被展开，这使得转换器无法填充ID字段。我一定是做错了什么，但是是什么呢？

- Guilhem S

Answer 1

Spring Data MongoDB 3.x 版本以下会自动展开复合 ID（复合 ID 下的字段解包并放置在根对象中）。此功能在版本 3.0 及以上已被删除： https://docs.spring.io/spring-data/mongodb/docs/current/reference/html/#new-features.3.0

- vu le