Support get default value for proc via env

Signed-off-by: Viet Anh Duong <anh.duong@kyber.network>
vietanhduong · Dec 30, 2023 · 7b83058 · 7b83058
1 parent bdb312b
commit 7b83058
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pkg/proc/proc.go b/pkg/proc/proc.go
@@ -8,13 +8,14 @@ import (
 	"github.com/sirupsen/logrus"
 	"github.com/vietanhduong/wbpf/pkg/logging"
 	"github.com/vietanhduong/wbpf/pkg/logging/logfields"
+	"github.com/vietanhduong/wbpf/pkg/utils"
 )
 
 var log = logging.DefaultLogger.WithFields(logrus.Fields{logfields.LogSubsys: "proc"})
 
 var (
-	procPath = flag.String("proc-path", "/proc", "Path to proc directory")
-	hostPath = flag.String("host-path", "/", "The host directory. Useful in container.")
+	procPath = flag.String("proc-path", utils.GetEnvOrDefault("PROC_PATH", "/proc"), "Path to proc directory")
+	hostPath = flag.String("host-path", utils.GetEnvOrDefault("HOST_PATH", "/"), "The host directory. Useful in container.")
 )
 
 func ProcPath(paths ...string) string {