zhangmeng
2019-08-30 4f3bd6a122917eed4b5a5ead80a05ef9c143e395
shm.go
@@ -4,7 +4,9 @@
   "errors"
   "fmt"
   "io"
   "reflect"
   "time"
   "unsafe"
   "basic.com/valib/shm.git"
)
@@ -13,6 +15,8 @@
type SHM struct {
   rw  *shm.ReadWriteCloser
   typ td
   recvData []byte
}
// Send impl interface Diliver
@@ -36,6 +40,8 @@
      return nil, errors.New("please open shm consumer first")
   }
   // orig
   // data := make([]byte, maxRecvSize)
   // n, err := s.rw.Read(data)
   // if err == nil || err == io.EOF {
@@ -43,10 +49,28 @@
   //    return data, nil
   // }
   data, err := s.rw.DirectRead()
   if err == nil || err == io.EOF {
      return data, nil
   // cur
   // data, err := s.rw.DirectRead()
   // if err == nil || err == io.EOF {
   //    return data, nil
   // }
   // new
   if s.recvData == nil {
      s.recvData = make([]byte, maxRecvSize)
   }
   sliceHeader := (*reflect.SliceHeader)(unsafe.Pointer(&s.recvData))
   sliceHeader.Cap = maxRecvSize
   sliceHeader.Len = 0
   n, err := s.rw.Read(s.recvData)
   if err == nil || err == io.EOF {
      sliceHeader.Len = n
      return s.recvData, nil
   }
   return nil, err
}