override fun close() {
          sourceClosed = true
          super.close()
        }
      }.buffer()

    override val sink: BufferedSink =
      object : ForwardingSink(delegate.sink) {
        override fun close() {
          sinkClosed = true
          super.close()
        }
      }.buffer()

    fun initWebSocket(
      random: Random?,

      object : RequestBody() {
        override fun contentType(): MediaType? = null

        override fun writeTo(sink: BufferedSink) {
          sink.writeUtf8("abc")
          sink.flush()
          makeFailingCall()
          sink.writeUtf8("def")
          sink.flush()
        }
      }
    val call =
      client.newCall(
        Request(
          url = server.url("/"),

This directory contains the microbenchmark suite of Elasticsearch. It relies on [JMH](http://openjdk.java.net/projects/code-tools/jmh/).

## Purpose

We do not want to microbenchmark everything but the kitchen sink and should typically rely on our
[macrobenchmarks](https://elasticsearch-benchmarks.elastic.co/app/kibana#/dashboard/Nightly-Benchmark-Overview) with

          url = server.url("/"),
          body =
            object : ForwardingRequestBody(transferKind.newRequestBody("def")) {
              override fun writeTo(sink: BufferedSink) {
                sinkReference.set(sink)
                super.writeTo(sink)
              }
            },
        ),
      )
    assertThat(readAscii(response.body.byteStream(), Int.MAX_VALUE))
      .isEqualTo("abc")

              override fun contentLength(): Long = postBytes.size.toLong()

              override fun writeTo(sink: BufferedSink) {
                sink.write(postBytes) // push bytes into the stream's buffer
                sink.flush() // Http2Connection.writeData subject to write window
                sink.close() // Http2Connection.writeData empty frame
              }
            },
        ),
      )

    override fun read(
      sink: Buffer,
      byteCount: Long,
    ): Long {
      while (left == 0) {
        source.skip(padding.toLong())
        padding = 0
        if (flags and FLAG_END_HEADERS != 0) return -1L
        readContinuationHeader()
        // TODO: test case for empty continuation header?
      }

      val read = source.read(sink, minOf(byteCount, left.toLong()))

      OverrideParam.WriteTimeout -> {
        val body =
          object : RequestBody() {
            override fun contentType(): MediaType? = null

            override fun writeTo(sink: BufferedSink) {
              if (sink
                  .timeout()
                  .timeoutNanos()
                  .nanoseconds.inWholeMilliseconds == 10L
              ) {
                throw IOException()
              }

    val extensions = this.extensions!!
    synchronized(this) {
      this.name = name
      this.socket = socket
      this.writer =
        WebSocketWriter(
          isClient = client,
          sink = socket.sink,
          random = random,
          perMessageDeflate = extensions.perMessageDeflate,
          noContextTakeover = extensions.noContextTakeover(client),
          minimumDeflateSize = minimumDeflateSize,

    return client.newCall(request).execute()
  }

  private fun writeFile(
    directory: Path,
    file: String,
    content: String,
  ) {
    val sink = fileSystem.sink(directory / file).buffer()
    sink.writeUtf8(content)
    sink.close()
  }

  /**
   * @param delta the offset from the current date to use. Negative values yield dates in the past;
   * positive values yield dates in the future.

    } else {
      javaNetSocket.soTimeout = chain.readTimeoutMillis()
      okHttpSocket.source.timeout().timeout(chain.readTimeoutMillis.toLong(), MILLISECONDS)
      okHttpSocket.sink.timeout().timeout(chain.writeTimeoutMillis.toLong(), MILLISECONDS)
      Http1ExchangeCodec(client, this, okHttpSocket)
    }
  }

  internal fun useAsSocket() {
    javaNetSocket.soTimeout = 0