From c2ef56536ce95b2623b5e649c3df88bca14f28c0 Mon Sep 17 00:00:00 2001
From: panlei <2799247126@qq.com>
Date: 星期四, 20 六月 2019 17:49:33 +0800
Subject: [PATCH] 解压缩图片

---
 ruleserver/ruleToformula.go |    8 ++++----
 main.go                     |   23 ++++++++++++++++-------
 2 files changed, 20 insertions(+), 11 deletions(-)

diff --git a/main.go b/main.go
index 64321ca..6b5506e 100644
--- a/main.go
+++ b/main.go
@@ -4,7 +4,7 @@
 	"fmt"
 	"ruleprocess/ruleserver"
 	"sync"
-
+	"analysis/work"
 	"basic.com/pubsub/protomsg.git"
 	"basic.com/valib/deliver.git"
 	"github.com/golang/protobuf/proto"
@@ -39,7 +39,8 @@
 			} else {
 				arg := ruleserver.ArgsFromSdk{}
 				m := paramFormat(msg,&arg)
-				arg.MainJudge()
+				fmt.Println("瑙f瀽鍑烘潵鐨勬暟鎹細",arg)
+				ruleserver.MainJudge(&arg)
 				// 鎶奱rg閲岀殑鎵撶殑鏍囩鎷垮嚭鏉ョ粰m鍐嶅皝瑁呬竴灞�
 				resultMag := ruleserver.ResultMsg{SdkMessage: m, RuleResult: arg.RuleResult}
 				fmt.Println("鎵撳畬鏍囩鍚庣殑缁撴灉锛�",resultMag)
@@ -50,18 +51,26 @@
 
 // 灏嗗閮ㄤ紶杩涙潵鐨剆dk鏁版嵁鍖呰В鎴� ArgsFromSdk
 func paramFormat(msg []byte,arg *ruleserver.ArgsFromSdk) protomsg.SdkMessage {
+	defer func() {
+		if err := recover();err != nil{
+			fmt.Println("瑙e寘杩囩▼鐨勯敊璇�",err.(string))
+		}
+
+	}()
 	// 鍙嶅簭鍒楀寲鏁版嵁寰楀埌sdk鍏ュ弬
 	m := protomsg.SdkMessage{}
 	err := proto.Unmarshal(msg, &m)
 	if err != nil {
-		fmt.Println("瑙f瀽msg鏃跺嚭鐜伴敊璇�", err)
-		return m
+		panic("瑙f瀽msg鏃跺嚭鐜伴敊璇�")
 	}
 	arg.CameraId = m.Cid
 	arg.TaskId = m.Tasklab.Taskid
-
+	bdata,err := UnCompress(m.Data)
+	if err != nil {
+		panic("瑙e帇缂╁浘鐗囨椂鍑虹幇閿欒")
+	}
 	i := protomsg.Image{}
-	err = proto.Unmarshal(m.Data, &i)
+	err = proto.Unmarshal(bdata, &i)
 	arg.ImageWidth = int(i.Width)
 	arg.ImageHeight = int(i.Height)
 	// 鏆傛椂鍐欐锛宻dk杩樻病鏈夎繖淇╃畻娉�
@@ -69,7 +78,7 @@
 	arg.IsStatic = false
 	fmt.Println("浠巑ongos涓嬁鍒扮殑鏁版嵁鍖呴暱搴︿负锛�", len(msg))
 	for _, sdkinfo := range m.Tasklab.Sdkinfos {
-		if sdkinfo.Sdktype == "yolo" {
+		if sdkinfo.Sdktype == "Yolo" {
 			yoloParam := protomsg.ParamYoloObj{}
 			err = proto.Unmarshal(sdkinfo.Sdkdata, &yoloParam)
 			if err != nil {
diff --git a/ruleserver/ruleToformula.go b/ruleserver/ruleToformula.go
index 6eaf1c1..dd8b75f 100644
--- a/ruleserver/ruleToformula.go
+++ b/ruleserver/ruleToformula.go
@@ -146,7 +146,7 @@
 }
 
 // 瑙勫垯涓诲嚱鏁板叆鍙�
-func (arg *ArgsFromSdk) MainJudge() {
+func MainJudge(arg *ArgsFromSdk) {
 	cameraPolygons := GetPolygons(arg.CameraId)
 	list := AreaMapList{}
 	for _, polygon := range cameraPolygons {
@@ -257,7 +257,7 @@
 		// 鏍规嵁cameraId鍘绘洿鏂版垨鑰呮彃鍏ョ粨鏋�,鐒跺悗鍒ゆ柇鏄惁鏁扮粍鏄惁鍙互寰楀嚭鎶ヨ鐨勭粨璁�
 		// 寰�鑱斿姩浠诲姟鐨勭粨鏋滄暟缁勯噷鏀惧�兼垨鏇存柊
 		for _, va := range timeEle.RuleResults {
-			if aml.areaMapList[0].cameraId != "" && va.CameraId == aml.areaMapList[0].cameraId {
+			if arg.CameraId != "" && va.CameraId == arg.CameraId {
 				va.Result = strconv.FormatBool(isOk)
 			}
 		}
@@ -291,7 +291,7 @@
 		fmt.Println("杩欏抚鍥惧儚鍦ㄤ换鍔′笅鐨勪竴鏁存潯瑙勫垯涓嬶紙鑱斿姩浠诲姟涓嬪氨鏄窡鏈憚鍍忔満鍍忕浉鍏崇殑灏忚鍒欙級鐨勫垽鏂粨鏋滀负false")
 		// 鎵�浠ヤ篃瑕佸幓缁撴灉鏁扮粍閲屾斁鍊兼垨鏇存柊
 		for _, va := range timeEle.RuleResults {
-			if aml.areaMapList[0].cameraId != "" && va.CameraId == aml.areaMapList[0].cameraId { // aml.areaMapList[0].cameraId 闅忎究鎵句竴涓暟鎹�
+			if arg.CameraId != "" && va.CameraId == arg.CameraId { // arg.CameraId 闅忎究鎵句竴涓暟鎹�
 				va.Result = strconv.FormatBool(isOk)
 			}
 		}
@@ -369,7 +369,7 @@
 // 瀵瑰崟甯у浘鍍忕殑鍒ゆ柇 鏄垗寮冿紙鎴栬�呰瀵逛簬鏌愪簺闇�姹傚彲浠ユ斁ES鏁版嵁搴撲竴浠斤級杩樻槸杩斿洖
 func judge(aml *AreaMapList, arg *ArgsFromSdk) {
 	// 寰楀埌灞炰簬璇ユ憚鍍忔満鐨勮嫢骞茬粍浠诲姟鐨勫畬鏁磋鍒欙紙璺熸瘡涓�鏉″畬鏁磋鍒欐瘮杈冧箣鍚庡緱鍑烘湰寮犲浘鍍忓浜庢煇涓鍒欐槸鍚︽姤璀︾殑缁撴灉銆傛斁杩沵ap锛屾瘮濡傛湰甯у浘鍍忕殑id锛屾墍纰版挒鎴愬姛鐨勮鍒檌d锛�
-	taskRuleList := GetRuleGroup(aml.areaMapList[0].cameraId)
+	taskRuleList := GetRuleGroup(arg.CameraId)
 	if len(taskRuleList) > 0 {
 		for _, taskRule := range taskRuleList {
 			ruleList := taskRule.GroupRules // 鑾峰彇鐨勬槸task涓嬮潰鐨勪换鍔$粍

--
Gitblit v1.8.0