From 9303b69ea569bcb5e581147543a3fd58e90d0d25 Mon Sep 17 00:00:00 2001
From: sunty <1172534965@qq.com>
Date: 星期四, 20 八月 2020 20:05:23 +0800
Subject: [PATCH] add get buckets contrl

---
 extend/code/code.go    |    3 +
 go.sum                 |    8 ++--
 go.mod                 |    6 +-
 controllers/buckets.go |   39 +++++++++++++++++++
 router/router.go       |   44 +++++++++++----------
 5 files changed, 71 insertions(+), 29 deletions(-)

diff --git a/controllers/buckets.go b/controllers/buckets.go
new file mode 100644
index 0000000..ca4ab28
--- /dev/null
+++ b/controllers/buckets.go
@@ -0,0 +1,39 @@
+package controllers
+
+import (
+	"basic.com/pubsub/esutil.git"
+	"basic.com/valib/logger.git"
+	"github.com/gin-gonic/gin"
+	"strconv"
+	"webserver/cache"
+	"webserver/extend/code"
+	"webserver/extend/config"
+	"webserver/extend/util"
+)
+
+type Buckers struct {
+}
+
+func (b *Buckers) GetBuckets(c *gin.Context) {
+	var info interface{}
+	localConf, err := cache.GetServerInfo()
+	if err != nil || localConf.AlarmIp == "" || localConf.ServerId == "" {
+		logger.Debug("localConfig is wrong!!!")
+		util.ResponseFormat(c, code.ComError, "es config err")
+		return
+	}
+	ip := localConf.AlarmIp
+	port := strconv.Itoa(int(localConf.AlarmPort))
+	c.BindJSON(&info)
+	inf := info.(map[string]interface{})
+	startTime := inf["startTime"].(string)
+	endTime := inf["endTime"].(string)
+	cameraId := inf["cameraId"].([]string)
+	thresholdTime := inf["thresholdTime"].(float64)
+	resDate, err := esutil.GetfaceDataBucketsBycameraIdAndTime(cameraId, startTime, endTime, thresholdTime, ip, port, config.EsInfo.EsIndex.AiOcean.IndexName)
+	if err != nil {
+		util.ResponseFormat(c, code.FailedQuery, err)
+		return
+	}
+	util.ResponseFormat(c, code.Success, resDate)
+}
diff --git a/extend/code/code.go b/extend/code/code.go
index 3e18ab7..f0c08c4 100644
--- a/extend/code/code.go
+++ b/extend/code/code.go
@@ -14,7 +14,7 @@
 	// Success 璇锋眰澶勭悊鎴愬姛
 	Success       = &Code{http.StatusOK, true, "璇锋眰澶勭悊鎴愬姛"}
 	AddSuccess    = &Code{http.StatusOK, true, "娣诲姞鎴愬姛"}
-	UploadSuccess = &Code{ http.StatusOK, true, "涓婁紶鎴愬姛"}
+	UploadSuccess = &Code{http.StatusOK, true, "涓婁紶鎴愬姛"}
 	UpdateSuccess = &Code{http.StatusOK, true, "鏇存柊鎴愬姛"}
 	UpdateFail    = &Code{http.StatusBadRequest, false, "鏇存柊澶辫触"}
 	DelSuccess    = &Code{http.StatusOK, true, "鍒犻櫎鎴愬姛"}
@@ -70,4 +70,5 @@
 
 	UpgradeSuccess = &Code{http.StatusOK, true, "鍗囩骇鎴愬姛"}
 	UpgradeFail    = &Code{http.StatusInternalServerError, false, "鍗囩骇澶辫触"}
+	FailedQuery    = &Code{http.StatusInternalServerError, false, "鏌ヨ澶辫触"}
 )
diff --git a/go.mod b/go.mod
index 9f5de4d..b2ac753 100644
--- a/go.mod
+++ b/go.mod
@@ -7,7 +7,7 @@
 	basic.com/fileServer/WeedFSClient.git v0.0.0-20190919054037-0182b6c3f5cb
 	basic.com/gb28181api.git v0.0.0-20191028082253-472438a8407b
 	basic.com/pubsub/cache.git v0.0.0-20190718093725-6a413e1d7d48
-	basic.com/pubsub/esutil.git v0.0.0-20200114073900-ad9de8362777
+	basic.com/pubsub/esutil.git v0.0.0-20200820115648-7b6d324ced5d
 	basic.com/pubsub/protomsg.git v0.0.0-20200605082339-fe3f28d45337
 	basic.com/valib/capture.git v0.0.0-20191204103802-89c923cf2abe
 	basic.com/valib/deliver.git v0.0.0-20190531095353-25d8c3b20051
@@ -15,7 +15,7 @@
 	basic.com/valib/goffmpeg.git v0.0.0-20200525032638-bc3239dc0121 // indirect
 	basic.com/valib/gogpu.git v0.0.0-20190711044327-62043b070865
 	basic.com/valib/gopherdiscovery.git v0.0.0-20190605034340-15d89d8b4e28
-	basic.com/valib/licence.git v2.0.2+incompatible // indirect
+	basic.com/valib/licence.git v2.0.2+incompatible
 	basic.com/valib/logger.git v0.0.0-20190928113028-4907b08c4159
 	github.com/Microsoft/go-winio v0.4.12 // indirect
 	github.com/StackExchange/wmi v0.0.0-20190523213315-cbe66965904d // indirect
@@ -23,7 +23,7 @@
 	github.com/alecthomas/template v0.0.0-20160405071501-a0175ee3bccc
 	github.com/dgrijalva/jwt-go v3.2.0+incompatible
 	github.com/disintegration/imaging v1.6.2 // indirect
-	github.com/dustin/go-humanize v1.0.0 // indirect
+	github.com/dustin/go-humanize v1.0.0
 	github.com/gin-gonic/gin v1.4.0
 	github.com/go-ole/go-ole v1.2.4 // indirect
 	github.com/gogo/protobuf v1.2.1
diff --git a/go.sum b/go.sum
index 82c2b41..f76e507 100644
--- a/go.sum
+++ b/go.sum
@@ -6,10 +6,9 @@
 basic.com/gb28181api.git v0.0.0-20191028082253-472438a8407b/go.mod h1:iKnzuRRqUEgt17894pX3oYcOG2fLYFVcXiZbPoMK7+4=
 basic.com/pubsub/cache.git v0.0.0-20190718093725-6a413e1d7d48 h1:BBA30Rgljn6MRieC4gUncETJDyna3ObyubTo9HEQ2M0=
 basic.com/pubsub/cache.git v0.0.0-20190718093725-6a413e1d7d48/go.mod h1:gHLJZz2ee1cGL0X0ae69fs56bAxkDgEQwDhhXZJNUcY=
-basic.com/pubsub/esutil.git v0.0.0-20191120125514-865efa73a9ae h1:/j1dIDLxzEp51N+ZHZIq1xeYVK9zz8epWEAfw01uWe8=
-basic.com/pubsub/esutil.git v0.0.0-20191120125514-865efa73a9ae/go.mod h1:yIvppFPFGC61DOdm71ujnsxZBMFUu2yKjr5O43bMWCw=
-basic.com/pubsub/esutil.git v0.0.0-20200114073900-ad9de8362777 h1:gTeuhepfLgOchD6bqydsGGV6KCj/UaseQQgo4DFyhGQ=
-basic.com/pubsub/esutil.git v0.0.0-20200114073900-ad9de8362777/go.mod h1:yIvppFPFGC61DOdm71ujnsxZBMFUu2yKjr5O43bMWCw=
+basic.com/pubsub/esutil.git v0.0.0-20200820114117-66d9e889d745/go.mod h1:yIvppFPFGC61DOdm71ujnsxZBMFUu2yKjr5O43bMWCw=
+basic.com/pubsub/esutil.git v0.0.0-20200820115648-7b6d324ced5d h1:8Q4LZypFfm1SD79yxueXxrM5HmGlQBo3KvwOb1vdJ+Y=
+basic.com/pubsub/esutil.git v0.0.0-20200820115648-7b6d324ced5d/go.mod h1:yIvppFPFGC61DOdm71ujnsxZBMFUu2yKjr5O43bMWCw=
 basic.com/pubsub/protomsg.git v0.0.0-20200605082339-fe3f28d45337 h1:6LTdIfHUZkfemhkuQnXXaeXRdnm4459PyxMd36AXV20=
 basic.com/pubsub/protomsg.git v0.0.0-20200605082339-fe3f28d45337/go.mod h1:un5NV5VWQoblVLZfx1Rt5vyLgwR0jI92d3VJhfrJhWU=
 basic.com/valib/capture.git v0.0.0-20191204103802-89c923cf2abe h1:uh3u7DuSOw6AwzvPC1EM19sw1Skks1EUJddcbHDKI9M=
@@ -32,6 +31,7 @@
 cloud.google.com/go v0.37.4/go.mod h1:NHPJ89PdicEuT9hdPXMROBD91xc5uRDxsMtSB16k7hw=
 code.cloudfoundry.org/bytefmt v0.0.0-20180906201452-2aa6f33b730c/go.mod h1:wN/zk7mhREp/oviagqUXY3EwuHhWyOvAdsn5Y4CzOrc=
 github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU=
+github.com/Microsoft/go-winio v0.4.12 h1:xAfWHN1IrQ0NJ9TBC0KBZoqLjzDTr1ML+4MywiUOryc=
 github.com/Microsoft/go-winio v0.4.12/go.mod h1:VhR8bwka0BXejwEJY73c50VrPtXAaKcyvVC4A4RozmA=
 github.com/OneOfOne/xxhash v1.2.2/go.mod h1:HSdplMjZKSmBqAxg5vPj2TmRDmfkzw+cTzAElWljhcU=
 github.com/PuerkitoBio/purell v1.1.0 h1:rmGxhojJlM0tuKtfdvliR84CFHljx9ag64t2xmVkjK4=
diff --git a/router/router.go b/router/router.go
index d45ed44..5d55f7a 100644
--- a/router/router.go
+++ b/router/router.go
@@ -25,6 +25,7 @@
 	dbPersonCont := new(controllers.DbPersonController)
 	ssController := new(controllers.SysSetController)
 	sdkController := new(controllers.SdkController)
+	bucketsController := new(controllers.Buckers)
 	esSearchController := new(controllers.EsSearchController)
 	esManagementController := new(controllers.EsManagementController)
 	realTimeController := new(controllers.RealTimeController)
@@ -45,9 +46,9 @@
 	licenseController := new(controllers.LicenseController)
 	fileAnalysisC := new(controllers.FileAnalysisController)
 	fileSettingC := new(controllers.FileAnalysisSettingController)
-    icController := new(controllers.InfoController)
-    gb28181C := new(controllers.Gb28181Controller)
-    fStackC := new(controllers.FileStackController)
+	icController := new(controllers.InfoController)
+	gb28181C := new(controllers.Gb28181Controller)
+	fStackC := new(controllers.FileStackController)
 
 	urlPrefix := "/data/api-v" // wp 娣诲姞 璺緞 鍓嶇紑
 
@@ -94,11 +95,11 @@
 		area.POST("/gb28181TreeDelete", areaController.Gb28181TreeDelete)
 	}
 
-    gbApi := r.Group(urlPrefix + "/gb28181")
-    {
-        gbApi.GET("/findAreaByParentId", gb28181C.FindAreaByParentId)
-        gbApi.GET("/newGbId",gb28181C.NewGbId)
-    }
+	gbApi := r.Group(urlPrefix + "/gb28181")
+	{
+		gbApi.GET("/findAreaByParentId", gb28181C.FindAreaByParentId)
+		gbApi.GET("/newGbId", gb28181C.NewGbId)
+	}
 
 	//鎽勫儚鏈虹鐞�
 	camera := r.Group(urlPrefix + "/camera")
@@ -107,7 +108,7 @@
 		camera.PUT("/update", cameraController.CameraUpdate)
 		camera.DELETE("/del/:cid", cameraController.CameraDel)
 		camera.GET("/show/:cid", cameraController.CameraSel)
-        camera.POST("/updateCameraArea", cameraController.UpdateCameraArea)
+		camera.POST("/updateCameraArea", cameraController.UpdateCameraArea)
 		camera.GET("/updateSnapshotUrl/:cid", cameraController.UpdateSnapshotUrl)
 
 		camera.POST("/alltask", cameraTaskController.FindTasksByCameraIds)
@@ -182,6 +183,7 @@
 		es.POST("/addCluster", esManagementController.AddCluster)
 		es.POST("/createNode", esManagementController.CreateNode)
 		es.POST("/updateEsHosts", esManagementController.UpdateEsHosts)
+		es.POST("/getBuckets", bucketsController.GetBuckets)
 	}
 
 	//瀹炴椂琚皟鏁版嵁
@@ -259,10 +261,10 @@
 		vsset.POST("/upgrade", ssController.Upgrade)
 	}
 
-    ic := r.Group(urlPrefix + "/info")
-    {
-        ic.GET("/getServerName", icController.GetServerName)
-    }
+	ic := r.Group(urlPrefix + "/info")
+	{
+		ic.GET("/getServerName", icController.GetServerName)
+	}
 
 	//绠楁硶搴撴搷浣�
 	sdkApi := r.Group(urlPrefix + "/sdk")
@@ -340,14 +342,14 @@
 		fileSettingApi.POST("/changeEnable", fileSettingC.ChangeEnable)
 	}
 
-    fStackApi := r.Group(urlPrefix +"/fileStack")
-    {
-        fStackApi.GET("/findAllByPage",fStackC.FindAllByPage)
-        fStackApi.POST("/save",fStackC.Save)
-        fStackApi.GET("/show/:id", fStackC.Show)
-        fStackApi.POST("/changeEnable", fStackC.ChangeEnable)
-        fStackApi.DELETE("/delete/:id", fStackC.Delete)
-    }
+	fStackApi := r.Group(urlPrefix + "/fileStack")
+	{
+		fStackApi.GET("/findAllByPage", fStackC.FindAllByPage)
+		fStackApi.POST("/save", fStackC.Save)
+		fStackApi.GET("/show/:id", fStackC.Show)
+		fStackApi.POST("/changeEnable", fStackC.ChangeEnable)
+		fStackApi.DELETE("/delete/:id", fStackC.Delete)
+	}
 
 	// 鏂囦欢 涓婁紶
 	r.Static("static", "./static") // 闈欐�佹枃浠�

--
Gitblit v1.8.0