reid from https://github.com/michuanhaohao/reid-strong-baseline
554325746@qq.com
2020-02-05 7c6a958f00e327a8991f48093456de7cb84cf918
common/helper.go
@@ -1,11 +1,24 @@
package common
import (
   "context"
   "encoding/json"
   "fmt"
   "io/ioutil"
   "strconv"
   "time"
   "basic.com/pubsub/protomsg.git"
   "github.com/gogo/protobuf/proto"
)
// GetIpcAddress get ipc
func GetIpcAddress(shm bool, id string) string {
   if shm {
      return id
   }
   return `ipc:///tmp/` + id + `.ipc`
}
// SubConfig sub
type SubConfig struct {
@@ -41,3 +54,43 @@
   i, _ := strconv.Atoi(s)
   return i
}
// UnserilizeProto un
func UnserilizeProto(ctx context.Context, data <-chan []byte, out chan<- protomsg.SdkMessage, fn func(...interface{})) {
   for {
      select {
      case <-ctx.Done():
         return
      case d := <-data:
         if len(d) < 100 {
            continue
         }
         msg := protomsg.SdkMessage{}
         if err := proto.Unmarshal(d, &msg); err != nil {
            fn("UnserilizeProto Unmarshal msg 处理异常:", err)
            continue
         }
         out <- msg
      default:
         time.Sleep(10 * time.Millisecond)
      }
   }
}
// UnpackImage unpack
func UnpackImage(msg protomsg.SdkMessage, fnName string, fn func(...interface{})) *protomsg.Image {
   // 反序列化数据得到sdk入参
   i := &protomsg.Image{}
   err := proto.Unmarshal(msg.Data, i)
   if err != nil {
      fn(fnName, " protobuf Unmarshal decode CameraImage error: ", err.Error())
      return nil
   }
   if i.Data == nil {
      fn(fnName, " protomsg.Image data null")
      return nil
   }
   return i
}