首页 > 代码库 > Cats(2)- Free语法组合,Coproduct-ADT composition

Cats(2)- Free语法组合,Coproduct-ADT composition

    上篇我们介绍了Free类型可以作为一种嵌入式编程语言DSL在函数式编程中对某种特定功能需求进行描述。一个完整的应用可能会涉及多样的关联功能,但如果我们为每个应用都设计一套DSL的话,那么在我们的函数式编程中将会不断重复的功能相似的DSL。我们应该秉承函数式编程的核心思想:函数组合(compositionality)来实现DSL的组合:把DSL拆解成最基础语句ADT,然后用这些ADT来组合成适合应用功能要求的完整DSL。我们还是使用上篇那个Interact DSL,这次再增加一个Login功能:

 1 package demo.app 2 import cats.free.{Free,Inject} 3 object FreeModules { 4   object ADTs { 5     sealed trait Interact[+A] 6     object Interact { 7       case class Ask(prompt: String) extends Interact[String] 8       case class Tell(msg: String) extends Interact[Unit] 9       type FreeInteract[A] = Free[Interact,A]10       def ask(prompt: String): FreeInteract[String] = Free.liftF(Ask(prompt))11       def tell(msg: String): FreeInteract[Unit] = Free.liftF(Tell(msg))12     }13 14     sealed trait Login[+A]15     object Login {16       type FreeLogin[A] = Free[Login,A]17       case class Authenticate(user: String, pswd: String) extends Login[Boolean]18       def authenticate(user: String, pswd: String): FreeLogin[Boolean] =19         Free.liftF(Authenticate(user,pswd))20     }21 22   }23 24 }

上面我们增加了个Login类。我们先来进行DSL编程:

 1   object DSLs { 2     import ADTs._ 3     import Interact._ 4     import Login._ 5     val interactDSL: FreeInteract[Unit]  = for { 6       first <- ask("What‘s your first name?") 7       last <- ask("What‘s your last name?") 8       _ <- tell(s"Hello, $first $last!") 9     } yield()10 11     val loginDSL: FreeLogin[Boolean] = for {12       login <- authenticate("Tiger","123")13     } yield login14   }

很明显,用一种DSL编程是无法满足Login功能需要的。我们需要像下面这样的DSL:

1  val interactLoginDSL: Free[???,Boolean] = for {2       uid <- ask("Enter your User ID:")3       psw <- ask("Enter your Password:")4       aut <- authenticate(uid,pwd)5     } yield aut

不过上面的???应该是什么呢?它应该是Interact和Login的集合。cats提供了Coproduct,它是一个树形数据结构:

/** `F` on the left and `G` on the right of [[scala.util.Either]]. * * @param run The underlying [[scala.util.Either]]. */final case class Coproduct[F[_], G[_], A](run: Either[F[A], G[A]]) {...}

Coproduct 的每一个节点(Either[F[A],G[A]])都是一个ADT,F[A]或者G[A]。我们可以用多层递归Coproduce结构来构建一个多语法的树形结构,如:

1 type H[A] = Coproduct[F,G,A]2 type I[A] = Coproduct[H,X,A]3 type J[A] = Coproduct[J,Y,A]  //ADT(F,G,X,Y)

用Coproduct的树形结构可以容纳多种DSL的ADT。在上面的例子里我们需要一个组合的语法InteractLogin:

1 type InteractLogin[A] = Coproduct[Interact,Login,A]

cats提供了Inject类来构建Coproduct:

sealed abstract class Inject[F[_], G[_]] {  def inj[A](fa: F[A]): G[A]  def prj[A](ga: G[A]): Option[F[A]]}private[free] sealed abstract class InjectInstances {  implicit def catsFreeReflexiveInjectInstance[F[_]]: Inject[F, F] =    new Inject[F, F] {      def inj[A](fa: F[A]): F[A] = fa      def prj[A](ga: F[A]): Option[F[A]] = Some(ga)    }  implicit def catsFreeLeftInjectInstance[F[_], G[_]]: Inject[F, Coproduct[F, G, ?]] =    new Inject[F, Coproduct[F, G, ?]] {      def inj[A](fa: F[A]): Coproduct[F, G, A] = Coproduct.leftc(fa)      def prj[A](ga: Coproduct[F, G, A]): Option[F[A]] = ga.run.fold(Some(_), _ => None)    }  implicit def catsFreeRightInjectInstance[F[_], G[_], H[_]](implicit I: Inject[F, G]): Inject[F, Coproduct[H, G, ?]] =    new Inject[F, Coproduct[H, G, ?]] {      def inj[A](fa: F[A]): Coproduct[H, G, A] = Coproduct.rightc(I.inj(fa))      def prj[A](ga: Coproduct[H, G, A]): Option[F[A]] = ga.run.fold(_ => None, I.prj)    }}

inj[A](fa: F[A]):G[A]代表将F[A]注入更大的语法集G[A]。cats提供了三种实现了ink函数的Inject隐式实例:

1、catsFreeReflexiveInjectInstance:Inject[F,F]:对单一语法,无须构建Coproduct

2、catsFreeLeftInjectInstance:Inject[F,Coproduct[F,G,?]]:构建Coproduct结构并将F放在左边

3、catsFreeRightInjectInstance:Inject[F,Coproduct[H,G,?]]:把F注入到已经包含H,G的Coproduct[H,G,?]

有了这三种实例后我们可以根据解析到的隐式实例类型使用inj函数通过Coproduct构建更大的语法集了。我们可以通过implicitly来验证一下Interact和Login语法的Inject隐式实例:

1     val selfInj = implicitly[Inject[Interact,Interact]]2     type LeftInterLogin[A] = Coproduct[Interact,Login,A]3     val leftInj = implicitly[Inject[Interact,LeftInterLogin]]4     type RightInterLogin[A] = Coproduct[Login,LeftInterLogin,A]5     val rightInj = implicitly[Inject[Interact,RightInterLogin]]

现在我们可以用Inject.inj和Free.liftF把Interact和Login升格成Free[G,A]。G是个类型变量,Interact和Login在Coproduct的最终左右位置由当前Inject隐式实例类型决定:

 1   object ADTs { 2     sealed trait Interact[+A] 3     object Interact { 4       case class Ask(prompt: String) extends Interact[String] 5       case class Tell(msg: String) extends Interact[Unit] 6       type FreeInteract[A] = Free[Interact,A] 7       //def ask(prompt: String): FreeInteract[String] = Free.liftF(Ask(prompt)) 8       //def tell(msg: String): FreeInteract[Unit] = Free.liftF(Tell(msg)) 9       def ask[G[_]](prompt: String)(implicit I: Inject[Interact,G]): Free[G,String] =10         Free.liftF(I.inj(Ask(prompt)))11       def tell[G[_]](msg: String)(implicit I: Inject[Interact,G]): Free[G,Unit] =12         Free.liftF(I.inj(Tell(msg)))13     }14 15     sealed trait Login[+A]16     object Login {17       type FreeLogin[A] = Free[Login,A]18       case class Authenticate(user: String, pswd: String) extends Login[Boolean]19       //def authenticate(user: String, pswd: String): FreeLogin[Boolean] =20       //  Free.liftF(Authenticate(user,pswd))21       def authenticate[G[_]](user: String, pswd: String)(implicit I: Inject[Login,G]): Free[G,Boolean] =22         Free.liftF(I.inj(Authenticate(user,pswd)))23     }

现在我们可以用混合语法的DSL来编程了:

 1   object DSLs { 2     import ADTs._ 3     import Interact._ 4     import Login._ 5     val interactDSL: FreeInteract[Unit] = for { 6       first <- ask("What‘s your first name?") 7       last <- ask("What‘s your last name?") 8       _ <- tell(s"Hello, $first $last!") 9     } yield()10 11     val loginDSL: FreeLogin[Boolean] = for {12       login <- authenticate("Tiger","123")13     } yield login14 15     type InteractLogin[A] = Coproduct[Interact,Login,A]16     val interactLoginDSL: Free[InteractLogin,Boolean] = for {17       uid <- ask[InteractLogin]("Enter your User ID:")18       pwd <- ask[InteractLogin]("Enter your Password:")19       aut <- authenticate[InteractLogin](uid,pwd)20     } yield aut21   }

在interactLoginDSL里所有ADT通过Inject隐式实例都被自动升格成统一的Free[Coproduct[Interact,Login,A]]。

interactLogin的功能实现方式之一示范如下:

 1   object IMPLs { 2     import cats.{Id,~>} 3     import ADTs._,Interact._,Login._ 4     import DSLs._ 5     object InteractConsole extends (Interact ~> Id) { 6       def apply[A](ia: Interact[A]): Id[A] = ia match { 7         case Ask(p) => {println(p); readLine} 8         case Tell(m) => println(m) 9       }10     }11     object LoginMock extends (Login ~> Id) {12       def apply[A](la: Login[A]): Id[A] = la match {13         case Authenticate(u,p) => if (u == "Tiger" && p == "123") true else false14       }15     }16     val interactLoginMock: (InteractLogin ~> Id) = InteractConsole.or(LoginMock)17   }

这个interactLoginMock就是一个Interact,Login混合语法程序的功能实现。不过我们还是应该赋予Login一个比较实在点的实现:我们可以用一种依赖注入方式通过Reader数据类型把外部系统的用户密码验证的方法传入:

 1   import Dependencies._ 2     import cats.data.Reader 3     type ReaderPass[A] = Reader[PasswordControl,A] 4     object LoginToReader extends (Login ~> ReaderPass) { 5       def apply[A](la: Login[A]): ReaderPass[A] = la match { 6         case Authenticate(u,p) => Reader{pc => pc.matchUserPassword(u,p)} 7       } 8     } 9     object InteractToReader extends (Interact ~> ReaderPass) {10       def apply[A](ia: Interact[A]): ReaderPass[A] = ia match {11         case Ask(p) => {println(p); Reader(pc => readLine)}12         case Tell(m) => {println(m); Reader(pc => ())}13       }14     }15     val userLogin: (InteractLogin ~> ReaderPass) = InteractToReader or LoginToReader

假设用户密码验证由外部另一个系统负责,PasswordControl是与这个外部系统的界面(interface):

1 object Dependencies {2   trait PasswordControl {3     val mapPasswords: Map[String,String]4     def matchUserPassword(uid: String, pwd: String): Boolean5   }6 }

我们用Reader来注入PasswordControl这个外部依赖(dependency injection IOC)。因为Interact和Login结合形成的是一个统一的语句集,所以我们必须进行Interact与?ReaderPass对应。下面我们先构建一个PasswordControl对象作为模拟数据,然后试运行:

 1 object catsComposeFree extends App { 2   import Dependencies._ 3   import FreeModules._ 4   import DSLs._ 5   import IMPLs._ 6   object UserPasswords extends PasswordControl { 7     override val mapPasswords: Map[String, String] = Map( 8       "Tiger" -> "123", 9       "John" -> "456"10     )11     override def matchUserPassword(uid: String, pwd: String): Boolean =12       mapPasswords.getOrElse(uid,pwd+"!") == pwd13   }14 15   val r = interactLoginDSL.foldMap(userLogin).run(UserPasswords)16   println(r)17 18 }

运算结果:

 1 Enter your User ID: 2 Tiger 3 Enter your Password: 4 123 5 true 6 ... 7 Enter your User ID: 8 Chan 9 Enter your Password:10 12311 false

我们再用这个混合的DSL编个稍微完整点的程序:

1     val userLoginDSL: Free[InteractLogin,Unit] = for {2       uid <- ask[InteractLogin]("Enter your User ID:")3       pwd <- ask[InteractLogin]("Enter your Password:")4       aut <- authenticate[InteractLogin](uid,pwd)5       _ <- if (aut) tell[InteractLogin](s"Hello $uid")6            else tell[InteractLogin]("Sorry, who are you?")7     } yield()

运算这个程序不需要任何修改:

1   //val r = interactLoginDSL.foldMap(userLogin).run(UserPasswords)2   //println(r)3   userLoginDSL.foldMap(userLogin).run(UserPasswords)

现在结果变成了:

 1 Enter your User ID: 2 Tiger 3 Enter your Password: 4 123 5 Hello Tiger 6 ... 7 Enter your User ID: 8 CHAN 9 Enter your Password:10 12311 Sorry, who are you?

如果我们在这两个语法的基础上再增加一个模拟权限管理的语法,ADT设计如下:

1     sealed trait Auth[+A]2     object Auth {3       case class Authorize(uid: String) extends Auth[Boolean]4       def authorize[G[_]](uid:String)(implicit I: Inject[Auth,G]): Free[G,Boolean] =5         Free.liftF(I.inj(Authorize(uid)))6     }

假设实际的权限管理依赖外部系统,我们先定义它的界面:

 1 object Dependencies { 2   trait PasswordControl { 3     val mapPasswords: Map[String,String] 4     def matchUserPassword(uid: String, pwd: String): Boolean 5   } 6   trait PermControl { 7     val mapAuthorized: Map[String,Boolean] 8     def authorized(uid: String): Boolean 9   }10 }

再用三种语法合成的DSL来编一段程序:

 1     import Auth._ 2     type Permit[A] = Coproduct[Auth,InteractLogin,A] 3     val userPermitDSL: Free[Permit,Unit] = for { 4       uid <- ask[Permit]("Enter your User ID:") 5       pwd <- ask[Permit]("Enter your Password:") 6       auth <- authenticate[Permit](uid,pwd) 7       perm <- if(auth) authorize[Permit](uid) 8               else Free.pure[Permit,Boolean](false) 9       _ <- if (perm) tell[Permit](s"Hello $uid, welcome to the program!")10            else tell[Permit]("Sorry, no no no!")11     } yield()

很遗憾,这段代码无法通过编译,cats还无法处理多层递归Coproduct。对Coproduct的处理scalaz还是比较成熟的,我在之前写过一篇scalaz Coproduct Free的博客,里面用的例子就是三种语法的DSL。实际上不单只是Coproduct的问题,现在看来cats.Free对即使很简单的应用功能也有着很复杂无聊的代码需求,这是我们无法接受的。由于Free编程在函数式编程里占据着如此重要的位置,我们暂时还没有其它选择,所以必须寻找一个更好的编程工具才行,freeK就是个这样的函数组件库。我们将在下篇讨论里用freeK来实现多种语法DSL编程。

无论如何,我还是把这篇讨论的示范代码附在下面:

 

  1 import cats.data.Coproduct  2 import cats.free.{Free, Inject}  3 object FreeModules {  4   object ADTs {  5     sealed trait Interact[+A]  6     object Interact {  7       case class Ask(prompt: String) extends Interact[String]  8       case class Tell(msg: String) extends Interact[Unit]  9       type FreeInteract[A] = Free[Interact,A] 10       //def ask(prompt: String): FreeInteract[String] = Free.liftF(Ask(prompt)) 11       //def tell(msg: String): FreeInteract[Unit] = Free.liftF(Tell(msg)) 12       def ask[G[_]](prompt: String)(implicit I: Inject[Interact,G]): Free[G,String] = 13         Free.liftF(I.inj(Ask(prompt))) 14       def tell[G[_]](msg: String)(implicit I: Inject[Interact,G]): Free[G,Unit] = 15         Free.liftF(I.inj(Tell(msg))) 16     } 17  18     sealed trait Login[+A] 19     object Login { 20       type FreeLogin[A] = Free[Login,A] 21       case class Authenticate(user: String, pswd: String) extends Login[Boolean] 22       //def authenticate(user: String, pswd: String): FreeLogin[Boolean] = 23       //  Free.liftF(Authenticate(user,pswd)) 24       def authenticate[G[_]](user: String, pswd: String)(implicit I: Inject[Login,G]): Free[G,Boolean] = 25         Free.liftF(I.inj(Authenticate(user,pswd))) 26     } 27  28     sealed trait Auth[+A] 29     object Auth { 30       case class Authorize(uid: String) extends Auth[Boolean] 31       def authorize[G[_]](uid:String)(implicit I: Inject[Auth,G]): Free[G,Boolean] = 32         Free.liftF(I.inj(Authorize(uid))) 33     } 34     val selfInj = implicitly[Inject[Interact,Interact]] 35     type LeftInterLogin[A] = Coproduct[Interact,Login,A] 36     val leftInj = implicitly[Inject[Interact,LeftInterLogin]] 37     type RightInterLogin[A] = Coproduct[Login,LeftInterLogin,A] 38     val rightInj = implicitly[Inject[Interact,RightInterLogin]] 39   } 40  41   object DSLs { 42     import ADTs._ 43     import Interact._ 44     import Login._ 45     val interactDSL: FreeInteract[Unit] = for { 46       first <- ask("What‘s your first name?") 47       last <- ask("What‘s your last name?") 48       _ <- tell(s"Hello, $first $last!") 49     } yield() 50  51     val loginDSL: FreeLogin[Boolean] = for { 52       login <- authenticate("Tiger","123") 53     } yield login 54  55     type InteractLogin[A] = Coproduct[Interact,Login,A] 56     val interactLoginDSL: Free[InteractLogin,Boolean] = for { 57       uid <- ask[InteractLogin]("Enter your User ID:") 58       pwd <- ask[InteractLogin]("Enter your Password:") 59       aut <- authenticate[InteractLogin](uid,pwd) 60     } yield aut 61     val userLoginDSL: Free[InteractLogin,Unit] = for { 62       uid <- ask[InteractLogin]("Enter your User ID:") 63       pwd <- ask[InteractLogin]("Enter your Password:") 64       aut <- authenticate[InteractLogin](uid,pwd) 65       _ <- if (aut) tell[InteractLogin](s"Hello $uid") 66            else tell[InteractLogin]("Sorry, who are you?") 67     } yield() 68   /*  import Auth._ 69     type Permit[A] = Coproduct[Auth,InteractLogin,A] 70     val userPermitDSL: Free[Permit,Unit] = for { 71       uid <- ask[Permit]("Enter your User ID:") 72       pwd <- ask[Permit]("Enter your Password:") 73       auth <- authenticate[Permit](uid,pwd) 74       perm <- if(auth) authorize[Permit](uid) 75               else Free.pure[Permit,Boolean](false) 76       _ <- if (perm) tell[Permit](s"Hello $uid, welcome to the program!") 77            else tell[Permit]("Sorry, no no no!") 78     } yield() */ 79   } 80   object IMPLs { 81     import cats.{Id,~>} 82     import ADTs._,Interact._,Login._ 83     import DSLs._ 84     object InteractConsole extends (Interact ~> Id) { 85       def apply[A](ia: Interact[A]): Id[A] = ia match { 86         case Ask(p) => {println(p); readLine} 87         case Tell(m) => println(m) 88       } 89     } 90     object LoginMock extends (Login ~> Id) { 91       def apply[A](la: Login[A]): Id[A] = la match { 92         case Authenticate(u,p) => if (u == "Tiger" && p == "123") true else false 93       } 94     } 95     val interactLoginMock: (InteractLogin ~> Id) = InteractConsole.or(LoginMock) 96     import Dependencies._ 97     import cats.data.Reader 98     type ReaderPass[A] = Reader[PasswordControl,A] 99     object LoginToReader extends (Login ~> ReaderPass) {100       def apply[A](la: Login[A]): ReaderPass[A] = la match {101         case Authenticate(u,p) => Reader{pc => pc.matchUserPassword(u,p)}102       }103     }104     object InteractToReader extends (Interact ~> ReaderPass) {105       def apply[A](ia: Interact[A]): ReaderPass[A] = ia match {106         case Ask(p) => {println(p); Reader(pc => readLine)}107         case Tell(m) => {println(m); Reader(pc => ())}108       }109     }110     val userLogin: (InteractLogin ~> ReaderPass) = InteractToReader or LoginToReader111 112   }113 }114 object Dependencies {115   trait PasswordControl {116     val mapPasswords: Map[String,String]117     def matchUserPassword(uid: String, pwd: String): Boolean118   }119   trait PermControl {120     val mapAuthorized: Map[String,Boolean]121     def authorized(uid: String): Boolean122   }123 }124 125 object catsComposeFree extends App {126   import Dependencies._127   import FreeModules._128   import DSLs._129   import IMPLs._130   object UserPasswords extends PasswordControl {131     override val mapPasswords: Map[String, String] = Map(132       "Tiger" -> "123",133       "John" -> "456"134     )135     override def matchUserPassword(uid: String, pwd: String): Boolean =136       mapPasswords.getOrElse(uid,pwd+"!") == pwd137   }138 139   //val r = interactLoginDSL.foldMap(userLogin).run(UserPasswords)140   //println(r)141   userLoginDSL.foldMap(userLogin).run(UserPasswords)142 143 }

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

Cats(2)- Free语法组合,Coproduct-ADT composition