times.go: Add test query filter

Usage: `./tools/run cts time -query QUERY ...`

Change-Id: Ifac16ee2405d1b5d03d246bf1556245e2dafe8c1
Reviewed-on: https://dawn-review.googlesource.com/c/dawn/+/93304
Commit-Queue: Austin Eng <enga@chromium.org>
Reviewed-by: Ben Clayton <bclayton@google.com>
This commit is contained in:
Austin Eng 2022-06-10 17:24:23 +00:00 committed by Dawn LUCI CQ
parent 09b79e8950
commit 737ff5b482
2 changed files with 16 additions and 0 deletions

View File

@ -38,6 +38,7 @@ type cmd struct {
source common.ResultSource source common.ResultSource
auth authcli.Flags auth authcli.Flags
tags string tags string
query string
aggregate bool aggregate bool
topN int topN int
histogram bool histogram bool
@ -57,6 +58,7 @@ func (c *cmd) RegisterFlags(ctx context.Context, cfg common.Config) ([]string, e
c.flags.auth.Register(flag.CommandLine, common.DefaultAuthOptions()) c.flags.auth.Register(flag.CommandLine, common.DefaultAuthOptions())
flag.IntVar(&c.flags.topN, "top", 0, "print the top N slowest tests") flag.IntVar(&c.flags.topN, "top", 0, "print the top N slowest tests")
flag.BoolVar(&c.flags.histogram, "histogram", false, "print a histogram of test timings") flag.BoolVar(&c.flags.histogram, "histogram", false, "print a histogram of test timings")
flag.StringVar(&c.flags.query, "query", "", "test query to filter results")
flag.StringVar(&c.flags.tags, "tags", "", "comma-separated list of tags to filter results") flag.StringVar(&c.flags.tags, "tags", "", "comma-separated list of tags to filter results")
flag.BoolVar(&c.flags.aggregate, "aggregate", false, "aggregate times by test") flag.BoolVar(&c.flags.aggregate, "aggregate", false, "aggregate times by test")
return nil, nil return nil, nil
@ -87,6 +89,13 @@ func (c *cmd) Run(ctx context.Context, cfg common.Config) error {
} }
} }
if c.flags.query != "" {
results = results.FilterByQuery(query.Parse(c.flags.query))
if len(results) == 0 {
return fmt.Errorf("no results after filtering by test query")
}
}
if c.flags.aggregate { if c.flags.aggregate {
type Key struct { type Key struct {
Query query.Query Query query.Query

View File

@ -270,6 +270,13 @@ func (l List) FilterByVariant(tags Tags) List {
}) })
} }
/// FilterByQuery returns the results that match the given query
func (l List) FilterByQuery(q query.Query) List {
return l.Filter(func(r Result) bool {
return q.Contains(r.Query)
})
}
// Statuses is a set of Status // Statuses is a set of Status
type Statuses = container.Set[Status] type Statuses = container.Set[Status]