gofmt generated file

1 files changed, 12 insertions, 2 deletions

diff --git a/generate_dns_types.go b/generate_dns_types.go
index 89ba251..4e2aade 100644
--- a/generate_dns_types.go
+++ b/generate_dns_types.go
@@ -1,9 +1,11 @@
-//+build ignore
+//go:build ignore
 
 package main
 
 import (
+        "bytes"
         "fmt"
+        "go/format"
         "go/types"
         "log"
         "os"
@@ -182,10 +184,18 @@ func main() {
                 localTypes[name] = fields
         }
 
+        // gofmt it!
+        buf := bytes.Buffer{}
+        tpl.Execute(&buf, localTypes)
+        src, err := format.Source(buf.Bytes())
+        if err != nil {
+                panic(err)
+        }
+
         fp, err := os.Create("zzz_types.go")
         if err != nil {
                 panic(err)
         }
         defer fp.Close()
-        tpl.Execute(fp, localTypes)
+        fp.Write(src)
 }