MongoDB
 sql >> Datenbank >  >> NoSQL >> MongoDB

Spring Data MongoDB Lookup mit Pipeline-Aggregation

Aufbauend auf den Informationen von @dnickless konnte ich dies lösen. Ich werde die vollständige Lösung in der Hoffnung posten, dass sie in Zukunft jemand anderem hilft.

Ich verwende mongodb-driver:3.6.4

Zuerst musste ich eine benutzerdefinierte Aggregationsoperationsklasse erstellen, damit ich eine benutzerdefinierte JSON-Mongodb-Abfrage übergeben konnte, die in der Aggregationsoperation verwendet werden sollte. Dadurch kann ich pipeline verwenden innerhalb eines $lookup die von der von mir verwendeten Treiberversion nicht unterstützt wird.

public class CustomProjectAggregationOperation implements AggregationOperation {
    private String jsonOperation;

    public CustomProjectAggregationOperation(String jsonOperation) {
        this.jsonOperation = jsonOperation;
    }

    @Override
    public Document toDocument(AggregationOperationContext aggregationOperationContext) {
        return aggregationOperationContext.getMappedObject(Document.parse(jsonOperation));
    }
}

Jetzt, da wir die Möglichkeit haben, eine benutzerdefinierte JSON-Abfrage an unsere mongodb-Spring-Implementierung zu übergeben, müssen diese Werte nur noch in eine TypedAggregation-Abfrage eingefügt werden.

public List<FulfillmentChannel> getFulfillmentChannels(
    String SOME_VARIABLE_STRING_1, 
    String SOME_VARIABLE_STRING_2) {

    AggregationOperation match = Aggregation.match(
            Criteria.where("dayOfWeek").is(SOME_VARIABLE_STRING_1));
    AggregationOperation match2 = Aggregation.match(
            Criteria.where("deliveryZipCodeTimings").ne(Collections.EMPTY_LIST));
    String query =
            "{ $lookup: { " +
                    "from: 'deliveryZipCodeTiming'," +
                    "let: { location_id: '$fulfillmentLocationId' }," +
                    "pipeline: [{" +
                    "$match: {$expr: {$and: [" +
                    "{ $eq: ['$fulfillmentLocationId', '$$location_id']}," +
                    "{ $eq: ['$zipCode', '" + SOME_VARIABLE_STRING_2 + "']}]}}}," +
                    "{ $project: { _id: 0, zipCode: 1, cutoffTime: 1 } }]," +
                    "as: 'deliveryZipCodeTimings'}}";

    TypedAggregation<FulfillmentChannel> aggregation = Aggregation.newAggregation(
            FulfillmentChannel.class,
            match,
            new CustomProjectAggregationOperation(query),
            match2
    );

    AggregationResults<FulfillmentChannel> results = 
        mongoTemplate.aggregate(aggregation, FulfillmentChannel.class);
    return results.getMappedResults();
}