cabe/go/backends/swarming.go

package backends

import (
	"context"
	"net/http"

	"go.opencensus.io/plugin/ochttp"
	"go.skia.org/infra/go/auth"
	"go.skia.org/infra/go/httputils"
	"go.skia.org/infra/go/sklog"
	"go.skia.org/infra/go/swarming"
	swarmingv2 "go.skia.org/infra/go/swarming/v2"
	"golang.org/x/oauth2/google"
)

const (
	swarmingServiceAddress = "chrome-swarming.appspot.com:443"
)

func DialSwarming(ctx context.Context) (swarmingv2.SwarmingV2Client, error) {
	// Create authenticated HTTP client.
	httpClientTokenSource, err := google.DefaultTokenSource(ctx, auth.ScopeReadOnly, swarming.AUTH_SCOPE)
	if err != nil {
		sklog.Fatalf("Problem setting up default token source: %s", err)
	}

	client := httputils.DefaultClientConfig().WithTokenSource(httpClientTokenSource).With2xxOnly().Client()
	tracedClient := &http.Client{Transport: &ochttp.Transport{
		Base: client.Transport,
	}}

	ret := swarmingv2.NewDefaultClient(tracedClient, swarmingServiceAddress)
	return ret, nil
}