From 3e04892020c074668264ac387bf5165e96c11c0b Mon Sep 17 00:00:00 2001
From: Romain Reignier <romain.reignier@exail.com>
Date: Mon, 3 Feb 2025 18:54:39 +0100
Subject: [PATCH] Add Publisher::get_subscription_count

---
 rclrs/src/publisher.rs | 37 +++++++++++++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)

diff --git a/rclrs/src/publisher.rs b/rclrs/src/publisher.rs
index b1cdd93b9..eb0bbafe9 100644
--- a/rclrs/src/publisher.rs
+++ b/rclrs/src/publisher.rs
@@ -146,6 +146,20 @@ where
         }
     }
 
+    /// Returns the number of subscriptions of the publisher.
+    pub fn get_subscription_count(&self) -> Result<usize, RclrsError> {
+        let mut subscription_count = 0;
+        // SAFETY: No preconditions for the function called.
+        unsafe {
+            rcl_publisher_get_subscription_count(
+                &*self.handle.rcl_publisher.lock().unwrap(),
+                &mut subscription_count,
+            )
+            .ok()?
+        };
+        Ok(subscription_count)
+    }
+
     /// Publishes a message.
     ///
     /// The [`MessageCow`] trait is implemented by any
@@ -327,6 +341,29 @@ mod tests {
             expected_publishers_info
         );
 
+        // Test get_subscription_count()
+        assert_eq!(node_1_empty_publisher.get_subscription_count(), Ok(0));
+        assert_eq!(node_1_basic_types_publisher.get_subscription_count(), Ok(0));
+        assert_eq!(node_2_default_publisher.get_subscription_count(), Ok(0));
+        let _node_1_empty_subscriber = graph.node1.create_subscription(
+            "graph_test_topic_1",
+            QOS_PROFILE_SYSTEM_DEFAULT,
+            |_msg: msg::Empty| {},
+        );
+        let _node_1_basic_types_subscriber = graph.node1.create_subscription(
+            "graph_test_topic_2",
+            QOS_PROFILE_SYSTEM_DEFAULT,
+            |_msg: msg::BasicTypes| {},
+        );
+        let _node_2_default_subscriber = graph.node1.create_subscription(
+            "graph_test_topic_3",
+            QOS_PROFILE_SYSTEM_DEFAULT,
+            |_msg: msg::Defaults| {},
+        );
+        assert_eq!(node_1_empty_publisher.get_subscription_count(), Ok(1));
+        assert_eq!(node_1_basic_types_publisher.get_subscription_count(), Ok(1));
+        assert_eq!(node_2_default_publisher.get_subscription_count(), Ok(1));
+
         Ok(())
     }
 }