From cc6d3763795b62d353df0c1eb58cbb736b709829 Mon Sep 17 00:00:00 2001
From: zhangzengfei <zhangzengfei@smartai.com>
Date: 星期五, 20 十月 2023 09:48:37 +0800
Subject: [PATCH] 添加主节点退出时, 重新选举master的功能

---
 system-service/controllers/cluster.go |   46 +++++++++++++++++++++++++++++++++++++++++++++-
 1 files changed, 45 insertions(+), 1 deletions(-)

diff --git a/system-service/controllers/cluster.go b/system-service/controllers/cluster.go
index da13d0a..df7d326 100644
--- a/system-service/controllers/cluster.go
+++ b/system-service/controllers/cluster.go
@@ -1,6 +1,7 @@
 package controllers
 
 import (
+	"vamicro/config"
 	"vamicro/system-service/models"
 	"vamicro/system-service/service"
 	"vamicro/system-service/vo"
@@ -11,6 +12,49 @@
 )
 
 type ClusterController struct {
+}
+
+// @Summary 鏌ヨ褰撳墠闆嗙兢鐘舵��
+// @Description 鏌ヨ鏈湴闆嗙兢
+// @Produce json
+// @Tags cluster
+// @Success 200 {string} json "{"code":200, success:true, msg:"", data:""}"
+// @Failure 500 {string} json "{"code":500, success:false, msg:"",data:""}"
+// @Router /data/api-v/cluster/status [get]
+func (cc ClusterController) GetClusterStat(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply {
+	var clusterE models.Cluster
+	var reply = bhomeclient.Reply{
+		Success: false,
+		Msg:     "",
+		Data:    nil,
+	}
+
+	arr, err := clusterE.FindAll()
+	if err == nil {
+		if arr != nil && len(arr) > 0 {
+			// 琛ㄧず宸插姞鍏ラ泦缇�
+			reply.Success = true
+
+			var nodeE models.Node
+			nodes, _ := nodeE.FindNodesByClusterId(arr[0].ClusterId)
+			//logger.Debug("鏌ヨ闆嗙兢鑺傜偣:", nodes)
+			for _, node := range nodes {
+				//logger.Debug("鑺傜偣:", node.NodeId, " serverId:", config.Server.AnalyServerId, " stat:", node.DriftState)
+				if node.NodeId == config.Server.AnalyServerId {
+					if node.DriftState == "master" {
+						reply.Msg = "master"
+
+					} else {
+						reply.Msg = "slave"
+					}
+					break
+				}
+			}
+			reply.Data = nodes
+		}
+	}
+
+	return &reply
 }
 
 // @Summary 鏌ヨ鏈湴闆嗙兢
@@ -27,7 +71,7 @@
 		if arr != nil && len(arr) > 0 {
 			var nodeE models.Node
 			nodes, _ := nodeE.FindNodesByClusterId(arr[0].ClusterId)
-			logger.Debugf("FindCluster nodes=%v", nodes)
+			//logger.Debugf("FindCluster nodes=%v", nodes)
 			return &bhomeclient.Reply{Success: true, Data: map[string]interface{}{
 				"clusterId":   arr[0].ClusterId,
 				"clusterName": arr[0].ClusterName,

--
Gitblit v1.8.0