From b3a335f79fd9a6ad91705e2ea293115681484d69 Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期五, 31 五月 2019 17:36:27 +0800
Subject: [PATCH] add mode
---
deliver.go | 22 ++++++++++++++++++----
1 files changed, 18 insertions(+), 4 deletions(-)
diff --git a/deliver.go b/deliver.go
index 5d8167f..d518da2 100644
--- a/deliver.go
+++ b/deliver.go
@@ -21,7 +21,10 @@
if m > ModeStart && m < ModeNNG {
return nngServer(m, url, args...)
} else if m == Shm {
- s, _ := shmServer(m, url, args...)
+ s, err := shmServer(m, url, args...)
+ if err != nil {
+ s = nil
+ }
return s
}
return nil
@@ -33,7 +36,10 @@
if m > ModeStart && m < ModeNNG {
return nngClient(m, url, args...)
} else if m == Shm {
- s, _ := shmClient(m, url, args...)
+ s, err := shmClient(m, url, args...)
+ if err != nil {
+ s = nil
+ }
return s
}
@@ -44,7 +50,11 @@
func NewServerWithError(m Mode, url string, args ...interface{}) (Deliver, error) {
if m > ModeStart && m < ModeNNG {
- return nngServer(m, url, args...), nil
+ 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...)
}
@@ -55,7 +65,11 @@
func NewClientWithError(m Mode, url string, args ...interface{}) (Deliver, error) {
if m > ModeStart && m < ModeNNG {
- return nngClient(m, url, args...), nil
+ 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...)
}
--
Gitblit v1.8.0