From e3c33684b4907cc338866623d2362eb85100387d Mon Sep 17 00:00:00 2001
From: panlei <2799247126@qq.com>
Date: 星期一, 28 十月 2019 16:35:08 +0800
Subject: [PATCH] 修改一下推送成功条件

---
 labelFilter/req.go |   14 ++++++++++----
 1 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/labelFilter/req.go b/labelFilter/req.go
index e0c47c6..7fb7bc2 100644
--- a/labelFilter/req.go
+++ b/labelFilter/req.go
@@ -9,14 +9,13 @@
 	"nanomsg.org/go-mangos/protocol/req"
 	"nanomsg.org/go-mangos/transport/ipc"
 	"nanomsg.org/go-mangos/transport/tcp"
-	"os"
 	"ruleprocess/structure"
 	"time"
 )
 
 func Die(format string, v ...interface{}) {
 	logger.Info("+++++++",format)
-	os.Exit(1)
+	//os.Exit(1)
 }
 
 func date() string {
@@ -66,6 +65,11 @@
 		fmt.Errorf("Failed set MaxRecvSize: %v", err)
 		return
 	}
+	errTimeOut := sock.SetOption(mangos.OptionRecvDeadline,time.Millisecond * 2000)
+	if errTimeOut != nil {
+		fmt.Errorf("Failed set MaxRecvDeadline: %v", err)
+		return
+	}
 	//sock.AddTransport(ipc.NewTransport())
 	sock.AddTransport(tcp.NewTransport())
 	if err = sock.Dial(url); err != nil {
@@ -82,10 +86,12 @@
 	if err = sock.Send(bytes); err != nil {
 		Die("鎺ㄩ�乻ocket鍙戦�佹暟鎹け璐�: %s", err.Error())
 	}
-	if msg, err = sock.Recv(); err != nil {
+	msg, err = sock.Recv();
+	if err != nil {
 		Die("鎺ユ敹鍝嶅簲澶辫触: %s", err.Error())
+	} else {
+		logger.Debug("浜嬩欢鎺ㄩ�佹垚鍔燂紒鏀跺埌鍝嶅簲",string(msg))
 	}
-	logger.Debug("浜嬩欢鎺ㄩ�佹垚鍔燂紒鏀跺埌鍝嶅簲",string(msg))
 	sock.Close()
 }
 

--
Gitblit v1.8.0