MongoDB聚合中如何对数组内字符串使用$substr实现匹配排序

我写了一段MongoDB聚合查询，用来找出tags字段中以指定query开头的内容，并且优先返回精确匹配的结果。当tags是单个字符串时查询正常，但tags是字符串数组时，$substr函数因为无法把数组转成字符串报错。得修改代码让$substr作用于数组里的每个字符串，实现查询“123”时的预期排序结果。

原聚合查询代码 #

results = await Post.aggregate([
    // 查找所有tags以给定query开头的文档
    { $match: { 'tags': new RegExp("^" + query, 'i') } },
    { $addFields: {
        // 标记是否有tags值与query精确匹配
        'exactMatch': { $eq: [ '$tags', query ] },
        // 标记是否有tags值的前3个字符与query匹配
        'startsWith': { $eq: [ { $substr: [ '$tags', 0, 3 ] }, query ] },
    } },
    // 优先返回精确匹配的结果
    { $sort: { 'exactMatch': -1, 'startsWith': -1 } },
    // 移除添加的字段
    { $project: { 'exactMatch': 0, 'startsWith': 0 } }
])

报错信息 #

MongoServerError: PlanExecutor error during aggregation :: caused by :: can't convert from BSON type array to String

示例数据、输入及预期结果 #

示例数据 #

[
    {
        title: 'Post1',
        content: 'This is test post 1.',
        tags: [
            'post', '12345'
        ]
    },
    {
        title: 'Post2',
        content: 'This is test post 2.',
        tags: [
            'post', '789'
        ]
    },
    {
        title: 'Post3',
        content: 'This is test post 3.',
        tags: [
            'post', '123'
        ]
    }
]

输入与预期结果 #

输入query为“123”时，预期结果：

[
    {
        title: 'Post3',
        content: 'This is test post 3.',
        tags: [
            'post', '123'
        ]
    },
    {
        title: 'Post1',
        content: 'This is test post 1.',
        tags: [
            'post', '12345'
        ]
    }
]

修改后的聚合查询代码 #

const query = "123";
const queryLength = query.length;

results = await Post.aggregate([
    // 筛选tags数组中包含以query开头元素的文档
    { $match: { 'tags': new RegExp("^" + query, 'i') } },
    { $addFields: {
        // 判断tags数组中是否存在与query精确匹配的元素
        'exactMatch': {
            $anyElementTrue: {
                $map: {
                    input: '$tags',
                    as: 'tag',
                    in: { $eq: ['$$tag', query] }
                }
            }
        },
        // 判断tags数组中是否存在元素以query开头
        'startsWith': {
            $anyElementTrue: {
                $map: {
                    input: '$tags',
                    as: 'tag',
                    in: { $eq: [ { $substr: ['$$tag', 0, queryLength] }, query ] }
                }
            }
        }
    } },
    // 排序规则：精确匹配优先，其次是前缀匹配
    { $sort: { 'exactMatch': -1, 'startsWith': -1 } },
    // 移除临时字段
    { $project: { 'exactMatch': 0, 'startsWith': 0 } }
])

关键修改说明 #

1. 精确匹配判断：用$map遍历tags数组的每个元素，逐个与query比较，再通过$anyElementTrue判断是否存在匹配项，解决了原代码直接用数组和字符串比较的错误。
2. 前缀匹配判断：同样用$map遍历数组，对每个元素使用$substr取前缀（前缀长度设为query的实际长度，避免写死3导致的通用性问题），再判断是否与query相等，最后用$anyElementTrue确认是否存在符合条件的元素。
3. 保持原有逻辑：$match和$sort、$project阶段的核心逻辑不变，仅修正了处理数组字段的错误。

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