Skip to content

S
so-operation-api
Commit 743b94a7 authored by 陈子龙's avatar 陈子龙
Browse files
Options

Merge branch 'dev-czl' into dev

parents 3a56f2f8 0966d83f
Hide whitespace changes
Inline Side-by-side
Showing with 100 additions and 27 deletions
src/bean/vo/request/task_manage.go
View file @ 743b94a7
......@@ -36,5 +36,5 @@ type ExecScriptReq struct {
HostGroupId int `form:"host_group_id" vd:"$>0;msg:'请输入主机分组id'"` //主机分组id
Type int `form:"type"` //脚本额外变量类型1yaml 2json
Value string `form:"value"` //脚本额外变量值
Script string `form:"script"` //执行脚本
Script string `form:"script"` //执行脚本/文件名
}
src/common/tools/hosts
View file @ 743b94a7
[HostGroup]
HostIp192.168.0.1 ansible_ssh_host=192.168.0.1 ansible_ssh_port=22 ansible_ssh_user="admin" ansible_ssh_pass="123456"
HostIp192.186.10.1 ansible_ssh_host=192.186.10.1 ansible_ssh_port=22 ansible_ssh_user="admin" ansible_ssh_pass="123456"
HostIp101.168.0.1 ansible_ssh_host=101.168.0.1 ansible_ssh_port=22 ansible_ssh_user="root" ansible_ssh_pass=""
HostIp101.186.10.1 ansible_ssh_host=101.186.10.1 ansible_ssh_port=22 ansible_ssh_user="root" ansible_ssh_pass=""
[webGroup1]
192.168.0.11 ansible_ssh_host=192.168.0.11 ansible_ssh_user=root ansible_ssh_pass=123.com ansible_ssh_port=3333
192.168.0.12 ansible_ssh_host=192.168.0.12 ansible_ssh_user=root ansible_ssh_pass=1234.com ansible_ssh_port=2222
[webGroup2]
192.168.1.11 ansible_ssh_host=192.168.1.11 ansible_ssh_user=admin ansible_ssh_pass=ccc.com ansible_ssh_port=11
192.168.1.12 ansible_ssh_host=192.168.1.12 ansible_ssh_user=admin ansible_ssh_pass=111.com ansible_ssh_port=22
\ No newline at end of file
192.168.1.12 ansible_ssh_host=192.168.1.12 ansible_ssh_user=admin ansible_ssh_pass=111.com ansible_ssh_port=22
[HostGroup15]
HostIp22 ansible_ssh_host=192.168.0.1 ansible_ssh_port=22 ansible_ssh_user="admin" ansible_ssh_pass="123456"
HostIp23 ansible_ssh_host=192.186.10.1 ansible_ssh_port=22 ansible_ssh_user="admin" ansible_ssh_pass="123456"
HostIp24 ansible_ssh_host=101.168.0.1 ansible_ssh_port=22 ansible_ssh_user="root" ansible_ssh_pass=""
HostIp25 ansible_ssh_host=101.186.10.1 ansible_ssh_port=22 ansible_ssh_user="root" ansible_ssh_pass=""
src/common/tools/hosts.go
View file @ 743b94a7
......@@ -8,7 +8,9 @@ import (
)
func HostsToJson() (data map[string][]string, err error) {
f, err := os.Open(`/etc/ansible/hosts`)
//f, err := os.Open(`/etc/ansible/hosts`)
f, err := os.Open(`D:\work\goWork\智能运维平台\so-operation-api\src\common\tools\hosts`)
if err != nil {
return nil, err
}
......@@ -35,13 +37,15 @@ func HostsToJson() (data map[string][]string, err error) {
}
func MapToSaveHosts(data map[string][]string) error {
f, err := os.OpenFile("/etc/ansible/hosts", os.O_APPEND|os.O_CREATE|os.O_RDWR|os.O_TRUNC, 0777)
//f, err := os.OpenFile("/etc/ansible/hosts", os.O_APPEND|os.O_CREATE|os.O_RDWR|os.O_TRUNC, 0777)
f, err := os.OpenFile("D:/work/goWork/智能运维平台/so-operation-api/src/common/tools/hosts", os.O_APPEND|os.O_CREATE|os.O_RDWR|os.O_TRUNC, 0777)
if err != nil {
return err
}
defer f.Close()
for k, v := range data {
_, err := f.Write([]byte(fmt.Sprintf("%s\n", k)))
_, err = f.Write([]byte(fmt.Sprintf("%s\n", k)))
if err != nil {
return err
}
......@@ -49,6 +53,10 @@ func MapToSaveHosts(data map[string][]string) error {
if err != nil {
return err
}
_, err = f.Write([]byte("\n"))
if err != nil {
return err
}
}
return nil
}
src/controller/task_manage.go
View file @ 743b94a7
......@@ -123,6 +123,7 @@ func ExecScript(c *gin.Context) {
SendJsonResponse(c, resp.InvalidParam.WithError(err), nil)
return
}
taskManageSvc := service.TaskManageSvc{}
data, err := taskManageSvc.ExecScript(req)
if err != nil {
......
src/service/host_manage.go
View file @ 743b94a7
......@@ -11,8 +11,8 @@ import (
"gitlab.wodcloud.com/smart-operation/so-operation-api/src/bean/vo/response"
"gitlab.wodcloud.com/smart-operation/so-operation-api/src/common/client"
"gitlab.wodcloud.com/smart-operation/so-operation-api/src/common/conf"
"gitlab.wodcloud.com/smart-operation/so-operation-api/src/common/tools"
"gitlab.wodcloud.com/smart-operation/so-operation-api/src/pkg/beagle/resp"
"os"
"os/exec"
"strings"
"time"
......@@ -22,7 +22,8 @@ type HostManageSvc struct {
User *entity.SystemUser
}
const AnsibleGroup string = "HostManage"
const AnsibleGroup string = "HostGroup"
const AnsibleIp string = "HostIp"
// AddHostManage 新增主机
func (h *HostManageSvc) AddHostManage(req request.AddHostManageReq) (err error) {
......@@ -84,12 +85,22 @@ func (h *HostManageSvc) AddHostManage(req request.AddHostManageReq) (err error)
return
}
//读取hosts中的主机组
hosts, err := tools.HostsToJson()
if err != nil {
err = resp.MarshalError.WithError(err)
session.Rollback()
return
}
hosts["["+AnsibleGroup+fmt.Sprintf("%d", hostManage.Id)+"]"] = nil
//新增主机分组列表
for _, v := range hostManageLists {
port := "22"
if v.Port != "" {
port = v.Port
}
hostManageList := entity.HostManageList{
Ip: v.Ip,
Port: port,
......@@ -106,6 +117,18 @@ func (h *HostManageSvc) AddHostManage(req request.AddHostManageReq) (err error)
session.Rollback()
return
}
hostsIp := fmt.Sprintf("%s ansible_ssh_host=%s ansible_ssh_port=%s ansible_ssh_user=\"%s\" ansible_ssh_pass=\"%s\"",
AnsibleIp+fmt.Sprintf("%d", hostManageList.Id), v.Ip, v.Port, v.UserName, v.Password)
hosts["["+AnsibleGroup+fmt.Sprintf("%d", hostManage.Id)+"]"] = append(hosts["["+AnsibleGroup+fmt.Sprintf("%d", hostManage.Id)+"]"], hostsIp)
}
//写入hosts
err = tools.MapToSaveHosts(hosts)
if err != nil {
err = resp.MarshalError.WithError(err)
session.Rollback()
return
}
session.Commit()
......@@ -166,6 +189,14 @@ func (h *HostManageSvc) EditHostManage(req request.EditHostManageReq) (err error
return
}
//读取hosts中的主机组
hosts, err := tools.HostsToJson()
if err != nil {
err = resp.MarshalError.WithError(err)
return
}
hosts["["+AnsibleGroup+fmt.Sprintf("%d", req.Id)+"]"] = nil
//新增主机分组列表
for _, v := range hostManageLists {
port := "22"
......@@ -188,8 +219,19 @@ func (h *HostManageSvc) EditHostManage(req request.EditHostManageReq) (err error
session.Rollback()
return
}
hostsIp := fmt.Sprintf("%s ansible_ssh_host=%s ansible_ssh_port=%s ansible_ssh_user=\"%s\" ansible_ssh_pass=\"%s\"",
AnsibleIp+fmt.Sprintf("%d", hostManageList.Id), v.Ip, v.Port, v.UserName, v.Password)
hosts["["+AnsibleGroup+fmt.Sprintf("%d", req.Id)+"]"] = append(hosts["["+AnsibleGroup+fmt.Sprintf("%d", req.Id)+"]"], hostsIp)
}
//写入hosts
err = tools.MapToSaveHosts(hosts)
if err != nil {
err = resp.MarshalError.WithError(err)
return
}
session.Commit()
return
}
......@@ -393,7 +435,7 @@ func (h *HostManageSvc) SaveStateHostManage(hostManageList []request.HostManageL
}
//处理ip列表
ipGroup := 1
HostManageListCaches := make([]entity.HostManageListCache, 0)
hostManageListCaches := make([]entity.HostManageListCache, 0)
for _, v := range hostManageList {
port := "22"
if v.Port != "" {
......@@ -410,40 +452,52 @@ func (h *HostManageSvc) SaveStateHostManage(hostManageList []request.HostManageL
Password: v.Password,
IpGroup: ipGroup,
}
HostManageListCaches = append(HostManageListCaches, HostManageListCache)
hostManageListCaches = append(hostManageListCaches, HostManageListCache)
}
ipGroup += 1
}
//往hosts文件中写入主机组ip
f, err := os.Open("/etc/ansible/hosts")
//读取hosts中的主机组
hosts, err := tools.HostsToJson()
if err != nil {
err = resp.FileExecError.WithError(err)
err = resp.MarshalError.WithError(err)
return
}
defer f.Close()
for _, v := range hostManageList {
_, err = f.Write([]byte(fmt.Sprintf("%s ansible_ssh_host=%s ansible_ssh_port=%s ansible_ssh_user=\"%s\" ansible_ssh_pass=\"%s\"\n", v.Ip, v.Ip, v.Port, v.UserName, v.Password)))
if err != nil {
err = resp.FileExecError.WithError(err)
return
}
if _, ok := hosts["["+AnsibleGroup+"]"]; !ok {
// 不存在
hosts["["+AnsibleGroup+"]"] = nil
}
//状态检测
id = uuid.New().String()
for _, v := range hostManageListCaches {
hostsIp := fmt.Sprintf("%s ansible_ssh_host=%s ansible_ssh_port=%s ansible_ssh_user=\"%s\" ansible_ssh_pass=\"%s\"", AnsibleIp+v.Ip, v.Ip, v.Port, v.UserName, v.Password)
flag := 0
for _, v1 := range hosts["["+AnsibleGroup+"]"] {
if v1 == hostsIp {
flag = 1
}
}
if flag == 0 {
hosts["["+AnsibleGroup+"]"] = append(hosts["["+AnsibleGroup+"]"], hostsIp)
}
}
//写入hosts
err = tools.MapToSaveHosts(hosts)
if err != nil {
err = resp.FAIL.WithError(err)
err = resp.MarshalError.WithError(err)
return
}
for k, v := range HostManageListCaches {
//状态检测
id = uuid.New().String()
for k, v := range hostManageListCaches {
//调用状态检测函数
HostManageListCaches[k].ConnStatus = StatusDetection(v.Ip)
HostManageListCaches[k].Id = id
hostManageListCaches[k].ConnStatus = StatusDetection(v.Ip)
hostManageListCaches[k].Id = id
}
//存入数据库
_, err = db.Table("host_manage_list_cache").Insert(&HostManageListCaches)
_, err = db.Table("host_manage_list_cache").Insert(&hostManageListCaches)
if err != nil {
err = resp.DbInsertError.WithError(err)
return
......@@ -509,7 +563,7 @@ func (h *HostManageSvc) SaveIpExceptionList(req request.HostIpExceptionListReq)
// StatusDetection 状态检测
func StatusDetection(ip string) (ipConn int) {
var cmd *exec.Cmd
cmd = exec.Command("ansible", fmt.Sprintf("%s", ip), "-m", "ping")
cmd = exec.Command("ansible", fmt.Sprintf("%s", AnsibleIp+ip), "-m", "ping")
output, err := cmd.Output()
if err != nil {
fmt.Println("ping：", string(output))
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment