How to return a function in scala

前端 未结 5 1138
忘了有多久
忘了有多久 2021-02-02 14:31

How can I return a function side-effecting lexical closure1 in Scala?

For instance, I was looking at this code sample in Go:

5条回答
  •  半阙折子戏
    2021-02-02 14:49

    Gah! Mutable variables?!

    val fib: Stream[Int] =
      1 #:: 1 #:: (fib zip fib.tail map Function.tupled(_+_))
    

    You can return a literal function that gets the nth fib, for example:

    val fibAt: Int => Int = fib drop _ head
    

    EDIT: Since you asked for the functional way of "getting a different value each time you call f", here's how you would do that. This uses Scalaz's State monad:

    import scalaz._
    import Scalaz._
    
    def uncons[A](s: Stream[A]) = (s.tail, s.head)
    val f = state(uncons[Int])
    

    The value f is a state transition function. Given a stream, it will return its head, and "mutate" the stream on the side by taking its tail. Note that f is totally oblivious to fib. Here's a REPL session illustrating how this works:

    scala> (for { _ <- f; _ <- f; _ <- f; _ <- f; x <- f } yield x)
    res29: scalaz.State[scala.collection.immutable.Stream[Int],Int] = scalaz.States$$anon$1@d53513
    
    scala> (for { _ <- f; _ <- f; _ <- f; x <- f } yield x)
    res30: scalaz.State[scala.collection.immutable.Stream[Int],Int]  = scalaz.States$$anon$1@1ad0ff8
    
    scala> res29 ! fib
    res31: Int = 5
    
    scala> res30 ! fib
    res32: Int = 3
    

    Clearly, the value you get out depends on the number of times you call f. But this is all purely functional and therefore modular and composable. For example, we can pass any nonempty Stream, not just fib.

    So you see, you can have effects without side-effects.

提交回复
热议问题