| | |
| | | import ( |
| | | "errors" |
| | | "fmt" |
| | | "io" |
| | | // "io" |
| | | |
| | | "time" |
| | | |
| | | "github.com/tmthrgd/shm-go" |
| | | "basic.com/valib/shm.git" |
| | | ) |
| | | |
| | | // SHM share memory |
| | | type SHM struct { |
| | | rw *shm.ReadWriteCloser |
| | | typ td |
| | | |
| | | data []byte |
| | | } |
| | | |
| | | // Send impl interface Diliver |
| | |
| | | return nil, errors.New("please open shm consumer first") |
| | | } |
| | | |
| | | ch := make(chan []byte) |
| | | ch := make(chan int) |
| | | go func(){ |
| | | data := make([]byte, maxRecvSize) |
| | | n, err := s.rw.Read(data) |
| | | if err == nil || err == io.EOF { |
| | | data = data[:n:n] |
| | | } |
| | | ch <- data |
| | | n, _ := s.rw.Read(s.data) |
| | | ch <- n |
| | | }() |
| | | select{ |
| | | case d := <-ch: |
| | | return d, nil |
| | | if d > 0{ |
| | | data := make([]byte, d) |
| | | copy(data, s.data) |
| | | return data, nil |
| | | } |
| | | case <- time.After(3 * time.Second): |
| | | return nil, errors.New("recv time out") |
| | | } |
| | |
| | | return &SHM{ |
| | | rw, |
| | | agent, |
| | | make([]byte, maxRecvSize), |
| | | }, nil |
| | | } |
| | | |
| | |
| | | return &SHM{ |
| | | rw, |
| | | coactee, |
| | | make([]byte, maxRecvSize), |
| | | }, nil |
| | | } |
| | | return nil, err |