diff options
| author | Taras Madan <tarasmadan@google.com> | 2024-07-15 08:16:51 +0200 |
|---|---|---|
| committer | Taras Madan <tarasmadan@google.com> | 2024-07-15 20:31:23 +0000 |
| commit | 1b8d04d0f39d8a579d6797de8364c0b3c0614b0a (patch) | |
| tree | a5b7f9f63c75f48ab303c984280f219bf5714f87 | |
| parent | 1e9b4efa8053e3d27be8a5803c91bf37923ffe3b (diff) | |
syz-cover: make project-id configurable, defaults to syzkaller
| -rw-r--r-- | pkg/cover/heatmap.go | 8 | ||||
| -rw-r--r-- | tools/syz-cover/syz-cover.go | 3 |
2 files changed, 6 insertions, 5 deletions
diff --git a/pkg/cover/heatmap.go b/pkg/cover/heatmap.go index 9cc3286e5..7745a1d49 100644 --- a/pkg/cover/heatmap.go +++ b/pkg/cover/heatmap.go @@ -119,9 +119,9 @@ func filesCoverageToTemplateData(fCov []*fileCoverageAndDate) *templateHeatmap { return &res } -func filesCoverageAndDates(ctx context.Context, ns string, fromDate, toDate civil.Date, +func filesCoverageAndDates(ctx context.Context, projectID, ns string, fromDate, toDate civil.Date, ) ([]*fileCoverageAndDate, error) { - client, err := coveragedb.NewClient(ctx, "syzkaller") + client, err := coveragedb.NewClient(ctx, projectID) if err != nil { return nil, fmt.Errorf("spanner.NewClient() failed: %s", err.Error()) } @@ -165,8 +165,8 @@ where namespace=$1 and dateto>=$2 and dateto<=$3 return res, nil } -func DoHeatMap(w io.Writer, ns string, dateFrom, dateTo civil.Date) error { - covAndDates, err := filesCoverageAndDates(context.Background(), ns, dateFrom, dateTo) +func DoHeatMap(w io.Writer, projectID, ns string, dateFrom, dateTo civil.Date) error { + covAndDates, err := filesCoverageAndDates(context.Background(), projectID, ns, dateFrom, dateTo) if err != nil { panic(err) } diff --git a/tools/syz-cover/syz-cover.go b/tools/syz-cover/syz-cover.go index bf1c02ae4..14de498ad 100644 --- a/tools/syz-cover/syz-cover.go +++ b/tools/syz-cover/syz-cover.go @@ -52,6 +52,7 @@ var ( civil.DateOf(time.Now().Add(-14*24*time.Hour)).String(), "heatmap date from(optional)") flagDateTo = flag.String("to", civil.DateOf(time.Now()).String(), "heatmap date to(optional)") + flagProjectID = flag.String("project", "syzkaller", "spanner db project name") ) func toolBuildNsHeatmap() { @@ -64,7 +65,7 @@ func toolBuildNsHeatmap() { if dateTo, err = civil.ParseDate(*flagDateTo); err != nil { tool.Failf("failed to parse date to: %v", err.Error()) } - if err = cover.DoHeatMap(buf, *flagNsHeatmap, dateFrom, dateTo); err != nil { + if err = cover.DoHeatMap(buf, *flagProjectID, *flagNsHeatmap, dateFrom, dateTo); err != nil { tool.Fail(err) } if err = osutil.WriteFile(*flagNsHeatmap+".html", buf.Bytes()); err != nil { |