[ Cmii ] [ Octopus ] - hot fix manual shut down init queue

agent-go/a_init/AgentInitialization.go

@@ -294,8 +294,8 @@ func handleInitMsgFromServer(initFromServerMsg *rabbitmq.OctopusMessage, initToS
 // shutdownRegisterQueueConnection 关闭初始化连接的两个队列
 func shutdownRegisterQueueConnection(initFromServerQueue *rabbitmq.RabbitQueue, initToServerQueue *rabbitmq.RabbitQueue) {
 
-	_ = initFromServerQueue.Close()
-	_ = initToServerQueue.Close()
+	initFromServerQueue.ConsumeOK = false
+	initToServerQueue.ConsumeOK = false
 
 	log.InfoF("Octopus Agent Init Queue has disconnected!")
 }

agent-go/rabbitmq/RabbitMsgQueue.go

@@ -46,6 +46,8 @@ type RabbitQueue struct {
 	RabbitConnectInfo *RabbitTCPConnectInfo
 	// 返回消息队列
 	ReceiveChan *RabbitReceiveChan
+	// Consume 是否继续
+	ConsumeOK bool
 }
 
 // RabbitMQConn is a struct that holds the connection and channel objects
@@ -192,6 +194,7 @@ func (r *RabbitQueue) Close() error {
 func (r *RabbitQueue) Handle() chan bool {
 
 	deliveries := r.Read(true)
+
 	foreverHandle := make(chan bool)
 	// 死循环，处理Octopus Message
 	P.Submit(func() {
@@ -208,6 +211,13 @@ func (r *RabbitQueue) Handle() chan bool {
 			P.Submit(func() {
 				om.HandleMsg(r.ReceiveChan)
 			})
+
+			// 2024年4月9日
+			if !r.ConsumeOK {
+				marshal, _ := json.Marshal(r.RabbitProp)
+				log.InfoF("Rabbit Queue of %s disconnect!", marshal)
+				break
+			}
 		}
 	})
 	return foreverHandle

agent-go/shell/octopus-update.sh

@@ -0,0 +1 @@
+#!/bin/bash