从 List中提取对象基于一些其他列表属性集合

Question

我有一个类 -

public class Data implements Identifiable{
    private Integer id;

    public Integer getId(){
        return id;
    }
}

现在我有两个集合 -

List<Data> data1 = // few hundred Objects

Set<Integer> dataIds = // few object ids

我想从 data1 中提取 List ，它的 id 在 dataIds 中>

我的方法应该如何？我在我的类路径中使用了番石榴，因此如果性能/效率具有可比性，可以使用番石榴的功能方法。

Answer 1

除非您想要做的只是迭代结果一次或者您需要可重用的实时筛选视图，否则您可能需要一个包含匹配项的非视图列表。创建 List 或 Set 来存储结果，然后迭代数据列表并添加匹配项，这是一种非常好的方法，而且易于理解！

List<Data> result = Lists.newArrayList();
for (Data data : data1) {
  if (dataIds.contains(data.getId()))
    result.add(data);
}

我看到您的 Data 类实现了 Identifying 接口。鉴于此，您可以创建一个 Function 来获取 ID...Identifyings.getIdFunction() 或其他内容。这很好，因为它可能在其他各个地方都很有用（我在博客文章 此处）。有了这个，使用 Guava 执行此操作也将相当简单：

Predicate<Identifiable> predicate = Predicates.compose(
    Predicates.in(dataIds), Identifiables.getIdFunction());
List<Data> filtered = Lists.newArrayList(Iterables.filter(data1, predicate));

这在功能上基本上等同于第一个示例，但似乎更难理解。由于这样做没有任何明显的好处（与您只想使用实时视图的情况不同），我的建议是选择第一个。

Answer 2

怎么样

Collections2.filter(
    data1,
    new Predicate<Data>() {
       public boolean apply(Data d) {
         return dataIds.contains(d.getId());
       }
    }
)

？ 记住不要让事情变得过于复杂，除非确实有必要。

Answer 3

使用 LambdaJ 您可以编写：

List<Data> result = extract(data1, on(Data.class).getId());