From 912a0a1894caa4666e2bae9c92c99bb261415479 Mon Sep 17 00:00:00 2001
From: panlei <2799247126@qq.com>
Date: 星期六, 29 六月 2019 19:15:04 +0800
Subject: [PATCH] ---
---
insertdata/insertDataToEs.go | 272 +++++++++++++++++++++++++++++------------------------
1 files changed, 149 insertions(+), 123 deletions(-)
diff --git a/insertdata/insertDataToEs.go b/insertdata/insertDataToEs.go
index 082e6b3..f7ab22a 100644
--- a/insertdata/insertDataToEs.go
+++ b/insertdata/insertDataToEs.go
@@ -1,19 +1,23 @@
package insertdata
import (
+ "encoding/json"
"errors"
"fmt"
- "encoding/json"
+ "log"
"net"
+ "strings"
"time"
- "basic.com/pubsub/protomsg.git"
"basic.com/dbapi.git"
+ "basic.com/pubsub/protomsg.git"
"github.com/golang/protobuf/proto"
- "ruleprocess/ruleserver"
"github.com/satori/go.uuid"
+ "ruleprocess/ruleserver"
+ "ruleprocess/util"
)
+var weedfsUrl = "http://192.168.1.182:6333/submit"
// 浜鸿劯鐨勬暟鎹粨鏋�
type PerVideoPicture struct {
Id string `json:"id"`
@@ -62,165 +66,187 @@
// yolo琛屼负鐨勬暟鎹粨鏋�
type Personaction struct {
- Id string `json:"id"`
- CameraId string `json:"cameraId"`
- CameraName string `json:"cameraName"`
- CameraAddr string `json:"cameraAddr"`
- TaskId string `json:"taskId"`
- TaskName string `json:"taskName"`
- SdkName string `json:"sdkName"`
- Content string `json:"content"`
- AlarmRules []AlarmRule
- AnalyServerId string `json:"analyServerId"`
- AnalyServerName string `json:"analyServerName"`
- AnalyServerIp string `json:"analyServerIp"`
- ClusterId string `json:"clusterId"`
- PicUrl string `json:"picUrl"`
- PicDate string `json:"picDate"`
- VideoUrl string `json:"videoUrl"`
- IsAlarm string `json:"isAlarm"`
- IsAckAlarm string `json:"isAckAlarm"`
- IsCollect string `json:"isCollect"`
- IsDelete int `json:"isDelete"`
+ Id string `json:"id"`
+ CameraId string `json:"cameraId"`
+ CameraName string `json:"cameraName"`
+ CameraAddr string `json:"cameraAddr"`
+ TaskId string `json:"taskId"`
+ TaskName string `json:"taskName"`
+ SdkName string `json:"sdkName"`
+ Content string `json:"content"`
+ AlarmRules []AlarmRule `json: "alramRules"`
+ AnalyServerId string `json:"analyServerId"`
+ AnalyServerName string `json:"analyServerName"`
+ AnalyServerIp string `json:"analyServerIp"`
+ ClusterId string `json:"clusterId"`
+ PicSmUrl []string `json:"picSmUrl"`
+ PicDate string `json:"picDate"`
+ VideoUrl string `json:"videoUrl"`
+ IsAlarm string `json:"isAlarm"`
+ IsAckAlarm string `json:"isAckAlarm"`
+ IsCollect string `json:"isCollect"`
+ IsDelete int `json:"isDelete"`
}
type AlarmRule struct {
GroupId string `json:"groupId"`
- AlarmLevel int32 `json:"alarmLevel"`
+ AlarmLevel int32 `json:"alarmLevel"`
RuleText string `json:"ruleText"`
}
+// 寰�ES鎻掓暟鎹�
func InsertToEs(msg ruleserver.ResultMsg) {
- fmt.Println("寰�ES鎻掓暟鎹�")
- for _, sdkinfo := range msg.Tasklab.Sdkinfos {
- if sdkinfo.Sdktype == "FaceDetect" {
- faceParam := protomsg.ParamFacePos{}
- err1 := proto.Unmarshal(sdkinfo.Sdkdata, &faceParam)
- if err1 != nil {
- fmt.Println("瑙ｆ瀽FACE sdk鏈夎", err1)
- continue
+ // 鐩存帴浠庤鍒欑殑鏍囩鏁版嵁閲屾嬁绗﹀悎瑙勫垯鐨勪汉鑴哥粨鏋�
+ if msg.RuleResult["faces"] != nil && len(msg.RuleResult["faces"].([]ruleserver.Arg)) > 0 {
+ log.Println("寰�ES鎻掍汉鑴告暟鎹�")
+ for _, face := range msg.RuleResult["faces"].([]ruleserver.Arg) {
+ // 涓婁紶澶у浘
+ // 瑙ｅ帇缂╁苟涓婁紶鍥剧墖
+ bdata, err := util.UnCompress(msg.Data)
+ if err != nil {
+ panic("瑙ｅ帇缂╁浘鐗囨椂鍑虹幇閿欒")
}
- for _, face := range faceParam.Faces {
+ i := protomsg.Image{}
+ err = proto.Unmarshal(bdata, &i)
+ bigPhotoUrl := make(map[string]interface{})
+ bigPhotoUrl, err = util.PostFormBufferData(weedfsUrl, i, uuid.NewV4().String())
+ fmt.Println(bigPhotoUrl)
+ if len(face.Liker) == 0 {
+ // 浜鸿劯妫�娴嬶紝娌℃湁鐩镐技鐨勫簳搴撲汉鍛�
+ flag, localConfig := dbapi.SysSetApi{}.GetServerInfo()
+ if !flag {
+ log.Println("鏌ヨ鏈満淇℃伅澶辫触锛�")
+ }
+ serverIp, err := GetLocalIP()
+ // 瑙ｅ帇缂╁苟涓婁紶鍥剧墖
+ bdata, err := util.UnCompress(msg.Data)
+ if err != nil {
+ panic("瑙ｅ帇缂╁浘鐗囨椂鍑虹幇閿欒")
+ }
+ i := protomsg.Image{}
+ err = proto.Unmarshal(bdata, &i)
+ i1 := protomsg.Image{}
+ i1 = util.Subimg(i.Data, int(face.Location.X), int(face.Location.Y), int(face.Location.X+face.Location.Width), int(face.Location.Y+face.Location.Height))
+ resp, err := util.PostFormBufferData(weedfsUrl, i1, uuid.NewV4().String())
+ if err != nil {
+ log.Println("涓婁紶灏忓浘鍑洪敊")
+ }
pervideo := PerVideoPicture{
uuid.NewV4().String(),
msg.Cid,
msg.Caddr,
time.Now().Format("2006-01-02 15:04:05"),
- "",
+ bigPhotoUrl["fileUrl"].(string),
msg.Tasklab.Taskid,
msg.Tasklab.Taskname,
- sdkinfo.SdkName,
+ face.SdkName,
"",
+ "", // 鍙娴嬶紝娌℃湁姣斿鏃堕棿
+ face.ThftRes.Gender,
+ face.ThftRes.Age,
"",
- face.Result.Gender,
- face.Result.Age,
- "",
- face.Result.Race,
- face.Result.Smile,
- face.Result.Beauty,
+ face.ThftRes.Race,
+ face.ThftRes.Smile,
+ face.ThftRes.Beauty,
"涓嶆槸姣忎釜浜鸿劯绠楁硶閮芥湁",
- "---",
- "",
- "",
- "",
- "",
+ resp["fileUrl"].(string),
+ "鏆傛棤闆嗙兢",
+ localConfig.ServerId,
+ localConfig.ServerName,
+ serverIp,
"",
"",
"",
"",
0,
- Base{
- "鏄瘡涓汉鑴哥畻娉曢兘鏈夊悧",
- "",
- 0,
- "",
- "",
- "",
- "",
- "",
- "",
- "",
- "",
- },
+ Base{},
}
requstbody, err := json.Marshal(pervideo)
if err != nil {
- fmt.Println("json parse error ", err)
+ log.Println("json parse error ", err)
return
}
err = EsReq("POST", "http://192.168.1.182:9200/videopersons/perVideoPicture", requstbody)
if err != nil {
- fmt.Println("es can not execute right.")
+ log.Println("es can not execute right.")
}
}
}
- if sdkinfo.Sdktype == "Yolo" {
- yoloObj := protomsg.ParamYoloObj{}
- err1 := proto.Unmarshal(sdkinfo.Sdkdata, &yoloObj)
- if err1 != nil {
- fmt.Println("瑙ｆ瀽YOLO sdk鏈夎", err1)
- //continue
- }
- alarmRules := []AlarmRule{}
- for _,result := range msg.RuleResult {
- alarmRules = append(alarmRules,AlarmRule{result.RuleGroupId,result.AlarmLevel,result.RuleText})
- }
- isAlarm := ""
- if len(alarmRules) > 0 {
- isAlarm = "1"
- }else {
- isAlarm = "0"
- continue
- }
- // 鏌ヨ鏈満淇℃伅
- flag,localConfig := dbapi.SysSetApi{}.GetServerInfo()
- if !flag {
- fmt.Println("鏌ヨ鏈満淇℃伅澶辫触锛�")
- }
- // 鏌ヨcameraName
- camera,err := dbapi.CameraApi{}.GetCameraById(msg.Cid)
- if err == nil {
- fmt.Println("鏌ヨ鎽勫儚鏈轰俊鎭け璐�")
- }
- serverIp,err := GetLocalIP()
- peraction := Personaction{
- uuid.NewV4().String(),
- msg.Cid,
- camera.Name,
- msg.Caddr,
- msg.Tasklab.Taskid,
- msg.Tasklab.Taskname,
- sdkinfo.SdkName,
- "",
- alarmRules,
- localConfig.ServerId,
- localConfig.ServerName,
- serverIp,
- "",
- "",
- time.Now().Format("2006-01-02 15:04:05"),
- "",
- isAlarm,
- "",
- "",
- 0,
- }
- requstbody, err := json.Marshal(peraction)
-
- if err != nil {
- fmt.Println("json parse error ", err)
- return
-
- }
- err = EsReq("POST", "http://192.168.1.182:9200/personaction/perVideoAction", requstbody)
- if err != nil {
- fmt.Println("es can not execute right.")
- }
+ }
+ if msg.RuleResult["yolo"] != nil && len(msg.RuleResult["yolo"].([]ruleserver.Result)) > 0 {
+ log.Println("寰�ES鎻抷olo鏁版嵁")
+ var sdkNames string = ""
+ alarmRules := []AlarmRule{}
+ for _, yoloResult := range msg.RuleResult["yolo"].([]ruleserver.Result) {
+ sdkNames = sdkNames + yoloResult.SdkName
+ alarmRules = append(alarmRules, AlarmRule{yoloResult.RuleGroupId, yoloResult.AlarmLevel, yoloResult.RuleText})
}
+ isAlarm := ""
+ resp := make(map[string]interface{})
+ if len(alarmRules) > 0 {
+ isAlarm = "1"
+ // 瑙ｅ帇缂╁苟涓婁紶鍥剧墖
+ bdata, err := util.UnCompress(msg.Data)
+ if err != nil {
+ panic("瑙ｅ帇缂╁浘鐗囨椂鍑虹幇閿欒")
+ }
+ i := protomsg.Image{}
+ err = proto.Unmarshal(bdata, &i)
+ resp, err = util.PostFormBufferData(weedfsUrl, i, uuid.NewV4().String())
+ log.Println("宸叉姤璀﹀苟涓婁紶鏀瑰抚鍥剧墖鍒版湇鍔″櫒")
+ } else {
+ isAlarm = "0"
+ // 涓嶆槸鎶ヨ鏁版嵁涓嶅瓨
+ return
+ }
+ log.Println("鍥剧墖涓婁紶杩斿洖鍊硷細", resp)
+ // 鏌ヨ鏈満淇℃伅
+ flag, localConfig := dbapi.SysSetApi{}.GetServerInfo()
+ if !flag {
+ log.Println("鏌ヨ鏈満淇℃伅澶辫触锛�")
+ }
+ // 鏌ヨcameraName
+ camera, err := dbapi.CameraApi{}.GetCameraById(msg.Cid)
+ if err != nil {
+ log.Println("鏌ヨ鎽勫儚鏈轰俊鎭け璐�")
+ }
+ serverIp, err := GetLocalIP()
+ peraction := Personaction{
+ uuid.NewV4().String(),
+ msg.Cid,
+ camera.Name,
+ msg.Caddr,
+ msg.Tasklab.Taskid,
+ msg.Tasklab.Taskname,
+ sdkNames,
+ "",
+ alarmRules,
+ localConfig.ServerId,
+ localConfig.ServerName,
+ serverIp,
+ "",
+ []string{strings.Split(resp["fileUrl"].(string), "/")[1]},
+ time.Now().Format("2006-01-02 15:04:05"),
+ "",
+ isAlarm,
+ "",
+ "",
+ 0,
+ }
+ requstbody, err := json.Marshal(peraction)
+ if err != nil {
+ log.Println("json parse error ", err)
+ return
+
+ }
+ err = EsReq("POST", "http://192.168.1.182:9200/personaction/perVideoAction", requstbody)
+ if err != nil {
+ log.Println("es can not execute right.")
+ }
}
}
--
Gitblit v1.8.0