过滤Java Collection的最佳方法是什么?

问题

我想基于谓词过滤ajava.util.Collection


#1 热门回答(523 赞)

Java 8(2014)在一行代码中使用流和lambda解决了这个问题:

List<Person> beerDrinkers = persons.stream()
    .filter(p -> p.getAge() > 16).collect(Collectors.toList());

这是atutorial

使用Collection#removeIf修改集合到位。 (注意:在这种情况下,谓词将删除满足谓词的对象):

persons.removeIf(p -> p.getAge() <= 16);

lambdaj允许过滤集合而无需编写循环或内部类:

List<Person> beerDrinkers = select(persons, having(on(Person.class).getAge(),
    greaterThan(16)));

你能想象更具可读性的东西吗?

**免责声明:**我是lambdaj的贡献者


#2 热门回答(211 赞)

假设你使用的是Java 1.5,并且你无法添加Google Collections,我会做一些非常类似于Google员工的事情。这是Jon的评论略有不同。

首先将此接口添加到你的代码库。

public interface IPredicate<T> { boolean apply(T type); }

当某个谓词对某种类型成立时,它的实现者可以回答它。例如。如果TwereUserAuthorizedUserPredicate<User>implementsIPredicate<T>,那么AuthorizedUserPredicate#apply返回是否已通过inUser授权。

然后在一些实用类中,你可以说

public static <T> Collection<T> filter(Collection<T> target, IPredicate<T> predicate) {
    Collection<T> result = new ArrayList<T>();
    for (T element: target) {
        if (predicate.apply(element)) {
            result.add(element);
        }
    }
    return result;
}

所以,假设你已经使用了以上可能

Predicate<User> isAuthorized = new Predicate<User>() {
    public boolean apply(User user) {
        // binds a boolean method in User to a reference
        return user.isAuthorized();
    }
};
// allUsers is a Collection<User>
Collection<User> authorizedUsers = filter(allUsers, isAuthorized);

如果关注线性检查的性能,那么我可能希望拥有一个具有目标集合的域对象。具有目标集合的域对象将具有用于初始化,添加和设置目标集合的方法的过滤逻辑。

更新:

在实用程序类(比如谓词)中,我添加了一个select方法,当谓词没有返回预期值时,默认值选项,以及在新IPredicate中使用的params的静态属性。

public class Predicate {
    public static Object predicateParams;

    public static <T> Collection<T> filter(Collection<T> target, IPredicate<T> predicate) {
        Collection<T> result = new ArrayList<T>();
        for (T element : target) {
            if (predicate.apply(element)) {
                result.add(element);
            }
        }
        return result;
    }

    public static <T> T select(Collection<T> target, IPredicate<T> predicate) {
        T result = null;
        for (T element : target) {
            if (!predicate.apply(element))
                continue;
            result = element;
            break;
        }
        return result;
    }

    public static <T> T select(Collection<T> target, IPredicate<T> predicate, T defaultValue) {
        T result = defaultValue;
        for (T element : target) {
            if (!predicate.apply(element))
                continue;
            result = element;
            break;
        }
        return result;
    }
}

以下示例查找集合之间缺少的对象:

List<MyTypeA> missingObjects = (List<MyTypeA>) Predicate.filter(myCollectionOfA,
    new IPredicate<MyTypeA>() {
        public boolean apply(MyTypeA objectOfA) {
            Predicate.predicateParams = objectOfA.getName();
            return Predicate.select(myCollectionB, new IPredicate<MyTypeB>() {
                public boolean apply(MyTypeB objectOfB) {
                    return objectOfB.getName().equals(Predicate.predicateParams.toString());
                }
            }) == null;
        }
    });

以下示例在集合中查找实例,并在未找到实例时将集合的第一个元素作为默认值返回:

MyType myObject = Predicate.select(collectionOfMyType, new IPredicate<MyType>() {
public boolean apply(MyType objectOfMyType) {
    return objectOfMyType.isDefault();
}}, collectionOfMyType.get(0));

更新(在Java 8发布之后):

我(艾伦)第一次发布这个答案已经好几年了,我仍然无法相信我正在为这个答案收集SO分。无论如何,现在Java 8已经引入了语言的闭包,我的答案现在会有很大的不同,而且更简单。使用Java 8,不需要一个独特的静态实用程序类。所以如果你想找到与你的谓词匹配的第一个元素。

final UserService userService = ... // perhaps injected IoC
final Optional<UserModel> userOption = userCollection.stream().filter(u -> {
    boolean isAuthorized = userService.isAuthorized(u);
    return isAuthorized;
}).findFirst();

可选项的JDK 8 API具有get(),isPresent(),orElse(defaultUser),orElseGet(userSupplier)orElseThrow(exceptionSupplier)以及其他"monadic"函数,如map,flatMapfilter

如果你只想收集与谓词匹配的所有用户,请使用Collectors终止所需集合中的流。

final UserService userService = ... // perhaps injected IoC
final List<UserModel> userOption = userCollection.stream().filter(u -> {
    boolean isAuthorized = userService.isAuthorized(u);
    return isAuthorized;
}).collect(Collectors.toList());

有关Java 8流如何工作的更多示例,请参见here


#3 热门回答(87 赞)

使用来自Apache Commons的CollectionUtils.filter(Collection,Predicate)