notion.api.v1.http.JavaNetHttpClient.kt Maven / Gradle / Ivy

Go to download
package notion.api.v1.http

import java.net.URI
import java.net.http.HttpClient
import java.net.http.HttpRequest
import java.net.http.HttpResponse
import java.time.Duration
import notion.api.v1.logging.NotionLogger

// TODO: proxy support
class JavaNetHttpClient(
    connectTimeoutMillis: Int = 3_000,
    private val readTimeoutMillis: Int = 30_000,
) : NotionHttpClient {
  private val client: HttpClient =
      HttpClient.newBuilder()
          .connectTimeout(Duration.ofMillis(connectTimeoutMillis.toLong()))
          .build()

  override fun get(
      logger: NotionLogger,
      url: String,
      query: Map,
      headers: Map
  ): NotionHttpResponse {
    val startTimeMillis = System.currentTimeMillis()
    val fullUrl = buildFullUrl(url, buildQueryString(query))
    val req =
        HttpRequest.newBuilder()
            .GET()
            .uri(URI(fullUrl))
            .timeout(Duration.ofMillis(readTimeoutMillis.toLong()))
    headers.forEach { (name, value) -> req.header(name, value) }
    val request = req.build()
    debugLogStart(logger, request.method(), fullUrl, "")
    try {
      val resp = client.send(request, HttpResponse.BodyHandlers.ofString())
      val response =
          NotionHttpResponse(
              status = resp.statusCode(), headers = resp.headers().map(), body = resp.body())
      debugLogSuccess(logger, startTimeMillis, response)
      return response
    } catch (e: Exception) {
      warnLogFailure(logger, e)
      throw e
    }
  }

  override fun postTextBody(
      logger: NotionLogger,
      url: String,
      query: Map,
      body: String,
      headers: Map
  ): NotionHttpResponse {
    val startTimeMillis = System.currentTimeMillis()
    val fullUrl = buildFullUrl(url, buildQueryString(query))
    val req =
        HttpRequest.newBuilder()
            .POST(HttpRequest.BodyPublishers.ofString(body, Charsets.UTF_8))
            .uri(URI(fullUrl))
            .timeout(Duration.ofMillis(readTimeoutMillis.toLong()))
    headers.forEach { (name, value) -> req.header(name, value) }
    val request = req.build()
    debugLogStart(logger, request.method(), fullUrl, body)
    try {
      val resp = client.send(request, HttpResponse.BodyHandlers.ofString())
      val response =
          NotionHttpResponse(
              status = resp.statusCode(), headers = resp.headers().map(), body = resp.body())
      debugLogSuccess(logger, startTimeMillis, response)
      return response
    } catch (e: Exception) {
      warnLogFailure(logger, e)
      throw e
    }
  }

  override fun patchTextBody(
      logger: NotionLogger,
      url: String,
      query: Map,
      body: String,
      headers: Map
  ): NotionHttpResponse {
    val startTimeMillis = System.currentTimeMillis()
    val fullUrl = buildFullUrl(url, buildQueryString(query))
    val req =
        HttpRequest.newBuilder()
            .method("PATCH", HttpRequest.BodyPublishers.ofString(body, Charsets.UTF_8))
            .uri(URI(buildFullUrl(url, buildQueryString(query))))
            .timeout(Duration.ofMillis(readTimeoutMillis.toLong()))
    headers.forEach { (name, value) -> req.header(name, value) }
    val request = req.build()
    debugLogStart(logger, request.method(), fullUrl, body)
    try {
      val resp = client.send(request, HttpResponse.BodyHandlers.ofString())
      val response =
          NotionHttpResponse(
              status = resp.statusCode(), headers = resp.headers().map(), body = resp.body())
      debugLogSuccess(logger, startTimeMillis, response)
      return response
    } catch (e: Exception) {
      warnLogFailure(logger, e)
      throw e
    }
  }

  override fun delete(
      logger: NotionLogger,
      url: String,
      query: Map,
      headers: Map
  ): NotionHttpResponse {
    val startTimeMillis = System.currentTimeMillis()
    val fullUrl = buildFullUrl(url, buildQueryString(query))
    val req =
        HttpRequest.newBuilder()
            .DELETE()
            .uri(URI(fullUrl))
            .timeout(Duration.ofMillis(readTimeoutMillis.toLong()))
    headers.forEach { (name, value) -> req.header(name, value) }
    val request = req.build()
    debugLogStart(logger, request.method(), fullUrl, "")
    try {
      val resp = client.send(request, HttpResponse.BodyHandlers.ofString())
      val response =
          NotionHttpResponse(
              status = resp.statusCode(), headers = resp.headers().map(), body = resp.body())
      debugLogSuccess(logger, startTimeMillis, response)
      return response
    } catch (e: Exception) {
      warnLogFailure(logger, e)
      throw e
    }
  }
}