fix(templates): headers with refs (#242)

Fixes: #241
lerenn · Jun 27, 2024 · 14e42a7 · 14e42a7
1 parent fe9359f
commit 14e42a7
Show file tree

Hide file tree

Showing 4 changed files with 403 additions and 0 deletions.
diff --git a/pkg/codegen/generators/v3/templates/message.tmpl b/pkg/codegen/generators/v3/templates/message.tmpl
@@ -131,7 +131,11 @@ func brokerMessageTo{{namify .Name}}(bMsg extensions.BrokerMessage) ({{namify .N
                         }
                         msg.Headers.{{ namify $key}} = t
                     {{- else}}
+                        {{- if $value.Reference }}
+                        msg.Headers.{{ namify $key}} = {{$value.ReferenceTo.Name}}(v)
+                        {{- else }}
                         msg.Headers.{{ namify $key}} = {{$value.Type}}(v)
+                        {{- end}}
                     {{- end}}
                 {{- else}}
                     {{- if eq $value.Type "object" }}
@@ -146,7 +150,11 @@ func brokerMessageTo{{namify .Name}}(bMsg extensions.BrokerMessage) ({{namify .N
                         }
                         msg.Headers.{{ namify $key}} = &t
                     {{- else}}
+                        {{- if $value.Reference }}
+                        h := {{$value.ReferenceTo.Name}}(v)
+                        {{- else }}
                         h := {{$value.Type}}(v)
+                        {{- end}}
                         msg.Headers.{{ namify $key}} = &h
                     {{- end}}
                 {{- end}}

diff --git a/test/v3/issues/241/asyncapi.gen.go b/test/v3/issues/241/asyncapi.gen.go