add check var

openkruise · Oct 8, 2023 · f8e1705 · f8e1705
1 parent 6868bd0
commit f8e1705
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/main.go b/main.go
@@ -87,8 +87,8 @@ func main() {
 	flag.StringVar(&scaleServerAddr, "scale-server-bind-address", ":6000", "The address the scale server endpoint binds to.")
 	viper.SetDefault(apiServerSustainedQPSFlag, 20)
 	viper.SetDefault(apiServerBurstQPSFlag, 30)
-	viper.BindEnv(apiServerSustainedQPSFlag)
-	viper.BindEnv(apiServerBurstQPSFlag)
+	Must(viper.BindEnv(apiServerSustainedQPSFlag))
+	Must(viper.BindEnv(apiServerBurstQPSFlag))
 
 	// Add cloud provider flags
 	cloudprovider.InitCloudProviderFlags()
@@ -215,3 +215,10 @@ func main() {
 		os.Exit(1)
 	}
 }
+
+// Must panics if there is an error
+func Must(err error) {
+	if err != nil {
+		setupLog.Error(err, "viper.BindEnv error")
+	}
+}
diff --git a/pkg/util/runtime.go b/pkg/util/runtime.go