Write out unions

prashantv · prashantv · commit 1f45dcc45462 · 2015-09-10T11:39:17.000-07:00
Unions are the same as structs, so reuse writeStruct
diff --git a/generator/go.go b/generator/go.go
@@ -649,6 +649,13 @@ func (g *GoGenerator) generateSingle(out io.Writer, thriftPath string, thrift *p
 		}
 	}
 
+	for _, k := range sortedKeys(thrift.Unions) {
+		un := thrift.Unions[k]
+		if err := g.writeStruct(out, un); err != nil {
+			g.error(err)
+		}
+	}
+
 	for _, k := range sortedKeys(thrift.Services) {
 		svc := thrift.Services[k]
 		if err := g.writeService(out, svc); err != nil {

Original file line number	Diff line number	Diff line change
`@@ -649,6 +649,13 @@ func (g GoGenerator) generateSingle(out io.Writer, thriftPath string, thrift p`
`649`	`649`	`}`
`650`	`650`	`}`
`651`	`651`
	`652`	`+ for _, k := range sortedKeys(thrift.Unions) {`
	`653`	`+ un := thrift.Unions[k]`
	`654`	`+ if err := g.writeStruct(out, un); err != nil {`
	`655`	`+ g.error(err)`
	`656`	`+ }`
	`657`	`+ }`
	`658`	`+`
`652`	`659`	`for _, k := range sortedKeys(thrift.Services) {`
`653`	`660`	`svc := thrift.Services[k]`
`654`	`661`	`if err := g.writeService(out, svc); err != nil {`