From 3603e569db5a358cdb8ff7d645566e7718a815f9 Mon Sep 17 00:00:00 2001 From: 554325746@qq.com <554325746@qq.com> Date: 星期三, 21 八月 2019 15:05:25 +0800 Subject: [PATCH] bug fix reverse --- deliver.go | 32 ++++++++++++++++++++++++++++++++ 1 files changed, 32 insertions(+), 0 deletions(-) diff --git a/deliver.go b/deliver.go index d518da2..c22dc79 100644 --- a/deliver.go +++ b/deliver.go @@ -76,3 +76,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