Address comments from PR6370 (#6371)

Signed-off-by: Krishna Kondaka <krishkdk@amazon.com>

Author: kkondaka

data-prepper-core/src/main/java/org/opensearch/dataprepper/core/pipeline/router/Router.java

@@ -10,8 +10,8 @@
 import org.opensearch.dataprepper.model.record.Record;
 
 import java.util.Collection;
-import java.util.HashMap;
 import java.util.HashSet;
+import java.util.LinkedHashMap;
 import java.util.Map;
 import java.util.Objects;
 import java.util.Set;
@@ -42,7 +42,7 @@ public <C> void route(
         Objects.requireNonNull(dataFlowComponents);
         Objects.requireNonNull(componentRecordsConsumer);
 
-        final Map<Object, Collection<Record>> componentRecords = new HashMap<>();
+        final Map<C, Collection<Record>> componentRecords = new LinkedHashMap<>();
 
         final Map<Record, Set<String>> recordsToRoutes = routeEventEvaluator.evaluateEventRoutes(allRecords);
 
@@ -64,12 +64,12 @@ public <C> void route(
                         recordsUnRouted.remove(record);
                     }
                 }
-                componentRecords.put((C)component, records);
+                componentRecords.put(component, records);
             });
         }
 
-        for (Map.Entry<Object, Collection<Record>> entry : componentRecords.entrySet()) {
-            componentRecordsConsumer.accept((C)entry.getKey(), entry.getValue());
+        for (Map.Entry<C, Collection<Record>> entry : componentRecords.entrySet()) {
+            componentRecordsConsumer.accept(entry.getKey(), entry.getValue());
         }
 
         if (recordsUnRouted != null) {