package kotlinx.benchmarks.json

import com.fasterxml.jackson.databind.*
import com.fasterxml.jackson.module.kotlin.*
import kotlinx.serialization.*
import kotlinx.serialization.json.*
import kotlinx.serialization.json.okio.encodeToBufferedSink
import okio.blackholeSink
import okio.buffer
import org.openjdk.jmh.annotations.*
import java.io.ByteArrayInputStream
import java.io.OutputStream
import java.util.concurrent.*

@Warmup(iterations = 7, time = 1)
@Measurement(iterations = 5, time = 1)
@BenchmarkMode(Mode.Throughput)
@OutputTimeUnit(TimeUnit.MILLISECONDS)
@State(Scope.Benchmark)
@Fork(2)
open class JacksonComparisonBenchmark {

    @Serializable
    data class DefaultPixelEvent(
        val version: Int,
        val dateTime2: String,
        val serverName: String,
        val domain: String,
        val method: String,
        val clientIp: String,
        val queryString: String,
        val userAgent: String,
        val contentType: String,
        val browserLanguage: String,
        val postData: String,
        val cookies: String
    )

    private val objectMapper: ObjectMapper = jacksonObjectMapper()

    private val data = DefaultPixelEvent(
        version = 1,
        dateTime2 = System.currentTimeMillis().toString(),
        serverName = "some-endpoint-qwer",
        domain = "some.domain.com",
        method = "POST",
        clientIp = "127.0.0.1",
        queryString = "anxa=CASCative&anxv=13.901.16.34566&anxe=FoolbarActive&anxt=E7AFBF15-1761-4343-92C1-78167ED19B1C&anxtv=13.901.16.34566&anxp=%5ECQ6%5Expt292%5ES33656%5Eus&anxsi&anxd=2019-10-08T17%3A03%3A57.246Z&f=00400000&anxr=1571945992297&coid=66abafd0d49f42e58dc7536109395306&userSegment&cwsid=opgkcnbminncdgghighmimmphiooeohh",
        userAgent = "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.14; rv:70.0) Gecko/20100101 Firefox/70.0",
        contentType = "text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8",
        browserLanguage = "en-US,en;q=0.5",
        postData = "-",
        cookies = "_ga=GA1.2.971852807.1546968515"
    )

    private val dataWithEscapes = DefaultPixelEvent(
        version = 1,
        dateTime2 = System.currentTimeMillis().toString(),
        serverName = "some-endp\"oint-qwer",
        domain = "<a href=\"some.domain.com\">",
        method = "POST",
        clientIp = "127.0.0.1",
        queryString = "anxa=CASCative&anxv=13.901.16.34566&anxe=\"FoolbarActive\"&anxt=E7AFBF15-1761-4343-92C1-78167ED19B1C&anxtv=13.901.16.34566&anxp=%5ECQ6%5Expt292%5ES33656%5Eus&anxsi&anxd=2019-10-08T17%3A03%3A57.246Z&f=00400000&anxr=1571945992297&coid=\"66abafd0d49f42e58dc7536109395306\"&userSegment&cwsid=opgkcnbminncdgghighmimmphiooeohh",
        userAgent = "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.14; rv:70.0) Gecko/20100101 Firefox/70.0",
        contentType = "text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8",
        browserLanguage = "\"en\"-\"US\",en;\\q=0.5",
        postData = "-",
        cookies = "_ga=GA1.2.971852807.1546968515"
    )

    private val devNullSink = blackholeSink().buffer()
    private val devNullStream = object : OutputStream() {
        override fun write(b: Int) {}
        override fun write(b: ByteArray) {}
        override fun write(b: ByteArray, off: Int, len: Int) {}
    }

    private val stringData = Json.encodeToString(DefaultPixelEvent.serializer(), data)
    private val utf8BytesData = stringData.toByteArray()

    @Serializable
    private class SmallDataClass(val id: Int, val name: String)

    private val smallData = SmallDataClass(42, "Vincent")

    @Benchmark
    fun jacksonToString(): String = objectMapper.writeValueAsString(data)

    @Benchmark
    fun jacksonToStringWithEscapes(): String = objectMapper.writeValueAsString(dataWithEscapes)

    @Benchmark
    fun jacksonSmallToString(): String = objectMapper.writeValueAsString(smallData)

    @Benchmark
    fun kotlinToString(): String = Json.encodeToString(DefaultPixelEvent.serializer(), data)

    @Benchmark
    fun kotlinToStream() = Json.encodeToStream(DefaultPixelEvent.serializer(), data, devNullStream)

    @Benchmark
    fun kotlinFromStream() = Json.decodeFromStream(DefaultPixelEvent.serializer(), ByteArrayInputStream(utf8BytesData))

    @Benchmark
    fun kotlinToOkio() = Json.encodeToBufferedSink(DefaultPixelEvent.serializer(), data, devNullSink)

    @Benchmark
    fun kotlinToStringWithEscapes(): String = Json.encodeToString(DefaultPixelEvent.serializer(), dataWithEscapes)

    @Benchmark
    fun kotlinSmallToString(): String = Json.encodeToString(SmallDataClass.serializer(), smallData)

    @Benchmark
    fun kotlinSmallToStream() = Json.encodeToStream(SmallDataClass.serializer(), smallData, devNullStream)

    @Benchmark
    fun kotlinSmallToOkio() = Json.encodeToBufferedSink(SmallDataClass.serializer(), smallData, devNullSink)

    @Benchmark
    fun jacksonFromString(): DefaultPixelEvent = objectMapper.readValue(stringData, DefaultPixelEvent::class.java)

    @Benchmark
    fun kotlinFromString(): DefaultPixelEvent = Json.decodeFromString(DefaultPixelEvent.serializer(), stringData)
}