From 22955cd5a948ace0f497c9d618914607f86dba5c Mon Sep 17 00:00:00 2001
From: liuxiaolong <736321739@qq.com>
Date: 星期四, 06 六月 2019 15:41:19 +0800
Subject: [PATCH] add hub flag

---
 server.go          |    3 ++-
 esutil/EsClient.go |    8 ++++++--
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/esutil/EsClient.go b/esutil/EsClient.go
index 385698f..6b7c6d3 100644
--- a/esutil/EsClient.go
+++ b/esutil/EsClient.go
@@ -331,7 +331,7 @@
 	return value[i-1]
 }
 
-func PostAction(sec int, Eurl string, picurl string) []byte {
+func PostAction(sec int, Eurl string, picurl string, ishub string) []byte {
 	index := "videopersons,personaction"
 	url := fmt.Sprintf("%s%s%s", Eurl, index, "/_search")
 
@@ -340,7 +340,11 @@
 	if sec > 60 {
 		preSec = "-60s"
 	}
-	prama := "{\"query\":{\"bool\":{\"filter\":[{\"range\":{\"picDate\":{\"gte\":\"now+8h-" + seccond + "s\",\"lt\":\"now+8h"+preSec+"\"}}}]}},\"size\":\"1000\",\"sort\":[{\"picDate\":{\"order\":\"desc\"}}]," +
+	var ishubReq =""
+	if ishub == "hub" {
+		ishubReq = "{\"term\":{\"personIsHub\":\"1\"}},"
+	}
+	prama := "{\"query\":{\"bool\":{\"filter\":["+ishubReq+"{\"range\":{\"picDate\":{\"gte\":\"now+8h-" + seccond + "s\",\"lt\":\"now+8h"+preSec+"\"}}}]}},\"size\":\"1000\",\"sort\":[{\"picDate\":{\"order\":\"desc\"}}]," +
 		"\"_source\":[\"baseInfo\",\"Gender\",\"BaseName\",\"Age\",\"personId\",\"personPicUrl\",\"indeviceName\",\"imgKey\",\"sdkType\",\"ageDescription\",\"indeviceid\",\"content\",\"Id\",\"picAddress\",\"picMaxUrl\",\"picDate\",\"Race\",\"videoNum\",\"picSmUrl\",\"taskName\",\"personIsHub\",\"idcard\",\"videoIp\",\"videoReqNum\"]" +
 		"}"
 	err, tokenRes := GetEsDataReq(url, prama, picurl, true)
diff --git a/server.go b/server.go
index dee518e..acb238b 100644
--- a/server.go
+++ b/server.go
@@ -19,6 +19,7 @@
 var sec = flag.Int("sec", 10, "the second for query data. ")
 
 var Level = flag.String("level","ErrorLevel","log level")
+var IsHub = flag.String("hub", "hub", "hub is personIsHub=1")
 
 func main() {
 	flag.Parse()
@@ -83,7 +84,7 @@
 }
 
 func handleMessage(conn net.Conn, connArr []net.Conn) bool {
-	jsonstring := esutil.PostAction(*sec, *Eurl, *Picurl)
+	jsonstring := esutil.PostAction(*sec, *Eurl, *Picurl, *IsHub)
 	if jsonstring == nil {
 		log.Log.Infoln("the data is nil,remoteArr:",conn.RemoteAddr())
 		conn.Write([]byte("\000"))

--
Gitblit v1.8.0