Add Stats.Bytes()

confluentinc · Apr 18, 2024 · e3edc2b · e3edc2b
1 parent f0d709f
commit e3edc2b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/kafka/event.go b/kafka/event.go
@@ -97,6 +97,10 @@ func (e Stats) String() string {
 	return e.statsJSON
 }
 
+func (e Stats) Bytes() []byte {
+	return []byte(e.statsJSON)
+}
+
 // AssignedPartitions consumer group rebalance event: assigned partition set
 type AssignedPartitions struct {
 	Partitions []TopicPartition

diff --git a/kafka/stats_event_test.go b/kafka/stats_event_test.go
@@ -31,7 +31,7 @@ func handleStatsEvent(t *testing.T, eventCh chan Event, statsReceived chan bool)
 
 			// test if the stats string can be decoded into JSON
 			var raw map[string]interface{}
-			err := json.Unmarshal([]byte(e.String()), &raw) // convert json to map
+			err := json.Unmarshal(e.Bytes(), &raw) // convert json to map
 			if err != nil {
 				t.Fatalf("json unmarshall error: %s", err)
 			}