[agent-go] 完成RabbitMQ连接部分的代码

2023-03-21 17:08:22 +08:00
parent 1c57a631d9
commit fe8a4a03fc
19 changed files with 482 additions and 93 deletions
--- a/agent-go/register/AgentIntitilization.go
+++ b/agent-go/register/AgentIntitilization.go
@@ -0,0 +1,105 @@
+package register
+
+import (
+	"agent-go/g"
+	"agent-go/rabbitmq"
+	"encoding/json"
+	"fmt"
+	"gopkg.in/yaml.v3"
+	"io/ioutil"
+)
+
+var log = g.G.LOG
+var omType = g.InitOmType
+
+func INIT() {
+
+	// 获取系统的环境变量
+	agentServerInfo := parseAgentServerInfo()
+
+	nacosConfig := g.G.NacosConfig
+
+	initToServerProp := &rabbitmq.ConnectProperty{
+		ExchangeName: nacosConfig.GetString("octopus.message.init_exchange"),
+		QueueName:    nacosConfig.GetString("octopus.message.init_to_server"),
+		ExchangeType: g.QueueDirect,
+		TopicKey:     nacosConfig.GetString("octopus.message.init_to_server_key"),
+	}
+
+	initFromServerProp := &rabbitmq.ConnectProperty{
+		ExchangeName: nacosConfig.GetString("octopus.message.init_exchange"),
+		QueueName:    nacosConfig.GetString("octopus.message.init_from_server"),
+		ExchangeType: g.QueueDirect,
+		TopicKey:     nacosConfig.GetString("octopus.message.init_from_server_key"),
+	}
+
+	// 建立RabbitMQ的连接
+	// defer 关闭初始化连接
+	initToServer, err := rabbitmq.NewRabbitMQConn(
+		initToServerProp,
+	)
+	if err != nil {
+		log.Error("init to server queue established error!")
+		panic(err)
+	}
+	defer rabbitmq.CloseChannel(initToServer)
+
+	initFromServer, err := rabbitmq.NewRabbitMQConn(
+		initFromServerProp,
+	)
+	if err != nil {
+		log.Error("init from server queue established error!")
+		panic(err)
+	}
+	defer rabbitmq.CloseChannel(initFromServer)
+
+	// 组装OctopusMessage
+	var octopusMsg *rabbitmq.OctopusMessage
+	octopusMsg = octopusMsg.BuildOctopusMsg(
+		omType,
+		agentServerInfo,
+	)
+	msgBytes, err := json.Marshal(octopusMsg)
+	if err != nil {
+		log.Error(fmt.Sprintf("octopus message convert to json is wrong! msg is => %v", octopusMsg))
+	}
+	// 发送OM至MQ中O
+	rabbitmq.Send(
+		initToServer,
+		initToServerProp,
+		msgBytes,
+	)
+
+	// 监听初始化连接中的信息
+	// 建立运行时RabbitMQ连接
+	handleInitMsgFromServer()
+
+}
+
+func handleInitMsgFromServer() {
+
+}
+
+func parseAgentServerInfo() AgentServerInfo {
+
+	// 约定文件地址为 /etc/environment.d/octopus-agent.conf
+	// 目前使用
+	var agentServerInfo AgentServerInfo
+	yamlFile, err := ioutil.ReadFile("C:\\Users\\wdd\\IdeaProjects\\ProjectOctopus\\agent-go\\server-env.yaml")
+	if err != nil {
+		panic(fmt.Errorf("failed to read YAML file: %v", err))
+	}
+
+	err = yaml.Unmarshal(yamlFile, &agentServerInfo)
+	if err != nil {
+		panic(fmt.Errorf("failed to unmarshal YAML: %v", err))
+	}
+
+	jsonFormat, err := json.Marshal(agentServerInfo)
+	if err != nil {
+		return AgentServerInfo{}
+	}
+	log.Info(fmt.Sprintf("agent server info is %v", string(jsonFormat)))
+
+	return agentServerInfo
+}