Add additional logs

Author: trueangle

lambda-runtime/src/commonMain/kotlin/io/github/trueangle/knative/lambda/runtime/LambdaRuntime.kt

@@ -46,6 +46,8 @@ object LambdaRuntime {
 
     inline fun <reified I, reified O> run(crossinline initHandler: () -> LambdaHandler<I, O>) = runBlocking {
         val handler = try {
+            Log.info("Initializing Kotlin Native Lambda Runtime")
+
             initHandler()
         } catch (e: Exception) {
             Log.fatal(e)
@@ -54,11 +56,14 @@ object LambdaRuntime {
             exitProcess(1)
         }
 
+        val handlerName = handler::class.simpleName
         val inputTypeInfo = typeInfo<I>()
         val outputTypeInfo = typeInfo<O>()
 
         while (true) {
             try {
+                Log.info("Runtime is ready for a new event")
+
                 val (event, context) = client.retrieveNextEvent<I>(inputTypeInfo)
 
                 with(Log) {
@@ -68,13 +73,21 @@ object LambdaRuntime {
                     trace(context)
                 }
 
+                Log.info("$handlerName invocation started")
+
                 if (handler is LambdaStreamHandler<I, *>) {
                     val response = streamingResponse { handler.handleRequest(event, it, context) }
+
+                    Log.info("$handlerName started response streaming")
+
                     client.streamResponse(context, response)
                 } else {
                     handler as LambdaBufferedHandler<I, O>
                     val response = bufferedResponse(context) { handler.handleRequest(event, context) }
 
+                    Log.info("$handlerName invocation completed")
+                    Log.trace("$handlerName's buffered response: $response")
+
                     client.sendResponse(context, response, outputTypeInfo)
                 }
             } catch (e: LambdaRuntimeException) {
@@ -107,6 +120,7 @@ internal inline fun streamingResponse(crossinline handler: suspend (ByteWriteCha
                 handler(channel)
             } catch (e: Exception) {
                 Log.warn(e)
+
                 channel.writeStringUtf8(e.toTrailer())
             }
         }

lambda-runtime/src/commonMain/kotlin/io/github/trueangle/knative/lambda/runtime/api/LambdaRuntimeClient.kt

@@ -29,12 +29,13 @@ import io.ktor.http.ContentType.Application.Json as ContentTypeJson
 @PublishedApi
 internal class LambdaClient(private val httpClient: HttpClient) {
     private val invokeUrl = "http://${LambdaEnvironment.RUNTIME_API}/2018-06-01/runtime"
+    private val requestTimeout = 15.minutes.inWholeMilliseconds
 
     suspend fun <T> retrieveNextEvent(bodyType: TypeInfo): Pair<T, Context> {
         val response = httpClient.get {
             url("${invokeUrl}/invocation/next")
 
-            timeout { requestTimeoutMillis = 15.minutes.inWholeMilliseconds }
+            timeout { requestTimeoutMillis = requestTimeout }
         }
         val context = contextFromResponseHeaders(response)
         val body = try {
@@ -47,8 +48,6 @@ internal class LambdaClient(private val httpClient: HttpClient) {
     }
 
     suspend fun <T> sendResponse(event: Context, body: T, bodyType: TypeInfo): HttpResponse {
-        Log.trace("Response from handler: $body")
-
         val response = httpClient.post {
             url("${invokeUrl}/invocation/${event.awsRequestId}/response")
             contentType(ContentTypeJson)
@@ -70,7 +69,7 @@ internal class LambdaClient(private val httpClient: HttpClient) {
             url("${invokeUrl}/invocation/${event.awsRequestId}/response")
 
             timeout {
-                requestTimeoutMillis = 15.minutes.inWholeMilliseconds
+                requestTimeoutMillis = requestTimeout
             }
 
             headers {