SpringData-MongoDB按给定顺序通过ID获取对象

Question

我有一个这样的用例。我在Mongo DB中有问题，有CRUD微服务。在那里，我公开了一个API方法，通过查询参数提供的ID列表获取问题。比如说，为了简单起见，用户会提出一些问题？id=2，id=7，id=4，id=5

然后我需要按照同样的顺序返回问题列表，如下所示

questions: [
    {
       id: 2,
       prompt: "prompt one",
       ...
    },
    {
        id: 7,
        prompt: "prompt two",
        ...
    },
    {
        id: 4,
        ...
    },
    {
        id: 5
        ...
    }
]

但是请注意，这既不是ASC也不是DESC，而是可以是任何任意顺序，例如/api/问题？id=2， id=7， id=4， id=5

我正在使用org。springframework。数据mongodb。存储库。MongoRepository作为我的DAO类。目前，在通过基于spring数据的存储库获取数据后，我正在我的服务层内进行排序。这个解决方案有效。但我更愿意在DAO存储库级别本身完成这种排序，因为它成本更低，并且不会给服务层的业务逻辑增加不必要的复杂性。相反，我可以仅将责任委托给DB，假设它能够通过更多的优化来完成这些事情。

我可以使用spring数据实现这一点吗？如果是这样的话，怎么做呢？

Answer 1

如果您正在使用MongoRepository，请创建一个扩展它的接口。例子：

public interface MyRepository extends MongoRepository<Question, String>{

    @Query("{_id: { $in: ?0 } })")
    List<Question> findByIds(List<String> ids, Sort sort);
}

然后在您的Service类或您使用存储库的任何地方。添加以下内容：

Sort sort = new Sort(Direction.ASC,"_id"); // Or DESC
List<Question> questionsById = repository.findByIds(ids, sort);

当然，这里的问题是一个虚拟类，我创建这个类是为了测试它，用您的类替换它。

好的，如果您需要按输入用户顺序，那么您真的无法使用聚合框架。

如果我在数据库中按该顺序插入了以下数据集：

/* 1 */
{
    "_id" : ObjectId("5a103a434d8a2fe38bec5c5e"),
    "prompt" : "prompt one"
}

/* 2 */
{
    "_id" : ObjectId("5a103a434d8a2fe38bec5c60"),
    "prompt" : "prompt two"
}

/* 3 */
{
    "_id" : ObjectId("5a103a434d8a2fe38bec5c62"),
    "prompt" : "prompt three"
}

/* 4 */
{
    "_id" : ObjectId("5a103a434d8a2fe38bec5c64"),
    "prompt" : "prompt four"
}

那么聚合管道必须看起来像：

db.getCollection('questions').aggregate([{
        "$match": {
            "_id": {
                "$in": [ObjectId("5a103a434d8a2fe38bec5c62"), ObjectId("5a103a434d8a2fe38bec5c60"), ObjectId("5a103a434d8a2fe38bec5c64"), ObjectId("5a103a434d8a2fe38bec5c5e")]
            },
        }
    },
    {
        "$project": {
            "orderByInputId": {
                "$cond": [{
                        "$eq": ["$_id", ObjectId("5a103a434d8a2fe38bec5c62")]
                    },
                    1,
                    {
                        "$cond": [{
                                "$eq": ["$_id", ObjectId("5a103a434d8a2fe38bec5c60")]
                            },
                            2,
                            {
                                "$cond": [{
                                        "$eq": ["$_id", ObjectId("5a103a434d8a2fe38bec5c64")]
                                    },
                                    3,
                                    4
                                ]
                            }
                        ]
                    }
                ]
            },
            prompt: 1
        }
    },

    // Sort the results
    {
        "$sort": {
            "orderByInputId": 1
        }
    }
])

我得到的结果如下：

/* 1 */
{
    "_id" : ObjectId("5a103a434d8a2fe38bec5c62"),
    "prompt" : "prompt three",
    "orderByInputId" : 1.0
}

/* 2 */
{
    "_id" : ObjectId("5a103a434d8a2fe38bec5c60"),
    "prompt" : "prompt two",
    "orderByInputId" : 2.0
}

/* 3 */
{
    "_id" : ObjectId("5a103a434d8a2fe38bec5c64"),
    "prompt" : "prompt four",
    "orderByInputId" : 3.0
}

/* 4 */
{
    "_id" : ObjectId("5a103a434d8a2fe38bec5c5e"),
    "prompt" : "prompt one",
    "orderByInputId" : 4.0
}