X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/16f704326f44fd1e5e5e60b936c9b5895d6a6ff8..f690cb6af6d36a5105a846dd69e0fd624b4e7f0e:/lib/dispatchcloud/cmd.go diff --git a/lib/dispatchcloud/cmd.go b/lib/dispatchcloud/cmd.go index 7231e83947..0254c6526c 100644 --- a/lib/dispatchcloud/cmd.go +++ b/lib/dispatchcloud/cmd.go @@ -6,16 +6,28 @@ package dispatchcloud import ( "context" + "fmt" - "git.curoverse.com/arvados.git/lib/cmd" - "git.curoverse.com/arvados.git/lib/service" - "git.curoverse.com/arvados.git/sdk/go/arvados" + "git.arvados.org/arvados.git/lib/cmd" + "git.arvados.org/arvados.git/lib/service" + "git.arvados.org/arvados.git/sdk/go/arvados" + "github.com/prometheus/client_golang/prometheus" ) var Command cmd.Handler = service.Command(arvados.ServiceNameDispatchCloud, newHandler) -func newHandler(ctx context.Context, cluster *arvados.Cluster, _ *arvados.NodeProfile) service.Handler { - d := &dispatcher{Cluster: cluster, Context: ctx} +func newHandler(ctx context.Context, cluster *arvados.Cluster, token string, reg *prometheus.Registry) service.Handler { + ac, err := arvados.NewClientFromConfig(cluster) + if err != nil { + return service.ErrorHandler(ctx, cluster, fmt.Errorf("error initializing client from cluster config: %s", err)) + } + d := &dispatcher{ + Cluster: cluster, + Context: ctx, + ArvClient: ac, + AuthToken: token, + Registry: reg, + } go d.Start() return d }