generate setters

Author: Chris Busbey

_gen/generate-components/main.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"bytes"
 	"flag"
 	"fmt"
 	"github.com/quickfixgo/quickfix/_gen"
@@ -158,7 +159,7 @@ func genComponentImports() (fileOut string) {
 	}
 
 	fileOut += "import(\n"
-	for i, _ := range imports {
+	for i := range imports {
 		fileOut += fmt.Sprintf("\"%v\"\n", i)
 	}
 	fileOut += ")\n"
@@ -225,6 +226,12 @@ func genHeader(header *datadictionary.MessageDef) {
 	fileOut += genComponentImports()
 	fileOut += delayOut
 
+	writer := new(bytes.Buffer)
+	if err := gen.WriteFieldSetters(writer, "Header", header.FieldsInDeclarationOrder); err != nil {
+		panic(err)
+	}
+	fileOut += writer.String()
+
 	gen.WriteFile(path.Join(pkg, "header.go"), fileOut)
 }
 
@@ -238,6 +245,12 @@ func genTrailer(trailer *datadictionary.MessageDef) {
 	}
 	fileOut += "}\n"
 
+	writer := new(bytes.Buffer)
+	if err := gen.WriteFieldSetters(writer, "Trailer", trailer.FieldsInDeclarationOrder); err != nil {
+		panic(err)
+	}
+	fileOut += writer.String()
+
 	gen.WriteFile(path.Join(pkg, "trailer.go"), fileOut)
 }
 
@@ -258,9 +271,20 @@ func genComponent(name string, component *datadictionary.Component) {
 	fileOut += delayOut
 	fileOut += "func New() *Component { return new(Component)}\n"
 
+	fileOut += genComponentSetters(component)
+
 	gen.WriteFile(path.Join(pkg, strings.ToLower(name), name+".go"), fileOut)
 }
 
+func genComponentSetters(component *datadictionary.Component) string {
+	writer := new(bytes.Buffer)
+	if err := gen.WriteFieldSetters(writer, "Component", component.Fields); err != nil {
+		panic(err)
+	}
+
+	return writer.String()
+}
+
 func main() {
 	flag.Usage = usage
 	flag.Parse()

_gen/generate-messages/main.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"bytes"
 	"flag"
 	"fmt"
 	"github.com/quickfixgo/quickfix/_gen"
@@ -47,7 +48,7 @@ import(
 `
 	fileOut += fmt.Sprintf("\"github.com/quickfixgo/quickfix/%v\"\n", headerTrailerPkg())
 
-	for i, _ := range imports {
+	for i := range imports {
 		fileOut += fmt.Sprintf("\"%v\"\n", i)
 	}
 	fileOut += ")\n"
@@ -172,6 +173,13 @@ func genGroupDeclaration(field *datadictionary.FieldDef, parent string) (fileOut
 
 	fileOut += "}\n"
 
+	writer := new(bytes.Buffer)
+	if err := gen.WriteFieldSetters(writer, field.Name, field.ChildFields); err != nil {
+		panic(err)
+	}
+	fileOut += writer.String()
+	fileOut += "\n"
+
 	return
 }
 
@@ -231,6 +239,14 @@ func genMessage(msg *datadictionary.MessageDef, requiredFields []*datadictionary
 	return fileOut
 }
 
+func genMessageSetters(msg *datadictionary.MessageDef) string {
+	writer := new(bytes.Buffer)
+	if err := gen.WriteFieldSetters(writer, "Message", msg.FieldsInDeclarationOrder); err != nil {
+		panic(err)
+	}
+	return writer.String()
+}
+
 func genMessageRoute(msg *datadictionary.MessageDef) string {
 	var beginStringEnum string
 	if fixSpec.FIXType == "FIXT" {
@@ -284,6 +300,7 @@ func genMessagePkg(msg *datadictionary.MessageDef) {
 
 	fileOut += genMessageImports()
 	fileOut += delayOut
+	fileOut += genMessageSetters(msg)
 	fileOut += genMessageRoute(msg)
 
 	gen.WriteFile(path.Join(pkg, strings.ToLower(msg.Name), msg.Name+".go"), fileOut)

_gen/helpers.go

@@ -0,0 +1,120 @@
+package gen
+
+import (
+	"fmt"
+	"github.com/quickfixgo/quickfix/datadictionary"
+	"io"
+	"text/template"
+)
+
+var fieldSetterTemplate *template.Template
+
+func init() {
+	tmplFuncs := make(template.FuncMap)
+	tmplFuncs["fixFieldTypeToGoType"] = fixFieldTypeToGoType
+
+	fieldSetterTemplate = template.Must(template.New("Setters").Funcs(tmplFuncs).Parse(`
+func (m *{{.Receiver}}) Set{{.Name}}(v {{ if .IsGroup}}[]{{.Name}}{{else}}{{fixFieldTypeToGoType .Type}}{{end}}) {
+{{- if .IsGroup -}}m.{{.Name}} = v
+{{- else if .Required -}}m.{{.Name}} = v
+{{- else -}}m.{{.Name}} = &v
+{{- end}}}`))
+}
+
+//WriteFieldSetters generates setters appropriate for Messages, Components and Repeating Groups
+func WriteFieldSetters(writer io.Writer, receiver string, fields []*datadictionary.FieldDef) error {
+	type setter struct {
+		Receiver string
+		*datadictionary.FieldDef
+	}
+
+	for _, field := range fields {
+		if field.IsComponent() {
+			continue
+		}
+
+		if err := fieldSetterTemplate.Execute(writer, setter{receiver, field}); err != nil {
+			return err
+		}
+	}
+
+	return nil
+}
+
+func fixFieldTypeToGoType(fieldType string) string {
+	switch fieldType {
+	case "MULTIPLESTRINGVALUE", "MULTIPLEVALUESTRING":
+		fallthrough
+	case "MULTIPLECHARVALUE":
+		fallthrough
+	case "CHAR":
+		fallthrough
+	case "CURRENCY":
+		fallthrough
+	case "DATA":
+		fallthrough
+	case "MONTHYEAR":
+		fallthrough
+	case "LOCALMKTDATE":
+		fallthrough
+	case "DATE":
+		fallthrough
+	case "EXCHANGE":
+		fallthrough
+	case "LANGUAGE":
+		fallthrough
+	case "XMLDATA":
+		fallthrough
+	case "COUNTRY":
+		fallthrough
+	case "UTCTIMEONLY":
+		fallthrough
+	case "UTCDATE":
+		fallthrough
+	case "UTCDATEONLY":
+		fallthrough
+	case "TZTIMEONLY":
+		fallthrough
+	case "TZTIMESTAMP":
+		fallthrough
+	case "STRING":
+		return "string"
+
+	case "BOOLEAN":
+		return "bool"
+
+	case "LENGTH":
+		fallthrough
+	case "DAYOFMONTH":
+		fallthrough
+	case "NUMINGROUP":
+		fallthrough
+	case "SEQNUM":
+		fallthrough
+	case "INT":
+		return "int"
+
+	case "TIME":
+		fallthrough
+	case "UTCTIMESTAMP":
+		return "time.Time"
+
+	case "QTY":
+		fallthrough
+	case "QUANTITY":
+		fallthrough
+	case "AMT":
+		fallthrough
+	case "PRICE":
+		fallthrough
+	case "PRICEOFFSET":
+		fallthrough
+	case "PERCENTAGE":
+		fallthrough
+	case "FLOAT":
+		return "float64"
+
+	default:
+		panic(fmt.Sprintf("Unknown type '%v'\n", fieldType))
+	}
+}

fix40/advertisement/Advertisement.go

@@ -48,6 +48,22 @@ type Message struct {
 //Marshal converts Message to a quickfix.Message instance
 func (m Message) Marshal() quickfix.Message { return quickfix.Marshal(m) }
 
+func (m *Message) SetAdvId(v int)              { m.AdvId = v }
+func (m *Message) SetAdvTransType(v string)    { m.AdvTransType = v }
+func (m *Message) SetAdvRefID(v int)           { m.AdvRefID = &v }
+func (m *Message) SetSymbol(v string)          { m.Symbol = v }
+func (m *Message) SetSymbolSfx(v string)       { m.SymbolSfx = &v }
+func (m *Message) SetSecurityID(v string)      { m.SecurityID = &v }
+func (m *Message) SetIDSource(v string)        { m.IDSource = &v }
+func (m *Message) SetIssuer(v string)          { m.Issuer = &v }
+func (m *Message) SetSecurityDesc(v string)    { m.SecurityDesc = &v }
+func (m *Message) SetAdvSide(v string)         { m.AdvSide = v }
+func (m *Message) SetShares(v int)             { m.Shares = v }
+func (m *Message) SetPrice(v float64)          { m.Price = &v }
+func (m *Message) SetCurrency(v string)        { m.Currency = &v }
+func (m *Message) SetTransactTime(v time.Time) { m.TransactTime = &v }
+func (m *Message) SetText(v string)            { m.Text = &v }
+
 //A RouteOut is the callback type that should be implemented for routing Message
 type RouteOut func(msg Message, sessionID quickfix.SessionID) quickfix.MessageRejectError
 

fix40/allocation/Allocation.go

@@ -20,6 +20,11 @@ type NoOrders struct {
 	WaveNo *string `fix:"105"`
 }
 
+func (m *NoOrders) SetClOrdID(v string) { m.ClOrdID = v }
+func (m *NoOrders) SetOrderID(v string) { m.OrderID = &v }
+func (m *NoOrders) SetListID(v string)  { m.ListID = &v }
+func (m *NoOrders) SetWaveNo(v string)  { m.WaveNo = &v }
+
 //NoExecs is a repeating group in Allocation
 type NoExecs struct {
 	//ExecID is a non-required field for NoExecs.
@@ -32,6 +37,11 @@ type NoExecs struct {
 	LastMkt *string `fix:"30"`
 }
 
+func (m *NoExecs) SetExecID(v int)     { m.ExecID = &v }
+func (m *NoExecs) SetLastShares(v int) { m.LastShares = &v }
+func (m *NoExecs) SetLastPx(v float64) { m.LastPx = &v }
+func (m *NoExecs) SetLastMkt(v string) { m.LastMkt = &v }
+
 //NoMiscFees is a repeating group in Allocation
 type NoMiscFees struct {
 	//MiscFeeAmt is a non-required field for NoMiscFees.
@@ -42,6 +52,10 @@ type NoMiscFees struct {
 	MiscFeeType *string `fix:"139"`
 }
 
+func (m *NoMiscFees) SetMiscFeeAmt(v float64) { m.MiscFeeAmt = &v }
+func (m *NoMiscFees) SetMiscFeeCurr(v string) { m.MiscFeeCurr = &v }
+func (m *NoMiscFees) SetMiscFeeType(v string) { m.MiscFeeType = &v }
+
 //NoAllocs is a repeating group in Allocation
 type NoAllocs struct {
 	//AllocAccount is a required field for NoAllocs.
@@ -66,6 +80,17 @@ type NoAllocs struct {
 	DlvyInst *string `fix:"86"`
 }
 
+func (m *NoAllocs) SetAllocAccount(v string)   { m.AllocAccount = v }
+func (m *NoAllocs) SetAllocShares(v int)       { m.AllocShares = v }
+func (m *NoAllocs) SetProcessCode(v string)    { m.ProcessCode = &v }
+func (m *NoAllocs) SetExecBroker(v string)     { m.ExecBroker = &v }
+func (m *NoAllocs) SetClientID(v string)       { m.ClientID = &v }
+func (m *NoAllocs) SetCommission(v float64)    { m.Commission = &v }
+func (m *NoAllocs) SetCommType(v string)       { m.CommType = &v }
+func (m *NoAllocs) SetNoDlvyInst(v int)        { m.NoDlvyInst = &v }
+func (m *NoAllocs) SetBrokerOfCredit(v string) { m.BrokerOfCredit = &v }
+func (m *NoAllocs) SetDlvyInst(v string)       { m.DlvyInst = &v }
+
 //Message is a Allocation FIX Message
 type Message struct {
 	FIXMsgType string `fix:"J"`
@@ -130,6 +155,34 @@ type Message struct {
 //Marshal converts Message to a quickfix.Message instance
 func (m Message) Marshal() quickfix.Message { return quickfix.Marshal(m) }
 
+func (m *Message) SetAllocID(v int)             { m.AllocID = v }
+func (m *Message) SetAllocTransType(v string)   { m.AllocTransType = v }
+func (m *Message) SetRefAllocID(v int)          { m.RefAllocID = &v }
+func (m *Message) SetNoOrders(v []NoOrders)     { m.NoOrders = v }
+func (m *Message) SetNoExecs(v []NoExecs)       { m.NoExecs = v }
+func (m *Message) SetSide(v string)             { m.Side = v }
+func (m *Message) SetSymbol(v string)           { m.Symbol = v }
+func (m *Message) SetSymbolSfx(v string)        { m.SymbolSfx = &v }
+func (m *Message) SetSecurityID(v string)       { m.SecurityID = &v }
+func (m *Message) SetIDSource(v string)         { m.IDSource = &v }
+func (m *Message) SetIssuer(v string)           { m.Issuer = &v }
+func (m *Message) SetSecurityDesc(v string)     { m.SecurityDesc = &v }
+func (m *Message) SetShares(v int)              { m.Shares = v }
+func (m *Message) SetAvgPx(v float64)           { m.AvgPx = v }
+func (m *Message) SetCurrency(v string)         { m.Currency = &v }
+func (m *Message) SetAvgPrxPrecision(v int)     { m.AvgPrxPrecision = &v }
+func (m *Message) SetTradeDate(v string)        { m.TradeDate = v }
+func (m *Message) SetTransactTime(v time.Time)  { m.TransactTime = &v }
+func (m *Message) SetSettlmntTyp(v string)      { m.SettlmntTyp = &v }
+func (m *Message) SetFutSettDate(v string)      { m.FutSettDate = &v }
+func (m *Message) SetNetMoney(v float64)        { m.NetMoney = &v }
+func (m *Message) SetNoMiscFees(v []NoMiscFees) { m.NoMiscFees = v }
+func (m *Message) SetSettlCurrAmt(v float64)    { m.SettlCurrAmt = &v }
+func (m *Message) SetSettlCurrency(v string)    { m.SettlCurrency = &v }
+func (m *Message) SetOpenClose(v string)        { m.OpenClose = &v }
+func (m *Message) SetText(v string)             { m.Text = &v }
+func (m *Message) SetNoAllocs(v []NoAllocs)     { m.NoAllocs = v }
+
 //A RouteOut is the callback type that should be implemented for routing Message
 type RouteOut func(msg Message, sessionID quickfix.SessionID) quickfix.MessageRejectError
 

fix40/allocationack/AllocationACK.go

@@ -34,6 +34,15 @@ type Message struct {
 //Marshal converts Message to a quickfix.Message instance
 func (m Message) Marshal() quickfix.Message { return quickfix.Marshal(m) }
 
+func (m *Message) SetClientID(v string)        { m.ClientID = &v }
+func (m *Message) SetExecBroker(v string)      { m.ExecBroker = &v }
+func (m *Message) SetAllocID(v int)            { m.AllocID = v }
+func (m *Message) SetTradeDate(v string)       { m.TradeDate = v }
+func (m *Message) SetTransactTime(v time.Time) { m.TransactTime = &v }
+func (m *Message) SetAllocStatus(v int)        { m.AllocStatus = v }
+func (m *Message) SetAllocRejCode(v int)       { m.AllocRejCode = &v }
+func (m *Message) SetText(v string)            { m.Text = &v }
+
 //A RouteOut is the callback type that should be implemented for routing Message
 type RouteOut func(msg Message, sessionID quickfix.SessionID) quickfix.MessageRejectError
 

fix40/dontknowtrade/DontKnowTrade.go

@@ -35,6 +35,16 @@ type Message struct {
 //Marshal converts Message to a quickfix.Message instance
 func (m Message) Marshal() quickfix.Message { return quickfix.Marshal(m) }
 
+func (m *Message) SetOrderID(v string)  { m.OrderID = &v }
+func (m *Message) SetExecID(v int)      { m.ExecID = &v }
+func (m *Message) SetDKReason(v string) { m.DKReason = v }
+func (m *Message) SetSymbol(v string)   { m.Symbol = v }
+func (m *Message) SetSide(v string)     { m.Side = v }
+func (m *Message) SetOrderQty(v int)    { m.OrderQty = v }
+func (m *Message) SetLastShares(v int)  { m.LastShares = v }
+func (m *Message) SetLastPx(v float64)  { m.LastPx = v }
+func (m *Message) SetText(v string)     { m.Text = &v }
+
 //A RouteOut is the callback type that should be implemented for routing Message
 type RouteOut func(msg Message, sessionID quickfix.SessionID) quickfix.MessageRejectError
 

fix40/email/Email.go

@@ -36,6 +36,16 @@ type Message struct {
 //Marshal converts Message to a quickfix.Message instance
 func (m Message) Marshal() quickfix.Message { return quickfix.Marshal(m) }
 
+func (m *Message) SetEmailType(v string)   { m.EmailType = v }
+func (m *Message) SetOrigTime(v time.Time) { m.OrigTime = &v }
+func (m *Message) SetRelatdSym(v string)   { m.RelatdSym = &v }
+func (m *Message) SetOrderID(v string)     { m.OrderID = &v }
+func (m *Message) SetClOrdID(v string)     { m.ClOrdID = &v }
+func (m *Message) SetLinesOfText(v int)    { m.LinesOfText = v }
+func (m *Message) SetText(v string)        { m.Text = v }
+func (m *Message) SetRawDataLength(v int)  { m.RawDataLength = &v }
+func (m *Message) SetRawData(v string)     { m.RawData = &v }
+
 //A RouteOut is the callback type that should be implemented for routing Message
 type RouteOut func(msg Message, sessionID quickfix.SessionID) quickfix.MessageRejectError