前言
程序的目的:一个基于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 := "XXX@163.com"
authCode := "XXX"
host := "smtp.163.com"
portStr := "465"
mailTo := "XXX@qq.com"
sendName := "XXX@163.com"
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