Log user init and invoke exceptions as ERROR in JSON format (#49)

Co-authored-by: Daniel Torok <[email protected]>

Author: dtorok

aws-lambda-java-runtime-interface-client/src/main/java/com/amazonaws/services/lambda/runtime/api/client/AWSLambda.java

@@ -189,7 +189,7 @@ public static void main(String[] args) {
 
     private static void startRuntime(String handler) {
         try (LogSink logSink = createLogSink()) {
-            LambdaLogger logger = new LambdaContextLogger(
+            LambdaContextLogger logger = new LambdaContextLogger(
                     logSink,
                     LogLevel.fromString(LambdaEnvironment.LAMBDA_LOG_LEVEL),
                     LogFormat.fromString(LambdaEnvironment.LAMBDA_LOG_FORMAT)
@@ -200,7 +200,7 @@ private static void startRuntime(String handler) {
         }
     }
 
-    private static void startRuntime(String handler, LambdaLogger lambdaLogger) throws Throwable {
+    private static void startRuntime(String handler, LambdaContextLogger lambdaLogger) throws Throwable {
         UnsafeUtil.disableIllegalAccessWarning();
 
         System.setOut(new PrintStream(new LambdaOutputStream(System.out), false, "UTF-8"));
@@ -222,7 +222,7 @@ private static void startRuntime(String handler, LambdaLogger lambdaLogger) thro
         try {
             requestHandler = findRequestHandler(handler, customerClassLoader);
         } catch (UserFault userFault) {
-            lambdaLogger.log(userFault.reportableError());
+            lambdaLogger.log(userFault.reportableError(), lambdaLogger.getLogFormat() == LogFormat.JSON ? LogLevel.ERROR : LogLevel.UNDEFINED);
             reportInitError(new Failure(userFault), runtimeClient);
             System.exit(1);
             return;
@@ -265,13 +265,13 @@ private static void startRuntime(String handler, LambdaLogger lambdaLogger) thro
                         serializeAsXRayJson(t));
             } finally {
                 if (userFault != null) {
-                    lambdaLogger.log(userFault.reportableError());
+                    lambdaLogger.log(userFault.reportableError(), lambdaLogger.getLogFormat() == LogFormat.JSON ? LogLevel.ERROR : LogLevel.UNDEFINED);
                 }
             }
         }
     }
 
-    static void onInitComplete(final LambdaRuntimeClient runtimeClient, final LambdaLogger lambdaLogger) throws IOException {
+    static void onInitComplete(final LambdaRuntimeClient runtimeClient, final LambdaContextLogger lambdaLogger) throws IOException {
         try {
             Core.getGlobalContext().beforeCheckpoint(null);
             // Blocking call to RAPID /restore/next API, will return after taking snapshot.
@@ -292,10 +292,10 @@ static void onInitComplete(final LambdaRuntimeClient runtimeClient, final Lambda
         }
     }
 
-    private static void logExceptionCloudWatch(LambdaLogger lambdaLogger, Exception exc) {
+    private static void logExceptionCloudWatch(LambdaContextLogger lambdaLogger, Exception exc) {
         UserFault.filterStackTrace(exc);
         UserFault userFault = UserFault.makeUserFault(exc, true);
-        lambdaLogger.log(userFault.reportableError());
+        lambdaLogger.log(userFault.reportableError(), lambdaLogger.getLogFormat() == LogFormat.JSON ? LogLevel.ERROR : LogLevel.UNDEFINED);
     }
 
     static void reportInitError(final Failure failure,

aws-lambda-java-runtime-interface-client/src/main/java/com/amazonaws/services/lambda/runtime/api/client/logging/AbstractLambdaLogger.java

@@ -65,4 +65,8 @@ public void log(byte[] message) {
     public void setLambdaContext(LambdaContext lambdaContext) {
         this.logFormatter.setLambdaContext(lambdaContext);
     }
+
+    public LogFormat getLogFormat() {
+        return logFormat;
+    }
 }