rework classes

Author: maschnetwork

samples/spring-cloud-function-demo/src/main/java/com/amazonaws/springcloudfunctiondemo/SpringCloudFunctionDemoApplication.java

@@ -18,30 +18,35 @@ public static void main(String[] args) {
     }
 
     @Bean
-    public Function<String, String> lowerCase() {return String::toLowerCase;}
+    public Function<String, String> lowerCase() {
+        return String::toLowerCase;
+    }
 
     @Bean
-    public Function<String, String> upperCase(){
+    public Function<String, String> upperCase() {
         return String::toUpperCase;
     }
 
     @Bean
-    public Function<String, String> reverse(){
+    public Function<String, String> reverse() {
         return value -> new StringBuilder(value).reverse().toString();
     }
 
     @Bean
-    public Function<com.amazonaws.springcloudfunctiondemo.Unicorn, String> helloUnicorn(){
-        return value -> "Hello %s! You are %d years old!".formatted(value.name(), value.age());
+    public Function<Unicorn, String> helloUnicorn() {
+        return value -> "Hello %s! You are %d years old!"
+                .formatted(value.name(), value.age());
     }
 
     @Bean
-    public Consumer<SQSEvent> asyncProcessor(){
-        return value -> System.out.printf("Processed %d messages!", value.getRecords().size());
+    public Consumer<SQSEvent> asyncProcessor() {
+        return sqsEvent -> sqsEvent
+                .getRecords()
+                .forEach(it -> System.out.println(it.getBody()));
     }
 
     @Bean
-    public Function<String, String> noOpFunction(){
+    public Function<String, String> noOpFunction() {
         return value -> "No proper function found!";
     }
 
@@ -62,4 +67,6 @@ public String routingResult(Message<?> message) {
             }
         };
     }
+
+    public record Unicorn(String name, int age) { }
 }

samples/spring-cloud-function-demo/src/main/java/com/amazonaws/springcloudfunctiondemo/Unicorn.java

@@ -1,5 +1,6 @@
 package com.amazonaws.springcloudfunctiondemo;
 
+import com.amazonaws.springcloudfunctiondemo.SpringCloudFunctionDemoApplication.Unicorn;
 import org.junit.jupiter.api.Test;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.boot.test.context.SpringBootTest;
@@ -36,7 +37,7 @@ public void testUppercaseFunction() {
 
     @Test
     public void testHandleUnicornHelloFunction() {
-        var fluffy = new com.amazonaws.springcloudfunctiondemo.Unicorn("Fluffy", 3);
+        var fluffy = new Unicorn("Fluffy", 3);
         var headers = new HttpHeaders();
         headers.setContentType(MediaType.APPLICATION_JSON);
 

samples/spring-cloud-function-demo/src/test/java/com/amazonaws/springcloudfunctiondemo/HttpIntegrationTest.java

@@ -1,5 +1,6 @@
 package com.amazonaws.springcloudfunctiondemo;
 
+import com.amazonaws.springcloudfunctiondemo.SpringCloudFunctionDemoApplication.Unicorn;
 import org.junit.jupiter.api.Test;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.boot.test.context.SpringBootTest;
@@ -25,7 +26,7 @@ public class MessageCallBackTest {
     @Test
     public void testCustomRouter() {
         // Create test message
-        var unicorn = new com.amazonaws.springcloudfunctiondemo.Unicorn("RouterUnicorn", 2);
+        var unicorn = new Unicorn("RouterUnicorn", 2);
         var headers = new MessageHeaders(Map.of("x-routing-key", "unicorn"));
         var message = new GenericMessage<>(unicorn, headers);
 
@@ -36,7 +37,7 @@ public void testCustomRouter() {
         assertEquals("helloUnicorn", functionName);
 
         // Get the function and apply it
-        Function<com.amazonaws.springcloudfunctiondemo.Unicorn, String> function = catalog.lookup(Function.class, functionName);
+        Function<Unicorn, String> function = catalog.lookup(Function.class, functionName);
         var result = function.apply(unicorn);
 
         // Verify result