Google Guava - 过滤单个属性的多个输入

Question

我刚进入 Google Guava，它似乎是一个强大的工具，我了解了如何使用谓词并按特定属性进行过滤。如何在FluentIterable 中链接谓词我的问题是过滤单个属性的最佳方法是什么。

例如，如果我有一个 Cars 集合。如何过滤 Cars.getPaintColor() 以提供黑色、红色和黄色的汽车？创建 3 个单独的谓词并使用 FluentIterable 似乎很笨拙。特别是在我的使用中，我可能希望在同一个属性上使用 10 多个过滤器，并且我不想创建 10 个谓词。

谢谢！

        List<String> colorList = (List<String>)filterCriteria.get("Color");
        List<String> makeList = (List<String>)filterCriteria.get("Make");
        List<String> rimSizeList = (List<String>)filterCriteria.get("RimSize");

        Predicate<String> predColor = Predicates.in(ImmutableSet.copyOf(colorList));
        Predicate<CarObj> predDirection2 = Predicates.compose(predColor ,[????] );

        Predicate<String> predMakeList  = Predicates.in(ImmutableSet.copyOf(makeList));
        Predicate<CarObj> predMakeList2 = Predicates.compose(predMakeList, [????] );

        Predicate<String> predRimSize = Predicates.in(ImmutableSet.copyOf(rimSizeList));
        Predicate<CarObj> predRimSize2 = Predicates.compose(predRimSize, [????] );

        Collection<CarObj> filtered = FluentIterable.from(mAllCars)
                .filter(predDirection2)
                .filter(predMakeList2)
                .filter(predRimSize2)
                .toList();

由于我使用的是 List，所以我在创建 ImmutableSet 时使用了 copyOf 而不是 of。

我不确定在 compose 的第二个参数中放什么。我猜它是这样的……在 CarObj 类中。

static Predicate<CarObj> byColor= new Predicate<CarObj>() {
    public boolean apply(CarObj input) {

        // What do I put here?
    }
};

Answer 1

因此，要检查油漆颜色是黑色、浅色还是黄色之一，您需要创建一个 Predicate 来检查集合是否包含该颜色：

Predicate<PaintColor> p = Predicates.in(ImmutableSet.of(
    PaintColor.RED, PaintColor.BLACK, PaintColor.YELLOW));

然后您可以使用 Function<Car, PaintColor> 来组合它，它返回您的类的油漆颜色属性：

Predicate<Car> p2 = Predicates.compose(p, Car.GET_PAINT_COLOR_FUNCTION);

编辑：

Car.GET_PAINT_COLOR_FUNCTION 我的意思是这样的：

public static final Function<Car, PaintColor> GET_PAINT_COLOR_FUNCTION =
    new Function<Car, PaintColor>() {
      @Override public PaintColor apply(Car car) {
        return car.getPaintColor();
      }
    });

正如我在 cmets 中所说，您可以根据需要将其调整为您的实际类型。例如，改为 Function<CarObj, String>。

Answer 2

按照 ColinD 的建议，将提取的 Function<Car, PaintColor> 与 Predicates.in() 组合在一起的替代方法是编写参数化的 Predicate<Car>：

public class CarPaintColorPredicate implements Predicate<Car> {
    private final PaintColor paintColor;

    public CarPaintColorPredicate(PaintColor paintColor) {
        this.paintColor = paintColor;
    }

    @Override
    public boolean apply(@Nullable Car input) {
        return input != null && input.getPaintColor() == paintColor;
    }
}

你可以直接使用它：

FluentIterable.from(cars)
        .filter(new CarPaintColorPredicate(PaintColor.RED))
        .toList();

或组合多种颜色：

FluentIterable.from(cars)
        .filter(Predicates.or(
            new CarPaintColorPredicate(PaintColor.RED),
            new CarPaintColorPredicate(PaintColor.BLACK)))
        .toList();

甚至与其他类型的谓词结合：

FluentIterable.from(cars)
        .filter(new CarPaintColorPredicate(PaintColor.RED))
        .filter(new CarMakePredicate("Ferrari"))
        .toList();

---

为了完整，带有Function<Car, PaintColor>的版本如下：

public enum CarPaintColorFunction implements Function<Car, PaintColor> {
    INSTANCE;

    @Override
    public PaintColor apply(@Nullable Car input) {
        return input == null ? null : input.getPaintColor();
    }
}

Function 仅返回属性值，然后通过Predicate 组合将其与接受值的集合（希望是Set）进行比较：

FluentIterable.from(cars)
        .filter(Predicates.compose(
            Predicates.in(Sets.immutableEnumSet(PaintColor.RED, PaintColor.BLACK)),
            CarPaintColorFunction.INSTANCE))
        .toList();

所有这些都在 Guava Wiki 的 Functional Explained 页面中得到了真正的解释。