generates NewHeader for BeginString

Author: Chris Busbey

cmd/generate-fix/generate-fix.go

@@ -59,6 +59,7 @@ func genHeader(pkg string, spec *datadictionary.DataDictionary) {
 		Package:    pkg,
 		Name:       "Header",
 		MessageDef: spec.Header,
+		FIXSpec:    spec,
 	}
 	if err := internal.HeaderTemplate.Execute(writer, c); err != nil {
 		errors <- err

cmd/generate-fix/internal/template_helpers.go

@@ -171,6 +171,14 @@ func requiredFields(m *datadictionary.MessageDef) (required []*datadictionary.Fi
 	return
 }
 
+func beginString(spec *datadictionary.DataDictionary) string {
+	if spec.FIXType == "FIXT" || spec.Major == 5 {
+		return "FIXT.1.1"
+	}
+
+	return fmt.Sprintf("FIX.%v.%v", spec.Major, spec.Minor)
+}
+
 func routerBeginString(spec *datadictionary.DataDictionary) (routerBeginString string) {
 	switch {
 	case spec.FIXType == "FIXT":

cmd/generate-fix/internal/templates.go

@@ -18,6 +18,7 @@ func init() {
 	tmplFuncs := template.FuncMap{
 		"toLower":             strings.ToLower,
 		"requiredFields":      requiredFields,
+		"beginString":         beginString,
 		"routerBeginString":   routerBeginString,
 		"importRootPath":      getImportPathRoot,
 		"quickfixType":        quickfixType,
@@ -146,6 +147,13 @@ type Header struct {
 	quickfix.Header
 }
 
+//NewHeader returns a new, initialized Header instance
+func NewHeader() (h Header) {
+	h.Init()
+	h.SetBeginString("{{ beginString .FIXSpec }}")
+	return
+}
+
 {{ template "setters" .}}
 {{ template "getters" . }}
 {{ template "hasers" . }}
@@ -222,7 +230,7 @@ func (m {{ .Name }}) ToMessage() quickfix.Message {
 {{ $required_fields := requiredFields .MessageDef -}}
 //New returns a {{ .Name }} initialized with the required fields for {{ .Name }} 
 func New({{template "field_args" $required_fields }}) (m {{ .Name }}) {
-	m.Header.Init()
+	m.Header = {{ .TransportPackage }}.NewHeader()
 	m.Init()
 	m.Trailer.Init()