From aaae9917c3df122e2bbcf591417c1b58021d79fc Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期三, 15 五月 2019 16:58:03 +0800
Subject: [PATCH] add close

---
 deliver.go |    7 +++++--
 nng.go     |    7 +++++++
 2 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/deliver.go b/deliver.go
index 87effb0..21d6ea9 100644
--- a/deliver.go
+++ b/deliver.go
@@ -8,13 +8,16 @@
 
 	// Recv recv data
 	Recv() ([]byte, error)
+
+	// Close close Deliver
+	Close()
 }
 
 // NewProducer create producer args presentive for parameter with protocal, e.g. sub topic
 func NewProducer(m Mode, url string, args ...interface{}) Deliver {
 
 	if m > ModeStart && m < ModeNNG {
-		return nngProducer(m, url, args)
+		return nngProducer(m, url, args...)
 	}
 	return nil
 }
@@ -23,7 +26,7 @@
 func NewConsumer(m Mode, url string, args ...interface{}) Deliver {
 
 	if m > ModeStart && m < ModeNNG {
-		return nngConsumer(m, url, args)
+		return nngConsumer(m, url, args...)
 	}
 
 	return nil
diff --git a/nng.go b/nng.go
index 00a8dc2..61b5e76 100644
--- a/nng.go
+++ b/nng.go
@@ -58,6 +58,13 @@
 	return n.sock.Recv()
 }
 
+// Close impl interface Deliver
+func (n *NNG) Close() {
+	if n.sock != nil {
+		n.sock.Close()
+	}
+}
+
 // nngProducer create from deliver Mode
 func nngProducer(m Mode, url string, args ...interface{}) *NNG {
 

--
Gitblit v1.8.0