From 618269ec137d1749372d998ca3d8004301c1b57e Mon Sep 17 00:00:00 2001 From: Nicola Puppa Date: Wed, 17 Jul 2024 08:54:45 +0200 Subject: [PATCH] fix: add message correlation log to record stream --- .../test/filters/logger/RecordStreamLogger.java | 11 +++++++++++ 1 file changed, 11 insertions(+) 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 index 620566664..b1a42e21e 100644 --- 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, String>> getValueTypeLoggers() { return valueTypeLoggers; }