From 5abbb114d703b40d2a42e9e3eecdadcb6d341dcf Mon Sep 17 00:00:00 2001
From: zhangzengfei <zhangzengfei@smartai.com>
Date: 星期一, 29 四月 2024 03:51:11 +0800
Subject: [PATCH] 修复keepalive地址
---
repository/captureRepo.go | 170 +++++++++++++++++++++++++++++++++++++-------------------
1 files changed, 112 insertions(+), 58 deletions(-)
diff --git a/repository/captureRepo.go b/repository/captureRepo.go
index 56a1caf..c5516e1 100644
--- a/repository/captureRepo.go
+++ b/repository/captureRepo.go
@@ -3,6 +3,7 @@
import (
"encoding/base64"
"encoding/json"
+ "gat1400Exchange/client"
"time"
"gat1400Exchange/config"
@@ -22,8 +23,10 @@
}
func (c CaptureRepository) FaceForward(faceList []vo.FaceObject) {
+ var err error
+
if faceList == nil || len(faceList) == 0 {
- logger.Warn("faceList is nil")
+ logger.Warn("FaceList is nil")
return
}
@@ -35,21 +38,51 @@
var deviceId = face.DeviceID
var faceId = face.FaceID
- var faceImageStr string
+ var bgImageStr, imageType string
+ var bgImageBytes, faceImageBytes []byte = nil, nil
// 鑾峰彇澶у浘, 鐩墠娴峰悍鐨勫皬鍥惧垎杈ㄧ巼澶綆
for _, image := range face.SubImageList.SubImageInfoObject {
- if len(image.Data) > len(faceImageStr) {
- faceImageStr = image.Data
+ if len(image.Data) > len(bgImageStr) {
+ bgImageStr = image.Data
+ imageType = image.Type
+ }
+ }
+
+ bgImageBytes, err = base64.StdEncoding.DecodeString(bgImageStr)
+ if err != nil {
+ logger.Warn("Decode Image Base64 String failure, %s", err.Error())
+ continue
+ }
+
+ // 鍒ゆ柇鍥剧墖绫诲瀷鏄惁涓哄満鏅浘, 鏍规嵁浜鸿劯鍧愭爣鍒囧皬鍥�.
+ if imageType == "14" && config.ForwardConf.CutFaceImage {
+ faceRect := &vo.Rect{
+ Left: face.LeftTopX,
+ Top: face.LeftTopY,
+ Right: face.RightBtmX,
+ Bottom: face.RightBtmY,
+ }
+
+ faceImageBytes, err = util.SubCutImg(bgImageBytes, faceRect, 90)
+ if err != nil {
+ logger.Warn("Cut face image failure, %s", err.Error())
}
}
// 杞彂鍥惧儚
- logger.Debug("Prepare forward image, deviceId:%s, image len:%d, server:%s", deviceId, len(faceImageStr), config.ForwardConf.SyncServer)
- if deviceId != "" && faceImageStr != "" && config.ForwardConf.SyncServer != "" {
- pd := c.PackPushDataV2(deviceId, faceId, faceImageStr, face.FaceAppearTime)
+ logger.Debug("Prepare forward image, deviceId:%s, image len:%d, server:%s", deviceId, len(bgImageStr), config.ForwardConf.SyncServer)
+ if deviceId != "" && bgImageStr != "" && config.ForwardConf.SyncServer != "" {
+ pd := c.PackPushDataV2(deviceId, faceId, face.FaceAppearTime, bgImageBytes, faceImageBytes)
if pd == nil {
return
+ }
+
+ // 閮ㄧ讲鍦ㄦ湇鍔＄鐨�1400浠ｇ悊, 浠呮帴鏀�1400淇℃伅, 妤煎眰淇℃伅鏆傚瓨鍦╫herFeature瀛楁
+ if config.ServeConf.Role == "proxy" {
+ if face.OtherFeature != "" {
+ pd.CameraFloor = face.OtherFeature
+ }
}
payload, err := json.Marshal(pd)
@@ -58,51 +91,17 @@
return
}
- if !c.SendData(payload, config.ForwardConf.SyncServer) {
- c.CacheData(payload)
- logger.Warn("鏁版嵁杞彂澶辫触, 娣诲姞缂撳瓨")
+ if !util.SendData(payload, config.ForwardConf.SyncServer) {
+ cacheItem, _ := json.Marshal(pd)
+ c.CacheData(cacheItem, "basic")
+ logger.Warn("The data forwarding failed, adding to local cache.")
} else {
- logger.Debug("鏁版嵁杞彂鎴愬姛")
+ logger.Debug("The data forwarding successful. deviceId:%s", deviceId)
}
}
}
return
-}
-
-func (c CaptureRepository) PackPushDataV2(deviceId, faceId, faceImage, appearTime string) *vo.PushDataInfoV2 {
- var pd = new(vo.PushDataInfoV2)
- var device models.Device
-
- if err := device.FindById(deviceId); err != nil {
- logger.Warn("Can't find device in database, device:%s, %s", deviceId, err.Error())
- return pd
- }
-
- // 鍖归厤妤煎眰
- faceAppearTime, err := time.ParseInLocation("20060102150405", appearTime, time.Local)
- if err != nil {
- logger.Warn("Parse face appear time error, %s", err.Error())
- faceAppearTime = time.Now()
- }
-
- var devPos models.Positions
- _ = devPos.FindDevicePosition(deviceId, faceAppearTime.Unix()+5) // 鍔�5绉掔數姊叧闂ㄧ殑鏃堕棿
-
- imageBytes, err := base64.StdEncoding.DecodeString(faceImage)
- if err != nil {
- logger.Warn("Decode Image Base64 String failure, %s", err.Error())
- return pd
- }
-
- pd.PicMaxImages = append(pd.PicMaxImages, imageBytes)
- pd.PicId = faceId
- pd.PicDate = faceAppearTime.Format("2006-01-02 15:04:05")
- pd.DataSource = "camera"
- pd.CameraFloor = devPos.Pos
- pd.CameraId = deviceId
-
- return pd
}
func (c CaptureRepository) PackPushData(deviceId, faceId, faceImage, appearTime string) *vo.PushDataInfo {
@@ -159,30 +158,85 @@
return pd
}
-func (c CaptureRepository) SendData(payload []byte, url string) bool {
- body, err := util.HttpPost(url, nil, payload)
+func (c CaptureRepository) PackPushDataV2(deviceId, faceId, appearTime string, bgImgBytes, faceImgBytes []byte) *vo.PushDataInfoV2 {
+ var pd = new(vo.PushDataInfoV2)
+ var floor string
+
+ faceAppearTime, err := time.ParseInLocation("20060102150405", appearTime, time.Local)
if err != nil {
- logger.Error("Post request failure, url:%s, err:%s", err.Error())
- return false
+ logger.Warn("Parse face appear time error, %s", err.Error())
+ faceAppearTime = time.Now()
}
- var rsp vo.SyncServerResponse
- err = json.Unmarshal(body, &rsp)
- if err != nil {
- logger.Error("Can't parse response, url:%s, rsp: %v", url, body)
- return false
+ if config.ServeConf.Role == "server" {
+ var device models.Device
+
+ if err := device.FindById(deviceId); err != nil {
+ logger.Warn("Can't find device in database, device:%s, %s", deviceId, err.Error())
+ return pd
+ }
+
+ // 鍖归厤妤煎眰
+ var devPos models.Positions
+ _ = devPos.FindDevicePosition(deviceId, faceAppearTime.Unix()+5) // 鍔�5绉掔數姊叧闂ㄧ殑鏃堕棿
+ if devPos.Pos == "" {
+ devPos.Pos = device.Floor
+ }
+
+ floor = devPos.Pos
}
- return rsp.Success
+ pd.PicMaxImages = append(pd.PicMaxImages, bgImgBytes)
+
+ if faceImgBytes != nil {
+ pd.PicSmImages = append(pd.PicSmImages, faceImgBytes)
+ }
+
+ pd.PicId = faceId
+ pd.PicDate = faceAppearTime.Format("2006-01-02 15:04:05")
+ pd.DataSource = "camera"
+ pd.CameraFloor = floor
+ pd.CameraId = deviceId
+
+ return pd
}
-func (c CaptureRepository) CacheData(payload []byte) {
+func (c CaptureRepository) CacheData(payload []byte, msgType string) {
var cacheItem = models.Cache{
+ Type: msgType,
Data: string(payload),
CreateTime: time.Now().Unix(),
Retry: 0,
}
- models.GetDB().Table(cacheItem.TableName()).Create(&c)
+ cacheItem.Save()
+}
+func (c CaptureRepository) MsgForward(msg *vo.RequestFaceList) {
+ faceInfo := msg.FaceListObject.FaceObject[0]
+ // 鍖归厤妤煎眰
+ faceAppearTime, err := time.ParseInLocation("20060102150405", faceInfo.FaceAppearTime, time.Local)
+ if err != nil {
+ logger.Warn("Parse face appear time error, %s", err.Error())
+ faceAppearTime = time.Now()
+ }
+
+ var devPos models.Positions
+ _ = devPos.FindPositionByTime(faceAppearTime.Unix() + 5) // 鍔�5绉掔數姊叧闂ㄧ殑鏃堕棿
+ if devPos.Pos == "" {
+ devPos.Pos = "1F"
+ }
+
+ for idx, _ := range msg.FaceListObject.FaceObject {
+ msg.FaceListObject.FaceObject[idx].OtherFeature = devPos.Pos
+ }
+
+ b, _ := json.Marshal(msg)
+ if client.FaceCapture(b) != vo.StatusSuccess {
+ cacheItem, _ := json.Marshal(msg)
+ c.CacheData(cacheItem, "1400")
+ logger.Warn("The data forwarding failed, adding to local cache.")
+ }
+
+ return
}
--
Gitblit v1.8.0