如何定义类型:无限函数?

Zan*_*Jie 4 scala language-lawyer

如何定义类型InfiniteFunction,它是一个函数,调用时返回另一个InfiniteFunction

类型看起来像:

() => () => () => ... // infinite
Run Code Online (Sandbox Code Playgroud)

或递归:

type InfiniteFunction = () => InfiniteFunction
Run Code Online (Sandbox Code Playgroud)

这不起作用

scala> type InfiniteFunction = () => InfiniteFunction
<console>:11: error: illegal cyclic reference involving type InfiniteFunction
       type InfiniteFunction = () => InfiniteFunction
Run Code Online (Sandbox Code Playgroud)

问题

我想在此功能上进行cps转换:

def travel(tree: TreeNode): Unit = {
  if (tree != null) {
    travel(tree.left)
    println(tree.value)
    travel(tree.right)
  }
}
Run Code Online (Sandbox Code Playgroud)

在cps之后:

def r[T](f: => T): () => T = () => f
def travel(tree: TreeNode, cb: () => AnyRef): Unit = {
  if (tree != null) {
    travel(tree.left, r{
      println(tree.value)
      travel(tree.right, cb)
    })
  } else {
    cb()
  }
}
Run Code Online (Sandbox Code Playgroud)

然后,我想通过产生它们而不是调用它们来优化tail调用:

def r[T](f: => T): () => T = () => f
def travel(tree: TreeNode, cb: () => InfiniteFunction): InfiniteFunction = {
  if (tree != null) {
    r(travel(tree.left, r{
      println(tree.value)
      r(travel(tree.right, cb))
    }))
  } else {
    r(cb())
  }
}

// demonstration how to use travel
var f: InfiniteFunction = r(
  travel(tree, r(throw new RuntimeException("it is over")))
)
// this will end by the exception "it is over"
while (true) f = f()
Run Code Online (Sandbox Code Playgroud)

这里InfiniteFunction需要InfiniteFunction类型,没有类型,需要类型转换:

def r[T](f: => T): () => T = () => f
def travel(tree: TreeNode, cb: () => AnyRef): () => AnyRef = {
  if (tree != null) {
    r(travel(tree.left, r {
      println(tree.value)
      r(travel(tree.right, cb))
    }))
  } else {
    r(cb())
  }
}

var f: () => AnyRef = r(
  travel(tree, r(throw new RuntimeException("it is over")))
)
while (true) f = f().asInstanceOf[() => AnyRef]
Run Code Online (Sandbox Code Playgroud)

mic*_*hid 6

使用特征而不是类型别名来解决循环引用的问题:

trait Inf extends (Unit => Inf)
Run Code Online (Sandbox Code Playgroud)

也是Unit类型()