使用嵌套数组的投影创建Spring Data Aggregation查询

时间:2016-12-01 16:29:03

标签: spring mongodb spring-data-mongodb

以下是我的文档的样子:

{
"_id" : ObjectId("583cb6bcce047d1e68339b64"),
"variantDetails" : [ 
    {
        "variants" : {
            "_" : "_"
        },
        "sku" : "069563-59690"
    }, 
    {
        "variants" : {
            "size" : "35"
        },
        "sku" : "069563-59690-35",
        "barcode" : "809702246941"
    }, 
    {
        "variants" : {
            "size" : "36"
        },
        "sku" : "069563-59690-36",
        "barcode" : "809702246958"
    }
    ......
] }

我想使用像这样的复杂聚合查询:

db.getCollection('product').aggregate([
    { '$match': { 'variantDetails.sku': { '$in': ['069563-59690', '069563-59690-36', '069563-59690-37', '511534-01001'] } } },
    { '$project': {'_id': 1, 'variantDetails': 1, 'variantLength': { '$size': '$variantDetails' } } },
    { '$unwind': '$variantDetails' },
    { '$match': { 'variantDetails.sku': { '$in': ['069563-59690', '069563-59690-36', '069563-59690-37', '511534-01001'] } } },
    { '$match': { '$or': [
        {'variantLength': { '$ne': 1 }, 'variantDetails.variants._': { '$ne': '_' } },
        {'variantLength': 1 }
    ] } },
    { '$group': { '_id': '$_id', 'variantDetails': { '$push': '$variantDetails' } } },
    { '$project': {'_id': 1, 'variantDetails.sku': 1, 'variantDetails.barcode': 1} }
])

这是我的java代码:

    final Aggregation agg = Aggregation.newAggregation(
            Aggregation.match(Criteria.where("variantDetails.sku").in(skus)),
            Aggregation.project("_id", "variantDetails").and("variantDetails").project("size").as("variantLength"),
            Aggregation.unwind("variantDetails"),
            Aggregation.match(Criteria.where("variantDetails.sku").in(skus)),
            Aggregation.match(new Criteria().orOperator(Criteria.where("variantLength").is(1), Criteria.where("variantLength").ne(1).and("variantDetails.variants._").is("_"))),
            Aggregation.group("_id").push("variantDetails").as("variantDetails"),
            Aggregation.project("_id", "variantDetails.sku", "variantDetails.barcode")
            );

    final AggregationResults<Product> result =  this.mongo.aggregate(agg, this.mongo.getCollectionName(Product.class), Product.class);
    return result.getMappedResults();

问题是春天翻译

Aggregation.project("_id", "variantDetails.sku", "variantDetails.barcode")

{ "$project" : { "_id" : 1 , "sku" : "$variantDetails.sku" , "barcode" : "$variantDetails.barcode"}

但我期待

{ '$project': {'_id': 1, 'variantDetails.sku': 1, 'variantDetails.barcode': 1} }

有人能让我知道如何做对吗?

2 个答案:

答案 0 :(得分:0)

您只需要在投影操作中将标签指定为别名,即spring提供的默认值不匹配。使用Spring 1.8.5版本

Aggregation.project("_id")
                   .and(context -> new BasicDBObject("$arrayElemAt", Arrays.asList("variantDetails.sku", 0))).as("variantDetails.sku")
                   .and(context -> new BasicDBObject("$arrayElemAt", Arrays.asList("variantDetails.barcode", 0))).as("variantDetails.barcode"));

答案 1 :(得分:0)

我遇到了同样的问题,这种方式有效:

Aggregation.project("_id")
.andExpression("variantDetails.sku").as("variantDetails.sku")                
.andExpression("variantDetails.barcode").as("variantDetails.barcode"));

预测将是:

{'$project': {'_id': 1, 'variantDetails.sku': '$variantDetails.sku', 
'variantDetails.barcode': '$variantDetails.barcode'} }