Add receiveAll for ZSocket type

Author: Clonkk

examples/ex06_pollermultipart.nim

@@ -6,13 +6,6 @@ import ../zmq
 const address = "tcp://127.0.0.1:5559"
 const max_msg = 10
 
-proc receiveMultipart(socket: ZSocket, flags: ZSendRecvOptions): seq[string] =
-  # Little trick to receive all multipart message no matter how many parts there is using getsockopt
-  var hasMore: int = 1
-  while hasMore > 0:
-    result.add(socket.receive())
-    hasMore = getsockopt[int](socket, RCVMORE)
-
 
 proc client() =
   var d1 = connect(address, mode = DEALER)
@@ -35,7 +28,7 @@ proc client() =
     if res > 0:
       for i in 0..<len(poller):
         if events(poller[i]):
-          let buf = receiveMultipart(poller[i].socket, NOFLAGS)
+          let buf = receiveAll(poller[i].socket, NOFLAGS)
           for j, msg in buf.pairs:
             echo &"CLIENT> Socket{i} received \"{msg}\""
         else:

zmq/connections.nim

@@ -321,6 +321,19 @@ proc receive*(s: ZSocket, flags: ZSendRecvOptions = NOFLAGS): string =
   ## Return an empty string on EAGAIN
   tryReceive(s, flags).msg
 
+proc receiveAll*(s: ZSocket, flags: ZSendRecvOptions = NOFLAGS): seq[string] =
+  ## Receive all parts of a message
+  ##
+  ## If EAGAIN occurs without any data being received, it will be an empty seq
+  var expectMessage = true
+  while expectMessage:
+    let (msgAvailable, moreAvailable, msg) = tryReceive(s, flags)
+    if msgAvailable:
+      result.add msg
+      expectMessage = moreAvailable
+    else:
+      expectMessage = false
+
 proc tryReceive*(c: ZConnection, flags: ZSendRecvOptions = NOFLAGS): tuple[msgAvailable: bool, moreAvailable: bool, msg: string] =
   ## Receives a message from a connection.
   ##
@@ -337,14 +350,8 @@ proc receiveAll*(c: ZConnection, flags: ZSendRecvOptions = NOFLAGS): seq[string]
   ## Receive all parts of a message
   ##
   ## If EAGAIN occurs without any data being received, it will be an empty seq
-  var expectMessage = true
-  while expectMessage:
-    let (msgAvailable, moreAvailable, msg) = tryReceive(c, flags)
-    if msgAvailable:
-      result.add msg
-      expectMessage = moreAvailable
-    else:
-      expectMessage = false
+  receiveAll(c.socket, flags)
+
 
 proc proxy*(frontend, backend: ZConnection) =
   ## The proxy connects a frontend socket to a backend socket. Data flows from frontend to backend.