From 020e17cc5311d091d713eb4fabae2a3d50944916 Mon Sep 17 00:00:00 2001 From: zhangmeng <775834166@qq.com> Date: 星期二, 27 八月 2019 13:13:33 +0800 Subject: [PATCH] add shm interface 2 --- shm.go | 18 ++++++++++++++++-- 1 files changed, 16 insertions(+), 2 deletions(-) diff --git a/shm.go b/shm.go index 7997e7d..602bcfa 100644 --- a/shm.go +++ b/shm.go @@ -39,8 +39,7 @@ // data := make([]byte, maxRecvSize) // n, err := s.rw.Read(data) // if err == nil || err == io.EOF { - // data := make([]byte, maxRecvSize) - // copy(data, s.recvData) + // data := data[:n:n] // return data, nil // } @@ -51,6 +50,21 @@ return nil, err } +// Recv2 impl interface +func (s *SHM) Recv2(data []byte) error { + if s == nil || s.rw == nil { + return 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 err +} + // Close impl interface Deliver func (s *SHM) Close() { if s == nil { -- Gitblit v1.8.0