Add local testing support with RIE

Author: AntoniaSzecsi

aws-lambda-java-runtime-interface-client/Dockerfile.rie

@@ -0,0 +1,22 @@
+FROM public.ecr.aws/amazoncorretto/amazoncorretto:11
+
+# Install RIE
+RUN yum update -y && \
+    yum install -y curl && \
+    curl -Lo /usr/local/bin/aws-lambda-rie \
+    https://github.com/aws/aws-lambda-runtime-interface-emulator/releases/latest/download/aws-lambda-rie && \
+    chmod +x /usr/local/bin/aws-lambda-rie
+
+WORKDIR /function
+
+COPY target/aws-lambda-java-runtime-interface-client-*.jar ./
+COPY target/aws-lambda-java-core-*.jar ./
+COPY target/aws-lambda-java-serialization-*.jar ./
+
+COPY test-handlers/EchoHandler.class ./
+
+# Run RIE which will launch the Java RIC
+ENTRYPOINT ["/usr/local/bin/aws-lambda-rie", "/usr/bin/java", "-cp", "./*", "com.amazonaws.services.lambda.runtime.api.client.AWSLambda"]
+
+# Default handler can be overridden by passing a new one on `docker run`
+CMD ["EchoHandler::handleRequest"]

aws-lambda-java-runtime-interface-client/Makefile

@@ -65,6 +65,10 @@ publish:
 test-publish:
 	./ric-dev-environment/test-platform-specific-jar-snapshot.sh
 
+.PHONY: test-rie
+test-rie:
+	./scripts/test-rie.sh "EchoHandler::handleRequest" 
+
 define HELP_MESSAGE
 
 Usage: $ make [TARGETS]
@@ -74,5 +78,5 @@ TARGETS
 	dev         Run all development tests after a change.
 	pr          Perform all checks before submitting a Pull Request.
 	test        Run the Unit tests.
-
+	test-rie    Build and test RIC locally with Lambda Runtime Interface Emulator. (Requires building the project first)
 endef

aws-lambda-java-runtime-interface-client/scripts/test-rie.sh

@@ -0,0 +1,43 @@
+#!/bin/bash
+set -euo pipefail
+
+SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+PROJECT_ROOT="$(dirname "$SCRIPT_DIR")"
+SERIALIZATION_ROOT="$(dirname "$PROJECT_ROOT")/aws-lambda-java-serialization"
+
+if ! ls "$PROJECT_ROOT"/target/aws-lambda-java-runtime-interface-client-*.jar >/dev/null 2>&1; then
+  echo "RIC jar not found. Please build the project first with 'mvn package'."
+  exit 1
+fi
+
+IMAGE_TAG="java-ric-rie-test"
+
+HANDLER="${1:-EchoHandler::handleRequest}"
+
+echo "Starting RIE test setup for Java..."
+
+# Download required dependencies if not present
+if ! ls "$PROJECT_ROOT"/target/aws-lambda-java-core-*.jar >/dev/null 2>&1; then
+  echo "Downloading aws-lambda-java-core..."
+  (cd "$PROJECT_ROOT" && mvn dependency:copy -Dartifact=com.amazonaws:aws-lambda-java-core:1.3.0 -DoutputDirectory=target)
+fi
+
+if ! ls "$PROJECT_ROOT"/target/aws-lambda-java-serialization-*.jar >/dev/null 2>&1; then
+  echo "Downloading aws-lambda-java-serialization..."
+  (cd "$PROJECT_ROOT" && mvn dependency:copy -Dartifact=com.amazonaws:aws-lambda-java-serialization:1.1.6 -DoutputDirectory=target)
+fi
+
+echo "Compiling EchoHandler..."
+javac -source 11 -target 11 -cp "$(ls "$PROJECT_ROOT"/target/aws-lambda-java-runtime-interface-client-*.jar):$(ls "$PROJECT_ROOT"/target/aws-lambda-java-core-*.jar):$(ls "$PROJECT_ROOT"/target/aws-lambda-java-serialization-*.jar)" \
+  -d "$PROJECT_ROOT/test-handlers/" "$PROJECT_ROOT/test-handlers/EchoHandler.java"
+
+echo "Building test Docker image..."
+docker build -t "$IMAGE_TAG" -f "$PROJECT_ROOT/Dockerfile.rie" "$PROJECT_ROOT"
+
+echo "Starting test container on port 9000..."
+echo ""
+echo "In another terminal, invoke with:"
+echo "curl -s -X POST -H 'Content-Type: application/json' \"http://localhost:9000/2015-03-31/functions/function/invocations\" -d '{\"message\":\"test\"}'"
+echo ""
+
+exec docker run -it -p 9000:8080 "$IMAGE_TAG" "$HANDLER"

aws-lambda-java-runtime-interface-client/test-handlers/EchoHandler.java

@@ -0,0 +1,20 @@
+import com.amazonaws.services.lambda.runtime.Context;
+import com.amazonaws.services.lambda.runtime.RequestHandler;
+import java.util.Map;
+import java.util.HashMap;
+
+public class EchoHandler implements RequestHandler<Map<String, Object>, Map<String, Object>> {
+    
+    @Override
+    public Map<String, Object> handleRequest(Map<String, Object> event, Context context) {
+        context.getLogger().log("Processing event: " + event);
+        
+        Map<String, Object> response = new HashMap<>(event);
+        response.put("timestamp", System.currentTimeMillis());
+        response.put("requestId", context.getAwsRequestId());
+        response.put("functionName", context.getFunctionName());
+        response.put("remainingTimeInMillis", context.getRemainingTimeInMillis());
+        
+        return response;
+    }
+}