zhangmeng
2019-08-30 4f3bd6a122917eed4b5a5ead80a05ef9c143e395
test one data
1个文件已修改
30 ■■■■■ 已修改文件
shm.go 30 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
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
}