我在代码中发现了这种模式很多次:
if (doIt)
object.callAMethod
else
object
我想知道是否有一种语法上更令人愉快的方式来编写上面的代码,尤其是为了避免重复object
变量。类似的东西:
// using the Scalaz "pipe" operator
// and "pimping" f: T => T with a `when` method
object |> (_.callAMethod).when(doIt)
不幸的是,上面的行失败了,因为类型推断需要(_.callAMethod)
的参数类型。
我现在最好的方法是:
implicit def doItOptionally[T](t: =>T) = new DoItOptionally(t)
class DoItOptionally[T](t: =>T) {
def ?>(f: T => T)(implicit doIt: Boolean = true) =
if (doIt) f(t) else t
}
implicit val doIt = true
object ?> (_.callAMethod)
不是很好,因为我必须声明一个implicit val
,但是如果有多个链式调用,这会得到回报:
object ?> (_.callAMethod) ?> (_.callAnotherMethod)
有没有人有更好的主意?我在这里错过了一些Scalaz魔法吗?
答案 0 :(得分:18)
class When[A](a: A) {
def when(f: A => Boolean)(g: A => A) = if (f(a)) g(a) else a
}
implicit def whenever[A](a: A) = new When(a)
示例:
scala> "fish".when(_.length<5)(_.toUpperCase)
res2: java.lang.String = FISH
答案 1 :(得分:2)
我无法评论你的答案@Rex Kerr,但更简洁的方法是:
implicit class When[A](a: A) {
def when(f: A => Boolean)(g: A => A) = if (f(a)) g(a) else a
}
只需将implicit
放在类之前就可以完全省略隐式函数。
答案 2 :(得分:0)
如果您将callAMethod
表示为内同态,则可以使用幺半群功能。类似的东西:
object |> valueOrZero(doIt, Endo(_.callAMethod))
(可能需要Endo
上的类型参数)