分类
Java

Java8 Function和BiFunction

Function

  1. Function作为一个函数式接口,主要方法apply接收一个参数,返回一个值
@FunctionalInterface
public interface Function<T, R> {

    /**
     * Applies this function to the given argument.
     *
     * @param t the function argument
     * @return the function result
     */
    R apply(T t);
}

首先我们来写一个计算数字的方法

public int compute(int a, Function<Integer, Integer> function) {
    int result = function.apply(a);
    return result;
}

然后我们调用这个方法

test.compute(5,value -> value * value) //25 计算平方
test.compute(5,value -> value + value) //10 求和
test.compute(5,value -> value - 2) //3  

可以看到我们定义一个方法就可以实现多种功能,这就是前面说过的Lambda表达式传递的是一种行为,我们把想要做的事在调用的时候,以一种行为的方式传递进来,程序读起来也更加直观

  1. Function compose方法

compose方法是一个默认方法,这个方法接收一个function作为参数,将参数function执行的结果作为参数给调用的function,以此来实现两个function组合的功能。

public interface Function<T, R> {
    default <V> Function<V, R> compose(Function<? super V, ? extends T> before) {
        Objects.requireNonNull(before);
        return (V v) -> apply(before.apply(v));
    }
}

下面我们来举例看一下:

public int compute(int a, Function<Integer, Integer> function1, Function<Integer, Integer> function2) {
    return function1.compose(function2).apply(a);
}

调用这个方法:

test.compute(2, value -> value * 3, value -> value * value) 

大家可以猜一下上面的结果是多少?

我们来分析一下:

function1.compose(function2).apply(a);

compose方法内部代码是:

return (V v) -> apply(before.apply(v));

返回的是一个Function,输入一个参数,返回一个参数值,这个Function 在调用apply时首先执行的是 before.apply(v) before在这里就是value -> value * value,也就是 2*2,将得到的结果4,作为参数传递给function1,在这里就是value -> value * 3 ,所以结果是:12

3.Function andThen

了解了compose方法,我们再来看andThen方法就好理解了,听名字就是“接下来”,andThen方法也是接收一个function作为参数,与compse不同的是,先执行本身的apply方法,将执行的结果作为参数给参数中的function。

public interface Function<T, R> {
    default <V> Function<T, V> andThen(Function<? super R, ? extends V> after) {
        Objects.requireNonNull(after);
        return (T t) -> after.apply(apply(t));
    }
}

下面我们来举例看一下:

public int compute2(int a, Function<Integer, Integer> function1, Function<Integer, Integer> function2) {
    return function1.andThen(function2).apply(a);
}

调用这个方法:

test.compute2(2, value -> value * 3, value -> value * value) 

这次结果是多少呢?我想大家应该很容易就知道了,首先执行的是 value -> value * 3 结果是6,然后执行的是 value * value 最终结果是36

BiFunction

了解了Function以后,有的小伙伴可能会问了,Function只能接收一个参数,如果我要传递两个参数呢,这一点Java8也替我们考虑到了,就是我们截下来要讲到的 BiFunction,首先还是直接上源码:

@FunctionalInterface
public interface BiFunction<T, U, R> {
    R apply(T t, U u);
 
    default <V> BiFunction<T, U, V> andThen(Function<? super R, ? extends V> after) {
        Objects.requireNonNull(after);
        return (T t, U u) -> after.apply(apply(t, u));
    }
}

可以看到BiFunction的apply方法,接收两个参数,返回一个值,来看一个例子

public int compute3(int a, int b, BiFunction<Integer, Integer, Integer> biFunction) {
   return biFunction.apply(a, b);
}

我们定义了一个方法,可以用来计算两个数的很多运算

test.compute3(2, 3, (v1, v2) -> v1 + v2) //5
test.compute3(2, 3, (v1, v2) -> v1 - v2) //-1
test.compute3(2, 3, (v1, v2) -> v1 * v2) //6

这个还是蛮简单的,我们再来看一下,BiFunction中有一个andThen方法,参数是一个Function,方法主要是将BiFunction返回的结果作为Function的参数,得出一个结果,举例:

public int compute4(int a, int b, BiFunction<Integer, Integer, Integer> biFunction,Function<Integer, Integer> function) {
    return biFunction.andThen(function).apply(a, b);
}
test.compute4(2, 3, (v1, v2) -> v1 + v2, v1 -> v1 * v1)

首先执行(v1, v2) -> v1 + v2,然后执行 v1 -> v1 * v1。

有的同学可能会问为什么BiFunction没有compose方法呢,大家仔细想一想,如果有compose方法的话,那就是先执行Function的apply方法,但是执行完毕后只返回一个参数,而BiFunction需要两个参数,所以肯定是不行的。

作者:大鹏_xzlp
链接:https://www.jianshu.com/p/8dc46a2dc21d
来源:简书

发表评论

电子邮件地址不会被公开。