commonMain.com.apadmi.mockzilla.lib.internal.utils.ServerUtils.kt Maven / Gradle / Ivy

Go to download
package com.apadmi.mockzilla.lib.internal.utils

import co.touchlab.kermit.Logger
import io.ktor.http.*
import io.ktor.server.application.*
import io.ktor.server.engine.*
import io.ktor.server.response.*
import io.ktor.server.routing.RoutingCall
import io.ktor.server.routing.RoutingContext
import io.ktor.util.*
import io.ktor.util.pipeline.*
import io.ktor.utils.io.*

internal suspend fun RoutingContext.safeResponse(
    logger: Logger,
    block: suspend (RoutingCall) -> Unit,
) = try {
    this.call.response.headers.append("Connection", "close")
    block(call)
} catch (e: Exception) {
    logger.e("Unexpected exception", e)

    // Letting the server handle the 500 in the case the server is about to be killed, otherwise
    // the port isn't freed correctly.
    if (e !is CancellationException && e !is kotlin.coroutines.cancellation.CancellationException) {
        call.respondText(
            text = e.stackTraceToString(),
            status = HttpStatusCode.InternalServerError
        )
    }
    Unit
}