MongoDB
 sql >> Base de données >  >> NoSQL >> MongoDB

Comment utiliser l'opération $cond dans le framework d'agrégation Spring-MongoDb

Si vous utilisez la version actuelle de Spring Data qui prend en charge le $cond opérateur via le $project pipeline, alors cela peut être converti en (non testé):

import static org.springframework.data.mongodb.core.aggregation.Aggregation.*;
import static org.springframework.data.mongodb.core.aggregation.ConditionalOperators.Cond.*;
import org.springframework.data.mongodb.core.query.Criteria;

Cond condOperation = ConditionalOperators.when(Criteria.where("start").is("EARLY"))
                                    .thenValueOf("deltastart.start")
                                    .otherwise("deltastart.end");

Aggregation agg = newAggregation(project().and(condOperation).as("start"));
AggregationResults<MyClass> results = mongoTemplate.aggregate(agg, MyClass.class); 
List<MyClass> myList = results.getMappedResults();

Pour la version Spring-Data MongoDB qui ne prend pas en charge le $cond opérateur dans l'opération d'agrégation, il existe une solution de contournement qui consiste à implémenter l'AggregationOperation interface à prendre dans un DBObject :

public class CustomProjectAggregationOperation implements AggregationOperation {
    private DBObject operation;

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

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

Puis implémentez le $project opération en tant que DBObject dans le pipeline d'agrégation identique à celui que vous avez :

DBObject operation = (DBObject) new BasicDBObject(
    "$project", new BasicDBObject(
         "start", new BasicDBObject(
                "$cond", new Object[]{
                        new BasicDBObject(
                            "$eq", new Object[]{ "$start", "EARLY"}
                        ),
                        "$deltastart.start",
                        "$deltastart.end"
                 }
           )
     )
);

que vous pouvez ensuite utiliser dans TypeAggregation :

TypedAggregation<CustomClass> aggregation = newAggregation(CustomClass.class,
    new CustomProjectAggregationOperation(operation)
);
AggregationResults<CustomClass> result = mongoTemplate.aggregate(aggregation, CustomClass.class);