aboutsummaryrefslogtreecommitdiff
path: root/src/main/scala/xyz/driver/core/rest/HttpRestServiceTransport.scala
diff options
context:
space:
mode:
Diffstat (limited to 'src/main/scala/xyz/driver/core/rest/HttpRestServiceTransport.scala')
-rw-r--r--src/main/scala/xyz/driver/core/rest/HttpRestServiceTransport.scala103
1 files changed, 0 insertions, 103 deletions
diff --git a/src/main/scala/xyz/driver/core/rest/HttpRestServiceTransport.scala b/src/main/scala/xyz/driver/core/rest/HttpRestServiceTransport.scala
deleted file mode 100644
index e31635b..0000000
--- a/src/main/scala/xyz/driver/core/rest/HttpRestServiceTransport.scala
+++ /dev/null
@@ -1,103 +0,0 @@
-package xyz.driver.core.rest
-
-import akka.actor.ActorSystem
-import akka.http.scaladsl.model._
-import akka.http.scaladsl.model.headers.RawHeader
-import akka.http.scaladsl.unmarshalling.Unmarshal
-import akka.stream.Materializer
-import akka.stream.scaladsl.TcpIdleTimeoutException
-import org.slf4j.MDC
-import xyz.driver.core.Name
-import xyz.driver.core.reporting.Reporter
-import xyz.driver.core.rest.errors.{ExternalServiceException, ExternalServiceTimeoutException}
-
-import scala.concurrent.{ExecutionContext, Future}
-import scala.util.{Failure, Success}
-
-class HttpRestServiceTransport(
- applicationName: Name[App],
- applicationVersion: String,
- val actorSystem: ActorSystem,
- val executionContext: ExecutionContext,
- reporter: Reporter)
- extends ServiceTransport {
-
- protected implicit val execution: ExecutionContext = executionContext
-
- protected val httpClient: HttpClient = new SingleRequestHttpClient(applicationName, applicationVersion, actorSystem)
-
- def sendRequestGetResponse(context: ServiceRequestContext)(requestStub: HttpRequest): Future[HttpResponse] = {
- val tags = Map(
- // open tracing semantic tags
- "span.kind" -> "client",
- "service" -> applicationName.value,
- "http.url" -> requestStub.uri.toString,
- "http.method" -> requestStub.method.value,
- "peer.hostname" -> requestStub.uri.authority.host.toString,
- // google's tracing console provides extra search features if we define these tags
- "/http/path" -> requestStub.uri.path.toString,
- "/http/method" -> requestStub.method.value.toString,
- "/http/url" -> requestStub.uri.toString
- )
- reporter.traceAsync(s"http_call_rpc", tags) { implicit span =>
- val requestTime = System.currentTimeMillis()
-
- val request = requestStub
- .withHeaders(context.contextHeaders.toSeq.map {
- case (ContextHeaders.TrackingIdHeader, _) =>
- RawHeader(ContextHeaders.TrackingIdHeader, context.trackingId)
- case (ContextHeaders.StacktraceHeader, _) =>
- RawHeader(
- ContextHeaders.StacktraceHeader,
- Option(MDC.get("stack"))
- .orElse(context.contextHeaders.get(ContextHeaders.StacktraceHeader))
- .getOrElse(""))
- case (header, headerValue) => RawHeader(header, headerValue)
- }: _*)
-
- reporter.debug(s"Sending request to ${request.method} ${request.uri}")
-
- val response = httpClient.makeRequest(request)
-
- response.onComplete {
- case Success(r) =>
- val responseLatency = System.currentTimeMillis() - requestTime
- reporter.debug(
- s"Response from ${request.uri} to request $requestStub is successful in $responseLatency ms: $r")
-
- case Failure(t: Throwable) =>
- val responseLatency = System.currentTimeMillis() - requestTime
- reporter.warn(
- s"Failed to receive response from ${request.method.value} ${request.uri} in $responseLatency ms",
- t)
- }(executionContext)
-
- response.recoverWith {
- case _: TcpIdleTimeoutException =>
- val serviceCalled = s"${requestStub.method.value} ${requestStub.uri}"
- Future.failed(ExternalServiceTimeoutException(serviceCalled))
- case t: Throwable => Future.failed(t)
- }
- }(context.spanContext)
- }
-
- def sendRequest(context: ServiceRequestContext)(requestStub: HttpRequest)(
- implicit mat: Materializer): Future[Unmarshal[ResponseEntity]] = {
-
- sendRequestGetResponse(context)(requestStub) flatMap { response =>
- if (response.status == StatusCodes.NotFound) {
- Future.successful(Unmarshal(HttpEntity.Empty: ResponseEntity))
- } else if (response.status.isFailure()) {
- val serviceCalled = s"${requestStub.method} ${requestStub.uri}"
- Unmarshal(response.entity).to[String] flatMap { errorString =>
- import spray.json._
- import xyz.driver.core.json._
- val serviceException = util.Try(serviceExceptionFormat.read(errorString.parseJson)).toOption
- Future.failed(ExternalServiceException(serviceCalled, errorString, serviceException))
- }
- } else {
- Future.successful(Unmarshal(response.entity))
- }
- }
- }
-}