MongoDB加入两个集合

时间:2020-01-27 08:47:47

标签: mongodb aggregation-framework spring-mongodb spring-mongo

这是订单实体:

public class Order 
{
    @Id
    private String id;
    private String internalId;
    private String externalId;
    private String status;
}

public class OrderAssigned 
{
    @Id
    private String id;
    private String internalId;
    private String externalId;
    private String stationId;
    private String rejectStatus;
}

我想加入这两个收藏集。

  • 首先我要找到stationId
  • 然后加入Orders.internalIdOrderAssigned.internalIdstatus = "OrderCreated"

类似于此SQL查询:

select orders.* 
from PARTNER_RESTAURANT_ORDER orders 
inner join PARTNER_RESTAURANT_ORDER_ASSIGNED_STATION assigned_station on orders.internalId = assigned_station.internalId 
where assigned_station.stationId = "5e2968e2763e750001c8ba5f" 
  and orders.status = "OrderCreated"

1 个答案:

答案 0 :(得分:1)

好的MongoDB并不是最适合联接的,但是可以通过$lookup来实现。

假设您具有以下数据集:

ORDERS:
{
    _id: 'id-1',
    internalId: 'internal-id-1',
    externalId: 'external-id-1',
    status: 'OrderCreated'
}
{
    _id: 'id-2',
    internalId: 'internal-id-2',
    externalId: 'external-id-2',
    status: 'INACTIVE'
)

ORDERS ASSIGNED:
{
    _id: 'id-1',
    internalId: 'internal-id-1',
    externalId: 'external-id-1',
    stationId: 'station-id-1',
    rejectStatus: 'NONE'
}
{
    _id: 'id-2',
    internalId: 'internal-id-2',
    externalId: 'external-id-2',
    stationId: 'station-id-2',
    rejectStatus: 'NONE'
}

然后:

db.orderAssigned.aggregate([
 {$match: {stationId: {$in: ['station-id-1', 'station-id-2']}}},
 {$lookup: {from: 'order', localField: 'internalId', foreignField: 'internalId', as: 'orders'}}
])

会给您:

{
    "_id" : "id-1",
    "internalId" : "internal-id-1",
    "externalId" : "external-id-1",
    "stationId" : "station-id-1",
    "rejectStatus" : "NONE",
    "orders" : [
        {
            "_id" : "id-1",
            "internalId" : "internal-id-1",
            "externalId" : "external-id-1",
            "status" : "OrderCreated"
        }
    ]
}
{
    "_id" : "id-2",
    "internalId" : "internal-id-2",
    "externalId" : "external-id-2",
    "stationId" : "station-id-2",
    "rejectStatus" : "NONE",
    "orders" : [
        {
            "_id" : "id-2",
            "internalId" : "internal-id-2",
            "externalId" : "external-id-2",
            "status" : "INACTIVE"
        }
    ]
}

但是,如果您需要一些额外的连接条件,例如status = "OrderCreated",则更适合的是带有管道的第二版 $ lookup

db.orderAssigned.aggregate([
 {$match: {stationId: {$in: ['station-id-1', 'station-id-2']}}},
 {$lookup: 
    {
        from: 'order', 
        let: {joinValue: '$internalId'}, 
        pipeline: [
            { $match:
                 { $expr:
                    { $and:
                       [
                         { $eq: [ "$internalId",  "$$joinValue" ] },
                         { $eq: [ "$status", "OrderCreated" ] }
                       ]
                    }
                 }
                  }     
        ], 
        as: 'orders'}}
])

会给您:

{
    "_id" : "id-1",
    "internalId" : "internal-id-1",
    "externalId" : "external-id-1",
    "stationId" : "station-id-1",
    "rejectStatus" : "NONE",
    "orders" : [
        {
            "_id" : "id-1",
            "internalId" : "internal-id-1",
            "externalId" : "external-id-1",
            "status" : "OrderCreated"
        }
    ]
}
{
    "_id" : "id-2",
    "internalId" : "internal-id-2",
    "externalId" : "external-id-2",
    "stationId" : "station-id-2",
    "rejectStatus" : "NONE",
    "orders" : [ ]
}