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"
	"golang.org/x/oauth2/google"
)

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

func DialSwarming(ctx context.Context) (swarming.ApiClient, 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, err := swarming.NewApiClient(tracedClient, swarmingServiceAddress)

	return ret, err
}