如何将函数引用传递给我的方法并使用它们?

时间:2018-03-19 07:15:39

标签: java lambda java-8 method-reference

我有一个工作方法,我想使用lambdas或函数引用更通用。也就是说,对于在我的方法中调用的一些方法,我想将它们指定为我的方法的参数。但是我无法理解它。

假设我有以下基本类结构

class B {}
class C {
    public Set<Integer> getIds() { ... }
    public void addB(B b) { ... }
}

我有自己的工作方法

private void p1(Map<Integer, C> cs, Map<Integer, B> bs) {
    for (C c : cs.values()) {
        for (Integer id : c.getIds()) {
            if (bs.containsKey(id)) {
                c.addB(bs.get(id));
            }
        }
    }
}

因此被称为p1(cs, bs); - 到目前为止一直很好。

使其更通用的第一步是为getIds()方法提供lambda。我使用以下方法实现了这一目标:

private void p1(Map<Integer, C> cs, Map<Integer, B> bs, Function<C, Set<Integer>> f) {
    for (C c : cs.values()) {
        for (Integer id : f.apply(c)) {  // Change here
            if (bs.containsKey(id)) {
                c.addB(bs.get(id));
            }
        }
    }
}

这就是所谓的:p1(cs, bs, C::getIds); - 再次,到目前为止一切顺利。我现在正在传递getIds()作为我的函数的一个参数并使用它。

现在我想为addB()方法提供一个lambda作为p1()的第四个参数,我找不到正确的方法。我尝试过几件事,比如说。 FunctionConsumer<B>,但我无法弄清楚如何做到这一点。有什么建议吗?

1 个答案:

答案 0 :(得分:5)

BiConsumer<C,B>应该有效:

private void p1(Map<Integer, C> cs, Map<Integer, B> bs, Function<C, Set<Integer>> f, BiConsumer<C,B> cons) {
    for (C c : cs.values()) {
        for (Integer id : f.apply(c)) {
            if (bs.containsKey(id)) {
                cons.accept(c,bs.get(id));
            }
        }
    }
}

然后你可以传递lambda表达式(c,b)->c.addB(b)或方法引用C::addB作为方法的第四个参数。