From 0af498fd328f89961f9358ff25215c247c8f89d3 Mon Sep 17 00:00:00 2001 From: 554325746@qq.com <554325746@qq.com> Date: 星期三, 21 八月 2019 15:06:22 +0800 Subject: [PATCH] update --- shm.go | 56 +++++++++++++++++++++++++++++++++++++++++++++----------- 1 files changed, 45 insertions(+), 11 deletions(-) diff --git a/shm.go b/shm.go index 911c2d9..1eb9f56 100644 --- a/shm.go +++ b/shm.go @@ -5,6 +5,8 @@ "fmt" "io" + "time" + "github.com/tmthrgd/shm-go" ) @@ -20,12 +22,26 @@ return errors.New("please init shm producer first") } - n, err := s.rw.Write(data) - if n < 1 { - fmt.Println("recv data less than 1 length") + ch := make(chan int) + go func(){ + n, _ := s.rw.Write(data) + ch <- n + }() + select{ + case <-ch: + return nil + case <- time.After(3 * time.Second): + return errors.New("send time out") } - return err + return errors.New("send should't here") + + // n, err := s.rw.Write(data) + // if n < 1 { + // fmt.Println("recv data less than 1 length") + // } + + // return err } // Recv impl interface Diliver @@ -35,14 +51,33 @@ return nil, errors.New("please open shm consumer first") } - data := make([]byte, maxRecvSize) - n, err := s.rw.Read(data) - if err == nil || err == io.EOF { - data = data[:n:n] - return data, nil + ch := make(chan []byte) + go func(){ + data := make([]byte, maxRecvSize) + n, err := s.rw.Read(data) + if err == nil || err == io.EOF { + data = data[:n:n] + } + ch <- data + }() + select{ + case d := <-ch: + return d, nil + case <- time.After(3 * time.Second): + return nil, errors.New("recv time out") } - return nil, err + return nil, errors.New("recv should't here") + + + // data := make([]byte, maxRecvSize) + // n, err := s.rw.Read(data) + // if err == nil || err == io.EOF { + // data = data[:n:n] + // return data, nil + // } + + // return nil, err } // Close impl interface Deliver @@ -79,7 +114,6 @@ blocks, size = param[0], param[1] // return nil, errors.New("shmServer created recv too much parameters") } - shm.Unlink(url) rw, err := shm.CreateSimplex(url, 0644, blocks, size) -- Gitblit v1.8.0