如何用MongoDB正则匹配站点数组城市名筛选网约车订单

问题根源 #

你之前的查询失效是因为：同一个字段多次使用$elemMatch时，后一个条件会覆盖前一个，最终只会匹配下车点的正则条件，无法同时验证两个城市是否都存在于站点数组中。

解决方案 #

以下提供两种可行的聚合查询方案，均支持正则模糊匹配、不区分大小写，并能关联返回司机和车辆信息：

方案1：简洁的$and+数组字段正则匹配

利用MongoDB对数组字段的隐式匹配特性，直接通过$and组合两个数组元素匹配条件，确保两个城市都存在于stops数组中：

const pickupRegex = new RegExp(pickup, "i"); // 不区分大小写匹配包含该城市名
const dropRegex = new RegExp(drop, "i");

const rides = await Ride.aggregate([
  // 核心匹配逻辑：同时存在上车点和下车点对应的站点
  {
    $match: {
      $and: [
        { "stops.cityName": { $regex: pickupRegex } },
        { "stops.cityName": { $regex: dropRegex } }
      ]
    }
  },
  // 关联司机信息（假设行程表用driverId关联司机表）
  {
    $lookup: {
      from: "drivers",
      localField: "driverId",
      foreignField: "_id",
      as: "driver"
    }
  },
  // 关联车辆信息（假设行程表用vehicleId关联车辆表）
  {
    $lookup: {
      from: "vehicles",
      localField: "vehicleId",
      foreignField: "_id",
      as: "vehicle"
    }
  },
  // 可选：将数组格式的司机/车辆转为单个对象（单行程对应单司机/车辆场景）
  { $unwind: "$driver" },
  { $unwind: "$vehicle" }
]);

如果需要前缀精确匹配（比如只匹配以Lahore开头的城市名），可以把正则改为：

const pickupRegex = new RegExp(`^${pickup}`, "i");
const dropRegex = new RegExp(`^${drop}`, "i");

方案2：$expr+$filter精确验证（适合复杂场景）

如果需要确保匹配的是数组中的不同元素，或者需要更精细的过滤逻辑，可以用$expr配合$filter和$size来验证两个城市的匹配结果：

const pickupRegex = new RegExp(pickup, "i");
const dropRegex = new RegExp(drop, "i");

const rides = await Ride.aggregate([
  {
    $match: {
      $expr: {
        $and: [
          // 验证存在匹配上车点的站点
          {
            $gt: [
              {
                $size: {
                  $filter: {
                    input: "$stops",
                    cond: {
                      $regexMatch: {
                        input: "$$this.cityName",
                        regex: pickupRegex
                      }
                    }
                  }
                }
              },
              0
            ]
          },
          // 验证存在匹配下车点的站点
          {
            $gt: [
              {
                $size: {
                  $filter: {
                    input: "$stops",
                    cond: {
                      $regexMatch: {
                        input: "$$this.cityName",
                        regex: dropRegex
                      }
                    }
                  }
                }
              },
              0
            ]
          }
        ]
      }
    }
  },
  // 同方案1，关联司机和车辆信息
  {
    $lookup: {
      from: "drivers",
      localField: "driverId",
      foreignField: "_id",
      as: "driver"
    }
  },
  {
    $lookup: {
      from: "vehicles",
      localField: "vehicleId",
      foreignField: "_id",
      as: "vehicle"
    }
  },
  { $unwind: "$driver" },
  { $unwind: "$vehicle" }
]);

优化建议 #

1. 性能优化：如果城市名变体仅为后缀差异（如统一带国家名），建议在存储时额外添加cityShortName字段（仅存城市名），避免正则匹配的性能损耗。
2. 索引支持：若使用前缀正则匹配（^xxx），可为stops.cityName创建单字段索引，大幅提升查询速度；若需全文匹配，可创建文本索引。

内容的提问来源于stack exchange，提问作者Muhammad Sohaib