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

Question

在Haskell中,您可以>>=像这样使用bind运算符():

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

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

我读过flatMapScala的绑定运算符:

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也将具有>>=.