Merge branch 'kotlin-vertx-benchmarks-revamp-kotlinx-serialization-json-io' into kotlin-vertx-benchmarks-revamp

Use `encodeToSink` in "kotlinx-serialization-json-io" with a custom-implemented `VertxBufferRawSink` in JSON serialization

Author: ShreckYe

frameworks/Kotlin/vertx-web-kotlinx/README.md

@@ -27,6 +27,7 @@ The tests were run with:
 * [Vert.x Reactive PostgreSQL Client](https://vertx.io/docs/vertx-pg-client/java/)
 * [kotlinx.coroutines](https://github.com/Kotlin/kotlinx.coroutines)
 * [kotlinx.serialization](https://github.com/Kotlin/kotlinx.serialization)
+* [kotlinx-io](https://github.com/Kotlin/kotlinx-io)
 * [kotlinx.html](https://github.com/Kotlin/kotlinx.html)
 
 ## Test URLs

frameworks/Kotlin/vertx-web-kotlinx/build.gradle.kts

@@ -14,6 +14,7 @@ repositories {
 }
 
 val vertxVersion = "4.5.10"
+val kotlinxSerializationVersion = "1.7.3"
 dependencies {
     implementation(platform("io.vertx:vertx-stack-depchain:$vertxVersion"))
     implementation("io.vertx:vertx-web")
@@ -26,7 +27,11 @@ dependencies {
     runtimeOnly("io.netty.incubator:netty-incubator-transport-native-io_uring:0.0.25.Final:linux-x86_64")
 
     implementation("org.jetbrains.kotlinx:kotlinx-coroutines-core:1.9.0")
-    implementation("org.jetbrains.kotlinx:kotlinx-serialization-json:1.7.3")
+
+    implementation("org.jetbrains.kotlinx:kotlinx-serialization-json:$kotlinxSerializationVersion")
+    implementation("org.jetbrains.kotlinx:kotlinx-serialization-json-io:$kotlinxSerializationVersion")
+    implementation("org.jetbrains.kotlinx:kotlinx-io-core:0.5.4")
+
     implementation("org.jetbrains.kotlinx:kotlinx-html:0.11.0")
     //implementation("org.jetbrains.kotlinx:kotlinx-datetime:0.4.0") // the latest version is 0.6.1
 }

frameworks/Kotlin/vertx-web-kotlinx/src/main/kotlin/KotlinxIo.kt

@@ -0,0 +1,49 @@
+import io.vertx.core.streams.WriteStream
+import io.vertx.kotlin.coroutines.coAwait
+import kotlinx.coroutines.runBlocking
+import kotlinx.io.RawSink
+import kotlinx.io.readByteArray
+import io.vertx.core.buffer.Buffer as VertxBuffer
+import kotlinx.io.Buffer as KotlinxIoBuffer
+
+@Suppress("NOTHING_TO_INLINE")
+private inline fun Long.toIntOrThrow(): Int {
+    require(this in Int.MIN_VALUE.toLong()..Int.MAX_VALUE.toLong())
+    return toInt()
+}
+
+@JvmInline
+value class VertxBufferWriteStreamRawSink(val writeStream: WriteStream<VertxBuffer>) : RawSink {
+    override fun write(source: KotlinxIoBuffer, byteCount: Long) {
+        runBlocking {
+            // `source` is temporarily converted to a byte array because wrapping it as a Vert.x `Buffer` is too complicated to implement.
+            writeStream.write(VertxBuffer.buffer(source.readByteArray(byteCount.toIntOrThrow()))).coAwait()
+        }
+    }
+
+    override fun flush() {}
+
+    override fun close() {
+        writeStream.end()
+    }
+}
+
+// not used currently
+fun WriteStream<VertxBuffer>.toRawSink(): RawSink =
+    VertxBufferWriteStreamRawSink(this)
+
+
+@JvmInline
+value class VertxBufferRawSink(val vertxBuffer: VertxBuffer) : RawSink {
+    override fun write(source: KotlinxIoBuffer, byteCount: Long) {
+        // same problem as above
+        vertxBuffer.appendBytes(source.readByteArray(byteCount.toIntOrThrow()))
+    }
+
+    override fun flush() {}
+
+    override fun close() {}
+}
+
+fun VertxBuffer.toRawSink(): RawSink =
+    VertxBufferRawSink(this)

frameworks/Kotlin/vertx-web-kotlinx/src/main/kotlin/MainVerticle.kt

@@ -1,4 +1,5 @@
 import io.netty.channel.unix.Errors.NativeIoException
+import io.vertx.core.buffer.Buffer
 import io.vertx.core.http.HttpHeaders
 import io.vertx.core.http.HttpServer
 import io.vertx.core.http.HttpServerRequest
@@ -19,9 +20,12 @@ import kotlinx.coroutines.Dispatchers
 import kotlinx.coroutines.launch
 import kotlinx.html.*
 import kotlinx.html.stream.appendHTML
+import kotlinx.io.buffered
+import kotlinx.serialization.ExperimentalSerializationApi
 import kotlinx.serialization.Serializable
 import kotlinx.serialization.SerializationStrategy
 import kotlinx.serialization.json.Json
+import kotlinx.serialization.json.io.encodeToSink
 import java.time.ZonedDateTime
 import java.time.format.DateTimeFormatter
 
@@ -120,7 +124,28 @@ class MainVerticle(val hasDb: Boolean) : CoroutineVerticle() {
         checkedCoroutineHandlerUnconfined {
             it.response().run {
                 putJsonResponseHeader()
+
+                /*
+                // approach 1
                 end(Json.encodeToString(serializer, requestHandler(it)))/*.coAwait()*/
+                */
+
+                /*
+                // approach 2
+                // java.lang.IllegalStateException: You must set the Content-Length header to be the total size of the message body BEFORE sending any data if you are not using HTTP chunked encoding.
+                toRawSink().buffered().use { bufferedSink ->
+                    @OptIn(ExperimentalSerializationApi::class)
+                    Json.encodeToSink(serializer, requestHandler(it), bufferedSink)
+                }
+                */
+
+                // approach 3
+                end(Buffer.buffer().apply {
+                    toRawSink().buffered().use { bufferedSink ->
+                        @OptIn(ExperimentalSerializationApi::class)
+                        Json.encodeToSink(serializer, requestHandler(it), bufferedSink)
+                    }
+                })
             }
         }