From e98330371e72127945f974eaddef9d53a1a74f2b Mon Sep 17 00:00:00 2001
From: panlei <2799247126@qq.com>
Date: 星期二, 10 九月 2019 18:06:31 +0800
Subject: [PATCH] update call compare

---
 ruleserver/readyDataForRule.go |   75 ++++++++++++++++++++++++++++++++-----
 1 files changed, 65 insertions(+), 10 deletions(-)

diff --git a/ruleserver/readyDataForRule.go b/ruleserver/readyDataForRule.go
index b64d22b..e4430c7 100644
--- a/ruleserver/readyDataForRule.go
+++ b/ruleserver/readyDataForRule.go
@@ -51,12 +51,11 @@
 	}
 	logger.Debug("鏈満淇℃伅鍜宻erver淇℃伅锛�", localConfig, serverIp, serverPort)
 	bigCache.Init(dbTablePersons, serverIp, serverPort, localConfig.ServerId)
-	sock, err = req.NewSocket();
-	if err != nil {
-		logger.Error("鍒涘缓璇锋眰socket澶辫触: %s", err.Error())
-	}
+	go Push1()
 }
 
+var sender chan *protomsg.CompareArgs = make(chan *protomsg.CompareArgs)
+var receiver chan []byte = make(chan []byte)
 type BaseInfo struct {
 	TableId      string  `json:"tableId"`
 	TableName    string  `json:"tableName"`
@@ -175,12 +174,14 @@
 		CompareThreshold:compareThreshold,
 		Source:false,
 	}
-	serverIp, _ := GetLocalIP()
-	bytes := Push("tcp://"+serverIp+":40010",comArg,sock)
+
+	bytes := getCompareMsg(comArg)
+	//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{}
@@ -332,7 +333,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
@@ -345,11 +346,13 @@
 		logger.Error("Failed set MaxRecvSize: %v", err)
 		return nil
 	}
-	//sock.AddTransport(ipc.NewTransport())
-	sock.AddTransport(tcp.NewTransport())
-	if err = sock.Dial(url); err != nil {
+	serverIP, _ := GetLocalIP()
+	if err = sock.Dial("tcp://"+serverIP+":4010"); err != nil {
 		logger.Error("璇锋眰socket鎷ㄥ彿澶辫触: %s", err.Error())
 	}
+	//sock.AddTransport(ipc.NewTransport())
+	sock.AddTransport(tcp.NewTransport())
+
 	logger.Info("搴忓垪鍖栨暟鎹�")
 	bytes,err1 := proto.Marshal(data)
 	logger.Info("鏁版嵁闀垮害涓猴細",len(bytes))
@@ -370,6 +373,58 @@
 	return msg
 }
 
+
+func Push1(){
+	//var sock mangos.Socket
+	var err error
+	var msg []byte
+
+	if sock, err = req.NewSocket(); err != nil {
+		logger.Error("鍒涘缓璇锋眰socket澶辫触: %s", err.Error())
+	}
+	errSize := sock.SetOption(mangos.OptionMaxRecvSize,5*1024*1024)
+	if errSize != nil {
+		logger.Error("Failed set MaxRecvSize: %v", err)
+	}
+	//sock.AddTransport(ipc.NewTransport())
+	sock.AddTransport(tcp.NewTransport())
+	serverIP, _ := GetLocalIP()
+	if err = sock.Dial("tcp://"+serverIP+":4010"); err != nil {
+		logger.Error("璇锋眰socket鎷ㄥ彿澶辫触: %s", err.Error())
+	}
+	logger.Info("搴忓垪鍖栨暟鎹�")
+
+	for {
+		select {
+		// case <-ctx.Done():
+		// 	return
+		case data := <- sender:
+			bytes,err1 := proto.Marshal(data)
+			logger.Info("鏁版嵁闀垮害涓猴細",len(bytes))
+			if err1 != nil {
+				logger.Info("搴忓垪鍖栧け璐ワ細",err1)
+			}
+			logger.Debug("鎺ㄩ�佹暟鎹�")
+			//bytes := []byte("ndfasojdfaidsos")
+			if err = sock.Send(bytes); err != nil {
+				logger.Error("鎺ㄩ�乻ocket鍙戦�佹暟鎹け璐�: %s", err.Error())
+				//os.Exit(1)
+			}
+			if msg, err = sock.Recv(); err != nil {
+				logger.Error("鎺ユ敹鍝嶅簲澶辫触: %s", err.Error())
+				//os.Exit(1)
+			}
+			logger.Debug("鏁版嵁鎺ㄩ�佹垚鍔燂紒鏀跺埌鍝嶅簲,鏁版嵁闀垮害涓猴細",len(msg))
+			receiver <- msg
+		default:
+
+		}
+	}
+}
+func getCompareMsg(data *protomsg.CompareArgs) []byte{
+	sender <- data
+	return <-receiver
+}
 // 鑾峰彇鏈満ip
 func GetLocalIP() (ipv4 string, err error) {
 	var (

--
Gitblit v1.8.0