添加参数判断

bench/main.go

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