X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/c3c538444c15e68e96780f157935f2baa4ba0bc5..f07c059fca954e4d001cbf1cb36c845be9d884dd:/lib/dispatchcloud/cmd.go diff --git a/lib/dispatchcloud/cmd.go b/lib/dispatchcloud/cmd.go index a5a11d2fa3..0254c6526c 100644 --- a/lib/dispatchcloud/cmd.go +++ b/lib/dispatchcloud/cmd.go @@ -5,13 +5,29 @@ package dispatchcloud import ( - "git.curoverse.com/arvados.git/lib/cmd" - "git.curoverse.com/arvados.git/lib/service" - "git.curoverse.com/arvados.git/sdk/go/arvados" + "context" + "fmt" + + "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(cluster *arvados.Cluster, _ *arvados.NodeProfile) service.Handler { - return &dispatcher{Cluster: cluster} +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 }