Вычисляемые групповые поля в MongoDB

Ответ 1

На самом деле вы можете сделать что-то вроде этого с "проектом", но для меня это немного противоречит интуитивному требованию перед $project:

    Aggregation agg = newAggregation(
        project("quantity")
            .andExpression("dayOfMonth(date)").as("day")
            .andExpression("month(date)").as("month")
            .andExpression("year(date)").as("year")
            .andExpression("price * quantity").as("totalAmount"),
        group(fields().and("day").and("month").and("year"))
            .avg("quantity").as("averavgeQuantity")
            .sum("totalAmount").as("totalAmount")
            .count().as("count")
    );

Как я уже сказал, противоречит интуиции, так как вы должны просто объявить все это на этапе $group, но помощники, похоже, не работают таким образом. Сериализация выглядит немного смешно (обертывает аргументы оператора даты массивами), но, похоже, она работает. Но все же это две стадии трубопровода, а не одна.

В чем проблема с этим? Ну, разделив этапы этапов, часть "проект" заставляет обрабатывать все документы в конвейере, чтобы получить вычисленные поля, а это значит, что он проходит через все, прежде чем перейти на групповой этап.

Разницу во времени обработки можно четко увидеть, выполнив запросы в обеих формах. С отдельным этапом проекта на моем аппаратном обеспечении требуется в три раза больше времени, чем запрос, когда все поля вычисляются во время операции "группа".

Итак, кажется, что единственный способ построить это правильно - это построить сам объект конвейера:

    ApplicationContext ctx =
            new AnnotationConfigApplicationContext(SpringMongoConfig.class);
    MongoOperations mongoOperation = (MongoOperations) ctx.getBean("mongoTemplate");

    BasicDBList pipeline = new BasicDBList();
    String[] multiplier = { "$price", "$quantity" };

    pipeline.add(
        new BasicDBObject("$group",
            new BasicDBObject("_id",
                new BasicDBObject("month", new BasicDBObject("$month", "$date"))
                    .append("day", new BasicDBObject("$dayOfMonth", "$date"))
                    .append("year", new BasicDBObject("$year", "$date"))
            )
            .append("totalPrice", new BasicDBObject(
                "$sum", new BasicDBObject(
                    "$multiply", multiplier
                )
            ))
            .append("averageQuantity", new BasicDBObject("$avg", "$quantity"))
            .append("count",new BasicDBObject("$sum",1))
        )
    );

    BasicDBObject aggregation = new BasicDBObject("aggregate","collection")
        .append("pipeline",pipeline);

    System.out.println(aggregation);

    CommandResult commandResult = mongoOperation.executeCommand(aggregation);

Или, если все это похоже на вас, вы всегда можете работать с источником JSON и разбирать это. Но, конечно, это должно быть справедливо JSON:

    String json = "[" +
        "{ \"$group\": { "+
            "\"_id\": { " +
                "\"month\": { \"$month\": \"$date\" }, " +
                "\"day\": { \"$dayOfMonth\":\"$date\" }, " +
                "\"year\": { \"$year\": \"$date\" } " +
            "}, " +
            "\"totalPrice\": { \"$sum\": { \"$multiply\": [ \"$price\", \"$quantity\" ] } }, " +
            "\"averageQuantity\": { \"$avg\": \"$quantity\" }, " +
            "\"count\": { \"$sum\": 1 } " +
        "}}" +
    "]";

    BasicDBList pipeline = (BasicDBList)com.mongodb.util.JSON.parse(json);

Ответ 2

Другой альтернативой является использование настраиваемого класса операции агрегации, определенного следующим образом:

public class CustomAggregationOperation implements AggregationOperation {
    private DBObject operation;

    public CustomAggregationOperation (DBObject operation) {
        this.operation = operation;
    }

    @Override
    public DBObject toDBObject(AggregationOperationContext context) {
        return context.getMappedObject(operation);
    }
}

Затем реализуйте его в конвейере следующим образом:

Aggregation aggregation = newAggregation(
    new CustomAggregationOperation(
        new BasicDBObject(
            "$group",
            new BasicDBObject("_id",
                new BasicDBObject("day", new BasicDBObject("$dayOfMonth", "$date" ))
                .append("month", new BasicDBObject("$month", "$date"))
                .append("year", new BasicDBObject("$year", "$date"))
            )
            .append("totalPrice", new BasicDBObject(
                "$sum", new BasicDBObject(
                    "$multiply", Arrays.asList("$price","$quantity")
                )
            ))
            .append("averageQuantity", new BasicDBObject("$avg", "$quantity"))
            .append("count",new BasicDBObject("$sum",1))
        )
    )
)

Итак, это в основном просто интерфейс, который согласуется с интерфейсом, используемым существующими помощниками конвейера, но вместо использования других вспомогательных методов он непосредственно принимает определение DBObject для возврата в конвейерную конструкцию.