From bf1f069e78f1d46912d1695e6d3ff1186dacef5e Mon Sep 17 00:00:00 2001
From: AllenXu93 <151703335+AllenXu93@users.noreply.github.com>
Date: Thu, 25 Jan 2024 14:54:03 +0800
Subject: [PATCH] support diagnose custom args

---
 pkg/controller/diagnose/diagnose.go | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/pkg/controller/diagnose/diagnose.go b/pkg/controller/diagnose/diagnose.go
index 1f8d9635..059e72d8 100644
--- 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,
+		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 {