Dynamic renaming of keys

opensearch-project · Oct 21, 2024 · 1f76593 · 1f76593
1 parent 26c2449
commit 1f76593
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 5 deletions.
diff --git a/data-prepper-api/src/main/java/org/opensearch/dataprepper/model/event/JacksonEventKey.java b/data-prepper-api/src/main/java/org/opensearch/dataprepper/model/event/JacksonEventKey.java
@@ -161,7 +161,8 @@ private static boolean isValidKey(final String key) {
                     || c == '@'
                     || c == '/'
                     || c == '['
-                    || c == ']')) {
+                    || c == ']'
+                    || c == '|')) {
 
                 return false;
             }

diff --git a/...va/org/opensearch/dataprepper/plugins/processor/mutateevent/RenameKeyProcessorConfig.java b/...va/org/opensearch/dataprepper/plugins/processor/mutateevent/RenameKeyProcessorConfig.java
@@ -73,6 +73,9 @@ public String getRenameWhen() {
         }
 
         public Pattern getFromKeyCompiledPattern() {
+            if (fromKeyPattern != null && fromKeyCompiledPattern == null) {
+                fromKeyCompiledPattern = Pattern.compile(fromKeyPattern);
+            }
             return fromKeyCompiledPattern;
         }
 
@@ -82,9 +85,6 @@ public Entry(final EventKey fromKey, final String fromKeyPattern, final EventKey
             this.toKey = toKey;
             this.overwriteIfToKeyExists = overwriteIfKeyExists;
             this.renameWhen = renameWhen;
-            if (fromKeyPattern != null) {
-                fromKeyCompiledPattern = Pattern.compile(fromKeyPattern);
-            }
         }
 
         public Entry() {

diff --git a/...ava/org/opensearch/dataprepper/plugins/processor/mutateevent/RenameKeyProcessorTests.java b/...ava/org/opensearch/dataprepper/plugins/processor/mutateevent/RenameKeyProcessorTests.java
@@ -19,8 +19,15 @@
 import org.mockito.Mock;
 import org.mockito.junit.jupiter.MockitoExtension;
 
-import java.util.*;
+import java.util.Collection;
+import java.util.Collections;
+import java.util.List;
+import java.util.UUID;
+import java.util.LinkedList;
+import java.util.Arrays;
+import java.util.Map;
 import java.util.ArrayList;
+import java.util.HashMap;
 
 import static org.hamcrest.CoreMatchers.equalTo;
 import static org.hamcrest.CoreMatchers.is;