fix
zhangmeng
2019-05-16 5ff1f32410e0697e581f6c389c8c22c5abd474c0
nng.go
@@ -24,6 +24,7 @@
// NNG mangos wrap
type NNG struct {
   sock mangos.Socket
   raw  bool
}
// Send impl interface Diliver
@@ -32,12 +33,12 @@
      return errors.New("please init NNG first")
   }
   switch n.sock.GetProtocol().Number() {
   case mangos.ProtoSurveyor:
      time.Sleep(surveyorTime * 2)
   default:
   }
   if _, err := n.sock.GetOption(mangos.OptionRaw); err == nil {
   // switch n.sock.GetProtocol().Number() {
   // case mangos.ProtoSurveyor:
   //    time.Sleep(surveyorTime * 2)
   // default:
   // }
   if n.raw {
      msg := mangos.NewMessage(len(data))
      msg.Body = data
      return n.sock.SendMsg(msg)
@@ -51,11 +52,13 @@
   if n.sock == nil {
      return nil, errors.New("please init NNG first")
   }
   if _, err := n.sock.GetOption(mangos.OptionRaw); err == nil {
      if msg, err := n.sock.RecvMsg(); err == nil {
         return msg.Body, err
   if n.raw {
      var msg *mangos.Message
      var err error
      if msg, err = n.sock.RecvMsg(); err != nil {
         return nil, err
      }
      return nil, err
      return msg.Body, nil
   }
   return n.sock.Recv()
}
@@ -81,6 +84,7 @@
      }
      return &NNG{
         sock,
         true,
      }
   }
@@ -102,6 +106,7 @@
      return &NNG{
         sock,
         true,
      }
   }