2 files changed, 17 insertions, 5 deletions
diff --git a/tools/go_generics/go_merge/BUILD b/tools/go_generics/go_merge/BUILD
index 2fd5a200d..5e0487e93 100644
--- a/tools/go_generics/go_merge/BUILD
+++ b/tools/go_generics/go_merge/BUILD
@@ -6,4 +6,7 @@ go_binary(
     name = "go_merge",
     srcs = ["main.go"],
     visibility = ["//:sandbox"],
+    deps = [
+        "//tools/tags",
+    ],
 )
diff --git a/tools/go_generics/go_merge/main.go b/tools/go_generics/go_merge/main.go
index e0345500f..801f2354f 100644
--- a/tools/go_generics/go_merge/main.go
+++ b/tools/go_generics/go_merge/main.go
@@ -22,10 +22,12 @@ import (
 	"go/format"
 	"go/parser"
 	"go/token"
-	"io/ioutil"
 	"os"
 	"path/filepath"
 	"strconv"
+	"strings"
+
+	"gvisor.dev/gvisor/tools/tags"
 )
 
 var (
@@ -132,10 +134,17 @@ func main() {
 	// Write the output file.
 	var buf bytes.Buffer
 	if err := format.Node(&buf, fset, f); err != nil {
-		fatalf("%v\n", err)
+		fatalf("fomatting: %v\n", err)
 	}
-
-	if err := ioutil.WriteFile(*output, buf.Bytes(), 0644); err != nil {
-		fatalf("%v\n", err)
+	outf, err := os.OpenFile(*output, os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0644)
+	if err != nil {
+		fatalf("opening output: %v\n", err)
+	}
+	defer outf.Close()
+	if t := tags.Aggregate(flag.Args()); len(t) > 0 {
+		fmt.Fprintf(outf, "%s\n\n", strings.Join(t.Lines(), "\n"))
+	}
+	if _, err := outf.Write(buf.Bytes()); err != nil {
+		fatalf("write: %v\n", err)
 	}
 }