diff --git a/rcljava/include/org_ros2_rcljava_node_NodeImpl.h b/rcljava/include/org_ros2_rcljava_node_NodeImpl.h index 90c565ed..4a169cb8 100644 --- a/rcljava/include/org_ros2_rcljava_node_NodeImpl.h +++ b/rcljava/include/org_ros2_rcljava_node_NodeImpl.h @@ -146,6 +146,15 @@ JNIEXPORT void JNICALL Java_org_ros2_rcljava_node_NodeImpl_nativeGetPublisherNamesAndTypesByNode( JNIEnv *, jclass, jlong, jstring, jstring, jobject); +/* + * Class: org_ros2_rcljava_node_NodeImpl + * Method: nativeGetSubscriptionNamesAndTypesByNode + * Signature: (JLjava/lang/String;Ljava/lang/String;Ljava/util/Collection;)V + */ +JNIEXPORT void +JNICALL Java_org_ros2_rcljava_node_NodeImpl_nativeGetSubscriptionNamesAndTypesByNode( + JNIEnv *, jclass, jlong, jstring, jstring, jobject); + #ifdef __cplusplus } #endif diff --git a/rcljava/src/main/cpp/org_ros2_rcljava_node_NodeImpl.cpp b/rcljava/src/main/cpp/org_ros2_rcljava_node_NodeImpl.cpp index b3beedd9..7332bc2b 100644 --- a/rcljava/src/main/cpp/org_ros2_rcljava_node_NodeImpl.cpp +++ b/rcljava/src/main/cpp/org_ros2_rcljava_node_NodeImpl.cpp @@ -537,3 +537,41 @@ Java_org_ros2_rcljava_node_NodeImpl_nativeGetPublisherNamesAndTypesByNode( RCLJAVA_COMMON_THROW_FROM_RCL(env, ret, "failed to get publisher names and types"); fill_jnames_and_types(env, publisher_names_and_types, jnames_and_types); } + +JNIEXPORT void JNICALL +Java_org_ros2_rcljava_node_NodeImpl_nativeGetSubscriptionNamesAndTypesByNode( + JNIEnv * env, jclass, jlong handle, jstring jname, jstring jnamespace, jobject jnames_and_types) +{ + rcl_node_t * node = reinterpret_cast(handle); + if (!node) { + rcljava_throw_exception(env, "java/lang/IllegalArgumentException", "node handle is NULL"); + return; + } + + const char * name = env->GetStringUTFChars(jname, NULL); + auto release_jname = rcpputils::make_scope_exit( + [jname, name, env]() {env->ReleaseStringUTFChars(jname, name);}); + const char * namespace_ = env->GetStringUTFChars(jnamespace, NULL); + auto release_jnamespace = rcpputils::make_scope_exit( + [jnamespace, namespace_, env]() {env->ReleaseStringUTFChars(jnamespace, namespace_);}); + rcl_allocator_t allocator = rcl_get_default_allocator(); + rcl_names_and_types_t subscription_names_and_types = rcl_get_zero_initialized_names_and_types(); + auto fini_names_and_types = rcpputils::make_scope_exit( + [pnames_and_types = &subscription_names_and_types, env]() { + rcl_ret_t ret = rcl_names_and_types_fini(pnames_and_types); + if (!env->ExceptionCheck() && RCL_RET_OK != ret) { + rcljava_throw_rclexception( + env, ret, "failed to fini subscription names and types structure"); + } + }); + + rcl_ret_t ret = rcl_get_subscriber_names_and_types_by_node( + node, + &allocator, + false, + name, + namespace_, + &subscription_names_and_types); + RCLJAVA_COMMON_THROW_FROM_RCL(env, ret, "failed to get subscription names and types"); + fill_jnames_and_types(env, subscription_names_and_types, jnames_and_types); +} diff --git a/rcljava/src/main/java/org/ros2/rcljava/node/Node.java b/rcljava/src/main/java/org/ros2/rcljava/node/Node.java index 7cdbec41..3d1b81ab 100644 --- a/rcljava/src/main/java/org/ros2/rcljava/node/Node.java +++ b/rcljava/src/main/java/org/ros2/rcljava/node/Node.java @@ -631,4 +631,16 @@ Client createClient(final Class serviceType, * @return the detected publisher names and types. */ Collection getPublisherNamesAndTypesByNode(String nodeName, String nodeNamespace); + + /** + * Return the subscription names and types that were created from the node specified by the given + * node name and namespace. + * See @{link graph#NameAndTypes} for more information about the returned value. + * + * @param nodeName name of the node we want to know its subscriptions. + * @param nodeNamespace namespace of the node we want to know its subscriptions. + * @return the detected subscription names and types. + */ + Collection getSubscriptionNamesAndTypesByNode( + String nodeName, String nodeNamespace); } diff --git a/rcljava/src/main/java/org/ros2/rcljava/node/NodeImpl.java b/rcljava/src/main/java/org/ros2/rcljava/node/NodeImpl.java index 76320031..58ed6b8a 100644 --- a/rcljava/src/main/java/org/ros2/rcljava/node/NodeImpl.java +++ b/rcljava/src/main/java/org/ros2/rcljava/node/NodeImpl.java @@ -820,4 +820,15 @@ public final Collection getPublisherNamesAndTypesByNode( private static native final Collection nativeGetPublisherNamesAndTypesByNode( long handle, String nodeName, String nodeNamespace, Collection namesAndTypes); + + public final Collection getSubscriptionNamesAndTypesByNode( + String nodeName, String nodeNamespace) + { + Collection namesAndTypes = new ArrayList(); + nativeGetSubscriptionNamesAndTypesByNode(this.handle, nodeName, nodeNamespace, namesAndTypes); + return namesAndTypes; + } + + private static native final Collection nativeGetSubscriptionNamesAndTypesByNode( + long handle, String nodeName, String nodeNamespace, Collection namesAndTypes); } diff --git a/rcljava/src/test/java/org/ros2/rcljava/node/NodeTest.java b/rcljava/src/test/java/org/ros2/rcljava/node/NodeTest.java index 88c1c956..2634be8a 100644 --- a/rcljava/src/test/java/org/ros2/rcljava/node/NodeTest.java +++ b/rcljava/src/test/java/org/ros2/rcljava/node/NodeTest.java @@ -1260,4 +1260,98 @@ public void accept(final Collection local, Collection subscription1 = node.createSubscription( + rcljava.msg.Empty.class, "test_get_subscription_names_and_types_one", + new Consumer() { + public void accept(final rcljava.msg.Empty msg) {} + }); + Subscription subscription2 = node.createSubscription( + rcljava.msg.Empty.class, "test_get_subscription_names_and_types_two", + new Consumer() { + public void accept(final rcljava.msg.Empty msg) {} + }); + Subscription subscription3 = remoteNode.createSubscription( + rcljava.msg.Empty.class, "test_get_subscription_names_and_types_two", + new Consumer() { + public void accept(final rcljava.msg.Empty msg) {} + }); + Subscription subscription4 = remoteNode.createSubscription( + rcljava.msg.Empty.class, "test_get_subscription_names_and_types_three", + new Consumer() { + public void accept(final rcljava.msg.Empty msg) {} + }); + Publisher publisher = node.createPublisher( + rcljava.msg.UInt32.class, "test_get_topic_names_and_types_this_should_not_appear"); + + BiConsumer, Collection> validateNameAndTypes = + new BiConsumer, Collection>() { + public void accept(final Collection local, Collection remote) { + // TODO(ivanpauno): Using assertj may help a lot here https://assertj.github.io/doc/. + assertEquals(local.size(), 2); + assertTrue( + "topic 'test_get_subscription_names_and_types_one' was not discovered for local node", + local.contains( + new NameAndTypes( + "/test_get_subscription_names_and_types_one", + new ArrayList(Arrays.asList("rcljava/msg/Empty"))))); + assertTrue( + "topic 'test_get_subscription_names_and_types_two' was not discovered for local node", + local.contains( + new NameAndTypes( + "/test_get_subscription_names_and_types_two", + new ArrayList(Arrays.asList("rcljava/msg/Empty"))))); + + assertEquals(remote.size(), 2); + assertTrue( + "topic 'test_get_subscription_names_and_types_two' was not discovered for remote node", + remote.contains( + new NameAndTypes( + "/test_get_subscription_names_and_types_two", + new ArrayList(Arrays.asList("rcljava/msg/Empty"))))); + assertTrue( + "topic 'test_get_subscription_names_and_types_three' was not discovered for remote node", + remote.contains( + new NameAndTypes( + "/test_get_subscription_names_and_types_three", + new ArrayList(Arrays.asList("rcljava/msg/Empty"))))); + } + }; + + long start = System.currentTimeMillis(); + boolean ok = false; + Collection local = null; + Collection remote = null; + do { + local = this.node.getSubscriptionNamesAndTypesByNode("test_node", "/"); + remote = this.node.getSubscriptionNamesAndTypesByNode( + "test_get_subscription_names_and_types_remote_node", "/"); + try { + validateNameAndTypes.accept(local, remote); + ok = true; + } catch (AssertionError err) { + // ignore here, it's going to be validated again at the end. + } + // TODO(ivanpauno): We could wait for the graph guard condition to be triggered if that + // would be available. + try { + TimeUnit.MILLISECONDS.sleep(100); + } catch (InterruptedException err) { + // ignore + } + } while (!ok && System.currentTimeMillis() < start + 1000); + assertNotNull(local); + assertNotNull(remote); + validateNameAndTypes.accept(local, remote); + + subscription1.dispose(); + subscription2.dispose(); + subscription3.dispose(); + subscription4.dispose(); + publisher.dispose(); + remoteNode.dispose(); + } }