将Mongo聚合查询转换为java对象

时间:2017-04-04 06:32:12

标签: java mongodb mongodb-query aggregation-framework

我正在尝试将mongo聚合查询转换为java对象。当我在RoboMongo(工具)中运行查询时,我得到结果但转换为java对象会得到空结果。

Mongo查询:

db.getCollection('wb_physicians').aggregate([
    { 
        $match: { 
            $and: [ 
                { "product.mpoCode": "VA001"}, 
                { "product.npoCode": { $exists: true } }
            ] 
        }
    },
    { 
        "$project" : { 
            "product.specialties.code": 1, 
            "providerId": 1,
            "product.code": 1, 
            "_id" : 0
        }
    }, 
    { "$unwind" : "$product.specialties" },
    { 
        "$group" : { 
            "_id" : { 
                "providerId": "$providerId" , 
                "productCode": "$product.code"
            }, 
            "specialityCodeList": { "$addToSet": "$product.specialties.code" }
        }
    }
])

Java代码:

private static AggregationOutput findProviderandSpecialty(DBCollection collection) {
    DBObject match =  new BasicDBObject("$match" , 
        new BasicDBObject("$and", Arrays.asList(
            new BasicDBObject("product.mpoCode" , "VA001").append("product.npoCode", "$exists: true")
        ))
    );

    DBObject project =  new BasicDBObject("$project" , 
        new BasicDBObject("product.specialties.code" , 1)
            .append("providerId" , 1)
            .append("product.code", 1)
            .append("_id", 0)
    );  

    DBObject unwind = new BasicDBObject("$unwind" , "$product.specialties");

    DBObject group = new BasicDBObject("$group",
        new BasicDBObject("_id", new BasicDBObject("providerId" , "$providerId"))
            .append("specialityCodeList", 
                new BasicDBObject("$addToSet", "$product.specialties.code")
            )
    );

    AggregationOutput output = collection.aggregate(match,project,unwind,group);    

    return output;
}   

你能帮助我做出错误的映射吗?

1 个答案:

答案 0 :(得分:2)

问题出在$match管道上:

DBObject match =  new BasicDBObject("$match" , 
    new BasicDBObject("$and", Arrays.asList(
        new BasicDBObject("product.mpoCode" , "VA001")
           .append("product.npoCode", "$exists: true")
    ))
);

应该是

DBObject match =  new BasicDBObject("$match" , 
    new BasicDBObject("$and", Arrays.asList(
        new BasicDBObject("product.mpoCode" , "VA001"),
        new BasicDBObject("product.npoCode", 
            new BasicDBObject("$exists", "true")
        )
    ))
);

尽管如此,您可以通过指定文件的逗号分隔表达式以及删除 $and $project 逻辑> $group 之前的管道 因为它是不必要的,所以你修改后的管道可以运行:

db.getCollection('wb_physicians').aggregate([
    { 
        "$match": { 
            "product.mpoCode": "VA001", 
            "product.npoCode": { "$exists": true }             
        }
    },    
    { "$unwind" : "$product.specialties" },
    { 
        "$group" : { 
            "_id" : { 
                "providerId": "$providerId" , 
                "productCode": "$product.code"
            }, 
            "specialityCodeList": { "$addToSet": "$product.specialties.code" }
        }
    }
])

最终的Java代码:

private static AggregationOutput findProviderandSpecialty(DBCollection collection) {
    DBObject match =  new BasicDBObject("$match" , 
        new BasicDBObject("product.mpoCode" , "VA001").append("product.npoCode",             
            new BasicDBObject("$exists", "true")
        )        
    );

    DBObject unwind = new BasicDBObject("$unwind" , "$product.specialties");

    DBObject group = new BasicDBObject("$group",
        new BasicDBObject("_id", new BasicDBObject("providerId" , "$providerId"))
            .append("specialityCodeList", 
                new BasicDBObject("$addToSet", "$product.specialties.code")
            )
    );

    List<DBObject> pipeline = Arrays.<DBObject>asList(match, unwind, group);
    AggregationOutput output = collection.aggregate(pipeline);    

    return output;
}