blob: 31e97779dc3afa3c72d776877e8678f7182a6ded [file] [log] [blame]
// Copyright 2022 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package worker
import (
"fmt"
"net/http"
"time"
"cloud.google.com/go/civil"
"golang.org/x/exp/event"
"golang.org/x/pkgsite-metrics/internal/bigquery"
"golang.org/x/pkgsite-metrics/internal/derrors"
"golang.org/x/pkgsite-metrics/internal/log"
"golang.org/x/pkgsite-metrics/internal/scan"
)
var insertResultsCounter = event.NewCounter("insert-results", &event.MetricOptions{Namespace: metricNamespace})
func (h *VulncheckServer) handleInsertResults(w http.ResponseWriter, r *http.Request) (err error) {
defer func() {
insertResultsCounter.Record(r.Context(), 1, event.Bool("success", err == nil))
}()
if h.bqClient == nil {
return errBQDisabled
}
var date civil.Date
if d := r.FormValue("date"); d != "" {
var err error
date, err = civil.ParseDate(d)
if err != nil {
return fmt.Errorf("%w: parsing 'date' query param: %v", derrors.InvalidArgument, err)
}
} else {
date = civil.DateOf(time.Now())
}
allowDuplicates, err := scan.ParseOptionalBoolParam(r, "allow-dups", false)
if err != nil {
return fmt.Errorf("%w: %v", derrors.InvalidArgument, err)
}
ctx := r.Context()
log.Infof(ctx, "reading results")
results, err := bigquery.FetchVulncheckResults(ctx, h.bqClient)
if err != nil {
return err
}
log.Infof(ctx, "inserting %d results for %s", len(results), date)
if err := bigquery.InsertVulncheckResults(ctx, h.bqClient, results, date, allowDuplicates); err != nil {
return err
}
fmt.Fprintf(w, "%d results for %s inserted successfully.\n", len(results), date)
return nil
}