mirror of
https://gitee.com/lauix/HFish
synced 2025-05-14 13:48:03 +08:00
119 lines
3.1 KiB
Go
119 lines
3.1 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, Custom 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, customStatus string) {
|
||
if (rpcClient != nil) {
|
||
status := Status{
|
||
ipAddr,
|
||
rpcName,
|
||
webStatus,
|
||
darkStatus,
|
||
sshStatus,
|
||
redisStatus,
|
||
mysqlStatus,
|
||
httpStatus,
|
||
telnetStatus,
|
||
ftpStatus,
|
||
memCacheStatus,
|
||
plugStatus,
|
||
esStatus,
|
||
tftpStatus,
|
||
vncStatus,
|
||
customStatus,
|
||
}
|
||
|
||
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, customStatus string) {
|
||
reportStatus(rpcName, ftpStatus, telnetStatus, httpStatus, mysqlStatus, redisStatus, sshStatus, webStatus, darkStatus, memCacheStatus, plugStatus, esStatus, tftpStatus, vncStatus, customStatus)
|
||
}
|