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