From a59cbf5e0cc306d862385e8d2670d2b0537ea9eb Mon Sep 17 00:00:00 2001 From: zhangmeng <775834166@qq.com> Date: 星期二, 27 八月 2019 15:01:00 +0800 Subject: [PATCH] recover --- deliver.go | 35 +++++++++++++++++++++++++++++++++++ 1 files changed, 35 insertions(+), 0 deletions(-) diff --git a/deliver.go b/deliver.go index d518da2..465a6c1 100644 --- a/deliver.go +++ b/deliver.go @@ -11,6 +11,9 @@ // Recv recv data Recv() ([]byte, error) + // Recv2 recv to param + Recv2([]byte) error + // Close close Deliver Close() } @@ -76,3 +79,35 @@ return nil, errors.New("please choose a suitable deliver.mode") } + +// NewServerWithTimeout with timeout +func NewServerWithTimeout(m Mode, url string, tm int, args ...interface{}) (Deliver, error) { + if m > ModeStart && m < ModeNNG { + setTimeOut(tm) + s := nngServer(m, url, args...) + if s == nil { + return nil, errors.New("nng create server error") + } + return s, nil + } else if m == Shm { + return shmServer(m, url, args...) + } + return nil, errors.New("please choose a suitable deliver.mode") +} + +// NewClientWithTimeout with timeout, tm unit millisecond +func NewClientWithTimeout(m Mode, url string, tm int, args ...interface{}) (Deliver, error) { + if m > ModeStart && m < ModeNNG { + setTimeOut(tm) + + s := nngClient(m, url, args...) + if s == nil { + return nil, errors.New("nng create client error") + } + return s, nil + } else if m == Shm { + return shmClient(m, url, args...) + } + + return nil, errors.New("please choose a suitable deliver.mode") +} -- Gitblit v1.8.0