package controllers
|
|
import (
|
"fmt"
|
"github.com/gin-gonic/gin"
|
"strings"
|
"test/config"
|
"test/util"
|
"time"
|
)
|
|
type SeaweedfsController struct{}
|
|
type SWFSInfo struct {
|
Role string `json:"role"`
|
}
|
|
//修改
|
func (sc *SeaweedfsController) UpdateSWFSServiceController(c *gin.Context) {
|
oldPeers := GetOldPeers()
|
newPeers := GetNewPeers()
|
UpdatePeers(oldPeers, newPeers)
|
}
|
|
func (sc *SeaweedfsController) AddSWFSNodeController(c *gin.Context) {
|
var body SWFSInfo
|
c.BindJSON(&body)
|
role := body.Role
|
if role == "master" {
|
AsMaster()
|
} else if role == "volume" {
|
AsVolume()
|
} else {
|
|
return
|
}
|
|
}
|
|
func (sc *SeaweedfsController) RestartMaster(c *gin.Context) {
|
end := "sh /opt/vasystem/script/seaweedfs_stop.sh"
|
start := "sh /opt/vasystem/script/seaweedfs_start.sh"
|
util.RunScript(end)
|
util.RunScript(start)
|
}
|
|
func RequestMasterNodesOperation() {
|
nowPeers := GetNowPeersList()
|
coreThread := len(nowPeers)/100 + 1
|
masterIp := make([]string, 0)
|
for i, val := range nowPeers {
|
ip := val.(string)
|
if (i+1)%coreThread == 0 {
|
masterIp = append(masterIp, strings.Split(ip, ":")[0])
|
for _, val := range masterIp {
|
RestartOrtherMaster(val)
|
masterIp = append(masterIp[:0], masterIp[1:]...)
|
}
|
time.Sleep(time.Second * 2)
|
} else {
|
masterIp = append(masterIp, strings.Split(ip, ":")[0])
|
if len(nowPeers) == i+1 {
|
for _, val := range masterIp {
|
RestartOrtherMaster(val)
|
}
|
break
|
}
|
continue
|
}
|
}
|
}
|
|
func GetOldPeers() string {
|
str := "cat /opt/vasystem/seaweedfs_start.sh | grep peers="
|
peers := strings.Split(util.RunScript(str), "\n")[0]
|
return peers
|
}
|
|
func AsVolume() {
|
nowPeers := GetNowPeersList()
|
fmt.Println(nowPeers)
|
}
|
|
func AsMaster() {
|
nowPeers := GetNowPeersList()
|
coreThread := len(nowPeers) / 100
|
fmt.Println(coreThread)
|
|
}
|
|
func GetNowPeersList() []interface{} {
|
|
getUrl := "http://" + config.Server.EsServerIp + ":" + config.Server.EsServerPort + "/" + config.BasicFS.IndexName + "/_search"
|
getJson := `{
|
"query": {
|
"bool": {
|
"filter": [
|
{
|
"term": {
|
"application":"nodeOperation"
|
}
|
}
|
]
|
}
|
},
|
"size": 1
|
}`
|
|
buf, _ := util.EsReq("POST", getUrl, []byte(getJson))
|
source, _ := util.Sourcelist(buf)
|
//fmt.Println(source)
|
peers := source[0]["peers"].([]interface{})
|
return peers
|
}
|
|
func GetNewPeers() string {
|
|
getUrl := "http://" + config.Server.EsServerIp + ":" + config.Server.EsServerPort + "/" + config.BasicFS.IndexName + "/_search"
|
getJson := `{
|
"query": {
|
"bool": {
|
"filter": [
|
{
|
"term": {
|
"application":"nodeOperation"
|
}
|
}
|
]
|
}
|
},
|
"size": 1
|
}`
|
|
buf, _ := util.EsReq("POST", getUrl, []byte(getJson))
|
source, _ := util.Sourcelist(buf)
|
//fmt.Println(source)
|
peers := source[0]["peers"].([]interface{})
|
fmt.Println(peers)
|
p := "peers=" + strings.Replace(strings.Trim(fmt.Sprint(peers), "[]"), " ", ",", -1)
|
return p
|
}
|
|
func UpdatePeers(oldPeers string, newPeers string) {
|
str := "sed -ie 's/" + oldPeers + "/" + newPeers + "/g' /opt/vasystem/seaweedfs_start.sh"
|
util.RunScript(str)
|
}
|
|
func AddNewMasterToPeers() (result bool) {
|
peer := config.Server.EsServerIp + ":6333"
|
addUrl := "http://" + config.Server.EsServerIp + ":" + config.Server.EsServerPort + "/" + config.BasicFS.IndexName + "/_update_by_query"
|
addJson := `{
|
"script": {
|
"lang": "painless",
|
"inline": "ctx._source.peers.add(params.newpeer)",
|
"params": {
|
"newpeer": "` + peer + `"
|
}
|
},
|
"query": {
|
"bool": {
|
"filter": [
|
{
|
"term": {
|
"application": "nodeOperation"
|
}
|
}
|
]
|
}
|
}
|
}`
|
buf, _ := util.EsReq("POST", addUrl, []byte(addJson))
|
updateRes, _ := util.SourceUpdated(buf)
|
if updateRes == -1 {
|
result = false
|
} else {
|
result = true
|
}
|
return result
|
}
|
