From 6e262c17a5463b02d7440620d13f2eeb98743223 Mon Sep 17 00:00:00 2001
From: zhangzengfei <zhangzengfei@smartai.com>
Date: 星期二, 14 十一月 2023 14:44:00 +0800
Subject: [PATCH] 切换主节点添加密码
---
system-service/vo/cluster.go | 1 +
system-service/controllers/cluster.go | 14 ++++++++++++++
2 files changed, 15 insertions(+), 0 deletions(-)
diff --git a/system-service/controllers/cluster.go b/system-service/controllers/cluster.go
index 90c5ee3..1a6ca4f 100644
--- a/system-service/controllers/cluster.go
+++ b/system-service/controllers/cluster.go
@@ -122,6 +122,20 @@
return &bhomeclient.Reply{Success: false, Msg: "鍙傛暟鏈夎"}
}
+ var clusterE models.Cluster
+ arr, err := clusterE.FindAll()
+ if err != nil {
+ return &bhomeclient.Reply{Success: false, Msg: "闆嗙兢鏌ヨ澶辫触"}
+ }
+
+ if arr != nil && len(arr) > 0 {
+ if config.ClusterSet.PwdPre+nodeVo.Password != arr[0].Password {
+ return &bhomeclient.Reply{Success: false, Msg: "瀵嗙爜閿欒"}
+ }
+ } else {
+ return &bhomeclient.Reply{Success: false, Msg: "闆嗙兢鏌ヨ澶辫触"}
+ }
+
sv := service.NewClusterService(h.Bk)
b, _ := sv.UpdateDriftStateByNodeId(nodeVo.ClusterId, nodeVo.NodeId, "master")
if b {
diff --git a/system-service/vo/cluster.go b/system-service/vo/cluster.go
index 328f323..3b9116f 100644
--- a/system-service/vo/cluster.go
+++ b/system-service/vo/cluster.go
@@ -21,6 +21,7 @@
type UpdateClusterVo struct {
ClusterId string `json:"clusterId"`
NodeId string `json:"nodeId"`
+ Password string `json:"password"`
}
type ClusterJoinVo struct {
--
Gitblit v1.8.0