From 383557064fd77c028fd71facf8ec1cf6b278d3e6 Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期五, 30 八月 2019 09:39:08 +0800
Subject: [PATCH] add send msg param
---
nng.go | 25 ++++++++++++++++++++-----
1 files changed, 20 insertions(+), 5 deletions(-)
diff --git a/nng.go b/nng.go
index 2ce7b68..032d929 100644
--- a/nng.go
+++ b/nng.go
@@ -28,6 +28,8 @@
mode Mode
url string
+ sendMsg *mangos.Message
+
arguments []interface{}
}
@@ -49,9 +51,16 @@
time.Sleep(time.Duration(surveyorTime*2) * time.Second)
}
- msg := mangos.NewMessage(len(data))
- msg.Body = data
- return n.sock.SendMsg(msg)
+ // msg := mangos.NewMessage(len(data))
+ // msg.Body = data
+ // return n.sock.SendMsg(msg)
+
+ if n.sendMsg == nil {
+ n.sendMsg = &mangos.Message{}
+ n.sendMsg.Header = make([]byte, 32)
+ }
+ n.sendMsg.Body = data
+ return n.sock.SendMsg(n.sendMsg)
}
@@ -77,6 +86,13 @@
}
return msg.Body, nil
+}
+
+// Recv2 impl interface
+func (n *NNG) Recv2(data []byte) (l int, err error) {
+ data, err = n.Recv()
+ l = len(data)
+ return l, err
}
// Close impl interface Deliver
@@ -127,8 +143,7 @@
switch n.mode {
case Bus:
sock, err = n.busMakeNNG(typ)
- case ReqRep:
- case SurvResp:
+ case ReqRep, SurvResp:
sock, err = n.rrMakeNNG(typ)
default:
sock, err = n.ppMakeNNG(typ)
--
Gitblit v1.8.0