From fa924718b1e8d5f566f7655807e77df1bcfe8f86 Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期二, 27 八月 2019 14:41:54 +0800
Subject: [PATCH] add recv2 return len
---
shm.go | 8 ++++----
deliver.go | 2 +-
nng.go | 4 ++--
3 files changed, 7 insertions(+), 7 deletions(-)
diff --git a/deliver.go b/deliver.go
index 465a6c1..9f566eb 100644
--- a/deliver.go
+++ b/deliver.go
@@ -12,7 +12,7 @@
Recv() ([]byte, error)
// Recv2 recv to param
- Recv2([]byte) error
+ Recv2([]byte) (int, error)
// Close close Deliver
Close()
diff --git a/nng.go b/nng.go
index 0e89b6f..fe82e3d 100644
--- a/nng.go
+++ b/nng.go
@@ -80,9 +80,9 @@
}
// Recv2 impl interface
-func (n *NNG) Recv2(data []byte) (err error) {
+func (n *NNG) Recv2(data []byte) (l int, err error) {
data, err = n.Recv()
- return err
+ return len(data), err
}
// Close impl interface Deliver
diff --git a/shm.go b/shm.go
index 602bcfa..64e58e6 100644
--- a/shm.go
+++ b/shm.go
@@ -51,18 +51,18 @@
}
// Recv2 impl interface
-func (s *SHM) Recv2(data []byte) error {
+func (s *SHM) Recv2(data []byte) (int, error) {
if s == nil || s.rw == nil {
- return errors.New("please open shm consumer first")
+ return 0, errors.New("please open shm consumer first")
}
n, err := s.rw.Read(data)
if err == nil || err == io.EOF {
data = data[:n:n]
- return nil
+ return n, nil
}
- return err
+ return 0, err
}
// Close impl interface Deliver
--
Gitblit v1.8.0