binary pack option

Author: devsnek

lib/manifold.ex

@@ -57,6 +57,8 @@ defmodule Manifold do
   def send([pid], message, options), do: __MODULE__.send(pid, message, options)
 
   def send(pids, message, options) when is_list(pids) do
+    message = Utils.pack_message(options[:pack_mode], message)
+
     case options[:send_mode] do
       :offload ->
         Sender.send(current_sender(), current_partitioner(), pids, message)

lib/manifold/utils.ex

@@ -56,4 +56,12 @@ defmodule Manifold.Utils do
 
     hibernate_delay + :rand.uniform(hibernate_jitter)
   end
+
+  @spec pack_message(mode :: atom(), message :: term()) :: term()
+  def pack_message(:binary, message), do: {:manifold_binary, :erlang.term_to_binary(message)}
+  def pack_message(_mode, message), do: message
+
+  @spec unpack_message(message :: term()) :: term()
+  def unpack_message({:manifold_binary, binary}), do: :erlang.binary_to_term(binary)
+  def unpack_message(message), do: message
 end

lib/manifold/worker.ex

@@ -17,11 +17,13 @@ defmodule Manifold.Worker do
   end
 
   def handle_cast({:send, [pid], message}, nil) do
+    message = Utils.unpack_message(message)
     send(pid, message)
     {:noreply, nil}
   end
 
   def handle_cast({:send, pids, message}, nil) do
+    message = Utils.unpack_message(message)
     for pid <- pids, do: send(pid, message)
     {:noreply, nil}
   end

test/manifold_test.exs

@@ -29,6 +29,22 @@ defmodule ManifoldTest do
     end
   end
 
+  test "pack_mode option" do
+    me = self()
+    message = :hello
+    pids = for _ <- 0..10000 do
+      spawn_link fn ->
+        receive do
+          message -> send(me, {self(), message})
+        end
+      end
+    end
+    Manifold.send(pids, message, pack_mode: :binary)
+    for pid <- pids do
+      assert_receive {^pid, ^message},  1000
+    end
+  end
+
   test "send to list of one" do
     me = self()
     message = :hello