Merge pull request #4 from xstr1/feature/move-const-to-arg

move cgroupPath to args
groove-x · Feb 11, 2021 · 8af63f7 · 8af63f7
2 parents 0a2f43b + f9a6c01
commit 8af63f7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/main.go b/main.go
@@ -27,6 +27,7 @@ var (
 	git         string
 
 	address      = flag.String("address", ":48900", "address")
+	cgroupPath   = flag.String("cgroup-path", "/system.slice", "path to cgroup")
 	enableDocker = flag.Bool("metrics.docker", false, "docker container metrics")
 )
 
@@ -40,7 +41,7 @@ func main() {
 	sig := make(chan os.Signal, 1)
 	signal.Notify(sig, syscall.SIGTERM, syscall.SIGINT)
 
-	system, err := cgroups.Load(subsystem, cgroups.StaticPath("/system.slice"))
+	system, err := cgroups.Load(subsystem, cgroups.StaticPath(*cgroupPath))
 	if err != nil {
 		log.Fatalf("cgroups load: %s", err)
 	}