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