diff --git a/cmd/register.go b/cmd/register.go index 3a7635b..6a6d46b 100644 --- a/cmd/register.go +++ b/cmd/register.go @@ -295,6 +295,7 @@ func doRegister(cfg *config.Config, inputs *registerInputs) error { ) cfg.Runner.Name = inputs.RunnerName cfg.Runner.Token = inputs.Token + cfg.Runner.Labels = inputs.CustomLabels _, err := register.Register(ctx, cfg.Runner) if err != nil { log.WithError(err).Errorln("Cannot register the runner") diff --git a/core/runner.go b/core/runner.go index a948639..6ae15dc 100644 --- a/core/runner.go +++ b/core/runner.go @@ -7,9 +7,10 @@ const ( // Runner struct type Runner struct { - ID int64 `json:"id"` - UUID string `json:"uuid"` - Name string `json:"name"` - Token string `json:"token"` - Address string `json:"address"` + ID int64 `json:"id"` + UUID string `json:"uuid"` + Name string `json:"name"` + Token string `json:"token"` + Address string `json:"address"` + Labels []string `json:"labels"` } diff --git a/register/register.go b/register/register.go index 93ea631..c5556e7 100644 --- a/register/register.go +++ b/register/register.go @@ -47,6 +47,7 @@ func (p *Register) Register(ctx context.Context, cfg config.Runner) (*core.Runne Name: resp.Msg.Runner.Name, Token: resp.Msg.Runner.Token, Address: p.Client.Address(), + Labels: cfg.Labels, } file, err := json.MarshalIndent(data, "", " ")