cleanup: rename package and vendor dependencies for old go versions
Fixes #8 and begins to address deployability.
This commit is contained in:
206
vendor/github.com/pebbe/zmq4/utils.go
generated
vendored
Normal file
206
vendor/github.com/pebbe/zmq4/utils.go
generated
vendored
Normal file
@@ -0,0 +1,206 @@
|
||||
package zmq4
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
)
|
||||
|
||||
/*
|
||||
Send multi-part message on socket.
|
||||
|
||||
Any `[]string' or `[][]byte' is split into separate `string's or `[]byte's
|
||||
|
||||
Any other part that isn't a `string' or `[]byte' is converted
|
||||
to `string' with `fmt.Sprintf("%v", part)'.
|
||||
|
||||
Returns total bytes sent.
|
||||
*/
|
||||
func (soc *Socket) SendMessage(parts ...interface{}) (total int, err error) {
|
||||
return soc.sendMessage(0, parts...)
|
||||
}
|
||||
|
||||
/*
|
||||
Like SendMessage(), but adding the DONTWAIT flag.
|
||||
*/
|
||||
func (soc *Socket) SendMessageDontwait(parts ...interface{}) (total int, err error) {
|
||||
return soc.sendMessage(DONTWAIT, parts...)
|
||||
}
|
||||
|
||||
func (soc *Socket) sendMessage(dontwait Flag, parts ...interface{}) (total int, err error) {
|
||||
|
||||
var last int
|
||||
PARTS:
|
||||
for last = len(parts) - 1; last >= 0; last-- {
|
||||
switch t := parts[last].(type) {
|
||||
case []string:
|
||||
if len(t) > 0 {
|
||||
break PARTS
|
||||
}
|
||||
case [][]byte:
|
||||
if len(t) > 0 {
|
||||
break PARTS
|
||||
}
|
||||
default:
|
||||
break PARTS
|
||||
}
|
||||
}
|
||||
|
||||
opt := SNDMORE | dontwait
|
||||
for i := 0; i <= last; i++ {
|
||||
if i == last {
|
||||
opt = dontwait
|
||||
}
|
||||
switch t := parts[i].(type) {
|
||||
case []string:
|
||||
opt = SNDMORE | dontwait
|
||||
n := len(t) - 1
|
||||
for j, s := range t {
|
||||
if j == n && i == last {
|
||||
opt = dontwait
|
||||
}
|
||||
c, e := soc.Send(s, opt)
|
||||
if e == nil {
|
||||
total += c
|
||||
} else {
|
||||
return -1, e
|
||||
}
|
||||
}
|
||||
case [][]byte:
|
||||
opt = SNDMORE | dontwait
|
||||
n := len(t) - 1
|
||||
for j, b := range t {
|
||||
if j == n && i == last {
|
||||
opt = dontwait
|
||||
}
|
||||
c, e := soc.SendBytes(b, opt)
|
||||
if e == nil {
|
||||
total += c
|
||||
} else {
|
||||
return -1, e
|
||||
}
|
||||
}
|
||||
case string:
|
||||
c, e := soc.Send(t, opt)
|
||||
if e == nil {
|
||||
total += c
|
||||
} else {
|
||||
return -1, e
|
||||
}
|
||||
case []byte:
|
||||
c, e := soc.SendBytes(t, opt)
|
||||
if e == nil {
|
||||
total += c
|
||||
} else {
|
||||
return -1, e
|
||||
}
|
||||
default:
|
||||
c, e := soc.Send(fmt.Sprintf("%v", t), opt)
|
||||
if e == nil {
|
||||
total += c
|
||||
} else {
|
||||
return -1, e
|
||||
}
|
||||
}
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
/*
|
||||
Receive parts as message from socket.
|
||||
|
||||
Returns last non-nil error code.
|
||||
*/
|
||||
func (soc *Socket) RecvMessage(flags Flag) (msg []string, err error) {
|
||||
msg = make([]string, 0)
|
||||
for {
|
||||
s, e := soc.Recv(flags)
|
||||
if e == nil {
|
||||
msg = append(msg, s)
|
||||
} else {
|
||||
return msg[0:0], e
|
||||
}
|
||||
more, e := soc.GetRcvmore()
|
||||
if e == nil {
|
||||
if !more {
|
||||
break
|
||||
}
|
||||
} else {
|
||||
return msg[0:0], e
|
||||
}
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
/*
|
||||
Receive parts as message from socket.
|
||||
|
||||
Returns last non-nil error code.
|
||||
*/
|
||||
func (soc *Socket) RecvMessageBytes(flags Flag) (msg [][]byte, err error) {
|
||||
msg = make([][]byte, 0)
|
||||
for {
|
||||
b, e := soc.RecvBytes(flags)
|
||||
if e == nil {
|
||||
msg = append(msg, b)
|
||||
} else {
|
||||
return msg[0:0], e
|
||||
}
|
||||
more, e := soc.GetRcvmore()
|
||||
if e == nil {
|
||||
if !more {
|
||||
break
|
||||
}
|
||||
} else {
|
||||
return msg[0:0], e
|
||||
}
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
/*
|
||||
Receive parts as message from socket, including metadata.
|
||||
|
||||
Metadata is picked from the first message part.
|
||||
|
||||
For details about metadata, see RecvWithMetadata().
|
||||
|
||||
Returns last non-nil error code.
|
||||
*/
|
||||
func (soc *Socket) RecvMessageWithMetadata(flags Flag, properties ...string) (msg []string, metadata map[string]string, err error) {
|
||||
b, p, err := soc.RecvMessageBytesWithMetadata(flags, properties...)
|
||||
m := make([]string, len(b))
|
||||
for i, bt := range b {
|
||||
m[i] = string(bt)
|
||||
}
|
||||
return m, p, err
|
||||
}
|
||||
|
||||
/*
|
||||
Receive parts as message from socket, including metadata.
|
||||
|
||||
Metadata is picked from the first message part.
|
||||
|
||||
For details about metadata, see RecvBytesWithMetadata().
|
||||
|
||||
Returns last non-nil error code.
|
||||
*/
|
||||
func (soc *Socket) RecvMessageBytesWithMetadata(flags Flag, properties ...string) (msg [][]byte, metadata map[string]string, err error) {
|
||||
bb := make([][]byte, 0)
|
||||
b, p, err := soc.RecvBytesWithMetadata(flags, properties...)
|
||||
if err != nil {
|
||||
return bb, p, err
|
||||
}
|
||||
for {
|
||||
bb = append(bb, b)
|
||||
|
||||
var more bool
|
||||
more, err = soc.GetRcvmore()
|
||||
if err != nil || !more {
|
||||
break
|
||||
}
|
||||
b, err = soc.RecvBytes(flags)
|
||||
if err != nil {
|
||||
break
|
||||
}
|
||||
}
|
||||
return bb, p, err
|
||||
}
|
||||
Reference in New Issue
Block a user