Merge pull request #1123 from lefkarag/db_setup_handle_error

Return error exit code when db setup Lambda function fails

Author: Dennis Kieselhorst

labs/unicorn-store/infrastructure/db-setup/src/main/java/com/amazon/aws/DBSetupHandler.java

@@ -65,7 +65,7 @@ public APIGatewayProxyResponseEvent handleRequest(final APIGatewayProxyRequestEv
                             .withBody("DB Setup successful");
                 }
             }
-        } catch (SQLException | IOException sqlException) {
+        } catch (Exception sqlException) {
             logger.error("Error connection to the database:" + sqlException.getMessage());
             return new APIGatewayProxyResponseEvent()
                     .withStatusCode(500)

labs/unicorn-store/setup-infrastructure.sh

@@ -19,7 +19,16 @@ cdk bootstrap
 cdk deploy UnicornStoreInfrastructure --require-approval never --outputs-file target/output.json
 
 # Execute the DB Setup function to create the table
-aws lambda invoke --function-name $(cat target/output.json | jq -r '.UnicornStoreInfrastructure.DbSetupArn') /dev/stdout | cat;
+lambda_result=$(aws lambda invoke --function-name $(cat target/output.json | jq -r '.UnicornStoreInfrastructure.DbSetupArn') /dev/stdout 2>&1)
+# Extract the status code from the response payload
+lambda_status_code=$(echo "$lambda_result" | jq 'first(.. | objects | select(has("statusCode"))) | .statusCode')
+
+if [ "$lambda_status_code" == "200" ]; then
+    echo "DB Setup Lambda function executed successfully"
+else
+    echo "DB Setup Lambda function execution failed"
+    exit 1
+fi
 
 cd ~/environment/aws-lambda-java-workshop/labs/unicorn-store
 ./setup-vpc-peering.sh