From 0c853f9c71a13d2a43a74524843476db08e15c8c Mon Sep 17 00:00:00 2001
From: panlei <2799247126@qq.com>
Date: 星期二, 02 七月 2019 20:56:54 +0800
Subject: [PATCH] 人脸大图路径截取
---
insertdata/insertDataToEs.go | 47 +++++++++++++++++++++++++++++++++--------------
1 files changed, 33 insertions(+), 14 deletions(-)
diff --git a/insertdata/insertDataToEs.go b/insertdata/insertDataToEs.go
index d73a0ec..2d6e35b 100644
--- a/insertdata/insertDataToEs.go
+++ b/insertdata/insertDataToEs.go
@@ -90,7 +90,7 @@
type AlarmRule struct {
GroupId string `json:"groupId"`
- AlarmLevel int32 `json:"alarmLevel"`
+ AlarmLevel string `json:"alarmLevel"`
RuleText string `json:"ruleText"`
}
@@ -98,9 +98,9 @@
func InsertToEs(msg ruleserver.ResultMsg) {
// 鐩存帴浠庤鍒欑殑鏍囩鏁版嵁閲屾嬁绗﹀悎瑙勫垯鐨勪汉鑴哥粨鏋�
- if msg.RuleResult["faces"] != nil && len(msg.RuleResult["faces"].([]ruleserver.Arg)) > 0 {
+ if msg.RuleResult["face"] != nil && len(msg.RuleResult["face"].([]ruleserver.Arg)) > 0 {
log.Println("寰�ES鎻掍汉鑴告暟鎹�")
- for _, face := range msg.RuleResult["faces"].([]ruleserver.Arg) {
+ for _, face := range msg.RuleResult["face"].([]ruleserver.Arg) {
// 涓婁紶澶у浘
// 瑙ｅ帇缂╁苟涓婁紶鍥剧墖
bdata, err := util.UnCompress(msg.Data)
@@ -126,9 +126,8 @@
}
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())
+ bytes := util.SubImg(i, 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.PostFormBufferData1(weedfsUrl, bytes, uuid.NewV4().String())
if err != nil {
log.Println("涓婁紶灏忓浘鍑洪敊")
}
@@ -137,12 +136,12 @@
msg.Cid,
msg.Caddr,
time.Now().Format("2006-01-02 15:04:05"),
- bigPhotoUrl["fileUrl"].(string),
+ strings.Split(bigPhotoUrl["fileUrl"].(string), "/")[1],
msg.Tasklab.Taskid,
msg.Tasklab.Taskname,
face.SdkName,
"",
- "", // 鍙娴嬶紝娌℃湁姣斿鏃堕棿
+ time.Now().Format("2006-01-02 15:04:05"), // 鍙娴嬶紝娌℃湁姣斿鏃堕棿
face.ThftRes.Gender,
face.ThftRes.Age,
"",
@@ -150,7 +149,7 @@
face.ThftRes.Smile,
face.ThftRes.Beauty,
"涓嶆槸姣忎釜浜鸿劯绠楁硶閮芥湁",
- resp["fileUrl"].(string),
+ strings.Split(resp["fileUrl"].(string), "/")[1],
"鏆傛棤闆嗙兢",
localConfig.ServerId,
localConfig.ServerName,
@@ -170,9 +169,7 @@
}
err = EsReq("POST", "http://192.168.1.182:9200/videopersons/perVideoPicture", requstbody)
- if err != nil {
- log.Println("es can not execute right.")
- }
+ log.Println(err)
}else {
log.Println("璺熷簳搴撶殑鐩镐技鏁版嵁---------锛�",face.Liker)
}
@@ -184,7 +181,8 @@
alarmRules := []AlarmRule{}
for _, yoloResult := range msg.RuleResult["yolo"].([]ruleserver.Result) {
sdkNames = sdkNames + yoloResult.SdkName
- alarmRules = append(alarmRules, AlarmRule{yoloResult.RuleGroupId, yoloResult.AlarmLevel, yoloResult.RuleText})
+ alarm := ChangeToString(yoloResult.AlarmLevel)
+ alarmRules = append(alarmRules, AlarmRule{yoloResult.RuleGroupId, alarm, yoloResult.RuleText})
}
isAlarm := ""
resp := make(map[string]interface{})
@@ -254,7 +252,7 @@
if err != nil {
log.Println("寰�ES鎻掑叆鏁版嵁澶辫触",err)
} else {
- log.Println("__________________________________________寰�ES鎻掑叆鏁版嵁鎴愬姛")
+ log.Println("__________________________________________寰�ES鎻掑叆鏁版嵁yolo鎴愬姛")
}
}
}
@@ -286,3 +284,24 @@
err = errors.New("ipv4 not found")
return
}
+
+// 鎶婃姤璀︾瓑绾ц浆鍖栨垚姹夊瓧
+func ChangeToString(i int32) string{
+ alarm := ""
+ if i == 1 {
+ alarm = "涓�绾�"
+ }
+ if i == 2 {
+ alarm = "浜岀骇"
+ }
+ if i == 3 {
+ alarm = "涓夌骇"
+ }
+ if i == 4 {
+ alarm = "鍥涚骇"
+ }
+ if i == 5 {
+ alarm = "浜旂骇"
+ }
+ return alarm
+}
\ No newline at end of file
--
Gitblit v1.8.0