STTP的基本使用(3):Response 转为 Future[( List[E], List[T] )]

2022-05-30 14:05:57 浏览数 (1)

如果使用 AkkaHttp 作为 STTP 的 backend 来并发地处理 list of url,就会得到类似 List[Future[Response[Either[ResponseError[io.circe.Error], T]]]],这样的结果。

一般地,我们更期望给上层调用者返回 Future[(List[Exception], List[T])]这样的类型。

下面代码演示了如何把:

List[Future[Response[Either[ResponseError[io.circe.Error], NasaData]]]]

准换为

Future[(List[Throwable], List[NasaData])]。

其中 tuple 的第一部分表示 response 中所有的 exception,第二部分表示所有正常数据。

几个技术关键点是:

1)Future.sequence 用于把 List[Future] 转为 Future[List]。

2)忽略 STTP Response 里除去 body 数据的其它部分,并把 response 转为 Either[Throwable, NasaData]。

3)经过步骤 1,2,数据类型已经是 Future[List[Either[Throwable, NasaData]]]。

4)List[Either[Throwable, NasaData]] 转为 (List[Throwable], List[NasaData])的思路是:

a) 构造一个空的 Tuple (List[Throwable(), List[NasaData]())作为一个累计器。

b) 使用 List.folderLeft()遍历元素,根据是 left 还是 right 累加到累计器的_1 或_2。

下面链接的文章演示了两种转换方式,一种是 Scala 原生手写,一种是使用 CAT。https://pbassiner.github.io/blog/composing_future,_list_and_either.html。

但是,其中语句 case Left(err) => acc.left.map(_ => err) 的逻辑似乎与期望不符。如果 list of either 中有 Left,并不能工作。

代码语言:javascript复制
  val erA: Either[String, Int] = Right(1)
  val erB: Either[String, Int] = Right(2)
  val erC: Either[String, Int] = Right(3)
  val elA: Either[String, Int] = Left("error1")
  val elB: Either[String, Int] = Left("error2")
  val listOfEither = List(erA, elA, erB, erC, elB)
  private val eitherOfList: Either[String, List[Int]] = listOfEither.foldLeft(Right[String, List[Int]](List()).asInstanceOf[Either[String, List[Int]]])((acc, elem) => {
    elem match {
      case Right(v) => acc.map(l => l :  v)
      case Left(e) => acc.left.map(_   e)
    }
  })
  println(s"eitherOfList: $eitherOfList")

输出如下,并没有error部分的信息。

代码语言:javascript复制
eitherOfList: Right(List(1, 2, 3))

下面的代码演示了并发发出100个RestAPi请求获取数据的处理代码,结果保存在Tuple里,分别是所有失败和成功的数据。

代码语言:javascript复制
  import io.circe.generic.auto._
  import sttp.client._
  import sttp.client.akkahttp.AkkaHttpBackend
  import sttp.client.circe._

  import scala.concurrent.Future
  import scala.concurrent.duration._
  import scala.concurrent.Await
  import scala.concurrent.ExecutionContext.Implicits.global

  val NASA_API_KEY = "XXXXXXXXXXXXXXX"
  val baseUrl = s"https://api.nasa.gov/neo/rest/v1/neo/browse?api_key=${NASA_API_KEY}"

  var pageSize = 10
  var pageNumb = 100

  case class Links(self: String, next: String)
  case class Page(number: Int, size: Int, total_elements: Int, total_pages: Int)
  case class NearEarthObject(absolute_magnitude_h: Double, designation: String)
  case class NasaData(links: Links, page: Page, near_earth_objects: List[NearEarthObject])

  implicit val sttpBackend = AkkaHttpBackend()
  
  val listOfFutureResult: List[Future[Response[Either[ResponseError[io.circe.Error], NasaData]]]] =
    (0 until pageNumb).toList.map(pageIndex => {
      val pageUrl = s"http://www.neowsapp.com/rest/v1/neo/browse?page=$pageIndex&size=$pageSize&api_key=${NASA_API_KEY}"
      basicRequest.get(uri"$pageUrl")
        .response(asJson[NasaData])
        .send()
    })

  // Note: Future[List] --> List[Future]
  val futureOfList: Future[List[Response[Either[ResponseError[io.circe.Error], NasaData]]]] = Future.sequence(listOfFutureResult)

  // Note: 把Response[Either[ResponseError[io.circe.Error], NasaData]] 转为了 Either[Throwable, NasaData]
  val respOfEither2Either: Response[Either[ResponseError[io.circe.Error], NasaData]] => Either[Throwable, NasaData] = resp => {
    println(s"resp==>  code:${resp.code.code}  ${resp.statusText}")
    val newEither: Either[Throwable, NasaData] = resp.body match {
      case Left(respErr) => Left(respErr.getCause)
      case Right(nasaData) => Right(nasaData)
    }
    newEither
  }

  val futureOfListOfEither: Future[List[Either[Throwable, NasaData]]] = futureOfList.map(listOfResp => {
    listOfResp.map(respOfEither2Either)
  })

  val listOfEither2TupleOfList: List[Either[Throwable, NasaData]] => (List[Throwable], List[NasaData]) = listE => {
    listE.foldLeft(List[Throwable](), List[NasaData]())((acc, eData) => {
      eData match {
        case Right(nasaData) => (acc._1, acc._2 :  nasaData)
        case Left(t) => (acc._1 :  t, acc._2)
      }
    })
  }

  val futureOfTupleOfList = futureOfListOfEither map listOfEither2TupleOfList

  val result = Await.result(futureOfTupleOfList, 1.minute)
  println(s"Exceptions in Response: $result._1")
  println(s"Data in Response: $result._2")
  

另,Akka HTTP默认主机连接数是32个,所以需要修改applicaiton.confg配置以支持更多连接数。

代码语言:javascript复制
akka.http.host-connection-pool.max-open-requests = 128

0 人点赞