时间戳

首页 > 解决方案 > Mongodb在数组中排序

javascript - Mongodb在数组中排序

问题描述

有没有办法按给定的数组排序？

像这样的东西：

const somes = await SomeModel.find({}).sort({'_id': {'$in': [ObjectId('sdasdsd), ObjectId('sdasdsd), ObjectId('sdasdsd)]}}).exec()

我正在寻找一种获得解决方案的方法，以获取集合的所有文档并按文档的 _id 是否与给定数组之一匹配进行排序。

一个例子：

我们有专辑收藏和歌曲收藏。在专辑集合中，我们存储属于专辑的歌曲的 ID。

我想获取歌曲，但如果歌曲在专辑中，请将它们放在数组的前面。

我解决了这个如下，但它看起来有点hacky：

const songs = await SongMode.find({}).skipe(limit * page).limit(limit).exec();
const album = await AlbumModel.findById(id).exec();

if(album) {
    songArr = album.songs.slice(limit * page);

    for(let song of album.songs) {
        songs.unshift(song);
        songs.pop();
    }
}

标签： javascriptnode.jsmongodbmongoosenosql

解决方案

这不能使用普通的.find().sort(). 相反，您将需要使用 MongoDB 聚合管道 ( .aggregate())。具体来说，您需要执行以下操作：

  1. 执行一个$project离子，如果_id是$in数组，则新sort_field的值为1，否则为0。
  2. 执行一个$sort这样的命令，你正在对 new 进行降序排序sort_field。

如果您使用的是 MongoDB 3.4 或更高版本，那么这很容易，因为有以下$addFields运算符：

const your_array_of_ids = [
    ObjectId('objectid1'),
    ObjectId('objectid2'),
    ObjectId('objectid3')
];

SomeModel.aggregate([
    { '$addFields': {
        'sort_field': { '$cond': {
            'if': { '$in': [ '$_id', your_array_of_ids ] },
            'then': 1,
            'else': 0
        }}
    }},
    { '$sort': {
        'sort_field': -1
    }}
]);

如果您使用的是旧版本的 MongoDB，则解决方案类似，但$addFields您将使用$project. 此外，您将需要明确包含您想要包含的所有其他字段，否则它们将被排除在结果之外。

推荐阅读