From e82c4624bd045a01058845625081bbcf180d0286 Mon Sep 17 00:00:00 2001
From: sunty <1172534965@qq.com>
Date: 星期四, 16 四月 2020 13:48:04 +0800
Subject: [PATCH] uasa unit test

---
 controllers/swfsControllers.go |   27 +++++++++++++++++----------
 1 files changed, 17 insertions(+), 10 deletions(-)

diff --git a/controllers/swfsControllers.go b/controllers/swfsControllers.go
index 5006f36..641a800 100644
--- a/controllers/swfsControllers.go
+++ b/controllers/swfsControllers.go
@@ -57,7 +57,7 @@
 	c.BindJSON(&body)
 	role := body.Role
 	if role == "master" {
-		AsMaster()
+		AsMaster(role)
 		util.ResponseFormat(c, code.AddSuccess, "鍔犲叆鑺傜偣鎴愬姛")
 		return
 	} else if role == "volume" {
@@ -70,7 +70,7 @@
 			return
 		}
 	} else if role == "master+volume" {
-		AsMaVo()
+		AsMaVo(role)
 		util.ResponseFormat(c, code.AddSuccess, "鍔犲叆鑺傜偣鎴愬姛")
 	} else {
 		util.ResponseFormat(c, code.RequestParamError, "閫夋嫨鑺傜偣绫诲瀷閿欒")
@@ -79,13 +79,13 @@
 
 }
 
-func AsMaVo() {
-	AsMaster()
+func AsMaVo(role string) {
+	AsMaster(role)
 	ReplaceLineContentBySearch(Option+StartScriptAsMaVo, Option, config.Server.ScriptPath, StartServerScript)
 }
 
 func (sc *SeaweedfsController) RoleOfVolumeToMasterController(c *gin.Context) {
-	AsMaster()
+	AsMaster("master")
 	ReplaceLineContentBySearch(Option+StartScriptAsMaVo, Option, config.Server.ScriptPath, StartServerScript)
 }
 
@@ -131,14 +131,21 @@
 	}
 }
 
-//璇锋眰浣滀负涓昏妭鐐规搷浣滄祦绋�
-func RequestMasterNodesOperation(nowPeers []interface{}) {
+//璇锋眰浣滀负褰撳墠瑙掕壊鑺傜偣鎿嶄綔娴佺▼
+func RequestNodesOperation(nowPeers []interface{}, role string) {
 	peersIp := make([]string, 0)
 	for _, val := range nowPeers {
 		peersIp = append(peersIp, strings.Split(val.(string), ":")[0])
 	}
 	coreBaseUnit, _ := strconv.Atoi(config.Server.CoreBaseUnit)
 	UpdateAllNodesScriptArgument(peersIp)
+	replaceContent := ""
+	if role == "master" {
+		replaceContent = Option + StartScriptAsMaster
+	} else if role == "master+volume" {
+		replaceContent = Option + StartScriptAsMaVo
+	}
+	ReplaceLineContentBySearch(replaceContent, Option, config.Server.ScriptPath, StartServerScript)
 	RestartAllServer(peersIp, coreBaseUnit)
 }
 
@@ -161,7 +168,7 @@
 	tick := time.Tick(1 * time.Second)
 	fmt.Println("鍑嗗寮�濮嬮獙璇佽妭鐐规湇鍔�")
 	for countdown := timeOut; countdown > 0; countdown-- {
-		fmt.Println("绗竴娆¢獙璇�")
+		fmt.Println("绗�", countdown, "娆¢獙璇�")
 		result := Verification(startupItem, ip)
 		fmt.Println("绗竴娆¢獙璇乺esult缁撴灉锛�", result)
 		if result == true {
@@ -263,11 +270,11 @@
 }
 
 //浣滀负涓昏妭鐐瑰姞鍏ワ紙榛樿鍖呭惈鏁版嵁鑺傜偣锛�
-func AsMaster() bool {
+func AsMaster(role string) bool {
 	AddNewMasterToPeers()
 	nowPeers := GetNowPeersList()
 	fmt.Println("nowPeers: ", nowPeers)
-	RequestMasterNodesOperation(nowPeers)
+	RequestNodesOperation(nowPeers, role)
 	return true
 }
 

--
Gitblit v1.8.0