apigateway-lambda sample app

Author: srprash

sample-apps/apigateway-lambda/build.gradle.kts

@@ -0,0 +1,38 @@
+plugins {
+  java
+  application
+}
+
+application {
+  mainClass.set("com.amazon.sampleapp.LambdaHandler")
+}
+
+java {
+  toolchain {
+    languageVersion.set(JavaLanguageVersion.of(17))
+  }
+}
+
+dependencies {
+  implementation("com.amazonaws:aws-lambda-java-core:1.2.2")
+  implementation("com.squareup.okhttp3:okhttp:4.11.0")
+  implementation("software.amazon.awssdk:s3:2.20.0")
+  testImplementation("org.junit.jupiter:junit-jupiter:5.10.0")
+}
+
+tasks.jar {
+  manifest {
+    attributes["Main-Class"] = "com.amazon.sampleapp.LambdaHandler"
+  }
+}
+
+tasks.register<Zip>("createLambdaZip") {
+  dependsOn("build")
+  from(tasks.compileJava.get())
+  from(tasks.processResources.get())
+  into("lib") {
+    from(configurations.runtimeClasspath.get())
+  }
+  archiveFileName.set("lambda-function.zip")
+  destinationDirectory.set(layout.buildDirectory.dir("distributions"))
+}

sample-apps/apigateway-lambda/src/main/java/com/amazon/sampleapp/LambdaHandler.java

@@ -0,0 +1,49 @@
+package com.amazon.sampleapp;
+
+import com.amazonaws.services.lambda.runtime.Context;
+import com.amazonaws.services.lambda.runtime.RequestHandler;
+import okhttp3.OkHttpClient;
+import okhttp3.Request;
+import okhttp3.Response;
+import software.amazon.awssdk.services.s3.S3Client;
+import software.amazon.awssdk.services.s3.model.ListBucketsResponse;
+
+public class LambdaHandler implements RequestHandler<Object, String> {
+  private final OkHttpClient httpClient;
+  private final S3Client s3Client;
+
+  public LambdaHandler() {
+    this.httpClient = new OkHttpClient();
+    this.s3Client = S3Client.create();
+  }
+
+  @Override
+  public String handleRequest(Object o, Context context) {
+    makeRemoteCall();
+    listS3Buckets();
+
+    // Get the _X_AMZN_TRACE_ID environment variable
+    String traceId = System.getenv("_X_AMZN_TRACE_ID");
+
+    // Construct the response string
+    return "Trace ID: " + traceId;
+  }
+
+  private void makeRemoteCall() {
+    try {
+      Request request = new Request.Builder().url("https://aws.amazon.com/").build();
+      Response response = httpClient.newCall(request).execute();
+      response.close();
+    } catch (Exception e) {
+      e.printStackTrace();
+    }
+  }
+
+  private void listS3Buckets() {
+    ListBucketsResponse response = s3Client.listBuckets();
+    int bucketCount = response.buckets().size();
+
+    // Print bucket count
+    System.out.println("Number of S3 buckets: " + bucketCount);
+  }
+}

settings.gradle.kts

@@ -51,6 +51,7 @@ include(":smoke-tests:spring-boot")
 include(":sample-apps:springboot")
 include(":sample-apps:spark")
 include(":sample-apps:spark-awssdkv1")
+include(":sample-apps:apigateway-lambda")
 
 // Used for contract tests
 include("appsignals-tests:images:mock-collector")