2017-05-25 56 views
2

如何构造一个函数,其中apply需要多个参数?构成多个参数的Scala函数

这是一个人为的例子:

val sum: List[Int] => Int = l => l.sum 

val double: Int => Int = i => i * i 

double.compose(sum).apply(List(1,2,3)) //=> 36 

val sumAppend: (List[Int], Int) => Int = (l, i) => i :: l sum 

double.compose(sumAppend).apply(List(1,2,3), 1) // Attempt to append 1 to list then sum 

上面给我一个类型推断的错误?

回答

2

定义compose2,例如作为一个扩展方法Function1

implicit class ComposeFunction2[A, B, C, D](f1: Function1[C, D]) { 
    def compose2(f2: Function2[A, B, C]): Function2[A, B, D] = 
    (a: A, b: B) => f1(f2(a, b)) 
} 

这将是比快的替代品,因为它不分配的元组。用法:

scala> val double: Int => Int = i => i * i 
double: Int => Int = <function1> 

scala> val sumAppend: (List[Int], Int) => Int = (l, i) => i :: l sum 
sumAppend: (List[Int], Int) => Int = <function2> 

scala> double.compose2(sumAppend).apply(List(1,2,3), 1) 
res5: Int = 49 
1

在这种情况下,您可以只编写带有单个参数的函数。所以sumAppend必须是一个函数,它接受任何类型的单个参数,并且必须返回Int的结果(即_ => Int)。

您可以将一个带有两个参数的函数转换为一个curried函数,并部分应用该函数,如下所示。

scala> val sumAppend = (l: List[Int], i: Int) => (i :: l).sum 
sumAppend: (List[Int], Int) => Int = $$Lambda$1630/[email protected] 

scala> double.compose(sumAppend.curried.apply(List(1,2,3))).apply(10) 
res18: Int = 256 

另一个选择是让sumAppend接受一个单参数,它是List [Int]和Int的元组。

scala> val sumAppend: ((List[Int], Int)) => Int = l => (l._2 :: l._1).sum 
sumAppend: ((List[Int], Int)) => Int = $$Lambda$1597/[email protected] 

scala> val x = double.compose(sumAppend).apply((List(1,2,3),10)) 
x: Int = 256