From 43e52a52e24ad4fd00d351b21259b9a76affbaed Mon Sep 17 00:00:00 2001 From: 554325746@qq.com <554325746@qq.com> Date: 星期四, 22 八月 2019 14:40:06 +0800 Subject: [PATCH] memleak bug fix --- shm.go | 20 +++++++++++--------- 1 files changed, 11 insertions(+), 9 deletions(-) diff --git a/shm.go b/shm.go index 9260cdd..7899793 100644 --- a/shm.go +++ b/shm.go @@ -14,6 +14,8 @@ type SHM struct { rw *shm.ReadWriteCloser typ td + + data []byte } // Send impl interface Diliver @@ -53,12 +55,11 @@ ch := make(chan []byte) go func(){ - data := make([]byte, maxRecvSize) - n, err := s.rw.Read(data) + n, err := s.rw.Read(s.data) if err == nil || err == io.EOF { - data = data[:n:n] + s.data = s.data[:n:n] } - ch <- data + ch <- s.data }() select{ case d := <-ch: @@ -108,13 +109,12 @@ return nil, errors.New("shmServer created recv error parameters") } } + blocks, size := 2, maxRecvSize - if len(param) != 2 { - return nil, errors.New("shmServer created recv too much parameters") + if len(param) == 2 { + blocks, size = param[0], param[1] + // return nil, errors.New("shmServer created recv too much parameters") } - - blocks, size = param[0], param[1] - shm.Unlink(url) rw, err := shm.CreateSimplex(url, 0644, blocks, size) @@ -123,6 +123,7 @@ return &SHM{ rw, agent, + make([]byte, maxRecvSize), }, nil } @@ -139,6 +140,7 @@ return &SHM{ rw, coactee, + make([]byte, maxRecvSize), }, nil } return nil, err -- Gitblit v1.8.0