添加参数判断

bench/main.go

@@ -17,23 +17,23 @@ Available commands:
 `
 
 func main() {
-	if len(os.Args) > 1 {
-		dtmimp.Logf("starting dtm....")
-		if os.Args[1] == "http" {
-			fmt.Println("start bench server")
-			common.MustLoadConfig()
-			dtmcli.SetCurrentDBType(common.DtmConfig.DB["driver"])
-			common.WaitDBUp()
-			dtmsvr.PopulateDB(true)
-			examples.PopulateDB(true)
-			dtmsvr.StartSvr()              // 启动dtmsvr的api服务
-			go dtmsvr.CronExpiredTrans(-1) // 启动dtmsvr的定时过期查询
-			StartSvr()
-			select {}
-		}else {
-			fmt.Printf(hint)
-		}
-	}else {
+	if len(os.Args) <= 1 {
+		fmt.Printf(hint)
+		return
+	}
+	dtmimp.Logf("starting dtm....")
+	if os.Args[1] == "http" {
+		fmt.Println("start bench server")
+		common.MustLoadConfig()
+		dtmcli.SetCurrentDBType(common.DtmConfig.DB["driver"])
+		common.WaitDBUp()
+		dtmsvr.PopulateDB(true)
+		examples.PopulateDB(true)
+		dtmsvr.StartSvr()              // 启动dtmsvr的api服务
+		go dtmsvr.CronExpiredTrans(-1) // 启动dtmsvr的定时过期查询
+		StartSvr()
+		select {}
+	} else {
 		fmt.Printf(hint)
 	}
 }