From 25c652ebac68249dbaf72758ba9a6ec169a7fcad Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Michal=20Tich=C3=A1k?= <michal.tichak@cern.ch>
Date: Tue, 4 Mar 2025 10:41:17 +0100
Subject: [PATCH] [core] use Taskid for kafka task events instead of
 Environmentid

---
 common/event/writer.go | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/common/event/writer.go b/common/event/writer.go
index 49bb0d3c..a6444b10 100644
--- a/common/event/writer.go
+++ b/common/event/writer.go
@@ -124,7 +124,10 @@ func (w *KafkaWriter) WriteEventWithTimestamp(e interface{}, timestamp time.Time
 				Payload:       &pb.Event_FrameworkEvent{FrameworkEvent: e},
 			}
 		case *pb.Ev_TaskEvent:
-			key = extractAndConvertEnvID(e)
+			key = []byte(e.Taskid)
+			if len(key) == 0 {
+				key = nil
+			}
 			wrappedEvent = &pb.Event{
 				Timestamp:     timestamp.UnixMilli(),
 				TimestampNano: timestamp.UnixNano(),