mirror of
https://gitee.com/lauix/HFish
synced 2025-05-15 14:18:01 +08:00
118 lines
3.0 KiB
Go
118 lines
3.0 KiB
Go
package client
|
||
|
||
import (
|
||
"HFish/utils/log"
|
||
"HFish/utils/conf"
|
||
"HFish/core/rpc/core"
|
||
"strings"
|
||
"fmt"
|
||
)
|
||
|
||
// 上报状态结构
|
||
type Status struct {
|
||
AgentIp string
|
||
AgentName string
|
||
Web, Deep, Ssh, Redis, Mysql, Http, Telnet, Ftp, MemCahe, Plug, ES, TFtp, Vnc string
|
||
}
|
||
|
||
// 上报结果结构
|
||
type Result struct {
|
||
AgentIp string
|
||
AgentName string
|
||
Type string
|
||
ProjectName string
|
||
SourceIp string
|
||
Info string
|
||
Id string // 数据库ID，更新用 0 为新插入数据
|
||
}
|
||
|
||
var rpcClient *rpc.Client
|
||
var ipAddr string
|
||
|
||
func RpcInit() {
|
||
rpcAddr := conf.Get("rpc", "addr")
|
||
c, conn, err := rpc.Dial("tcp", rpcAddr)
|
||
|
||
if err != nil {
|
||
rpcClient = nil
|
||
log.Pr("RPC", "127.0.0.1", "连接 RPC Server 失败")
|
||
ipAddr = ""
|
||
} else {
|
||
rpcClient = c
|
||
ipAddr = strings.Split(conn.LocalAddr().String(), ":")[0]
|
||
fmt.Println("连接RPC Server 成功")
|
||
}
|
||
}
|
||
|
||
func reportStatus(rpcName string, ftpStatus string, telnetStatus string, httpStatus string, mysqlStatus string, redisStatus string, sshStatus string, webStatus string, darkStatus string, memCacheStatus string, plugStatus string, esStatus string, tftpStatus string, vncStatus string) {
|
||
if (rpcClient != nil) {
|
||
status := Status{
|
||
ipAddr,
|
||
rpcName,
|
||
webStatus,
|
||
darkStatus,
|
||
sshStatus,
|
||
redisStatus,
|
||
mysqlStatus,
|
||
httpStatus,
|
||
telnetStatus,
|
||
ftpStatus,
|
||
memCacheStatus,
|
||
plugStatus,
|
||
esStatus,
|
||
tftpStatus,
|
||
vncStatus,
|
||
}
|
||
|
||
var reply string
|
||
err := rpcClient.Call("HFishRPCService.ReportStatus", status, &reply)
|
||
|
||
if err != nil {
|
||
log.Pr("RPC", "127.0.0.1", "上报服务状态失败", err)
|
||
RpcInit()
|
||
} else {
|
||
fmt.Println("上报服务状态成功")
|
||
}
|
||
} else {
|
||
RpcInit()
|
||
}
|
||
}
|
||
|
||
func ReportResult(typex string, projectName string, sourceIp string, info string, id string) string {
|
||
var reply string
|
||
|
||
if (rpcClient != nil) {
|
||
// projectName 只有 WEB 才需要传项目名 其他协议空即可
|
||
// id 0 为 新插入数据，非 0 都是更新数据
|
||
// id 非 0 的时候 sourceIp 为空
|
||
rpcName := conf.Get("rpc", "name")
|
||
|
||
result := Result{
|
||
ipAddr,
|
||
rpcName,
|
||
typex,
|
||
projectName,
|
||
sourceIp,
|
||
info,
|
||
id,
|
||
}
|
||
|
||
err := rpcClient.Call("HFishRPCService.ReportResult", result, &reply)
|
||
|
||
if err != nil {
|
||
log.Pr("RPC", "127.0.0.1", "上报上钩结果失败")
|
||
RpcInit()
|
||
} else {
|
||
fmt.Println("上报上钩结果成功")
|
||
}
|
||
|
||
return reply
|
||
} else {
|
||
return "0"
|
||
}
|
||
}
|
||
|
||
func Start(rpcName string, ftpStatus string, telnetStatus string, httpStatus string, mysqlStatus string, redisStatus string, sshStatus string, webStatus string, darkStatus string, memCacheStatus string, plugStatus string, esStatus string, tftpStatus string, vncStatus string) {
|
||
reportStatus(rpcName, ftpStatus, telnetStatus, httpStatus, mysqlStatus, redisStatus, sshStatus, webStatus, darkStatus, memCacheStatus, plugStatus, esStatus, tftpStatus, vncStatus)
|
||
}
|