Use type map[string][]string for ProduceMessage headers

apache · Nov 3, 2023 · 78c85ad · 78c85ad
1 parent b6d277f
commit 78c85ad
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/sdks/go/pkg/beam/io/natsio/write.go b/sdks/go/pkg/beam/io/natsio/write.go
@@ -38,7 +38,7 @@ func init() {
 type ProducerMessage struct {
 	Subject string
 	ID      string
-	Headers nats.Header
+	Headers map[string][]string
 	Data    []byte
 }
 

diff --git a/sdks/go/pkg/beam/io/natsio/write_test.go b/sdks/go/pkg/beam/io/natsio/write_test.go
@@ -156,7 +156,7 @@ func TestWrite(t *testing.T) {
 				ProducerMessage{
 					Subject: subject,
 					ID:      "1",
-					Headers: nats.Header{"key": []string{"val"}},
+					Headers: map[string][]string{"key": {"val"}},
 					Data:    []byte("msg1"),
 				},
 			},