Skip to content

Commit

Permalink
Subscribe to gazebo topic on ROS topic connection
Browse files Browse the repository at this point in the history
We are currently subscribing to the /gazebo/performance_metrics topic
even if there are no subscribers to the ROS topic forwarding this data.
The link_states and model_states topics currently use an advertise
mechanism with callbacks when a subscriber connects or disconnects,
so I've used that same pattern for the performance_metrics topic.
This also helps workaround the deadlock documented in ros-simulation#1175 and
gazebosim/gazebo-classic#2902.

Signed-off-by: Steve Peters <scpeters@openrobotics.org>
  • Loading branch information
scpeters committed Dec 12, 2020
1 parent 60bc303 commit 52f2318
Show file tree
Hide file tree
Showing 2 changed files with 41 additions and 5 deletions.
9 changes: 9 additions & 0 deletions gazebo_ros/include/gazebo_ros/gazebo_ros_api_plugin.h
Original file line number Diff line number Diff line change
Expand Up @@ -150,6 +150,14 @@ class GazeboRosApiPlugin : public SystemPlugin
/// \brief Callback for a subscriber disconnecting from ModelStates ros topic.
void onModelStatesDisconnect();

#ifdef GAZEBO_ROS_HAS_PERFORMANCE_METRICS
/// \brief Callback for a subscriber connecting to PerformanceMetrics ros topic.
void onPerformanceMetricsConnect();

/// \brief Callback for a subscriber disconnecting from PerformanceMetrics ros topic.
void onPerformanceMetricsDisconnect();
#endif

/// \brief Function for inserting a URDF into Gazebo from ROS Service Call
bool spawnURDFModel(gazebo_msgs::SpawnModel::Request &req,
gazebo_msgs::SpawnModel::Response &res);
Expand Down Expand Up @@ -383,6 +391,7 @@ class GazeboRosApiPlugin : public SystemPlugin
ros::Publisher pub_performance_metrics_;
int pub_link_states_connection_count_;
int pub_model_states_connection_count_;
int pub_performance_metrics_connection_count_;

// ROS comm
boost::shared_ptr<ros::AsyncSpinner> async_ros_spin_;
Expand Down
37 changes: 32 additions & 5 deletions gazebo_ros/src/gazebo_ros_api_plugin.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -34,6 +34,7 @@ GazeboRosApiPlugin::GazeboRosApiPlugin() :
stop_(false),
pub_link_states_connection_count_(0),
pub_model_states_connection_count_(0),
pub_performance_metrics_connection_count_(0),
physics_reconfigure_initialized_(false),
pub_clock_frequency_(0),
world_created_(false),
Expand Down Expand Up @@ -198,13 +199,10 @@ void GazeboRosApiPlugin::loadGazeboRosApiPlugin(std::string world_name)
light_modify_pub_ = gazebonode_->Advertise<gazebo::msgs::Light>("~/light/modify");
request_pub_ = gazebonode_->Advertise<gazebo::msgs::Request>("~/request");
response_sub_ = gazebonode_->Subscribe("~/response",&GazeboRosApiPlugin::onResponse, this);
#ifdef GAZEBO_ROS_HAS_PERFORMANCE_METRICS
performance_metric_sub_ = gazebonode_->Subscribe("/gazebo/performance_metrics",
&GazeboRosApiPlugin::onPerformanceMetrics, this);
#endif
// reset topic connection counts
pub_link_states_connection_count_ = 0;
pub_model_states_connection_count_ = 0;
pub_performance_metrics_connection_count_ = 0;

// Manage clock for simulated ros time
pub_clock_ = nh_->advertise<rosgraph_msgs::Clock>("/clock", 10);
Expand Down Expand Up @@ -416,7 +414,13 @@ void GazeboRosApiPlugin::advertiseServices()

#ifdef GAZEBO_ROS_HAS_PERFORMANCE_METRICS
// publish performance metrics
pub_performance_metrics_ = nh_->advertise<gazebo_msgs::PerformanceMetrics>("performance_metrics", 10);
ros::AdvertiseOptions pub_performance_metrics_ao =
ros::AdvertiseOptions::create<gazebo_msgs::PerformanceMetrics>(
"performance_metrics",10,
boost::bind(&GazeboRosApiPlugin::onPerformanceMetricsConnect,this),
boost::bind(&GazeboRosApiPlugin::onPerformanceMetricsDisconnect,this),
ros::VoidPtr(), &gazebo_queue_);
pub_performance_metrics_ = nh_->advertise(pub_performance_metrics_ao);
#endif

// Advertise more services on the custom queue
Expand Down Expand Up @@ -577,6 +581,29 @@ void GazeboRosApiPlugin::onModelStatesConnect()
pub_model_states_event_ = gazebo::event::Events::ConnectWorldUpdateBegin(boost::bind(&GazeboRosApiPlugin::publishModelStates,this));
}

#ifdef GAZEBO_ROS_HAS_PERFORMANCE_METRICS
void GazeboRosApiPlugin::onPerformanceMetricsConnect()
{
pub_performance_metrics_connection_count_++;
if (pub_performance_metrics_connection_count_ == 1) // connect on first subscriber
{
performance_metric_sub_ = gazebonode_->Subscribe("/gazebo/performance_metrics",
&GazeboRosApiPlugin::onPerformanceMetrics, this);
}
}

void GazeboRosApiPlugin::onPerformanceMetricsDisconnect()
{
pub_performance_metrics_connection_count_--;
if (pub_performance_metrics_connection_count_ <= 0) // disconnect with no subscribers
{
performance_metric_sub_.reset();
if (pub_performance_metrics_connection_count_ < 0) // should not be possible
ROS_ERROR_NAMED("api_plugin", "One too many disconnect from pub_performance_metrics_ in gazebo_ros.cpp? something weird");
}
}
#endif

void GazeboRosApiPlugin::onLinkStatesDisconnect()
{
pub_link_states_connection_count_--;
Expand Down

0 comments on commit 52f2318

Please sign in to comment.