From 971bd16379d411f93eef96f935b9ad92c372569f Mon Sep 17 00:00:00 2001 From: zhangmeng <775834166@qq.com> Date: 星期五, 30 八月 2019 10:17:14 +0800 Subject: [PATCH] use mangos msgchache --- deliver.go | 80 ++++++++++++++++++++++++++++++++++++++++ 1 files changed, 80 insertions(+), 0 deletions(-) diff --git a/deliver.go b/deliver.go index 9220db3..9f566eb 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 { @@ -8,6 +10,9 @@ // Recv recv data Recv() ([]byte, error) + + // Recv2 recv to param + Recv2([]byte) (int, error) // Close close Deliver Close() @@ -18,6 +23,12 @@ if m > ModeStart && m < ModeNNG { return nngServer(m, url, args...) + } else if m == Shm { + s, err := shmServer(m, url, args...) + if err != nil { + s = nil + } + return s } return nil } @@ -27,7 +38,76 @@ if m > ModeStart && m < ModeNNG { 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 { + 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") +} + +// 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 { + 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") +} + +// 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