Question

在Haskell中，您可以使用绑定运算符（>>=），如下所示：

repli :: [a] -> [a]
repli xs = xs >>= \x -> [x,x]

*Main> repli [1,2,3]
[1,1,2,2,3,3]

我已经读过flatMap是Scala的绑定运算符：

def repli [A](xs: List[A]): List[A] =
  xs.flatMap { x => List(x,x) }

scala> repli (List(1,2,3))
res0: List[Int] = List(1, 1, 2, 2, 3, 3)

作为一项教学练习，我尝试向Scala添加对>>=的支持：

class MyList[T](list: List[T]) {
  def >>= [U](f: T => List[U]): List[U] = list.flatMap(f)
}
implicit def list2mylist[T](list: List[T]) = new MyList(list)

def repliNew [A](xs: List[A]): List[A] =
  xs >>= { x: A => List(x,x) }

scala> repliNew (List(1,2,3))
res1: List[Int] = List(1, 1, 2, 2, 3, 3)

这非常有效，但仅适用于列表。我真的想用flatMap方法支持任何类。什么是最好的解决方法？

Answer 1

Scalaz执行如下操作：

trait MA[M[_], A] {
  def value: M[A]
  def >>=(f: A => M[B])(implicit m: Monad[M]): M[B] =
    m.bind(value, f)
}

对所有M[A]和MA[M, A]隐式转换M[_]到A：

implicit def ma[M[_], A](m: => M[A]): MA[M, A] = new MA[M, A] {
  lazy val value = m
}

你只需要一个特质Monad，并为你关心的每个monad提供一个实例：

trait Monad[M[_]] {
  def pure[A](a: => A): M[A]
  def bind[A, B](m: M[A], f: A => M[B]): M[B]
}

Answer 2

如何为隐含类添加flatMap的同义词？

implicit class BindRich[A,B](fm:{def flatMap(a:A):B}) {
  def >>=(a:A):B = fm.flatMap(a)
}

在范围内，具有方法flatMap的任何对象也将具有>>=。

将Haskell的Monadic Bind运算符添加到Scala

2 个答案: