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