[agent-go] [status] - test for connect

agent-go/rabbitmq/OctopusMessage.go

@@ -3,9 +3,11 @@ package rabbitmq
 import (
 	"agent-go/executor"
 	"agent-go/g"
+	"agent-go/status"
 	"agent-go/utils"
 	"encoding/json"
 	"fmt"
+	"strings"
 	"time"
 )
 
@@ -105,10 +107,28 @@ func executorOMHandler(octopusMessage *OctopusMessage) {
 }
 
 func statusOMHandler(octopusMessage *OctopusMessage) {
-	log.Info("接收到查询Agent状态的请求，假装已经处理")
 
-	//statusMessageString := octopusMessage.Content.(string)
+	statusMsgString := octopusMessage.Content.(string)
+	var statusMessage *status.StatusMessage
+	err := json.Unmarshal([]byte(statusMsgString), &statusMessage)
+	if err != nil {
+		log.Error(fmt.Sprintf("status message convert to json is wrong! msg is => %s", statusMsgString))
+		return
+	}
 
+	var statusRes string
+	if strings.HasPrefix(statusMessage.Type, "p") {
+		// ping info
+		statusRes = status.Ping()
+	} else {
+		// status info
+		agentStatusString, _ := json.Marshal(status.ReportAppStatus())
+		statusRes = string(agentStatusString)
+	}
+
+	// 返回消息
+
+	log.InfoF("接收到查询Agent状态的请求，结果为 => %s", statusRes)
 }
 
 func blackHoleOMHandler(octopusMessage *OctopusMessage) {