bug fixed withintime

zhangmeng

2021-09-24 82a262ef5cf721e5a236c8b1d2ab3ff92ca33122

 shmData.go

@@ -5,9 +5,9 @@
   "unsafe"
)

// lock free queue struct
// []byte struct
type ShmDataInfo struct {
   Capacity uint32
   Len uint32
   Cache    []byte
}

@@ -18,41 +18,87 @@
   ))
}

const SHMSLICEOFFSET int = 32
// ConvertToSlice convert to []byte
func ptr2Slice(s unsafe.Pointer, size int) []byte {
func ptr2Slice(s unsafe.Pointer, size,cap int) []byte {
   var x reflect.SliceHeader
   x.Len = size
   x.Cap = size
   x.Cap = cap
   x.Data = uintptr(s)
   return *(*[]byte)(unsafe.Pointer(&x))
}

//WriteShmData   data待写入的数据;   shmId:共享内存id,只attach,不create
// 把data写入shmId对应的shm中,  流程: attach->写入数据->detach
// 适合shm是空的,解码使用
func WriteShmData(data []byte, shmId int) error {
   shmdata,err := Attach(shmId)
   shmData, err := Attach(shmId)
   if err != nil {
      return err
   }
   sdi := shmData2Info(shmdata)
   if len(data) <= len(shmdata) {
      sdi.Capacity = uint32(len(data))
   sdi := shmData2Info(shmData)
   if len(data) <= len(shmData)-SHMSLICEOFFSET {
      sdi.Len = uint32(len(data))
   } else {
      sdi.Capacity = uint32(len(shmdata))
      sdi.Len = uint32(len(shmData)-SHMSLICEOFFSET)
   }

   tmpData := ptr2Slice(unsafe.Pointer(&sdi.Cache), int(sdi.Capacity))
   tmpData := ptr2Slice(unsafe.Pointer(&sdi.Cache), int(sdi.Len), len(shmData)-SHMSLICEOFFSET)
   copy(tmpData, data)

   err = Detach(shmData)
   if err != nil {
      return err
   }

   return nil
}

func ReadShmData(shmId int) ([]byte,error) {
   shmdata,err := Attach(shmId)
//ReadShmData attach到shmId对应的共享内存,并读出数据[]byte
// 使用之后需要detach
func ReadShmData(shmId int) ([]byte, error) {
   shmData, err := Attach(shmId)
   if err != nil {
      return nil,err
      return nil, err
   }

   sdi := shmData2Info(shmdata)
   tmpData := ptr2Slice(unsafe.Pointer(&sdi.Cache), int(sdi.Capacity))
   sdi := shmData2Info(shmData)
   tmpData := ptr2Slice(unsafe.Pointer(&sdi.Cache), int(sdi.Len), len(shmData)-SHMSLICEOFFSET)

   return tmpData, nil
}

//DeepCopyDataToShm 深拷贝,把内存的数据拷贝attach之后共享内存中
// 输入: shmData是attach返回的[]byte, memdData是内存中的数据
// 写入数据到共享内存使用:attach->DeepCopyDataToShm->detach
// 适合解码使用
func DeepCopyDataToShm(memData,shmData []byte) {
   sdi := shmData2Info(shmData)
   if len(memData) <= len(shmData)-SHMSLICEOFFSET {
      sdi.Len = uint32(len(memData))
   } else {
      sdi.Len = uint32(len(shmData)-SHMSLICEOFFSET)
   }

   tmpData := ptr2Slice(unsafe.Pointer(&sdi.Cache), int(sdi.Len), len(shmData)-SHMSLICEOFFSET)
   copy(tmpData, memData)
}

//ReadShmData 把attach后的shmData转成内存data
// 输入:attach到共享内存得到的shmData
// 输出:转换后的data
// 从共享内存中读取数据使用:attach->ConvertShmToData->只读,业务处理结束后->detach
func ConvertShmToData(shmData []byte) []byte {
   sdi := shmData2Info(shmData)
   tmpData := ptr2Slice(unsafe.Pointer(&sdi.Cache), int(sdi.Len), len(shmData)-SHMSLICEOFFSET)

   return tmpData
}

//UpdateDataToShm (类似浅拷贝)更新数据到共享内存
// 输入: ConvertShmToData得到的数据data,经过业务处理后,该数据有变化,需要update到shm里
// 更新数据到共享内存使用: attach->ConvertShmToData->修改data数据,业务处理结束后->UpdateDataToShm->detach
func UpdateDataToShm(data, shmData []byte) {
   sdi := shmData2Info(shmData)
   sdi.Len = uint32(len(data))
}