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 := "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