[ Cmii ] [ Octopus ] - project a lot

agent-go/main.go

@@ -15,9 +15,15 @@ func main() {
 	// 解析命令行参数
 	var version string
 	var agentServerInfoConf string
+	var mode string
 	flag.StringVar(&version, "version", "", "config file version")
+	flag.StringVar(&mode, "mode", "agent", "agent run mode")
 	flag.StringVar(&agentServerInfoConf, "agentServerInfoConf", "", "agent server info conf file")
 	flag.Parse()
+	if mode == "bastion" {
+		BastionModeInit()
+		return
+	}
 	// 读取对应版本的配置文件
 	filename := fmt.Sprintf("octopus-agent-%s.yaml", version)
 	println("config file name is => " + filename)