From b6fe3714161846eba179fc3ffb8d528eb0fb900b Mon Sep 17 00:00:00 2001
From: Fedor Korotkov <fedor.korotkov@gmail.com>
Date: Fri, 11 Oct 2024 07:44:01 -0400
Subject: [PATCH] allow to set up worker name (#210)

---
 internal/command/worker/run.go | 4 ++++
 internal/worker/option.go      | 6 ++++++
 2 files changed, 10 insertions(+)

diff --git a/internal/command/worker/run.go b/internal/command/worker/run.go
index 2ecf54d..d7cee80 100644
--- a/internal/command/worker/run.go
+++ b/internal/command/worker/run.go
@@ -23,6 +23,7 @@ var (
 	ErrEmptyBootstrapTokenProvided = errors.New("empty bootstrap token was provided")
 )
 
+var name string
 var bootstrapTokenRaw string
 var bootstrapTokenStdin bool
 var logFilePath string
@@ -38,6 +39,8 @@ func newRunCommand() *cobra.Command {
 		Args:  cobra.ExactArgs(1),
 	}
 
+	cmd.PersistentFlags().StringVar(&name, "name", "",
+		"name of the worker (defaults to the hostname)")
 	cmd.PersistentFlags().StringVar(&bootstrapTokenRaw, "bootstrap-token", "",
 		"a bootstrap token retrieved via \"orchard get bootstrap-token <service-account-name-for-workers>\"")
 	cmd.PersistentFlags().BoolVar(&bootstrapTokenStdin, "bootstrap-token-stdin", false,
@@ -111,6 +114,7 @@ func runWorker(cmd *cobra.Command, args []string) (err error) {
 
 	workerInstance, err := worker.New(
 		controllerClient,
+		worker.WithName(name),
 		worker.WithResources(resources),
 		worker.WithLogger(logger),
 	)
diff --git a/internal/worker/option.go b/internal/worker/option.go
index 4ea85d0..d16cb5b 100644
--- a/internal/worker/option.go
+++ b/internal/worker/option.go
@@ -7,6 +7,12 @@ import (
 
 type Option func(*Worker)
 
+func WithName(name string) Option {
+	return func(worker *Worker) {
+		worker.name = name
+	}
+}
+
 func WithResources(resources v1.Resources) Option {
 	return func(worker *Worker) {
 		worker.resources = resources