add command after repair

Author: YZ775

cluster.go

@@ -304,6 +304,8 @@ type RepairOperation struct {
 	RepairSteps           []RepairStep `json:"repair_steps"`
 	HealthCheckCommand    []string     `json:"health_check_command"`
 	CommandTimeoutSeconds *int         `json:"command_timeout_seconds,omitempty"`
+	SuccessCommand        []string     `json:"success_command,omitempty"`
+	SuccessCommandTimeout *int         `json:"success_command_timeout,omitempty"`
 }
 
 type RepairStep struct {
@@ -319,6 +321,7 @@ const DefaultMaxConcurrentRepairs = 1
 const DefaultRepairEvictionTimeoutSeconds = 600
 const DefaultRepairHealthCheckCommandTimeoutSeconds = 30
 const DefaultRepairCommandTimeoutSeconds = 30
+const DefaultRepairSuccessCommandTimeoutSeconds = 30
 
 type Retire struct {
 	OptionalCommand               []string `json:"optional_command,omitempty"`

op/status.go

@@ -613,6 +613,29 @@ func GetRepairQueueStatus(ctx context.Context, inf cke.Infrastructure, n *cke.No
 		}
 		if healthy {
 			rqs.RepairCompleted[entry.Address] = true
+			//execute Success command
+			op, err := entry.GetMatchingRepairOperation(cluster)
+			if err != nil {
+				return cke.RepairQueueStatus{}, err
+			}
+			if op.SuccessCommand != nil {
+				err := func() error {
+					ctx := ctx
+					timeout := cke.DefaultRepairSuccessCommandTimeoutSeconds
+					if op.SuccessCommandTimeout != nil {
+						timeout = *op.SuccessCommandTimeout
+					}
+					if timeout != 0 {
+						var cancel context.CancelFunc
+						ctx, cancel = context.WithTimeout(ctx, time.Second*time.Duration(timeout))
+						defer cancel()
+					}
+					args := append(op.SuccessCommand[1:], entry.Address)
+					command := well.CommandContext(ctx, op.SuccessCommand[0], args...)
+					return command.Run()
+				}()
+				return cke.RepairQueueStatus{}, err
+			}
 		}
 	}