Scala Cats EitherT 和 foreach

Question

我正在尝试实现类似以下代码的功能

def getUsers: EitherT[Future, String, Response] = {
  for {
    users <- EitherT(userRepository.findAll) // this method would retrieve a Future[Either[String, Seq[User]]]
    user <- users
    friends <- EitherT(userRepository.findFriends(user.id))
  } yield new Response(user, friends)
}

我知道这是不可能的，因为EitherT 不管理foreach。

case class User(id: Long, name: String, email: String)

case class Response(user: User, friends: Seq[User])

解决这个问题的好方法是什么？

Answer 1

由于您要查找所有用户，因此您将有多个用户及其各自的朋友，因此我修改了您的返回值以返回 Responses 中的 List。那么也许你想做这样的事情：

import cats.data.EitherT
import cats.syntax.traverse._
import cats.instances.list._
import cats.instances.future._
import scala.concurrent.Future

def getUsers: EitherT[Future, String, List[Response]] = {
  EitherT(userRepository.findAll).flatMap { users =>
    users.toList.traverse(user => EitherT(userRepository.findFriends(user.id)).map(Response(user, _)))
  }
}

确保您在范围内具有隐式 ExecutionContext 或导入 scala.concurrent.ExecutionContext.Implicits.global。