diff options
| author | Kevin Krakauer <krakauer@google.com> | 2020-02-05 14:43:11 -0800 |
|---|---|---|
| committer | Kevin Krakauer <krakauer@google.com> | 2020-02-05 14:43:11 -0800 |
| commit | bf0ea204e9415a181c63ee10078cca753df14f7e (patch) | |
| tree | f4d149ac2bf2ea900f8b2a0cdea0a7a2ea7e6d0e /tools/go_marshal/gomarshal | |
| parent | 29ad5762e4549d961f48c65292cfdeb7256524f6 (diff) | |
| parent | f2d3efca1deded31a2929ea77c0eecf476764660 (diff) | |
Merge branch 'master' into tcp-matchers-submit
Diffstat (limited to 'tools/go_marshal/gomarshal')
| -rw-r--r-- | tools/go_marshal/gomarshal/BUILD | 1 | ||||
| -rw-r--r-- | tools/go_marshal/gomarshal/generator.go | 11 |
2 files changed, 12 insertions, 0 deletions
diff --git a/tools/go_marshal/gomarshal/BUILD b/tools/go_marshal/gomarshal/BUILD index c92b59dd6..b5d5a4487 100644 --- a/tools/go_marshal/gomarshal/BUILD +++ b/tools/go_marshal/gomarshal/BUILD @@ -14,4 +14,5 @@ go_library( visibility = [ "//:sandbox", ], + deps = ["//tools/tags"], ) diff --git a/tools/go_marshal/gomarshal/generator.go b/tools/go_marshal/gomarshal/generator.go index af90bdecb..0b3f600fe 100644 --- a/tools/go_marshal/gomarshal/generator.go +++ b/tools/go_marshal/gomarshal/generator.go @@ -23,6 +23,9 @@ import ( "go/token" "os" "sort" + "strings" + + "gvisor.dev/gvisor/tools/tags" ) const ( @@ -104,6 +107,14 @@ func NewGenerator(srcs []string, out, outTest, pkg string, imports []string) (*G func (g *Generator) writeHeader() error { var b sourceBuffer b.emit("// Automatically generated marshal implementation. See tools/go_marshal.\n\n") + + // Emit build tags. + if t := tags.Aggregate(g.inputs); len(t) > 0 { + b.emit(strings.Join(t.Lines(), "\n")) + b.emit("\n") + } + + // Package header. b.emit("package %s\n\n", g.pkg) if err := b.write(g.output); err != nil { return err |