add updateEsHosts

sunty

2019-11-07 b8e48712c2a521225d7f2ef7a982f766a6005615

 controllers/es.go

@@ -26,6 +26,7 @@
type EsClusterInfo struct {
   Ip string `json:"ip"`
}

// @Security ApiKeyAuth
// @Summary 比对数据查询
// @Description  比对数据查询
@@ -43,28 +44,28 @@
      util.ResponseFormat(c, code.RequestParamError, "参数有误")
      return
   }
   if searchBody.CompareNum !="" {
   if searchBody.CompareNum != "" {
      //二次搜索，不需要再比对了
      co := service.GetCompResultByNum(searchBody.CompareNum)
      if co != nil {
         //二次搜索和排序
         twiceM := GetCompareDataTwice(co,searchBody)
         util.ResponseFormat(c,code.Success,twiceM)
         twiceM := GetCompareDataTwice(co, searchBody)
         util.ResponseFormat(c, code.Success, twiceM)
         return
      } else {
         m :=make(map[string]interface{},0)
         m := make(map[string]interface{}, 0)
         m["compareNum"] = searchBody.CompareNum
         m["total"] = 0
         m["totalList"] = []CompareResult{}
         util.ResponseFormat(c,code.CompareResultGone,m)
         util.ResponseFormat(c, code.CompareResultGone, m)
         return
      }
   }
   m :=make(map[string]interface{},0)
   m := make(map[string]interface{}, 0)
   m["compareNum"] = searchBody.CompareNum
   m["total"] = 0
   m["totalList"] = []CompareResult{}
   util.ResponseFormat(c,code.CompareResultGone,m)
   util.ResponseFormat(c, code.CompareResultGone, m)
}

func searchEsData(searchBody models.EsSearch) map[string]interface{} {
@@ -114,7 +115,7 @@
   //使用es底层机制处理分页
   //请求头
   localConf, err2 := cache.GetServerInfo()
   if err2 !=nil || localConf.AlarmIp == "" || localConf.ServerId == "" {
   if err2 != nil || localConf.AlarmIp == "" || localConf.ServerId == "" {
      logger.Debug("localConfig is wrong!!!")
      return nil
   }
@@ -214,12 +215,12 @@
      nodeInfo := make(map[string]interface{})
      context := strings.Fields(infos[i])
      nodeIp := context[0]
      Type := context[8]
      Type := context[7]
      var nodeType string
      if Type == "*" {
      if Type == "mdi" {
         nodeType = "主节点"
      }
      if Type == "-" {
      if Type == "di" {
         nodeType = "从节点"
      }
      nodeInfo["ip"] = nodeIp
@@ -382,6 +383,82 @@
   return msg, err
}

//修改elasticsearch.yml映射文件
// @Security ApiKeyAuth
// @Summary 修改es集群Ip
// @Description  修改es集群Ip
// @Accept  json
// @Produce json
// @Tags es
// @Param obj body controllers.Hosts true "修改es集群Ip参数"
// @Success 200 {string} json "{"code":200, msg:"", success:true}"
// @Failure 500 {string} json "{"code":500, msg:"", success:false}"
// @Router /data/api-v/es/updateEsHosts [POST]
func (em *EsManagementController) UpdateEsHosts(c *gin.Context) {
   var hosts Hosts
   c.BindJSON(&hosts)
   nodeInfos, err := getEsClusterInfors(hosts.OldIp)
   var nodeType string
   if err != nil {
      logger.Fatal(err)
   }
   newMasterHosts := make([]string, 0)
   allHosts := make([]string, 0)
   for _, val := range nodeInfos {
      ip := val["ip"].(string)
      tmpType := val["nodeType"].(string)
      if ip == hosts.OldIp {
         nodeType = tmpType
      }
      if tmpType == "master" {
         newMasterHosts = append(newMasterHosts, ip)
      }
      if ip != hosts.OldIp {
         allHosts = append(allHosts, ip)
      }
   }

   if nodeType == "master" {
      str1 := "sed -ie 's/network.host: " + hosts.OldIp + "/network.host: " + hosts.NewIp + "/g' /opt/elasticsearch/config/elasticsearch.yml"
      runScript(str1)
      for i, host := range newMasterHosts {
         if host == hosts.OldIp {
            newMasterHosts[i] = hosts.NewIp
         }
      }
      for i, val := range newMasterHosts {
         val = "\\\"" + val + ":9300\\\""
         newMasterHosts[i] = val
      }
      newUnicastHost := strings.Replace(strings.Trim(fmt.Sprint(newMasterHosts), ""), " ", ",", -1)

      tmpStr := "cat /opt/elasticsearch/config/elasticsearch.yml | grep discovery.zen.ping.unicast.hosts:"
      rs := runScript(tmpStr)
      ts := strings.Split(rs, "\n")[0]
      ots := strings.Split(ts, " ")[1]
      outs := strings.Replace(ots, "\"", "\\\"", -1)
      oldUnicastHost := "\\" + strings.Replace(outs, "]", "\\]", -1)
      str2 := "sed -ie 's/discovery.zen.ping.unicast.hosts: " + oldUnicastHost + "/discovery.zen.ping.unicast.hosts: " + newUnicastHost + "/g' /opt/elasticsearch/config/elasticsearch.yml"
      runScript(str2)
      for _, host := range allHosts {
         str3 := "sshpass -p \"123\" ssh basic@" + host + " \"cd /opt/elasticsearch/config  ; " + str2 + " ; cat elasticsearch.yml\""
         runScript(str3)
      }
   } else {
      str1 := "sed -ie 's/network.host: " + hosts.OldIp + "/network.host: " + hosts.NewIp + "/g' /opt/elasticsearch/config/elasticsearch.yml"
      runScript(str1)
   }
}

type Hosts struct {
   NewIp string `json:newIp`
   OldIp string `json:oldIp`
}

func forwardCommand() {

}

//func AddEsCluster(hosts []string) (string){
//   msg := "加入失败"
//   for i,val := range hosts{