From 6be5f42ff0e8c9671dd0e99eb2988e8c7c305d91 Mon Sep 17 00:00:00 2001
From: panlei <2799247126@qq.com>
Date: 星期二, 10 九月 2019 17:06:35 +0800
Subject: [PATCH] dial fix

---
 ruleserver/readyDataForRule.go |   72 +++++++++++++++++++++++++++--------
 1 files changed, 55 insertions(+), 17 deletions(-)

diff --git a/ruleserver/readyDataForRule.go b/ruleserver/readyDataForRule.go
index 3520ea2..cbf9a5a 100644
--- a/ruleserver/readyDataForRule.go
+++ b/ruleserver/readyDataForRule.go
@@ -1,21 +1,22 @@
 package ruleserver
 
 import (
+	"basic.com/dbapi.git"
 	bigCache "basic.com/pubsub/cache.git"
-	"basic.com/pubsub/cache.git/esutil"
 	"basic.com/pubsub/protomsg.git"
 	"encoding/json"
+	"errors"
 	"fmt"
 	"github.com/go-yaml/yaml"
+	"github.com/golang/protobuf/proto"
 	"io/ioutil"
 	"math"
 	"nanomsg.org/go-mangos"
 	"nanomsg.org/go-mangos/protocol/req"
 	"nanomsg.org/go-mangos/transport/tcp"
-	"os"
+	"net"
 	"ruleprocess/cache"
 	"ruleprocess/logger"
-	"github.com/golang/protobuf/proto"
 	"strconv"
 	"time"
 )
@@ -54,6 +55,10 @@
 	if err != nil {
 		logger.Error("鍒涘缓璇锋眰socket澶辫触: %s", err.Error())
 	}
+	serverIP, _ := GetLocalIP()
+	if err = sock.Dial("tcp://"+serverIP+":4010"); err != nil {
+		logger.Error("璇锋眰socket鎷ㄥ彿澶辫触: %s", err.Error())
+	}
 }
 
 type BaseInfo struct {
@@ -81,7 +86,7 @@
 			taskGroup = task
 		}
 	}
-	logger.Debug("褰撳墠鏁版嵁甯ц鍖归厤鐨勮鍒欑粍锛�-------------------------")
+	logger.Debug("褰撳墠鏁版嵁甯ц鍖归厤鐨勮鍒欑粍锛�-------------------------","鎽勫儚鏈篿d:",cameraId,"浠诲姟id",taskId)
 	for _,ruleGroup := range taskGroup.GroupRules{
 		logger.Info(ruleGroup.GroupText)
 	}
@@ -174,11 +179,14 @@
 		CompareThreshold:compareThreshold,
 		Source:false,
 	}
-	bytes := Push("tcp://192.168.1.182:40010",comArg,sock)
+
+	bytes := Push(comArg,sock)
+	//bytes := []byte{}
 	var scResult protomsg.SdkCompareResult
 	err1 := proto.Unmarshal(bytes, &scResult)
 	if err1 != nil {
 		logger.Error("getBaseInfo瑙e帇閿欒", err1)
+		return
 	}
 	//logger.Info("----------------------------------------map鏄�", m)
 	ids := []string{}
@@ -190,21 +198,25 @@
 	logger.Info("------------------------------------------------------------------------------------------------------------------------")
 	//logger.Info("=====================浜哄憳id鐨勯泦鍚堜负锛�",ids)
 	if len(ids) > 0 {
-		baseinfos, err1 := esutil.Dbpersoninfosbyid(ids, dbTablePersons, serverIp, serverPort)
+		var dbApi dbapi.DbPersonApi
+		baseinfos, err1 := dbApi.Dbpersoninfosbyid(ids)
+		//baseinfos, err1 := esutil.Dbpersoninfosbyid(ids, dbTablePersons, serverIp, serverPort)
 		if err1 != nil {
 			logger.Error("鏌ヨ搴曞簱浜哄憳淇℃伅鍑洪敊", err1)
 		}
+		var dtapi dbapi.DbTableApi
 		for _, baseinfo := range baseinfos {
 			// 鏍规嵁tableId鏌ヨ搴曞簱淇℃伅缁檒iker璧嬪��
 			//logger.Info("---------鐪嬬湅姣忎釜搴曞簱浜哄憳鐨勪俊鎭細",baseinfo.Id,baseinfo.PersonName)
 			tableIds := []string{}
 			tableIds = append(tableIds, baseinfo.TableId) // 铏界劧鏄紶鍏ユ暟缁勮繑鍥炴暟缁勭殑鎺ュ彛锛屼絾鎴戞寜鍗曚釜鐨勪娇鐢ㄤ簡
-			table, err := esutil.Dbtablefosbyid(tableIds, "dbtables", serverIp, serverPort)
+			table, err := dtapi.DbtablesById(tableIds)
+			//table, err := esutil.Dbtablefosbyid(tableIds, "dbtables", serverIp, serverPort)
 			if err != nil || len(table) == 0 {
 				logger.Error("鏍规嵁id鏌ヨ搴曞簱淇℃伅鍑洪敊锛�", err, "--杩斿洖鍊奸暱搴︿负锛�", len(table))
 			}
-			logger.Debug("鐪嬬湅杩欎釜base鐨勫姣斿�兼槸澶氬皯锛�", Decimal(m[baseinfo.Id].CompareScore * 100))
-			base := BaseInfo{TableId: baseinfo.TableId, TableName: table[0].TableName, BwType: table[0].BwType, CompareScore: Decimal(m[baseinfo.Id].CompareScore * 100), PersonId: baseinfo.Id, PersonName: baseinfo.PersonName, PersonPicUrl: baseinfo.PersonPicUrl, PhoneNum: baseinfo.PhoneNum, Sex: baseinfo.Sex, IdCard: baseinfo.IdCard, MonitorLevel: baseinfo.MonitorLevel, Content: baseinfo.Reserved}
+			logger.Debug("鐪嬬湅杩欎釜base鐨勫姣斿�兼槸澶氬皯锛�", Decimal(m[baseinfo.Id].CompareScore))
+			base := BaseInfo{TableId: baseinfo.TableId, TableName: table[0].TableName, BwType: table[0].BwType, CompareScore: Decimal(m[baseinfo.Id].CompareScore), PersonId: baseinfo.Id, PersonName: baseinfo.PersonName, PersonPicUrl: baseinfo.PersonPicUrl, PhoneNum: baseinfo.PhoneNum, Sex: baseinfo.Sex, IdCard: baseinfo.IdCard, MonitorLevel: baseinfo.MonitorLevel, Content: baseinfo.Reserved}
 			//os.Exit(1)
 			arg.Liker = append(arg.Liker, &base)
 		}
@@ -216,7 +228,7 @@
 func Compare(args *SdkDatas, groupRule *protomsg.GroupRule) {
 	compareFlag := 0
 	var tableIds []string
-	var threshold float32 = 0.5 // 榛樿闃堝�间负0.5
+	var threshold float32 = 50 // 榛樿闃堝�间负50
 	// 鐪嬬湅鏄惁鏈夊彧閰嶄汉鑴告瘮瀵圭畻娉曚絾娌℃湁閰嶅姣斿簱鐨勮鍒欙紝濡傛灉鏈夛紝鍒欐瘮瀵瑰璞′负鍏ㄩ儴搴曞簱
 	for j := 0; j < len(groupRule.Rules); j++ {
 		if groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a972" { // 閰嶄簡浜鸿劯姣斿
@@ -304,7 +316,7 @@
 			a.filterData = append(a.filterData, &arg1)
 		}
 	}
-	//logger.Info("鍖哄煙鏄細",areaPoints,"鍖哄煙鍐呯洰鏍囨暟閲忎负锛�",a.targetNum,"---",len(a.filterData))
+	logger.Info("鍖哄煙鏄細",areaPoints,"鍖哄煙鍐呯洰鏍囨暟閲忎负锛�",a.targetNum,"---",len(a.filterData))
 	a.time = time.Unix(time.Now().Unix(), 0).String()[11:16]
 	a.keepRight = arg.KeepRight
 	a.isStatic = arg.IsStatic
@@ -326,7 +338,7 @@
 	}
 }
 
-func Push(url string,data *protomsg.CompareArgs,sock mangos.Socket) []byte{
+func Push(data *protomsg.CompareArgs,sock mangos.Socket) []byte{
 	//var sock mangos.Socket
 	var err error
 	var msg []byte
@@ -341,9 +353,7 @@
 	}
 	//sock.AddTransport(ipc.NewTransport())
 	sock.AddTransport(tcp.NewTransport())
-	if err = sock.Dial(url); err != nil {
-		logger.Error("璇锋眰socket鎷ㄥ彿澶辫触: %s", err.Error())
-	}
+
 	logger.Info("搴忓垪鍖栨暟鎹�")
 	bytes,err1 := proto.Marshal(data)
 	logger.Info("鏁版嵁闀垮害涓猴細",len(bytes))
@@ -354,12 +364,40 @@
 	//bytes := []byte("ndfasojdfaidsos")
 	if err = sock.Send(bytes); err != nil {
 		logger.Error("鎺ㄩ�乻ocket鍙戦�佹暟鎹け璐�: %s", err.Error())
-		os.Exit(1)
+		//os.Exit(1)
 	}
 	if msg, err = sock.Recv(); err != nil {
 		logger.Error("鎺ユ敹鍝嶅簲澶辫触: %s", err.Error())
-		os.Exit(1)
+		//os.Exit(1)
 	}
 	logger.Debug("鏁版嵁鎺ㄩ�佹垚鍔燂紒鏀跺埌鍝嶅簲,鏁版嵁闀垮害涓猴細",len(msg))
 	return msg
 }
+
+// 鑾峰彇鏈満ip
+func GetLocalIP() (ipv4 string, err error) {
+	var (
+		addrs   []net.Addr
+		addr    net.Addr
+		ipNet   *net.IPNet // IP鍦板潃
+		isIpNet bool
+	)
+	// 鑾峰彇鎵�鏈夌綉鍗�
+	if addrs, err = net.InterfaceAddrs(); err != nil {
+		return
+	}
+	// 鍙栫涓�涓潪lo鐨勭綉鍗P
+	for _, addr = range addrs {
+		// 杩欎釜缃戠粶鍦板潃鏄疘P鍦板潃: ipv4, ipv6
+		if ipNet, isIpNet = addr.(*net.IPNet); isIpNet && !ipNet.IP.IsLoopback() {
+			// 璺宠繃IPV6
+			if ipNet.IP.To4() != nil {
+				ipv4 = ipNet.IP.String() // 192.168.1.1
+				return
+			}
+		}
+	}
+
+	err = errors.New("ipv4 not found")
+	return
+}

--
Gitblit v1.8.0