Spring Data MongoDB Aggregation - переименовать _id в проекции - PullRequest
Новая
       75

Spring Data MongoDB Aggregation - переименовать _id в проекции

2 голосов
/ 20 июня 2020

В оболочке mon go (3.6) я могу выполнить следующую агрегацию для проецирования и переименовать "_id" во что-то еще: 

db.collection.aggregate([
  { $group : { _id : ... },
  { $project : {   '_id': 0,        // exclude the id field
                   'name' : '$_id', // project the "old" id value to the "name" field
                   ...
               }
  }
])

Мне нужно перевести это с помощью spring-data-mon go (через Spring Boot 2.3.1). Я могу исключить поле id с помощью следующей ProjectionOperation: 

project().andExclude("_id")

Но пока мне не удалось переименовать поле id. Ни одно из следующих действий не работает:

  • мое первое предположение 

      project().and("_id").as("name")

  • как предлагается здесь 

      project().andExpression("_id").as("name")

Это не должно быть так сложно, но я не могу понять, что мне не хватает.

РЕДАКТИРОВАТЬ: образец данных и конвейер полной агрегации для воспроизведения отказа

Со следующими документами: 

{ 
    "_id" : ObjectId("5a0c7a3135587511c9247db4"), 
    "_class" : "task", 
    "category" : "green", 
    "status" : "OK"
}
{ 
    "_id" : ObjectId("5a0cd21d35587511c9247db8"), 
    "_class" : "task", 
    "category" : "red", 
    "status" : "KO"
}

Объект домена: 

@Document(collection = "tasks")
@TypeAlias("task")
public class Task {

    @Id
    private String id;
    private String status;
    private String category;

    // getters/setters omitted
}

И следующее агрегирование : 

Aggregation a = Aggregation.newAggregation(
                group("status", "category").count().as("count"),
                group("_id.category").push(
                        new Document("k", "$_id.status").append("v", "$count"))
                        .as("counts"),
                project().and(arrayToObject("$counts")).as("counts"),
                // this final stage fails with: a java.lang.IllegalArgumentException: Invalid reference '_id'!
                project().and("_id").as("name").andExclude("_id")
        );

mongotemplate.aggregate(a, "tasks", org.bson.Document.class)

Перед последним этапом конвейера агрегация дает: 

[ { "_id" : "green", "counts" : { "OK" : 1 } },
  { "_id" : "red", "counts" : { "KO" : 1 } } ]

Таким образом, мы можем проецировать «_id» на «name», но вместо этого получаем следующее: 

java.lang.IllegalArgumentException: Invalid reference '_id'!
    at org.springframework.data.mongodb.core.aggregation.ExposedFieldsAggregationOperationContext.getReference(ExposedFieldsAggregationOperationContext.java:114) ~[spring-data-mongodb-3.0.1.RELEASE.jar:3.0.1.RELEASE]
    at org.springframework.data.mongodb.core.aggregation.ExposedFieldsAggregationOperationContext.getReference(ExposedFieldsAggregationOperationContext.java:77) ~[spring-data-mongodb-3.0.1.RELEASE.jar:3.0.1.RELEASE]
    at org.springframework.data.mongodb.core.aggregation.ProjectionOperation$ProjectionOperationBuilder$FieldProjection.renderFieldValue(ProjectionOperation.java:1445) ~[spring-data-mongodb-3.0.1.RELEASE.jar:3.0.1.RELEASE]
    at org.springframework.data.mongodb.core.aggregation.ProjectionOperation$ProjectionOperationBuilder$FieldProjection.toDocument(ProjectionOperation.java:1432) ~[spring-data-mongodb-3.0.1.RELEASE.jar:3.0.1.RELEASE]
    at org.springframework.data.mongodb.core.aggregation.ProjectionOperation.toDocument(ProjectionOperation.java:261) ~[spring-data-mongodb-3.0.1.RELEASE.jar:3.0.1.RELEASE]
    at org.springframework.data.mongodb.core.aggregation.AggregationOperation.toPipelineStages(AggregationOperation.java:55) ~[spring-data-mongodb-3.0.1.RELEASE.jar:3.0.1.RELEASE]
    at org.springframework.data.mongodb.core.aggregation.AggregationOperationRenderer.toDocument(AggregationOperationRenderer.java:56) ~[spring-data-mongodb-3.0.1.RELEASE.jar:3.0.1.RELEASE]
    at org.springframework.data.mongodb.core.aggregation.Aggregation.toPipeline(Aggregation.java:721) ~[spring-data-mongodb-3.0.1.RELEASE.jar:3.0.1.RELEASE]
    at org.springframework.data.mongodb.core.AggregationUtil.createPipeline(AggregationUtil.java:95) ~[spring-data-mongodb-3.0.1.RELEASE.jar:3.0.1.RELEASE]
    at org.springframework.data.mongodb.core.MongoTemplate.doAggregate(MongoTemplate.java:2118) ~[spring-data-mongodb-3.0.1.RELEASE.jar:3.0.1.RELEASE]
    at org.springframework.data.mongodb.core.MongoTemplate.aggregate(MongoTemplate.java:2093) ~[spring-data-mongodb-3.0.1.RELEASE.jar:3.0.1.RELEASE]
    at org.springframework.data.mongodb.core.MongoTemplate.aggregate(MongoTemplate.java:1992) ~[spring-data-mongodb-3.0.1.RELEASE.jar:3.0.1.RELEASE]

Я уже нашел обходной путь (см. Ниже), но мне все еще любопытно, почему агрегирование, показанное выше, не работает. Я не понимаю, почему ссылка "_id" не отображается должным образом.

Успешное решение: 

Aggregation a = Aggregation.newAggregation(
            group("status", "category").count().as("count"),
            group("_id.category").push(
                    new Document("k", "$_id.status").append("v", "$count"))
                    .as("counts"),
            project().and(arrayToObject("$counts")).as("counts"),
            // the following works
            addFields().addFieldWithValue("name", "$_id").build(),
            project().andExclude("_id").andExclude("counts")
    );

Для тех, кому нужно обходное решение, когда они не могут понять, кому преобразовать агрегацию MongoDB в Spring Data Mon go агрегация: { ссылка }

1 Ответ

2 голосов
/ 20 июня 2020

Попробуйте это: 

project("_id").and(arrayToObject("$counts")).as("counts"),
project().andExclude("_id").and("_id").as("name")

EDIT:

Spring-Mon go выполняет некоторые проверки вашего конвейера и ожидает, что _id присутствует в ваш предыдущий этап, поэтому причина java.lang.IllegalArgumentException: Invalid reference '_id'!

Aggregation agg = Aggregation.newAggregation(
        group("status", "category").count().as("count"),
        group("_id.category").push(new Document("k", "$_id.status").append("v", "$count")).as("counts"),
        project("_id").and(ArrayOperators.arrayOf("$counts").toObject()).as("counts"),
        project().and("_id").as("name").andExclude("_id"));

AggregationResults<Document> results = mongoTemplate.aggregate(agg, "tasks", Document.class);
results.getMappedResults().forEach(System.out::println);

--- Вывод --- 

Document{{name=green}}
Document{{name=red}}
...