java - 在实体集合中查找所有 id 集合的最有效方法

Question

我有一个实体:

public class Entity
{
    private long id;    
    private String data;

    public long getId() {
        return id;
    }

    public String getData() {
        return data;
    }
}

以及实体的集合:

Collection<Entity> entities= ...

找到Collection<Long> 的最有效方法是什么？ entities 中的所有 id ?

Answer 1

假设你有

class Entity {
    final long id;
    final String data;

    public long getId() {
        return id;
    }

    public String getData() {
        return data;
    }

    Entity(long id, String data) {
        this.id = id;
        this.data = data;
    }
}

在 Java 8 中你可以编写

Collection<Entity> entities = Arrays.asList(new Entity(1, "one"), 
                  new Entity(11, "eleven"), new Entity(100, "one hundred"));
// get a collection of all the ids.
List<Long> ids = entities.stream()
                         .map(Entity::getId).collect(Collectors.toList());

System.out.println(ids);

打印

[1, 10, 100]

您可以想象，这在 Java 7 或更低版本中相当难看。注意 Entity.getId 在应用于 map() 时意味着在每个元素上调用此方法。

现在，真正有趣的部分是你可以做到这一点。

List<Long> ids = entities.parallelStream()
                         .map(Entity::getId).collect(Collectors.toList());

在大多数情况下使用并行流会损害性能，但它使尝试并看到非常容易(可能太容易;)

---

最有效的方法是拥有或构建 map 。

Map<Long, Entity> entitiesMap = ...
// get all ids
Collection<Long> addIds = entitiesMap.keySet();

// look up entities by id.
List<Long> ids = ...
List<Entity> matching = new ArrayList<>();
for(Long id: ids)
    matching.add(entitiesMap.get(id));