105 lines
2.4 KiB
Go
105 lines
2.4 KiB
Go
|
/************************************************************
|
|||
|
|
** @Description: notify
|
|||
|
|
** @Author: george hao
|
|||
|
|
** @Date: 2019-06-26 15:17
|
|||
|
|
** @Last Modified by: george hao
|
|||
|
|
** @Last Modified time: 2019-06-26 15:17
|
|||
|
|
*************************************************************/
|
|||
|
|
package server
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"encoding/json"
|
|||
|
|
"fmt"
|
|||
|
|
"github.com/george518/PPGo_Job/libs"
|
|||
|
|
"strconv"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
//启动时注册
|
|||
|
|
func Register() error {
|
|||
|
|
//获取本机ip以及端口 todo ip合法性判断
|
|||
|
|
if C.TcpIp == "auto" {
|
|||
|
|
tcpIp := libs.GetHostIp(C.IpType)
|
|||
|
|
if tcpIp == "" {
|
|||
|
|
return fmt.Errorf("无法获取本机IP，请手工在配置文件里设置")
|
|||
|
|
}
|
|||
|
|
SaveConfig("TcpIp", tcpIp)
|
|||
|
|
}
|
|||
|
|
param := make(map[string]string, 0)
|
|||
|
|
if C.ServerName == "auto" {
|
|||
|
|
serverName := "agent-" + C.TcpIp + "-" + strconv.Itoa(C.TcpPort)
|
|||
|
|
SaveConfig("ServerName", serverName)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
param["server_ip"] = C.TcpIp
|
|||
|
|
param["port"] = strconv.Itoa(C.TcpPort)
|
|||
|
|
param["server_name"] = C.ServerName
|
|||
|
|
param["detail"] = "自动注册执行器"
|
|||
|
|
param["connection_type"] = "2"
|
|||
|
|
param["group_id"] = C.GroupId
|
|||
|
|
|
|||
|
|
if C.RegisterUrl == "" {
|
|||
|
|
return fmt.Errorf("自动注册地址配置错误")
|
|||
|
|
}
|
|||
|
|
body, err := libs.HttpGet(C.RegisterUrl, param)
|
|||
|
|
if err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
m := make(map[string]interface{})
|
|||
|
|
err = json.Unmarshal([]byte(body), &m)
|
|||
|
|
if err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if _, ok := m["status"]; ok {
|
|||
|
|
if m["status"] == float64(0) {
|
|||
|
|
//回写serverId
|
|||
|
|
serverId := int(m["message"].(float64))
|
|||
|
|
SaveConfig("ServerId", strconv.Itoa(serverId))
|
|||
|
|
return nil
|
|||
|
|
} else {
|
|||
|
|
return fmt.Errorf("自动注册失败：%v", m["message"])
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return fmt.Errorf("自动注册失败")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//程序异常退出的通知
|
|||
|
|
func Close() error {
|
|||
|
|
|
|||
|
|
param := make(map[string]string, 0)
|
|||
|
|
param["server_ip"] = C.TcpIp
|
|||
|
|
param["port"] = strconv.Itoa(C.TcpPort)
|
|||
|
|
param["status"] = "1"
|
|||
|
|
|
|||
|
|
if C.UpdateStatusUrl == "" {
|
|||
|
|
return fmt.Errorf("执行器退出通知异常，请到系统中修改状态")
|
|||
|
|
}
|
|||
|
|
body, err := libs.HttpGet(C.UpdateStatusUrl, param)
|
|||
|
|
if err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
m := make(map[string]interface{})
|
|||
|
|
err = json.Unmarshal([]byte(body), &m)
|
|||
|
|
if err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if _, ok := m["status"]; ok {
|
|||
|
|
if m["status"] == float64(0) {
|
|||
|
|
return nil
|
|||
|
|
} else {
|
|||
|
|
return fmt.Errorf("执行器退出通知异常：%v", m["message"])
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return fmt.Errorf("执行器退出通知异常：未知原因")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//心跳机制
|
|||
|
|
func Heartbeat() error {
|
|||
|
|
return nil
|
|||
|
|
}
|