HFish/core/rpc/client/client.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)
}