checks

Author: nadav3396

src/main/java/io/lumigo/core/SpansContainer.java

@@ -160,7 +160,8 @@ public void init(Map<String, String> env, Reporter reporter, Context context, Ob
                                         : null)
                         .event(
                                 Configuration.getInstance().isLumigoVerboseMode()
-                                        ? JsonUtils.getObjectAsJsonString(EventParserFactory.parseEvent(event))
+                                        ? JsonUtils.getObjectAsJsonString(
+                                                EventParserFactory.parseEvent(event))
                                         : null)
                         .build();
     }

src/main/java/io/lumigo/core/instrumentation/impl/ApacheKafkaProducerInstrumentation.java

@@ -6,10 +6,9 @@
 import io.lumigo.core.instrumentation.LumigoInstrumentationApi;
 import io.lumigo.core.instrumentation.agent.Loader;
 import io.lumigo.core.utils.LRUCache;
+import io.lumigo.models.KafkaSpan;
 import java.nio.charset.StandardCharsets;
 import java.util.UUID;
-
-import io.lumigo.models.KafkaSpan;
 import lombok.AllArgsConstructor;
 import net.bytebuddy.agent.builder.AgentBuilder;
 import net.bytebuddy.asm.Advice;

src/main/java/io/lumigo/core/parsers/event/EventParserFactory.java

@@ -1,7 +1,6 @@
 package io.lumigo.core.parsers.event;
 
 import com.amazonaws.services.lambda.runtime.events.APIGatewayProxyRequestEvent;
-import com.amazonaws.services.lambda.runtime.events.KafkaEvent;
 import com.amazonaws.services.lambda.runtime.events.SNSEvent;
 import com.amazonaws.services.lambda.runtime.events.SQSEvent;
 import org.pmw.tinylog.Logger;

src/main/java/io/lumigo/core/utils/AwsUtils.java

@@ -5,7 +5,6 @@
 import com.fasterxml.jackson.annotation.JsonInclude;
 import io.lumigo.models.KafkaSpan;
 import io.lumigo.models.Span;
-
 import java.nio.charset.StandardCharsets;
 import java.util.*;
 import java.util.regex.Matcher;
@@ -164,7 +163,10 @@ public static TriggeredBy extractTriggeredByFromEvent(Object event) {
                     for (KafkaEvent.KafkaEventRecord record : entry.getValue()) {
                         for (Map<String, byte[]> headers : record.getHeaders()) {
                             if (headers.containsKey(KafkaSpan.LUMIGO_MESSAGE_ID_KEY)) {
-                                messageIds.add(new String(headers.get(KafkaSpan.LUMIGO_MESSAGE_ID_KEY), StandardCharsets.UTF_8));
+                                messageIds.add(
+                                        new String(
+                                                headers.get(KafkaSpan.LUMIGO_MESSAGE_ID_KEY),
+                                                StandardCharsets.UTF_8));
                                 break;
                             }
                         }

src/main/java/io/lumigo/models/KafkaSpanFactory.java

@@ -119,7 +119,8 @@ public static KafkaSpan createConsume(
         List<KafkaSpan.KafkaConsumerRecord> records = new ArrayList<>();
         consumerRecords.forEach(
                 record -> {
-                    Header messageIdHeader = record.headers().lastHeader(KafkaSpan.LUMIGO_MESSAGE_ID_KEY);
+                    Header messageIdHeader =
+                            record.headers().lastHeader(KafkaSpan.LUMIGO_MESSAGE_ID_KEY);
                     String messageId =
                             messageIdHeader == null
                                     ? null

src/test/java/infa/AwsLambdaEventGenerator.java

@@ -8,8 +8,6 @@
 import com.amazonaws.services.lambda.runtime.events.models.dynamodb.StreamRecord;
 import com.amazonaws.services.lambda.runtime.events.models.s3.*;
 import com.sun.tools.javac.util.List;
-
-import java.text.CollationElementIterator;
 import java.util.Collections;
 import java.util.Date;
 import java.util.HashMap;
@@ -164,16 +162,19 @@ public KafkaEvent kafkaEvent() {
                 Collections.singletonMap(
                         "msk-topic-0",
                         Collections.singletonList(
-                                KafkaEvent.KafkaEventRecord
-                                        .builder()
+                                KafkaEvent.KafkaEventRecord.builder()
                                         .withTopic("msk-topic")
                                         .withPartition(0)
                                         .withOffset(1612)
                                         .withTimestamp(1716147063028L)
                                         .withTimestampType("CREATE_TIME")
                                         .withKey("a2V5")
                                         .withValue("TWVzc2FnZSBhdCAxNzE2MTQ3MDYzMDE1")
-                                        .withHeaders(Collections.singletonList(Collections.singletonMap("lumigoMessageId", "NjA1ZTU0YWItMA==".getBytes())))
+                                        .withHeaders(
+                                                Collections.singletonList(
+                                                        Collections.singletonMap(
+                                                                "lumigoMessageId",
+                                                                "NjA1ZTU0YWItMA==".getBytes())))
                                         .build())),
                 "aws:kafka",
                 "arn:aws:kafka:us-west-2:779055952581:cluster/MskLambdaStackCluster/8fff24de-4f6c-44ca-b072-61d7a1b450a4-2",

src/test/java/io/lumigo/core/utils/AwsUtilsTest.java

@@ -254,12 +254,13 @@ void test_extractTriggeredByFromEvent_KafkaEvent() throws JSONException {
         Object event = awsLambdaEventGenerator.kafkaEvent();
 
         JSONAssert.assertEquals(
-                "{" +
-                        "\"triggeredBy\": \"kafka\", " +
-                        "\"arn\": \"arn:aws:kafka:us-west-2:779055952581:cluster/MskLambdaStackCluster/8fff24de-4f6c-44ca-b072-61d7a1b450a4-2\", " +
-                        "\"messageIds\": [\"NjA1ZTU0YWItMA==\"]"+
-                        "}",
-                JsonUtils.getObjectAsJsonString(AwsUtils.extractTriggeredByFromEvent(event)), true);
+                "{"
+                        + "\"triggeredBy\": \"kafka\", "
+                        + "\"arn\": \"arn:aws:kafka:us-west-2:779055952581:cluster/MskLambdaStackCluster/8fff24de-4f6c-44ca-b072-61d7a1b450a4-2\", "
+                        + "\"messageIds\": [\"NjA1ZTU0YWItMA==\"]"
+                        + "}",
+                JsonUtils.getObjectAsJsonString(AwsUtils.extractTriggeredByFromEvent(event)),
+                true);
     }
 
     @ParameterizedTest