我有一个简化的代码版本。显而易见,我在概念上想要什么:
def heavyCalcMul: Int => Int = i => i * 2
def heavyCalcDiv: Int => Int = i => i / 2
def heavyCalcPls: Int => Int = i => i + 2
我这样使用它:
val x = 2
val midResult = heavyCalcMul(x)
val result = heavyCalcDiv(midResult) + heavyCalcPls(midResult)
但是我想用这种风格重写这段代码:
val x = 2
val result = heavyCalcMul(x) { midResult: Int =>
heavyCalcDiv(midResult) + heavyCalcPls(midResult)
}
有可能吗?
答案 0 :(得分:5)
您可以使用andThen
:
val calc = heavyCalcMul
.andThen(mid =>
heavyCalcDiv(mid) + heavyCalcPls(mid)
)
val result2 = calc(x)