4

我很想let在 Scala 中构建类似于 Haskell 中的构造。我尝试了几种方法,但似乎没有一个是好的。这是一些代码:

object CustomLet extends App {
  val data = for (i <- 1 to 1024; j <- 1 to 512) yield (i % j) * i * (i + 1) - 1

  def heavyCalc() = { println("heavyCalc called"); data.sum }

  def doSomethingWithRes(res: Int) = {
    println(s"${res * res}")
    1
  }

  def cond(value: Int): Boolean = value > 256

  // not really usable, even though it's an expression (2x heavyCalc calls)
  def withoutLet() = if (cond(heavyCalc())) doSomethingWithRes(heavyCalc()) else 0

  // not an expression
  def letWithVal(): Int = {
    val res = heavyCalc()
    if (cond(res)) doSomethingWithRes(res)
    else 0
  }

  // a lot of code to simulate "let", at least it is an expression
  def letWithMatch(): Int = heavyCalc() match {
    case res => if (cond(res)) doSomethingWithRes(res) else 0
  }

  // not perfect solution from
  // http://stackoverflow.com/questions/3241101/with-statement-equivalent-for-scala/3241249#3241249
  def let[A, B](param: A)(body: A => B): B = body(param)

  // not bad, but I'm not sure if it could handle more bindings at once
  def letWithApp(): Int = let(heavyCalc()) {res => if (cond(res)) doSomethingWithRes(res) else 0}

  List[(String, () => Int)](
    ("withoutLet", withoutLet),
    ("letWithVal", letWithVal),
    ("letWithMatch", letWithMatch),
    ("letWithApp", letWithApp)
  ).foreach(
    item => item match {
      case (title, func) => {
        println(s"executing $title")
        val ret = func()
        println(s"$title finished with $ret")
        println()
      }
    }
  )
}

这是它的理想外观(只有一个绑定,更多可以用,; 分隔,不确定in关键字):

  // desired look
  def letTest(): Int =
    let res = heavyCalc() in
      if (cond(res)) doSomethingWithRes(res) else 0

我不确定这是否可能,但我没有使用大多数高级 Scala 东西(如宏)的经验,所以我无法确定。

EDIT1:需要明确的是,我对它的主要期望是:表达和相对简单的语法(如上面概述的)。

4

2 回答 2

6

您可以使用正向管道

object ForwardPipeContainer {
  implicit class ForwardPipe[A](val value: A) extends AnyVal {
    def |>[B](f: A => B): B = f(value)
  }
}

像这样使用:

import ForwardPipeContainer._

def f(i: Int) = i * i

println( f(3) |> (x => x * x) )

您可以将多个参数放在一个元组中:

println( (f(2), f(3)) |> (x => x._1 * x._2) )

如果与部分函数 synatx 结合使用,效果会更好:

println( (f(2), f(3)) |> { case (x, y) => x * y } )

这个答案是What is a good way of reusing function result in Scala的变体,两者都基于Cache 一个中间变量在一个单行中,我从中得到了最初的想法。

于 2013-08-24T19:19:16.317 回答
3
def letTest(): Int =
    let res = heavyCalc() in
      if (cond(res)) doSomethingWithRes(res) else 0

我会这样写:

def letTest(): Int = {
  val res = heavyCalc()
  if (cond(res)) doSomethingWithRes(res) else 0
}

忽略惰性,let只是一个引入词法范围的构造,将一些术语绑定到一些名称,然后返回一个表达式。所以在Scala中你会做

{ // new lexical scope
  // bind terms section
  val a = f()
  def b = a + g() // may be I don't want g to be evaluated unless b is needed
  val c = h()
  // result expression
  if (c) b else a 
}

如果你想确保块中没有其他事情发生,宏应该能够强制执行这种语法布局。实际上有一个名为Spores的 SIP(​​Scala 改进过程)提案,它会强制执行一些相同的约束(还有一个额外的约束:您不会在不知不觉中捕获对封闭对象的引用)。

请注意,Scala 中的块是计算结果为块中最后一个表达式的表达式。因此,让我从 Haskell中随机举一个例子:

aaa = let y = 1+2
          z = 4+6
          in let f = 3
                 e = 3
             in e+f

这转化为:

val aaa = {
  val y = 1 + 2
  val z = 4 + 6
  val u = {
    val f = 3
    val e = 3
    e + f
  }
  u
}

如您所见,块语句可以用作表达式。

于 2013-08-24T14:06:54.773 回答