MongoDB中實現多表聯查的範例教學

2022-07-07 18:05:54

前些天遇到一個需求，不復雜，用 SQL 表現的話，大約如此：

SELECT *
FROM db1 LEFT JOIN db2 ON db1.a = db2.b
WHERE db1.userId='$me' AND db2.status=1

沒想到搜了半天，我廠的程式碼倉庫裡沒有這種用法，各種教學也多半隻針對合併查詢（即只篩選 db1，沒有 db2 的條件）。所以最後只好讀檔案+程式碼嘗試，終於找到答案，記錄一下。

我們用 mongoose 作為連線庫
聯查需要用 $lookup
如果宣告外來鍵的時候用 ObjectId，就很簡單：

// 假設下面兩個表 db1 和 db2
export const Db1Schema = new mongoose.Schema(
  {
    userId: { type: String, index: true },
    couponId: { type: ObjectId, ref: Db2Schema },
  },
  { versionKey: false, timestamps: true }
);
export const Db2Schema = new mongoose.Schema(
  {
    status: { type: Boolean, default: 0 },
  },
  { versionKey: false, timestamps: true }
);

// 那麼只要
db1Model.aggregate([
  {
    $lookup: {
      from: 'db2', // 目標表
      localField: 'couponId', // 本地欄位
      foreignField: '_id', // 對應的目標欄位
      as: 'source',
  },
  {
    $match: [ /* 各種條件 */ ],
  },
]);

但是我們沒有用 ObjectId，而是用 string 作為外來鍵，所以無法直接用上面的聯查。必須在 pipeline 裡手動轉換、聯合。此時，當前表（db1）的欄位不能直接使用，要配合 let，然後加上 $$ 字首；連表（db2）直接加 $ 字首即可。

最終程式碼如下：

mongo.ts

// 每次必有的條件，當前表的欄位用 `$$`，連表的欄位用 `$`
const filter = [{ $eq: ['$$userId', userId] }, { $eq: ['$isDeleted', false] }];
if (status === Expired) {
  dateOp = '$lte';
} else if (status === Normal) {
  dateOp = '$gte';
  filter.push({ $in: ['$$status', [Normal, Shared]] });
} else {
  dateOp = '$gte';
  filter.push({ $eq: ['$$status', status] });
}
const results = await myModel.aggregate([
  {
    $lookup: {
      from: 'coupons',
      // 當前表欄位必須 `let` 之後才能用
      let: { couponId: '$couponId', userId: '$userId', status: '$status' },
      // 在 pipeline 裡完成篩選
      pipeline: [
        {
          $match: {
            $expr: {
              // `$toString` 是內建方法，可以把 `ObjectId` 轉換成 `string`
              $and: [{ $eq: [{ $toString: '$_id' }, '$$couponId'] }, ...filter, { [dateOp]: ['$endAt', new Date()] }],
            },
          },
        },
        // 只要某些欄位，在這裡篩選
        {
          $project: couponFields,
        },
      ],
      as: 'source',
    },
  },
  {
    // 這種篩選相當 LEFT JOIN，所以需要去掉沒有連表內容的結果
    $match: {
      source: { $ne: [] },
    },
  },
  {
    // 為了一次查表出結果，要轉換一下輸出格式
    $facet: {
      results: [{ $skip: size * (page - 1) }, { $limit: size }],
      count: [
        {
          $count: 'count',
        },
      ],
    },
  },
]);

同事告訴我，這樣做的效率不一定高。我覺得，考慮到實際場景，他說的可能沒錯，不過，早晚要邁出這樣的一步。而且，未來我們也應該慢慢把外來鍵改成 ObjectId 型別。

總結

到此這篇關於MongoDB中實現多表聯查的文章就介紹到這了,更多相關MongoDB多表聯查內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com！

MongoDB中實現多表聯查的範例教學

熱門文章