From 50e51f5e0e1764ea02777f9ead6be89c08a6e982 Mon Sep 17 00:00:00 2001 From: zhangmeng <775834166@qq.com> Date: 星期一, 27 五月 2019 15:41:35 +0800 Subject: [PATCH] add shm nil --- deliver.go | 49 +++++++++++++++++++++++++++++++++++++++++++------ 1 files changed, 43 insertions(+), 6 deletions(-) diff --git a/deliver.go b/deliver.go index 21d6ea9..e4b7ba4 100644 --- a/deliver.go +++ b/deliver.go @@ -1,5 +1,7 @@ package deliver +import "errors" + // Deliver define a interface how to use mangos type Deliver interface { @@ -13,21 +15,56 @@ Close() } -// NewProducer create producer args presentive for parameter with protocal, e.g. sub topic -func NewProducer(m Mode, url string, args ...interface{}) Deliver { +// NewServer create listener args presentive for parameter with protocal, e.g. sub topic +func NewServer(m Mode, url string, args ...interface{}) Deliver { if m > ModeStart && m < ModeNNG { - return nngProducer(m, url, args...) + return nngServer(m, url, args...) + } else if m == Shm { + s, err := shmServer(m, url, args...) + if err != nil { + s = nil + } + return s } return nil } -// NewConsumer create consumer args presentive for parameter with protocal, e.g. sub topic -func NewConsumer(m Mode, url string, args ...interface{}) Deliver { +// NewClient create dialer args presentive for parameter with protocal, e.g. sub topic +func NewClient(m Mode, url string, args ...interface{}) Deliver { if m > ModeStart && m < ModeNNG { - return nngConsumer(m, url, args...) + return nngClient(m, url, args...) + } else if m == Shm { + s, err := shmClient(m, url, args...) + if err != nil { + s = nil + } + return s } return nil } + +// NewServerWithError create listener args presentive for parameter with protocal, e.g. sub topic +func NewServerWithError(m Mode, url string, args ...interface{}) (Deliver, error) { + + if m > ModeStart && m < ModeNNG { + return nngServer(m, url, args...), nil + } else if m == Shm { + return shmServer(m, url, args...) + } + return nil, errors.New("please choose a suitable deliver.mode") +} + +// NewClientWithError create dialer args presentive for parameter with protocal, e.g. sub topic +func NewClientWithError(m Mode, url string, args ...interface{}) (Deliver, error) { + + if m > ModeStart && m < ModeNNG { + return nngClient(m, url, args...), nil + } else if m == Shm { + return shmClient(m, url, args...) + } + + return nil, errors.New("please choose a suitable deliver.mode") +} -- Gitblit v1.8.0