chore: varied example of using core utilities

Author: pankajagrawal16

java/CoreUtilities/Function/pom.xml

@@ -2,7 +2,7 @@
     xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/maven-v4_0_0.xsd">
     <modelVersion>4.0.0</modelVersion>
     <groupId>powertools</groupId>
-    <artifactId>CoreUtilities</artifactId>
+    <artifactId>Function</artifactId>
     <version>1.0</version>
     <packaging>jar</packaging>
     <name>A sample Hello World created for SAM CLI using powertools core utilities</name>

java/CoreUtilities/Function/src/main/java/helloworld/App.java

@@ -12,51 +12,104 @@
 import com.amazonaws.services.lambda.runtime.RequestHandler;
 import com.amazonaws.services.lambda.runtime.events.APIGatewayProxyRequestEvent;
 import com.amazonaws.services.lambda.runtime.events.APIGatewayProxyResponseEvent;
+import com.amazonaws.xray.AWSXRay;
+import com.amazonaws.xray.entities.Entity;
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
+import software.amazon.cloudwatchlogs.emf.model.DimensionSet;
+import software.amazon.cloudwatchlogs.emf.model.Unit;
+import software.amazon.lambda.powertools.logging.LoggingUtils;
 import software.amazon.lambda.powertools.logging.Logging;
 import software.amazon.lambda.powertools.metrics.Metrics;
+import software.amazon.lambda.powertools.tracing.CaptureMode;
+import software.amazon.lambda.powertools.tracing.TracingUtils;
 import software.amazon.lambda.powertools.tracing.Tracing;
 
-import static software.amazon.lambda.powertools.tracing.CaptureMode.DISABLED;
+import static software.amazon.lambda.powertools.metrics.MetricsUtils.metricsLogger;
+import static software.amazon.lambda.powertools.metrics.MetricsUtils.withSingleMetric;
+import static software.amazon.lambda.powertools.tracing.TracingUtils.putMetadata;
+import static software.amazon.lambda.powertools.tracing.TracingUtils.withEntitySubsegment;
 
 /**
  * Handler for requests to Lambda function.
  */
 public class App implements RequestHandler<APIGatewayProxyRequestEvent, APIGatewayProxyResponseEvent> {
+    private final static Logger log = LogManager.getLogger();
 
-    Logger log = LogManager.getLogger();
-
-    @Logging(logEvent = true)
-    @Tracing(captureMode = DISABLED)
-    @Metrics(captureColdStart = true)
+    @Logging(logEvent = true, samplingRate = 0.7)
+    @Tracing(captureMode = CaptureMode.RESPONSE_AND_ERROR)
+    @Metrics(namespace = "ServerlessAirline", service = "payment", captureColdStart = true)
     public APIGatewayProxyResponseEvent handleRequest(final APIGatewayProxyRequestEvent input, final Context context) {
         Map<String, String> headers = new HashMap<>();
+
         headers.put("Content-Type", "application/json");
         headers.put("X-Custom-Header", "application/json");
 
+        metricsLogger().putMetric("CustomMetric1", 1, Unit.COUNT);
+
+        withSingleMetric("CustomMetrics2", 1, Unit.COUNT, "Another", (metric) -> {
+            metric.setDimensions(DimensionSet.of("AnotherService", "CustomService"));
+            metric.setDimensions(DimensionSet.of("AnotherService1", "CustomService1"));
+        });
+
+        LoggingUtils.appendKey("test", "willBeLogged");
+
         APIGatewayProxyResponseEvent response = new APIGatewayProxyResponseEvent()
                 .withHeaders(headers);
         try {
             final String pageContents = this.getPageContents("https://checkip.amazonaws.com");
+            log.info(pageContents);
+            TracingUtils.putAnnotation("Test", "New");
             String output = String.format("{ \"message\": \"hello world\", \"location\": \"%s\" }", pageContents);
 
+            TracingUtils.withSubsegment("loggingResponse", subsegment -> {
+                String sampled = "log something out";
+                log.info(sampled);
+                log.info(output);
+            });
+
+            threadOption1();
+
+            threadOption2();
+
+            log.info("After output");
             return response
                     .withStatusCode(200)
                     .withBody(output);
-        } catch (IOException e) {
+        } catch (IOException | InterruptedException e) {
             return response
                     .withBody("{}")
                     .withStatusCode(500);
         }
     }
 
-    @Tracing(namespace = "getPageContents")
+    private void threadOption1() throws InterruptedException {
+        final Entity traceEntity = AWSXRay.getTraceEntity();
+        assert traceEntity != null;
+        traceEntity.run(new Thread(this::log));
+    }
+
+    private void threadOption2() throws InterruptedException {
+        Entity traceEntity = AWSXRay.getTraceEntity();
+        Thread anotherThread = new Thread(() -> withEntitySubsegment("inlineLog", traceEntity, subsegment -> {
+            String var = "somethingToProcess";
+            log.info("inside threaded logging inline {}", var);
+        }));
+        anotherThread.start();
+        anotherThread.join();
+    }
+
+    @Tracing
+    private void log() {
+        log.info("inside threaded logging for function");
+    }
+
+    @Tracing(namespace = "getPageContents", captureMode = CaptureMode.DISABLED)
     private String getPageContents(String address) throws IOException {
-        log.info("Retrieving {}", address);
         URL url = new URL(address);
+        putMetadata("getPageContents", address);
         try (BufferedReader br = new BufferedReader(new InputStreamReader(url.openStream()))) {
             return br.lines().collect(Collectors.joining(System.lineSeparator()));
         }
     }
-}
+}

java/CoreUtilities/template.yaml

@@ -22,7 +22,7 @@ Resources:
   HelloWorldFunction:
     Type: AWS::Serverless::Function # More info about Function Resource: https://github.com/awslabs/serverless-application-model/blob/master/versions/2016-10-31.md#awsserverlessfunction
     Properties:
-      CodeUri: HelloWorldFunction
+      CodeUri: Function
       Handler: helloworld.App::handleRequest
       Environment: # More info about Env Vars: https://github.com/awslabs/serverless-application-model/blob/master/versions/2016-10-31.md#environment-object
         Variables: