package service
|
|
import (
|
"basic.com/valib/logger.git"
|
"encoding/json"
|
"fmt"
|
"strconv"
|
"strings"
|
"time"
|
"webserver/cache"
|
"webserver/extend/config"
|
"webserver/extend/esutil"
|
)
|
|
func GetVerificationData() []map[string]interface{} {
|
localConf, err2 := cache.GetServerInfo()
|
if err2 !=nil || localConf.AlarmIp == "" || localConf.ServerId == "" {
|
logger.Debug("localConfig is wrong!!!")
|
return nil
|
}
|
url := "http://" + localConf.AlarmIp + ":" + strconv.Itoa(int(localConf.AlarmPort)) +
|
"/" + config.EsInfo.EsIndex.DbTables.IndexName + "/_search"
|
|
jsonDSL := `{
|
"query": {
|
"bool": {
|
"filter": [{
|
"term": {
|
"isDelete": "0"
|
}
|
},
|
{
|
"terms": {
|
"analyServerId": ["` + localConf.ServerId + `",""]
|
}
|
}
|
]
|
}
|
},
|
"size": 10000
|
}
|
`
|
buf, err := esutil.EsReq("POST", url, []byte(jsonDSL))
|
if err != nil {
|
logger.Debug(err)
|
}
|
|
sources, err := esutil.Sourcelist(buf)
|
if err != nil {
|
logger.Debug(err)
|
}
|
return sources
|
}
|
|
func VerificationTime(source []map[string]interface{}) map[string][]string {
|
resultInfo := make(map[string][]string, 0)
|
currentTime := time.Now()
|
for _, v := range source {
|
id := v["id"].(string)
|
startTime := v["startTime"].(string)
|
endTime := ""
|
if v["endTime"] != nil {
|
endTime = v["endTime"].(string)
|
}
|
st, _ := time.ParseInLocation("2006-01-02 15:04:05", startTime, time.Local)
|
if endTime != "" {
|
et, _ := time.ParseInLocation("2006-01-02 15:04:05", endTime, time.Local)
|
if et.After(currentTime) && st.Before(currentTime) {
|
resultInfo["effective"] = append(resultInfo["effective"], id)
|
} else {
|
resultInfo["invalid"] = append(resultInfo["invalid"], id)
|
}
|
} else if st.Before(currentTime) && endTime == "" {
|
resultInfo["effective"] = append(resultInfo["effective"], id)
|
} else {
|
resultInfo["invalid"] = append(resultInfo["invalid"], id)
|
}
|
}
|
//fmt.Println(resultInfo)
|
return resultInfo
|
}
|
|
func UpdateEnableStatus(id []string, flag bool) (message string) {
|
ids := strings.Replace(strings.Trim(fmt.Sprint(id), "[]"), " ", "\",\"", -1)
|
status := 0
|
if flag == true {
|
status = 1
|
}
|
enable := strconv.Itoa(status)
|
localConf, err2 := cache.GetServerInfo()
|
if err2 !=nil || localConf.AlarmIp == "" || localConf.ServerId == "" {
|
logger.Debug("localConfig is wrong!!!")
|
return "localConf wrong"
|
}
|
url := "http://" + localConf.AlarmIp + ":" + strconv.Itoa(int(localConf.AlarmPort)) +
|
"/" + config.EsInfo.EsIndex.DbTables.IndexName + "/_update_by_query?refresh"
|
jsonDsl := `
|
{
|
"script": {
|
"lang": "painless",
|
"inline": "ctx._source.enable = ` + enable + `"
|
},
|
"query": {
|
"terms": {
|
"id": ["` + ids + `"]
|
}
|
}
|
}
|
`
|
buf, err := esutil.EsReq("POST", url, []byte(jsonDsl))
|
if err != nil {
|
logger.Debug("http request info is err!")
|
message = "修改失败"
|
}
|
var info interface{}
|
json.Unmarshal(buf, &info)
|
out, ok := info.(map[string]interface{})
|
if !ok {
|
logger.Debug("http response interface can not change map[string]interface{}")
|
message = "修改失败"
|
}
|
middle, ok := out["updated"].(float64)
|
if !ok {
|
logger.Debug("first result change error!")
|
message = "修改失败"
|
}
|
if middle > 0 {
|
logger.Debug("修改成功")
|
//if flag {//生效的
|
// discovery.AddDbMessage(&protomsg.EsPersonCacheChange{
|
// Type: protomsg.EsCacheChanged_T_DbTable,
|
// PersonId: "",
|
// TableId: id,
|
// Feature: "",
|
// Action: protomsg.DbAction_Insert,
|
// })
|
//} else {//失效的
|
// discovery.AddDbMessage(&protomsg.EsPersonCacheChange{
|
// Type: protomsg.EsCacheChanged_T_DbTable,
|
// PersonId: "",
|
// TableId: id,
|
// Feature: "",
|
// Action: protomsg.DbAction_Insert,
|
// })
|
//}
|
message = "修改成功"
|
}
|
return message
|
}
|
|
func InitEnableStatus() (messageInfo string) {
|
verificationData := GetVerificationData()
|
verificationTime := VerificationTime(verificationData)
|
idOfEffective := verificationTime["effective"]
|
idOfinvalid := verificationTime["invalid"]
|
mesInfo := false
|
effectiveMesInfo := ""
|
invalidMesInfo := ""
|
if len(idOfEffective) > 0 {
|
effectiveMesInfo = UpdateEnableStatus(idOfEffective, true)
|
} else {
|
effectiveMesInfo = "修改成功"
|
}
|
if len(idOfinvalid) > 0 {
|
invalidMesInfo = UpdateEnableStatus(idOfinvalid, false)
|
} else {
|
invalidMesInfo = "修改成功"
|
}
|
if effectiveMesInfo == "修改成功" && invalidMesInfo == "修改成功" {
|
mesInfo = true
|
}
|
if mesInfo == true {
|
messageInfo = "修改成功"
|
}
|
return messageInfo
|
}
|
