Как обращаться с монашкой "Читатель" и "Попробовать"?

Я читаю эту замечательную статью о инъекции зависимостей в scala с монадой Reader.

Исходный пример работает хорошо, но я немного изменил возвращаемые типы UserRepository.get/find. Это было User, но я изменил его на Try[User].

Тогда код не будет скомпилирован, я много раз пытался, но все равно не повезло.

import scala.util.Try
import scalaz.Reader

case class User(email: String, supervisorId: Int, firstName: String, lastName: String)

trait UserRepository {
  def get(id: Int): Try[User]

  def find(username: String): Try[User]
}

trait Users {

  def getUser(id: Int) = Reader((userRepository: UserRepository) =>
    userRepository.get(id)
  )

  def findUser(username: String) = Reader((userRepository: UserRepository) =>
    userRepository.find(username)
  )
}

object UserInfo extends Users {

  def userEmail(id: Int) = {
    getUser(id) map (ut => ut.map(_.email))
  }

  def userInfo(username: String) =
    for {
      userTry <- findUser(username)
      user <- userTry     // !!!!!!!! compilation error
      bossTry <- getUser(user.supervisorId)
      boss <- bossTry     // !!!!!!!! compilation error
    } yield Map(
      "fullName" -> s"${user.firstName} ${user.lastName}",
      "email" -> s"${user.email}",
      "boss" -> s"${boss.firstName} ${boss.lastName}"
    )
}

Ошибка компиляции:

Error:(34, 12) type mismatch;
 found   : scala.util.Try[Nothing]
 required: scalaz.Kleisli[scalaz.Id.Id,?,?]
      user <- userTry
           ^

и

Error:(36, 12) type mismatch;
 found   : scala.util.Try[scala.collection.immutable.Map[String,String]]
 required: scalaz.Kleisli[scalaz.Id.Id,?,?]
      boss <- bossTry
           ^

Я прочитал документ Kleisli.flatMap (Тип возврата findUser и getUser - Kleisli), для этого требуется тип параметра:

B => Kleisli[M, A, C]

Так как a Try не будет Kleisli, такие ошибки есть.

Я не уверен, как с этим справиться. Могу ли я использовать scala.util.Try здесь? Как я могу превратить его в тип Kleisli? Как я могу заставить этот пример работать?

Ответ 1

Вы можете использовать монадный трансформатор ReaderT для создания монады Reader и монады Try в одну монаду, в которой вы можете использовать for -понимание и т.д.

ReaderT - это просто псевдоним типа Kleisli, и вы можете использовать Kleisli.kleisli вместо Reader.apply для создания ваших вычислений Reader -y. Обратите внимание, что вам нужен scalaz-contrib для экземпляра monad для Try (или вы можете написать свой собственный - это довольно просто).

import scala.util.Try
import scalaz._, Scalaz._
import scalaz.contrib.std.utilTry._

case class User(
  email: String,
  supervisorId: Int,
  firstName: String,
  lastName: String
)

trait UserRepository {
  def get(id: Int): Try[User]

  def find(username: String): Try[User]
}

trait Users {
  def getUser(id: Int): ReaderT[Try, UserRepository, User] =
    Kleisli.kleisli(_.get(id))

  def findUser(username: String): ReaderT[Try, UserRepository, User] =
    Kleisli.kleisli(_.find(username))
}

Теперь, когда это сделано, UserInfo намного проще (и теперь он компилируется!):

object UserInfo extends Users {
  def userEmail(id: Int) = getUser(id).map(_.email)

  def userInfo(
    username: String
  ): ReaderT[Try, UserRepository, Map[String, String]] =
    for {
      user <- findUser(username)
      boss <- getUser(user.supervisorId)
    } yield Map(
      "fullName" -> s"${user.firstName} ${user.lastName}",
      "email" -> s"${user.email}",
      "boss" -> s"${boss.firstName} ${boss.lastName}"
    )
}

Мы можем показать, что он работает:

import scala.util.{ Failure, Success }

val repo = new UserRepository {
  val bar = User("[email protected]", 0, "Bar", "McFoo")
  val foo = User("[email protected]", 0, "Foo", "McBar")

  def get(id: Int) = id match {
    case 0 => Success(bar)
    case 1 => Success(foo)
    case i => Failure(new Exception(s"No user with id $i"))
  }

  def find(username: String) = username match {
    case "bar" => Success(bar)
    case "foo" => Success(foo)
    case other => Failure(new Exception(s"No user with name $other"))
  }
}

И затем:

UserInfo.userInfo("foo").run(repo).foreach(println)
Map(fullName -> Foo McBar, email -> [email protected], boss -> Bar McFoo)

Точно так же вы запускаете Reader, но в конце вы получите Try.