support diagnose custom args

alibaba · Jan 25, 2024 · fc37eb8 · fc37eb8
1 parent d012db3
commit fc37eb8
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/pkg/controller/diagnose/diagnose.go b/pkg/controller/diagnose/diagnose.go
@@ -15,14 +15,20 @@ type Controller interface {
 }
 
 func NewDiagnoseController(namespace string) Controller {
+	diagnoseArgs := ""
+	if diagnoseArgsFromEnv, ok := os.LookupEnv("KUBESKOOP_DIAGNOSE_ARGS"); ok {
+		diagnoseArgs = diagnoseArgsFromEnv
+	}
 	// 1. build skoop global context
 	return &Diagnostor{
 		namespace: namespace,
+		diagnoseArgs: diagnoseArgs,
 	}
 }
 
 type Diagnostor struct {
-	namespace string
+	namespace    string
+	diagnoseArgs string
 }
 
 func (d *Diagnostor) Diagnose(ctx context.Context, taskConfig *skoopContext.TaskConfig) (string, error) {
@@ -39,6 +45,7 @@ func (d *Diagnostor) Diagnose(ctx context.Context, taskConfig *skoopContext.Task
 		"--dport", strconv.FormatUint(uint64(taskConfig.Destination.Port), 10),
 		"--protocol", taskConfig.Protocol,
 		"--collector-namespace", d.namespace,
+		d.diagnoseArgs,
 	)
 	output, err := cmd.CombinedOutput()
 	if err != nil {