Add publisher_put_(string|integer|float)

biyooon-ex · Jan 21, 2024 · 48b748b · 48b748b
1 parent 36621d5
commit 48b748b
Show file tree

Hide file tree

Showing 3 changed files with 33 additions and 8 deletions.
diff --git a/lib/zenohex/nif.ex b/lib/zenohex/nif.ex
@@ -16,8 +16,10 @@ defmodule Zenohex.Nif do
     :erlang.nif_error(:nif_not_loaded)
   end
 
-  def publisher_put(_publisher, _value) do
-    :erlang.nif_error(:nif_not_loaded)
+  for type <- ["string", "integer", "float"] do
+    def unquote(:"publisher_put_#{type}")(_publisher, _value) do
+      :erlang.nif_error(:nif_not_loaded)
+    end
   end
 
   def declare_subscriber(_session, _key_expr) do

diff --git a/native/zenohex_nif/src/lib.rs b/native/zenohex_nif/src/lib.rs
@@ -53,7 +53,24 @@ fn declare_publisher(env: Env, resource: ResourceArc<ExSessionRef>, key_expr: St
 }
 
 #[rustler::nif]
-fn publisher_put(resource: ResourceArc<ExPublisherRef>, value: String) -> Atom {
+fn publisher_put_string(resource: ResourceArc<ExPublisherRef>, value: String) -> Atom {
+    publisher_put_impl(resource, value)
+}
+
+#[rustler::nif]
+fn publisher_put_integer(resource: ResourceArc<ExPublisherRef>, value: i64) -> Atom {
+    publisher_put_impl(resource, value)
+}
+
+#[rustler::nif]
+fn publisher_put_float(resource: ResourceArc<ExPublisherRef>, value: f64) -> Atom {
+    publisher_put_impl(resource, value)
+}
+
+fn publisher_put_impl<T: Into<zenoh::value::Value>>(
+    resource: ResourceArc<ExPublisherRef>,
+    value: T,
+) -> Atom {
     let publisher: &Publisher = &resource.0;
     match publisher.put(value).res_sync() {
         Ok(()) => atom::ok(),
@@ -103,7 +120,9 @@ rustler::init!(
         test_thread,
         zenoh_open,
         declare_publisher,
-        publisher_put,
+        publisher_put_string,
+        publisher_put_integer,
+        publisher_put_float,
         declare_subscriber,
         subscriber_recv_timeout
     ],

diff --git a/test/zenohex/nif_test.exs b/test/zenohex/nif_test.exs
@@ -22,9 +22,13 @@ defmodule Zenohex.NifTest do
       assert is_reference(Nif.declare_publisher(session, "key/expression"))
     end
 
-    test "publisher_put/2", %{session: session} do
-      publisher = Nif.declare_publisher(session, "key/expression")
-      assert Nif.publisher_put(publisher, "value") == :ok
+    for {type, value} <- [{"string", "value"}, {"integer", 0}, {"float", 0.0}] do
+      test "publisher_put_#{type}/2", %{session: session} do
+        type = unquote(type)
+        value = unquote(value)
+        publisher = Nif.declare_publisher(session, "key/expression")
+        assert apply(Nif, :"publisher_put_#{type}", [publisher, value]) == :ok
+      end
     end
   end
 
@@ -37,7 +41,7 @@ defmodule Zenohex.NifTest do
       publisher = Nif.declare_publisher(session, "key/expression")
       subscriber = Nif.declare_subscriber(session, "key/expression")
 
-      Nif.publisher_put(publisher, "value")
+      Nif.publisher_put_string(publisher, "value")
       assert Nif.subscriber_recv_timeout(subscriber, 1000) == "value"
       assert Nif.subscriber_recv_timeout(subscriber, 1000) == :timeout
     end