birb-integrated-bench/src/Application.kt

package com.kmlabz.k8s

import com.google.gson.Gson
import com.google.gson.GsonBuilder
import io.ktor.application.*
import io.ktor.client.*
import io.ktor.client.engine.apache.*
import io.ktor.client.request.*
import io.ktor.client.request.forms.*
import io.ktor.http.*
import io.ktor.http.content.*
import io.ktor.util.*
import kotlinx.coroutines.*
import java.io.File
import java.time.Duration
import java.time.Instant

fun main(args: Array<String>): Unit = io.ktor.server.tomcat.EngineMain.main(args)

@KtorExperimentalAPI
@Suppress("unused")
@kotlin.jvm.JvmOverloads
fun Application.module(testing: Boolean = false) {
    runBlocking {
        val cachedFile = File("/app/wave.wav").readBytes()
        val gson = GsonBuilder().setPrettyPrinting().create()
        val startTime = Instant.now()
        var elapsed = Duration.ofSeconds(0)
        val targetTime = Duration.ofSeconds(10)
        var currIter = 1
        val timestamps = LinkedHashMap<Int,Instant>()
        var currTimeStamp = Instant.now()
        while(elapsed > targetTime){
            uploadData(currIter, gson, cachedFile)
            currIter+=1
            currTimeStamp = Instant.now()
            timestamps[currIter] = currTimeStamp
            elapsed = Duration.between(startTime, currTimeStamp)
        }
    }
}

@KtorExperimentalAPI
suspend fun uploadData(currIteration: Int, gson: Gson, inputSound: ByteArray) {
    HttpClient(Apache).use { client ->

        val parts: List<PartData> = formData {
            val headersBuilder = HeadersBuilder()
            headersBuilder[HttpHeaders.ContentType] = "audio/wave"
            headersBuilder[HttpHeaders.ContentDisposition] = "filename=csirip.wav"
            this.append(
                "file",
                inputSound,
                headersBuilder.build()
            )
            val inputSvcApi = InputSvcApi("2020-02-02T02:02:02", currIteration)
            val jsonHeadersBuilder = HeadersBuilder()
            jsonHeadersBuilder[HttpHeaders.ContentType] = "application/json"
            this.append(
                "description",
                gson.toJson(inputSvcApi),
                jsonHeadersBuilder.build()
            )
        }

        client.submitFormWithBinaryData<Unit>(formData = parts) {
            url("https://birb.k8s.kmlabz.com/benchmark")
        }
    }
}

data class InputSvcApi(val date: String, val device_id: Int)