From 7c811247ecf143e08c576986a884bedadc57dd66 Mon Sep 17 00:00:00 2001 From: liuxiaolong <liuxiaolong@aiotlink.com> Date: 星期五, 05 六月 2020 18:29:41 +0800 Subject: [PATCH] add refresh token to resp --- controllers/initForData.go | 74 ++++++++++++++++++++++++------------ 1 files changed, 49 insertions(+), 25 deletions(-) diff --git a/controllers/initForData.go b/controllers/initForData.go index 8d595d9..abd5bbb 100644 --- a/controllers/initForData.go +++ b/controllers/initForData.go @@ -1,7 +1,6 @@ package controllers import ( - "basic.com/dbapi.git" "basic.com/valib/logger.git" "fmt" "github.com/gin-gonic/gin" @@ -17,6 +16,7 @@ type InitForData struct { } +// @Security ApiKeyAuth // @Summary 鎶撴媿瀹炴椂鍥炬暟鎹垵濮嬪寲 // @Description 瀹炴椂鑾峰彇鏁版嵁 // @Accept json @@ -31,44 +31,56 @@ //searchBody := make(map[string]interface{}, 0) var searchBody RealTimeArg c.BindJSON(&searchBody) - index := config.EsInfo.EsIndex.VideoPersons.IndexName + "," + config.EsInfo.EsIndex.Personaction.IndexName + index := config.EsInfo.EsIndex.AiOcean.IndexName localConf, err2 := cache.GetServerInfo() - if err2 !=nil || localConf.AlarmIp == "" { + if err2 != nil || localConf.AlarmIp == "" || localConf.ServerId == "" { logger.Debug("localConfig is wrong!!!") - util.ResponseFormat(c,code.ComError,"localConf wrong") + util.ResponseFormat(c, code.ComError, "localConf wrong") return } url := "http://" + localConf.AlarmIp + ":" + strconv.Itoa(int(localConf.AlarmPort)) + "/" + index + "/_search" cameraIdStr := "" - if searchBody.TreeNodes !=nil { + linkTagInfoCameraIdStr := "" + if searchBody.TreeNodes != nil { cameraId := searchBody.TreeNodes if cameraId != nil && len(cameraId) > 0 { esCameraId := strings.Replace(strings.Trim(fmt.Sprint(cameraId), "[]"), " ", "\",\"", -1) cameraIdStr = "{\"terms\":{\"cameraId\":[\"" + esCameraId + "\"]}}," + linkTagInfoCameraIdStr = "{\"terms\":{\"linkTagInfo.cameraId\":[\"" + esCameraId + "\"]}}," } } - var setApi dbapi.SysSetApi - _, sysconf := setApi.GetServerInfo() - analyServerFilterStr := "{\"term\":{\"analyServerId\":\"" + sysconf.ServerId + "\"}}" + analyServerFilterStr := "{\"term\":{\"analyServerId\":\"" + localConf.ServerId + "\"}}" + linkTagInfoAnalyServerFilterStr := "{\"term\":{\"linkTagInfo.analyServerId\":\"" + localConf.ServerId + "\"}}" - prama := "{\"query\":{\"bool\":{\"filter\":[{\"term\":{\"isAlarm\":\"1\"}}," + + prama := "{\"query\":{\"bool\":{\"should\":[" + + "{\"bool\":{\"filter\":[" + + "{\"term\":{\"isAlarm\":true}}," + cameraIdStr + analyServerFilterStr + - "]}},\"size\":\"20\",\"sort\":[{\"picDate\":{\"order\":\"desc\"}}]," + - "\"_source\":[\"baseInfo\",\"alarmRules\",\"sex\",\"analyServerName\",\"sdkName\",\"ageDescription\",\"content\",\"id\",\"cameraAddr\",\"picMaxUrl\",\"picDate\",\"race\",\"videoUrl\",\"picSmUrl\",\"taskName\",\"isAlarm\",\"isAlarm\",\"analyServerIp\",\"cameraId\"]" + + "]}}," + + "{\"bool\":{\"filter\":[" + + "{\"term\":{\"isAlarm\":true}}," + + linkTagInfoCameraIdStr + + linkTagInfoAnalyServerFilterStr + + "]}}" + + "],\"minimum_should_match\":1}}," + + "\"size\":\"20\",\"sort\":[{\"picDate\":{\"order\":\"desc\"}}]," + + "\"_source\":{\"includes\":[],\"excludes\":[\"*.feature\"]}" + "}" - //logger.Debug("InitForCaptureData:", prama) + logger.Debug("InitForCaptureData:", prama) tokenRes := esutil.GetEsDataReq(url, prama, true) - util.ResponseFormat(c, code.Success, tokenRes) + tmpAllDate := esutil.ResponseData(tokenRes) + util.ResponseFormat(c, code.Success, tmpAllDate) } type InitForMonitorArg struct { TreeNodes []string `json:"treeNodes"` - Tasks []string `json:"tasks"` + Tasks []string `json:"tasks"` } +// @Security ApiKeyAuth // @Summary 瀹炴椂浠诲姟鐩戞帶鏁版嵁鍒濆鍖� // @Description 瀹炴椂鐩戞帶姣斿鏁版嵁 // @Accept json @@ -82,43 +94,55 @@ //searchBody := make(map[string]interface{}, 0) var searchBody InitForMonitorArg c.BindJSON(&searchBody) - index := config.EsInfo.EsIndex.VideoPersons.IndexName + "," + config.EsInfo.EsIndex.Personaction.IndexName + index := config.EsInfo.EsIndex.AiOcean.IndexName cameraIdStr := "" + linkTagInfoCameraIdStr := "" cameraId := searchBody.TreeNodes if cameraId != nil && len(cameraId) > 0 { esCameraId := strings.Replace(strings.Trim(fmt.Sprint(cameraId), "[]"), " ", "\",\"", -1) cameraIdStr = "{\"terms\":{\"cameraId\":[\"" + esCameraId + "\"]}}," + linkTagInfoCameraIdStr = "{\"terms\":{\"linkTagInfo.cameraId\":[\"" + esCameraId + "\"]}}," } //鍒ゆ柇浠诲姟ID - taskIdStr := "{\"terms\":{\"taskId\":[]}}," + taskIdStr := "" + linkTagInfoTaskIdStr := "" taskId := searchBody.Tasks if taskId != nil && len(taskId) > 0 { esTaskId := strings.Replace(strings.Trim(fmt.Sprint(taskId), "[]"), " ", "\",\"", -1) taskIdStr = "{\"terms\":{\"taskId\":[\"" + esTaskId + "\"]}}," + linkTagInfoTaskIdStr = "{\"terms\":{\"linkTagInfo.taskId\":[\"" + esTaskId + "\"]}}," } //璇锋眰澶� localConf, err2 := cache.GetServerInfo() - if err2 !=nil || localConf.AlarmIp == "" { + if err2 != nil || localConf.AlarmIp == "" || localConf.ServerId == "" { logger.Debug("localConfig is wrong!!!") - util.ResponseFormat(c,code.ComError,"localConf wrong") + util.ResponseFormat(c, code.ComError, "localConf wrong") return } url := "http://" + localConf.AlarmIp + ":" + strconv.Itoa(int(localConf.AlarmPort)) + "/" + index + "/_search" - var setApi dbapi.SysSetApi - _, sysconf := setApi.GetServerInfo() - analyServerFilterStr := "{\"term\":{\"analyServerId\":\"" + sysconf.ServerId + "\"}}" - prama := "{\"query\":{\"bool\":{\"filter\":[" + - cameraIdStr + + analyServerFilterStr := "{\"term\":{\"analyServerId\":\"" + localConf.ServerId + "\"}}" + linkTagInfoAnalyServerFilterStr := "{\"term\":{\"linkTagInfo.analyServerId\":\"" + localConf.ServerId + "\"}}" + + prama := "{\"query\":{\"bool\":{\"should\":[" + + "{\"bool\":{\"filter\":[" + taskIdStr + + cameraIdStr + analyServerFilterStr + "]}}," + + "{\"bool\":{\"filter\":[" + + linkTagInfoTaskIdStr + + linkTagInfoCameraIdStr + + linkTagInfoAnalyServerFilterStr + + "]}}" + + "],\"minimum_should_match\":1}}," + "\"sort\":[{\"picDate\":{\"order\":\"desc\"}}]," + "\"size\":\"20\"," + - "\"_source\":[\"baseInfo\",\"alarmRules\",\"sex\",\"analyServerName\",\"sdkName\",\"ageDescription\",\"content\",\"id\",\"cameraAddr\",\"picMaxUrl\",\"picDate\",\"race\",\"videoUrl\",\"picSmUrl\",\"taskName\",\"personIsHub\",\"isAlarm\",\"analyServerIp\",\"cameraId\"]}" + "\"_source\":{\"includes\":[],\"excludes\":[\"*.feature\"]}}" //logger.Debug("url:", url) //logger.Debug("InitForMonitoringData:", prama) tokenRes := esutil.GetEsDataReq(url, prama, true) - util.ResponseFormat(c, code.Success, tokenRes) + tmpAllDate := esutil.ResponseData(tokenRes) + util.ResponseFormat(c, code.Success, tmpAllDate) } -- Gitblit v1.8.0