查詢匹配連續數組元素的文檔

Question

我有一個 MongoDB 集合，其中包含以下格式的文檔：

    { "_id" : 1, "tokens": [ "I", "have", "a", "dream" ] },
    { "_id" : 2, "tokens": [ "dream", "a", "little", "dream" ] },
    { "_id" : 3, "tokens": [ "dream", "a", "dream" ] },
    { "_id" : 4, "tokens": [ "a" , "little", "dream" ] },
    ...

我需要獲取所有“令牌”包含連續數組元素的文件： “a”、“dream” 。 因此，以下是匹配的文件：

    { "_id" : 1, "tokens": [ "I", "have", "a", "dream" ] },
    { "_id" : 3, "tokens": [ "dream", "a", "dream" ] },

有沒有辦法得到正確的結果？

Answer 1

一個技巧就是有一個正則表達式。

• $match獲取具有$all數組輸入的所有文檔
• $addFields具有重復的標記和輸入數組
• $reduce有助於連接所有字符串連接-
• $regexMatch匹配兩個字符串
• $match消除不需要的數據
• $project僅獲取必要的字段

代碼是

[{
    $match: {
        tokens: { $all: ["a", "dream"] }
    }
}, {
    $addFields: {
        duplicate: "$tokens",
        inputData: ["a", "dream"]
    }
}, {
    $addFields: {
        duplicate: {
            $reduce: {
                input: "$duplicate",
                initialValue: "",
                in: { $concat: ["$$value", "-", "$$this"] }
            }
        },
        inputData: {
            $reduce: {
                input: "$inputData",
                initialValue: "",
                in: { $concat: ["$$value", "-", "$$this"] }
            }
        }
    }
}, {
    $addFields: {
        match: {
            $regexMatch: { input: "$duplicate", regex: '$inputData' }
        }
    }
}, {
    $match: {
        match: true
    }
}, {
    $project: {  _id: 1,  tokens: 1 }
}]

工作蒙戈游樂場

注意：請檢查多個場景，盡管它適用於這種場景