MongoDB Atlas Search嵌入文档查询：仅返回匹配数组元素

问题背景 #

集合结构如下：

{
  "_id": ObjectId,
  "name": String,
  "companies": [
    {
      "name": String,
      "rating": Number,
      "address": String
    }
  ]
}

已针对companies.name创建嵌入式类型的Atlas搜索索引，使用该索引查询时，返回的完整文档会包含数组中所有未匹配的元素。无法使用高亮功能（嵌入式文档类型不支持），虽可通过$unwind后匹配name字段实现需求，但希望找到更优方案；重构文档会带来其他负面影响，因此不考虑。

现有查询代码尝试提取匹配的数组元素，但执行报错：

let query1 = {
  '$search': {
    'index': 'businessname',
    'embeddedDocument': {
      'path': 'companies', 
      'operator': {
        'queryString': {
          'defaultPath': 'companies.name', 
          'query': `${tags} ~ 2`
        }
      }
    }
  }
}
let project = {
  '$project': {
    '_id': 1,
    'displayName': 1,
    'email': 1,
    'companies.$': 1
  }
}

报错信息：MongoServerError: Invalid $project :: caused by :: Cannot use positional projection in aggregation projection

可行解决方案 #

方案1：使用$filter聚合操作符 #

在$project阶段用$filter筛选出匹配条件的companies元素，无需$unwind，性能更优：

let project = {
  '$project': {
    '_id': 1,
    'displayName': 1,
    'email': 1,
    'matchedCompanies': {
      '$filter': {
        'input': '$companies',
        'as': 'comp',
        'cond': {
          '$regexMatch': {
            'input': '$$comp.name',
            'regex': tags,
            'options': 'i' // 可选，忽略大小写
          }
        }
      }
    }
  }
}

注意：这里的正则匹配要和Atlas搜索的queryString逻辑对齐，确保筛选结果一致。如果是带编辑距离（~2）的模糊匹配，可借助第三方库在应用层计算，或用MongoDB 4.4+支持的$function自定义匹配逻辑（需注意性能）。

方案2：应用层后处理筛选 #

如果Atlas搜索返回的结果量不大，可以先获取完整文档，在应用层直接筛选匹配的companies元素：

// 执行聚合查询，获取完整文档
const results = await db.collection('your-collection').aggregate([query1]).toArray();
// 应用层筛选匹配项
const processedResults = results.map(doc => {
  return {
    ...doc,
    matchedCompanies: doc.companies.filter(comp => {
      // 实现和Atlas搜索一致的编辑距离匹配逻辑
      const leven = require('leven');
      return leven(comp.name, tags) <= 2;
    })
  };
});

此方案适合数据量较小的场景，避免数据库端的计算开销。

方案3：$search结合$match+$filter #

若需在数据库端完成完整筛选，可在$search之后添加$match阶段确保只保留有匹配元素的文档，再用$project提取匹配项：

const pipeline = [
  query1,
  {
    '$match': {
      '$expr': {
        '$gt': [
          {
            '$size': {
              '$filter': {
                'input': '$companies',
                'as': 'comp',
                'cond': {
                  '$regexMatch': {
                    'input': '$$comp.name',
                    'regex': tags,
                    'options': 'i'
                  }
                }
              }
            }
          },
          0
        ]
      }
    }
  },
  {
    '$project': {
      '_id': 1,
      'displayName': 1,
      'email': 1,
      'matchedCompanies': {
        '$filter': {
          'input': '$companies',
          'as': 'comp',
          'cond': {
            '$regexMatch': {
              'input': '$$comp.name',
              'regex': tags,
              'options': 'i'
            }
          }
        }
      }
    }
  }
];

关键说明 #

• 聚合管道中不能使用位置投影符$，这是MongoDB的固有规则，必须用$filter等数组操作符替代。
• 编辑距离类的复杂匹配，优先在应用层处理更高效；若需数据库端执行，需确认MongoDB版本支持$function，并评估性能影响。

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