From cca672e96d5a3f4a3b2516249f50a0ca5eb9667c Mon Sep 17 00:00:00 2001
From: Quang Nguyen <nguyenquang@microsoft.com>
Date: Tue, 19 Nov 2024 22:19:25 -0500
Subject: [PATCH] fix: toolbox build

---
 hack/tools/toolbox/go.mod  |  2 ++
 hack/tools/toolbox/go.sum  |  2 ++
 hack/tools/toolbox/main.go | 12 ++++++++++--
 3 files changed, 14 insertions(+), 2 deletions(-)
 create mode 100644 hack/tools/toolbox/go.sum

diff --git a/hack/tools/toolbox/go.mod b/hack/tools/toolbox/go.mod
index 31c4a5b76d..4c6c44f37a 100644
--- a/hack/tools/toolbox/go.mod
+++ b/hack/tools/toolbox/go.mod
@@ -5,3 +5,5 @@ go 1.22.5
 replace github.com/microsoft/retina/hack/tools/toolbox/kapinger => ../kapinger
 
 require github.com/microsoft/retina/hack/tools/toolbox/kapinger v0.0.0-00010101000000-000000000000
+
+require github.com/microsoft/retina/hack/tools/kapinger v0.0.0-20241119173012-8db4de4752fc
diff --git a/hack/tools/toolbox/go.sum b/hack/tools/toolbox/go.sum
new file mode 100644
index 0000000000..b6662d3f64
--- /dev/null
+++ b/hack/tools/toolbox/go.sum
@@ -0,0 +1,2 @@
+github.com/microsoft/retina/hack/tools/kapinger v0.0.0-20241119173012-8db4de4752fc h1:W1HBgyk9NNNQTyW2U7uC2TNr9/VfJMZ+uxFacv0LNQU=
+github.com/microsoft/retina/hack/tools/kapinger v0.0.0-20241119173012-8db4de4752fc/go.mod h1:mdq2Z9SePF3lj4VRJV6cfzuSFfp0z7a/zlX2SsVtou4=
diff --git a/hack/tools/toolbox/main.go b/hack/tools/toolbox/main.go
index 4cca6b50dc..f235f7490e 100644
--- a/hack/tools/toolbox/main.go
+++ b/hack/tools/toolbox/main.go
@@ -1,7 +1,15 @@
 package main
 
-import "github.com/microsoft/retina/hack/tools/toolbox/kapinger/servers"
+import (
+	"context"
+
+	"github.com/microsoft/retina/hack/tools/kapinger/config"
+	"github.com/microsoft/retina/hack/tools/toolbox/kapinger/servers"
+)
 
 func main() {
-	servers.StartAll()
+	cfg := config.LoadConfigFromEnv()
+
+	ctx := context.Background()
+	servers.StartAll(ctx, cfg)
 }