From 6887a385c690306a57e15b0ff3bad6138bd051c8 Mon Sep 17 00:00:00 2001 From: zhangmeng <775834166@qq.com> Date: 星期四, 16 五月 2019 17:18:45 +0800 Subject: [PATCH] change name --- nng.go | 27 +++++++++++++++++---------- 1 files changed, 17 insertions(+), 10 deletions(-) diff --git a/nng.go b/nng.go index 242409a..009ef8d 100644 --- a/nng.go +++ b/nng.go @@ -33,11 +33,10 @@ return errors.New("please init NNG first") } - // switch n.sock.GetProtocol().Number() { - // case mangos.ProtoSurveyor: - // time.Sleep(surveyorTime * 2) - // default: - // } + if surveyorTime > 0 { + time.Sleep(time.Duration(surveyorTime*2) * time.Second) + } + if n.raw { msg := mangos.NewMessage(len(data)) msg.Body = data @@ -71,11 +70,13 @@ } } -func nngListener(m Mode, url string, args ...interface{}) *NNG { +func nngServer(m Mode, url string, args ...interface{}) *NNG { rmExistedIpcName(url) if sock, err := newSocket(protoProducer(m)); err == nil { if err = setSocketOptions(sock, args); err != nil { + sock.Close() + sock = nil return nil } if err = sock.Listen(url); err != nil { @@ -92,10 +93,12 @@ return nil } -func nngDialer(m Mode, url string, args ...interface{}) *NNG { +func nngClient(m Mode, url string, args ...interface{}) *NNG { if sock, err := newSocket(protoConsumer(m)); err == nil { if err = setSocketOptions(sock, args); err != nil { + sock.Close() + sock = nil return nil } @@ -117,7 +120,7 @@ // maxRecvSize max recv size var ( maxRecvSize = 33 * 1024 * 1024 - surveyorTime = time.Second / 2 + surveyorTime = -1 ) func defualtSocketOptions() map[string]interface{} { @@ -152,10 +155,14 @@ for _, arg := range args { switch arg.(type) { case int: - surveyorTime = time.Duration(arg.(int)/2) * time.Second + if arg.(int) < 2 { + surveyorTime = 1 + } else { + surveyorTime = arg.(int) / 2 + } default: } - options[mangos.OptionSurveyTime] = surveyorTime + options[mangos.OptionSurveyTime] = time.Duration(surveyorTime) * time.Second } default: fmt.Println("no additional args") -- Gitblit v1.8.0