Check if appropriate bson module is installed or not just after init_…

…node. (#198)

Check whether mongodb oriented bson is installed or not just after init_node,
and if appropriate bson is not installed, shutdown node with error message.

See: #198

rosbridge_library/src/rosbridge_library/util/__init__.py

@@ -13,3 +13,18 @@
     string_types = (str,)
 else:
     string_types = (str, unicode)
+
+def appropriate_bson_installed():
+    """ Check whether appropriate bson module is installed or not
+
+    pymongo oriented bson module should be installed. See following issue for
+    detail.
+
+    * https://github.com/RobotWebTools/rosbridge_suite/issues/198
+    """
+    import bson
+    try:
+        bson.BSON
+    except AttributeError:
+        return False
+    return True

rosbridge_server/scripts/rosbridge_tcp.py

@@ -10,6 +10,7 @@
 from rosbridge_library.capabilities.advertise_service import AdvertiseService
 from rosbridge_library.capabilities.unadvertise_service import UnadvertiseService
 from rosbridge_library.capabilities.call_service import CallService
+from rosbridge_library.util import appropriate_bson_installed
 
 from functools import partial
 from signal import signal, SIGINT, SIG_DFL
@@ -41,6 +42,13 @@ def shutdown_hook(server):
             init_node("rosbridge_tcp")
             signal(SIGINT, SIG_DFL)
 
+            # Check whether appropriate bson module is installed or not
+            # See: https://github.com/RobotWebTools/rosbridge_suite/issues/198
+            if not appropriate_bson_installed():
+                rospy.logerr("BSON installation does not support all necessary features. "
+                             "Please use the MongoDB BSON implementation.")
+                rospy.signal_shutdown("shutdown")
+
             """
             Parameter handling:
              - try to get parameter from parameter server (..define those via launch-file)

rosbridge_server/scripts/rosbridge_udp.py

@@ -45,6 +45,7 @@
 from rosbridge_library.capabilities.advertise_service import AdvertiseService
 from rosbridge_library.capabilities.unadvertise_service import UnadvertiseService
 from rosbridge_library.capabilities.call_service import CallService
+from rosbridge_library.util import appropriate_bson_installed
 
 def shutdown_hook():
     reactor.stop()
@@ -53,6 +54,13 @@ def shutdown_hook():
     rospy.init_node("rosbridge_websocket")
     rospy.on_shutdown(shutdown_hook)    # register shutdown hook to stop the server
 
+    # Check whether appropriate bson module is installed or not
+    # See: https://github.com/RobotWebTools/rosbridge_suite/issues/198
+    if not appropriate_bson_installed():
+        rospy.logerr("BSON installation does not support all necessary features. "
+                     "Please use the MongoDB BSON implementation.")
+        rospy.signal_shutdown("shutdown")
+
     ##################################################
     # Parameter handling                             #
     ##################################################

rosbridge_server/scripts/rosbridge_websocket.py

@@ -49,6 +49,7 @@
 from rosbridge_library.capabilities.advertise_service import AdvertiseService
 from rosbridge_library.capabilities.unadvertise_service import UnadvertiseService
 from rosbridge_library.capabilities.call_service import CallService
+from rosbridge_library.util import appropriate_bson_installed
 
 def shutdown_hook():
     IOLoop.instance().stop()
@@ -57,6 +58,13 @@ def shutdown_hook():
     rospy.init_node("rosbridge_websocket")
     rospy.on_shutdown(shutdown_hook)    # register shutdown hook to stop the server
 
+    # Check whether appropriate bson module is installed or not
+    # See: https://github.com/RobotWebTools/rosbridge_suite/issues/198
+    if not appropriate_bson_installed():
+        rospy.logerr("BSON installation does not support all necessary features. "
+                     "Please use the MongoDB BSON implementation.")
+        rospy.signal_shutdown("shutdown")
+
     ##################################################
     # Parameter handling                             #
     ##################################################