fix: add message correlation log to record stream

camunda · Jul 17, 2024 · 618269e · 618269e
1 parent 4dc3edf
commit 618269e
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/filters/src/main/java/io/camunda/zeebe/process/test/filters/logger/RecordStreamLogger.java b/filters/src/main/java/io/camunda/zeebe/process/test/filters/logger/RecordStreamLogger.java
@@ -29,6 +29,7 @@
 import io.camunda.zeebe.protocol.record.value.JobBatchRecordValue;
 import io.camunda.zeebe.protocol.record.value.JobRecordValue;
 import io.camunda.zeebe.protocol.record.value.MessageBatchRecordValue;
+import io.camunda.zeebe.protocol.record.value.MessageCorrelationRecordValue;
 import io.camunda.zeebe.protocol.record.value.MessageRecordValue;
 import io.camunda.zeebe.protocol.record.value.MessageStartEventSubscriptionRecordValue;
 import io.camunda.zeebe.protocol.record.value.MessageSubscriptionRecordValue;
@@ -125,6 +126,7 @@ public RecordStreamLogger(final RecordStreamSource recordStreamSource) {
         ValueType.PROCESS_INSTANCE_MIGRATION, this::logProcessInstanceMigrationRecordValue);
     valueTypeLoggers.put(
         ValueType.COMPENSATION_SUBSCRIPTION, this::logCompensationSubscriptionRecordValue);
+    valueTypeLoggers.put(ValueType.MESSAGE_CORRELATION, this::logMessageCorrelationRecordValue);
   }
 
   public void log() {
@@ -488,6 +490,15 @@ private String logCompensationSubscriptionRecordValue(final Record<?> record) {
     return joiner.toString();
   }
 
+  private String logMessageCorrelationRecordValue(final Record<?> record) {
+    final MessageCorrelationRecordValue value = (MessageCorrelationRecordValue) record.getValue();
+    final StringJoiner joiner = new StringJoiner(", ", "", "");
+    joiner.add(String.format("(Process instance key: %d)", value.getProcessInstanceKey()));
+    joiner.add(String.format("(Message name: %s)", value.getName()));
+    joiner.add(String.format("(Message correlation key id: %s)", value.getCorrelationKey()));
+    return joiner.toString();
+  }
+
   protected Map<ValueType, Function<Record<?>, String>> getValueTypeLoggers() {
     return valueTypeLoggers;
   }