Add afterburner

Signed-off-by: Chase Engelbrecht <[email protected]>
engechas · Mar 20, 2024 · 96e03dd · 96e03dd
1 parent 9df58e2
commit 96e03dd
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/data-prepper-plugins/s3-source/build.gradle b/data-prepper-plugins/s3-source/build.gradle
@@ -27,6 +27,7 @@ dependencies {
     implementation 'org.hibernate.validator:hibernate-validator:8.0.1.Final'
     implementation 'com.fasterxml.jackson.dataformat:jackson-dataformat-csv'
     implementation 'com.fasterxml.jackson.datatype:jackson-datatype-jsr310'
+    implementation 'com.fasterxml.jackson.module:jackson-module-afterburner'
     implementation 'org.xerial.snappy:snappy-java:1.1.10.5'
     implementation 'org.apache.parquet:parquet-common:1.13.1'
     implementation 'dev.failsafe:failsafe:3.3.2'

diff --git a/...s3-source/src/main/java/org/opensearch/dataprepper/plugins/source/s3/codec/OCSFCodec.java b/...s3-source/src/main/java/org/opensearch/dataprepper/plugins/source/s3/codec/OCSFCodec.java
@@ -3,6 +3,7 @@
 import com.fasterxml.jackson.databind.MappingIterator;
 import com.fasterxml.jackson.databind.ObjectMapper;
 import com.fasterxml.jackson.databind.ObjectReader;
+import com.fasterxml.jackson.module.afterburner.AfterburnerModule;
 import org.opensearch.dataprepper.model.annotations.DataPrepperPlugin;
 import org.opensearch.dataprepper.model.annotations.DataPrepperPluginConstructor;
 import org.opensearch.dataprepper.model.codec.InputCodec;
@@ -16,7 +17,8 @@
 
 @DataPrepperPlugin(name = "ocsf", pluginType = InputCodec.class, pluginConfigurationType = OCSFCodecConfig.class)
 public class OCSFCodec implements InputCodec {
-    private static final ObjectReader OBJECT_READER = new ObjectMapper().readerFor(OCSF.class);
+    private static final ObjectMapper OBJECT_MAPPER = new ObjectMapper().registerModule(new AfterburnerModule());
+    private static final ObjectReader OBJECT_READER = OBJECT_MAPPER.readerFor(OCSF.class);
 
     @DataPrepperPluginConstructor
     public OCSFCodec(final OCSFCodecConfig config) {