Go编写服务监管程序

前言

        程序的目的：一个基于Linux系统下的进程监控与管理工具，它能够监控指定的进程或服务的运行情况，并在发现它们不存在或出现异常时自动进行重启操作。这个程序就像一个可靠的看门狗，时刻守护着系统的稳定运行。

        程序的本身是周期性通过命令获得程序状态，如果异常，执行指定的重启命令，并发送邮件。公司之前有过Java版本、QT版本的服务监控程序，现在提供GO语言版本。这个demo会涉及到结构体转换、读取json文件、发送邮箱、执行cmd命令、定时器、远程登录ssh等。

程序思路

1、新建host.json文件。存放服务器信息、服务监听命令、服务重启命令

2、编写 mailSend.go 用于发送邮件

3、编写 constant.go 用于结构体和常量

4、编写ReadFile方法、LocalExec方法、ssh登录方法等

核心代码

配置文件host.json

[{"sshHost": "127.0.0.1",   "sshUser": "",    "sshStatus": 1,    "sshPassword": "",   "sshType": "local",    "sshKeyPath": "",    "sshPort": 22,    "monitors": [{"name": "道闸后台java服务",       "type": "java",        "status": 1,        "findCmd": "ps -ef|grep test",        "keyword": "test.jar",        "starCmd": "nohup java -jar /home/test.jar >/dev/null  2>&1 &",        "fileDir": "/home/test.jar"      },      {"name": "docker nginx_server 服务",        "type": "docker",        "status": 0,        "findCmd": "docker ps|grep nginx_server",        "keyword": "Up",        "starCmd": "docker restart nginx_server",        "fileDir": ""      },]}
]

邮箱 mailSend.go

package constantimport ("gopkg.in/gomail.v2"    "strconv")func SendMailSelf(body string) {userName := "[e-mail]"    authCode := "XXX"    host := "smtp.163.com"    portStr := "465"    mailTo := "[e-mail]"    sendName := "[e-mail]"    subject := "服务器重启"    SendMail(userName, authCode, host, portStr, mailTo, sendName, subject, body)
}func SendMail(userName, authCode, host, portStr, mailTo, sendName string, subject, body string) error {port, _ := strconv.Atoi(portStr)m := gomail.NewMessage()m.SetHeader("From", m.FormatAddress(userName, sendName))m.SetHeader("To", mailTo)m.SetHeader("Subject", subject)m.SetBody("text/html", body)d := gomail.NewDialer(host, port, userName, authCode)err := d.DialAndSend(m)return err
}

核心方法 PasswordSsh

        此方法使用golang.org/x/crypto/ssh，远程登录服务器执行相关命令。此监管程序可以放在本地或者其他服务器，不在依赖当前的操作系统，即可监控所有的服务。

// PasswordSsh 用于登录ssh 
PasswordSsh(hostConfig *constant.SshConfig) {//创建ssh登陆配置    config := &ssh.ClientConfig{Timeout:         time.Second, //ssh 连接time out 时间一秒钟, 如果ssh验证错误 会在一秒内返回       User:            hostConfig.SshUser,       HostKeyCallback: ssh.InsecureIgnoreHostKey(), //这个可以， 但是不够安全       //HostKeyCallback: hostKeyCallBackFunc(h.Host),    }if hostConfig.SshType == "password" {config.Auth = []ssh.AuthMethod{ssh.Password(hostConfig.SshPassword)}} else {config.Auth = []ssh.AuthMethod{publicKeyAuthFunc(hostConfig.SshKeyPath)}}//dial 获取ssh client    addr := fmt.Sprintf("%s:%d", hostConfig.SshHost, hostConfig.SshPort)sshClient, err := ssh.Dial("tcp", addr, config)if err != nil {log.Fatal("创建ssh client 失败", err)}defer sshClient.Close()// 循环监控具体程序命令    for _, ServerItem := range hostConfig.Monitors {if ServerItem.Status == 1 {//创建ssh-session          session, err := sshClient.NewSession()if err != nil {log.Fatal("创建ssh session 失败", err)}defer session.Close()//执行远程命令          combo, err := session.CombinedOutput(ServerItem.FindCmd)if err != nil {log.Fatal("远程执行cmd 失败", err)}fmt.Println(string(combo))isStart := strToArr(string(combo), ServerItem.Keyword)fmt.Printf("this %s need restart :%v\n", ServerItem.Name, isStart)if isStart {//创建ssh-session             session, err := sshClient.NewSession()if err != nil {log.Fatal("创建ssh session 失败", err)}defer session.Close()session.CombinedOutput(ServerItem.StarCmd)log.Println("重启命令输出:", string(combo))// 发送邮箱             constant.SendMailSelf(ServerItem.StarCmd)}}}
}

github地址

https://github.com/cfh123/go-regulatory-procedures