MongoDB加入两个集合

Question

这是订单实体：

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.internalId和OrderAssigned.internalId和status = "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"

Answer 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"，那么更适合您的是带有 pipeline 的 $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" : [ ]
}