diff options
author | vlad <vlad@driver.xyz> | 2017-06-13 10:25:55 -0700 |
---|---|---|
committer | vlad <vlad@driver.xyz> | 2017-06-13 10:25:55 -0700 |
commit | 0000a65ab4479a2a40e2d6468036438e9705b4aa (patch) | |
tree | 60c868828741e7e5367aa7b6d167abbdaf91d5b8 /src/main/scala/xyz/driver/common/http/AsyncHttpClientUploader.scala | |
download | rest-query-0000a65ab4479a2a40e2d6468036438e9705b4aa.tar.gz rest-query-0000a65ab4479a2a40e2d6468036438e9705b4aa.tar.bz2 rest-query-0000a65ab4479a2a40e2d6468036438e9705b4aa.zip |
Initial extraction of Driver non-specific utilities
Diffstat (limited to 'src/main/scala/xyz/driver/common/http/AsyncHttpClientUploader.scala')
-rw-r--r-- | src/main/scala/xyz/driver/common/http/AsyncHttpClientUploader.scala | 116 |
1 files changed, 116 insertions, 0 deletions
diff --git a/src/main/scala/xyz/driver/common/http/AsyncHttpClientUploader.scala b/src/main/scala/xyz/driver/common/http/AsyncHttpClientUploader.scala new file mode 100644 index 0000000..97c96cd --- /dev/null +++ b/src/main/scala/xyz/driver/common/http/AsyncHttpClientUploader.scala @@ -0,0 +1,116 @@ +package xyz.driver.common.http + +import java.io.Closeable +import java.net.URI +import java.util.concurrent.{ExecutorService, Executors} + +import xyz.driver.common.concurrent.MdcThreadFactory +import xyz.driver.common.http.AsyncHttpClientUploader._ +import xyz.driver.common.utils.RandomUtils +import com.typesafe.scalalogging.StrictLogging +import org.asynchttpclient._ +import org.slf4j.MDC + +import scala.concurrent.duration.FiniteDuration +import scala.concurrent.{ExecutionContext, Future, Promise} + +class AsyncHttpClientUploader(settings: Settings) extends Closeable with StrictLogging { + + private val es: ExecutorService = { + val threadFactory = MdcThreadFactory.from(Executors.defaultThreadFactory()) + Executors.newSingleThreadExecutor(threadFactory) + } + + private implicit val executionContext = ExecutionContext.fromExecutor(es) + + private def httpClientConfig: DefaultAsyncHttpClientConfig = { + val builder = new DefaultAsyncHttpClientConfig.Builder() + builder.setConnectTimeout(settings.connectTimeout.toMillis.toInt) + builder.setRequestTimeout(settings.requestTimeout.toMillis.toInt) + // builder.setThreadFactory(threadFactory) // Doesn't help to push MDC context into AsyncCompletionHandler + builder.build() + } + + private val httpClient = new DefaultAsyncHttpClient(httpClientConfig) + + def run(method: Method, uri: URI, contentType: String, data: String): Future[Unit] = { + // log all outcome connections + val fingerPrint = RandomUtils.randomString(10) + logger.info("{}, apply(method={}, uri={}, contentType={})", fingerPrint, method, uri, contentType) + val promise = Promise[Response]() + + val q = new RequestBuilder(method.toString) + .setUrl(uri.toString) + .setBody(data) + + settings.defaultHeaders.foreach { + case (k, v) => + q.setHeader(k, v) + } + + q.addHeader("Content-Type", contentType) + + httpClient.prepareRequest(q).execute(new AsyncCompletionHandler[Unit] { + override def onCompleted(response: Response): Unit = { + promise.success(response) + } + + override def onThrowable(t: Throwable): Unit = { + promise.failure(t) + super.onThrowable(t) + } + }) + + // see AsyncHttpClientFetcher + val parentMdcContext = MDC.getCopyOfContextMap + promise.future.flatMap { response => + setContextMap(parentMdcContext) + + val statusCode = response.getStatusCode + // https://en.wikipedia.org/wiki/List_of_HTTP_status_codes#2xx_Success + if (statusCode >= 200 && statusCode < 300) { + logger.debug("{}, success", fingerPrint) + Future.successful(()) + } else { + logger.error( + "{}, HTTP {}, BODY:\n{}", + fingerPrint, + response.getStatusCode.asInstanceOf[AnyRef], + response.getResponseBody.take(100) + ) + Future.failed(new IllegalStateException("An unexpected response from the server")) + } + } + } + + private[this] def setContextMap(context: java.util.Map[String, String]): Unit = + Option(context).fold(MDC.clear())(MDC.setContextMap) + + override def close(): Unit = { + httpClient.close() + es.shutdown() + } + +} + +object AsyncHttpClientUploader { + + case class Settings(connectTimeout: FiniteDuration, + requestTimeout: FiniteDuration, + defaultHeaders: Map[String, String] = Map.empty) + + sealed trait Method + + object Method { + + case object Put extends Method { + override val toString = "PUT" + } + + case object Post extends Method { + override val toString = "POST" + } + + } + +} |