cob: Add -tags support for ad-hoc build tags. (#5)

Signed-off-by: Simarpreet Singh <[email protected]>

config.go

@@ -9,6 +9,7 @@ type config struct {
 	bench          string
 	benchmem       bool
 	benchtime      string
+	tags           string
 }
 
 func newConfig(c *cli.Context) config {
@@ -19,5 +20,6 @@ func newConfig(c *cli.Context) config {
 		bench:          c.String("bench"),
 		benchmem:       c.Bool("benchmem"),
 		benchtime:      c.String("benchtime"),
+		tags:           c.String("tags"),
 	}
 }

main.go

@@ -54,6 +54,10 @@ func main() {
 				Usage: "Run enough iterations of each benchmark to take t, specified as a time.Duration (for example, -benchtime 1h30s).",
 				Value: "1s",
 			},
+			&cli.StringFlag{
+				Name:  "tags",
+				Usage: "Run only those benchmarks with the specified build tags.",
+			},
 		},
 	}
 
@@ -167,6 +171,9 @@ func prepareBenchArgs(c config) []string {
 	if c.benchmem {
 		args = append(args, "-benchmem")
 	}
+	if c.tags != "" {
+		args = append(args, "-tags", c.tags)
+	}
 	args = append(args, c.args...)
 	return args
 }