From f183a58542f5e60907cc513d84d32a69674f511c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=AD=8F=E7=81=BF?= <weican@bd-apaas.com>
Date: Tue, 27 Jun 2023 11:06:35 +0800
Subject: [PATCH] =?UTF-8?q?=E5=88=9D=E5=A7=8B=E5=8C=96ansible=20hosts?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/main.go                          | 16 ++++++++++++++++
 src/router/automatedmaintenrouter.go |  1 +
 2 files changed, 17 insertions(+)

diff --git a/src/main.go b/src/main.go
index 7dce950..9e00fbc 100644
--- a/src/main.go
+++ b/src/main.go
@@ -3,6 +3,7 @@ package main
 import (
 	"fmt"
 	"net/http"
+	"os"
 	"time"
 
 	"github.com/gin-gonic/gin"
@@ -25,6 +26,8 @@ var (
 // main start
 func main() {
 	initTimeZone()
+	// init ansible hosts
+	initAnsibleHosts()
 	pflag.Parse() // init start args
 	initConfig()
 	// init log config
@@ -102,3 +105,16 @@ func initTimeZone() {
 	var cstZone = time.FixedZone("CST", 8*3600) // 固定东八区(CST: China Standard Time)
 	time.Local = cstZone
 }
+
+func initAnsibleHosts() {
+	f, err := os.Create("/etc/ansible/hosts")
+	defer f.Close()
+	if err != nil {
+		fmt.Println(err.Error())
+	} else {
+		_, err := f.Write([]byte("[web]\n"))
+		if err != nil {
+			fmt.Println(err.Error())
+		}
+	}
+}
diff --git a/src/router/automatedmaintenrouter.go b/src/router/automatedmaintenrouter.go
index 15a6f46..59ec8f5 100644
--- a/src/router/automatedmaintenrouter.go
+++ b/src/router/automatedmaintenrouter.go
@@ -18,6 +18,7 @@ func InitAutomatedMaintenRouter(e *gin.Engine) {
 		task.DELETE("/del", controller.DelTaskManage)      // 删除
 		task.GET("/details", controller.DetailsTaskManage) // 详情
 		task.GET("/list", controller.ListTaskManage)       // 列表
+		task.POST("/exec/script")                          // 立即执行
 	}
 	//主机管理
 	host := so.Group("/hostManage")
-- 
2.26.0