Как я могу считать элементы в коллекции mon go, которые имеют пересекающиеся подколлекции? - PullRequest
1 голос
/ 16 января 2020

Для каждого элемента в моей коллекции мне нужно найти количество других элементов, имеющих пересекающиеся подколлекции. Например, для данной коллекции

[{id:1,"sub":[1, 2, 3]},
{id:2,"sub":[2, 3, 4]},
{id:3,"sub":[4, 5, 6],
{id:4,"sub":[7, 8, 9]}]

ожидаемый результат равен

[{id:1,"count":1},
{id:2,"count":2},
{id:3,"count":1},
{id:4,"count":0"}]

1 Ответ

1 голос
/ 17 января 2020

Начиная с алгоритма на чистом языке запросов MongoDB: вы должны реструктурировать свои документы так, чтобы каждый документ содержал свой начальный массив sub и массив всех других значений sub. Для этого вам нужно запустить $ group вместе с $ unwind . Тогда становится просто запустить $ map с $ setIntersect $ filter из всех пустых и равных собственным массивам и получить размер, используя $ size

db.collection.aggregate([
    {
        $group: {
            _id: null,
            current: { $push: "$$ROOT" },
            all: { $push: "$sub" }
        }
    },
    {
        $unwind: "$current"
    },
    {
        $project: {
            id: "$current.id",
            count: {
                $size: {
                    $filter: {
                        input: {
                            $map: {
                                input: "$all",
                                in: { $setIntersection: [ "$$this", "$current.sub" ] }
                            }
                        },
                        cond: {
                            $and: [ 
                                { $ne: [ "$$this", [] ] },
                                { $ne: [ "$$this", "$current.sub" ]}
                            ]
                        }
                    }
                }
            }
        }
    }
])

Пн go Детская площадка

Поскольку агрегация довольно сложна, нет смысла запускать ее строго типизированным способом в C#. Все, что вы можете сделать, это использовать BsonDocument класс для построения вашего конвейера, например:

var groupDef = new BsonDocument()
        {
            { "_id", "" },
            { "current", new BsonDocument(){ { "$push", "$$ROOT" } } },
            { "all", new BsonDocument(){ { "$push", "$sub" } } },
        };

var projectDef = BsonDocument.Parse(@"{
        id: ""$current.id"",
        _id: 0,
        count: {
        $size: {
            $filter: {
            input: {
                $map: {
                input: ""$all"",
                in: {
                    $setIntersection: [
                    ""$$this"",
                    ""$current.sub""
                    ]
                }
                }
            },
            cond: {
                $and: [
                {
                    $ne: [
                    ""$$this"",
                    []
                    ]
                },
                {
                    $ne: [
                    ""$$this"",
                    ""$current.sub""
                    ]
                }
                ]
            }
            }
        }
        }
    }");

var result = mongoDBCollection.Aggregate()
                                .Group(groupDef)
                                .Unwind("current")
                                .Project(projectDef)
                                .ToList();
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...