container_images/concourse-metrics/cmd/publish-coverage/main.go

// Copyright 2021 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package main

import (
	"context"
	"flag"
	"fmt"
	"os"

	monitoring "cloud.google.com/go/monitoring/apiv3"
	"github.com/GoogleCloudPlatform/guest-test-infra/container_images/concourse-metrics/pkg/requests"
)

var (
	metricPath      = flag.String("metric-path", "", "Path of the custom metric name to use (custom.googleapis.com/[metric-path]).")
	packageName     = flag.String("package-name", "", "Name of package.")
	coveragePercent = flag.Int64("coverage", 0, "Line coverage value.")
	projectID       = flag.String("project-id", "", "GCP project Id.")
)

func main() {
	ctx := context.Background()

	flag.Parse()

	c, err := monitoring.NewMetricClient(ctx)
	if err != nil {
		fmt.Printf("Error creating a new Cloud Monitoring metric client: %+v.\n", err)
		os.Exit(1)
	}
	defer c.Close()

	req, err := requests.BuildCoverageRequest(requests.CoverageArgs{
		CoveragePercent: *coveragePercent,
		MetricPath:      *metricPath,
		PackageName:     *packageName,
		ProjectID:       *projectID,
	})
	if err != nil {
		fmt.Printf("Error creating request: %+v.\n", err)
		os.Exit(1)
	}

	err = c.CreateTimeSeries(ctx, req)
	if err != nil {
		fmt.Printf("Failed to write time series data: %+v.\n", err)
		os.Exit(1)
	}
}