protocol-zmtp 0.3.0 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a8774efa1927a863efbec0173b00b4642fe99f809ee0df07e3ab6a164829909e
-  data.tar.gz: a77071e52f4461323f7d839738ad56d0ed83af312ae1071b4f88ffb365ee2514
+  metadata.gz: f6f3ddb84f7c69bf776e615da15797cb815eaf663f357cf16cff60c744153bc8
+  data.tar.gz: fd410e2dc417e623254be530230e88fc3e428f62413d6465ee88465c82444b10
 SHA512:
-  metadata.gz: c0cc5712a1c5aabd5f360ea5fe052bb45e963f645551f0bea2a46f62d3d43d4f9f8a0a0cd3a2de63e563641d2463586b0574016db6ab394a66b78a0c7bd93924
-  data.tar.gz: 87ae97e2a6bc9448b16a9053d959e7b7707639bb15ad168bc020f83eae0d95309b82be6a0772a38d6e56307038de013d7544e836094c5d52b4dfb114175b5a28
+  metadata.gz: 0c84020c35c81c9b9609541019cb7433956bfed7a2bf8fd4ea615ad7560b9aecec5086efbab733576309ff7cb2febb4f5c98bef2388f55a056bcf38c78a0633d
+  data.tar.gz: 7251f30d057e56f97e6ffa3eb95af5727d7763a04829b3c93580e36998b1aa2591cdc0152ff4d53e7764ea80f75113585513d537147c3de238b498d50f25d15c

data/lib/protocol/zmtp/connection.rb

@@ -52,6 +52,11 @@ module Protocol
         @mutex            = Mutex.new
         @max_message_size = max_message_size
         @last_received_at = nil
+
+        # Reusable scratch buffer for frame headers. Array#pack(buffer:)
+        # writes in place so the per-message 2-or-9 byte String allocation
+        # in write_frames disappears on the hot send path.
+        @header_buf = String.new(capacity: 9, encoding: Encoding::BINARY)
       end
 
 
@@ -109,6 +114,27 @@ module Protocol
       end
 
 
+      # Writes a batch of multi-frame messages to the buffer under a
+      # single mutex acquisition. Used by work-stealing send pumps that
+      # dequeue up to N messages at once — avoids the N lock/unlock
+      # pairs per batch that a plain `batch.each { write_message }`
+      # would incur.
+      #
+      # @param messages [Array<Array<String>>] each element is one
+      #   multi-frame message
+      # @return [void]
+      def write_messages(messages)
+        @mutex.synchronize do
+          i = 0
+          n = messages.size
+          while i < n
+            write_frames(messages[i])
+            i += 1
+          end
+        end
+      end
+
+
       # Writes pre-encoded wire bytes to the buffer without flushing.
       # Used for fan-out: encode once, write to many connections.
       #
@@ -241,13 +267,32 @@ module Protocol
       private
 
       # Writes message parts as ZMTP frames, encrypting if needed.
+      #
+      # For the unencrypted path, writes the frame header and body
+      # separately to the IO instead of allocating a wire String. This
+      # avoids copying the body just to glue a 1- or 9-byte header onto
+      # it -- significant for large messages where the body copy was
+      # the dominant allocation per send.
       def write_frames(parts)
+        encrypted = @mechanism.encrypted?
+        buf       = @header_buf
+
         parts.each_with_index do |part, i|
           more = i < parts.size - 1
-          if @mechanism.encrypted?
+          if encrypted
             @io.write(@mechanism.encrypt(part.b, more: more))
           else
-            @io.write(Codec::Frame.new(part, more: more).to_wire)
+            body  = part.b
+            size  = body.bytesize
+            flags = more ? Codec::Frame::FLAGS_MORE : 0
+            buf.clear
+            if size > Codec::Frame::SHORT_MAX
+              [flags | Codec::Frame::FLAGS_LONG, size].pack("CQ>", buffer: buf)
+            else
+              [flags, size].pack("CC", buffer: buf)
+            end
+            @io.write(buf)
+            @io.write(body)
           end
         end
       end

data/lib/protocol/zmtp/version.rb

@@ -2,6 +2,6 @@
 
 module Protocol
   module ZMTP
-    VERSION = "0.3.0"
+    VERSION = "0.4.0"
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: protocol-zmtp
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.4.0
 platform: ruby
 authors:
 - Patrik Wenger