fix

zhangmeng

2019-05-16 5ff1f32410e0697e581f6c389c8c22c5abd474c0

fix

 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,
        }
    }