如何在 Spring Data Repo 中模仿 List<Optional<T>> 之类的行为？

Question

我正在尝试使用list of Id's query 数据库，如果它们不存在，请对它们做一些事情。

我的计划是做类似的事情

List<Optional<MyObject>> fingById(Set<String> myIds);

我知道这不会开箱即用，因为我认为即使所有 id 都存在，也无法保证它们会被订购，所以如果其中一些存在，我会得到一些随机的 null 选项将无法使用返回它的 ID。

基本上我最终需要做的是检索这些对象，更新一个字段或使用默认值（如果不存在）创建它们。

批量执行此操作的最佳方法是什么？

谢谢:)

Answer 1

您可以通过他们的 id 将结果映射到Map<String, Optional<MyObject>>，然后用Optional.empty() 填写空白。

List<MyObject> findByIdIn(Set<String> ids);

default Map<String, Optional<MyObject>> mapByIdIn(Set<String> ids) {
    Map<String, Optional<MyObject>> found = findByIdIn(ids)
            .stream()
            .collect(Collectors.toMap(MyObject::getId, Optional::of));
    ids.stream()
       .filter(id -> !found.containsKey(id))
       .forEach(id -> found.put(id, Optional.empty()));
    return found;
}

Answer 2

我认为使用 List 的 Optional 值不是一个好主意。相反，我建议只使用存储库的默认方法并查询所有现有实体。

List<MyObject> findAll();

之后，您可以遍历所有对象并检查缺少的 id。为了提高性能，我建议使用Map：

var objectMap = repository.findAll().stream()
    .collect(Collectors.toMap(MyObject::getId, Function::Identity);

然后，您可以简单地检查地图中是否包含任何 id：

myIds.stream().filter(objectMap::contains)
    .forEach({
        // do some stuff, present id
    });

myIds.stream().filter((id) -> !myMap.contains(id))
    .forEach({
        // do some stuff, not present id
    });