From 5bf9f5cc0ad2689de4e34d628d8ac3168df5142a Mon Sep 17 00:00:00 2001
From: zhangzengfei <zhangzengfei@smartai.com>
Date: 星期五, 23 八月 2024 17:41:41 +0800
Subject: [PATCH] 修复编译bug

---
 controller/captureCtl.go |   29 +++++++++++++++++++++++++++++
 1 files changed, 29 insertions(+), 0 deletions(-)

diff --git a/controller/captureCtl.go b/controller/captureCtl.go
index 7e53382..79efc0f 100644
--- a/controller/captureCtl.go
+++ b/controller/captureCtl.go
@@ -5,6 +5,7 @@
 	"fmt"
 	"math/rand"
 	"net/http"
+	"path"
 	"strconv"
 	"time"
 
@@ -13,6 +14,7 @@
 	"gat1400Exchange/pkg/logger"
 	"gat1400Exchange/repository"
 	"gat1400Exchange/service"
+	"gat1400Exchange/util"
 	"gat1400Exchange/vo"
 
 	"github.com/gin-gonic/gin"
@@ -46,6 +48,29 @@
 	face := req.FaceListObject.FaceObject[0]
 	logger.Debug("Receive new face message, ip:%s, device id:%s faceId:%s, LeftTopX:%d, appearTime:%s", c.RemoteIP(), face.DeviceID, face.FaceID, face.LeftTopX, face.FaceAppearTime)
 
+	if config.ClientConf.UploadType == "url" {
+		for idx, img := range face.SubImageList.SubImageInfoObject {
+			if img.Type != "14" {
+				continue
+			}
+
+			imageBytes, err := base64.StdEncoding.DecodeString(img.Data)
+			if err != nil {
+				logger.Warn("Decode Small Image Base64 String failure, %s", err.Error())
+				c.AbortWithStatus(http.StatusBadRequest)
+				return
+			}
+
+			today := time.Now().Format("2006-01-02")
+			imagePath := path.Join(config.LogConf.Path, "gat1400_face_images", today, img.ImageID+".jpeg")
+			err = util.WriteToFile(imagePath, imageBytes)
+			if err == nil {
+				face.SubImageList.SubImageInfoObject[idx].Data = ""
+				face.SubImageList.SubImageInfoObject[idx].StoragePath = config.ImageConf.ImageUriPrefix + "/" + today + "/" + img.ImageID + ".jpeg"
+			}
+		}
+	}
+
 	// 濡傛灉寮�鍚簡涓嬬骇, 韬唤搴旇鏄秷鎭唬鐞�, 涓嶅啀杞彂鍒版湇鍔″櫒
 	if config.ClientConf.Enable && config.ServeConf.Role == "agent" {
 		go a.Repository.VIIDFaceMsgForward(&req)
@@ -57,6 +82,10 @@
 		go a.Repository.FaceForward(req.FaceListObject.FaceObject)
 	}
 
+	if config.ForwardConf.RecordServer != "" {
+		go a.Repository.PubRecordMessage(face.DeviceID, face.FaceID)
+	}
+
 	rspMsg := vo.ResponseStatus{
 		RequestURL:   c.FullPath(),
 		StatusCode:   vo.StatusSuccess,

--
Gitblit v1.8.0