Trace and Pin Openstack Check (DataDog#2306)

* Add tracing to OpenStack

* Pin datadog_checks_base

Author: nmuesch

openstack/datadog_checks/openstack/openstack.py

@@ -13,7 +13,7 @@
 
 from datadog_checks.checks import AgentCheck
 from datadog_checks.config import is_affirmative
-
+from datadog_checks.utils.tracing import traced, add_trace_check
 
 try:
     # Agent >= 6.0: the check pushes tags invoking `set_external_tags`
@@ -531,6 +531,9 @@ class OpenStackCheck(AgentCheck):
     def __init__(self, name, init_config, agentConfig, instances=None):
         AgentCheck.__init__(self, name, init_config, agentConfig, instances)
 
+        if is_affirmative(init_config.get('trace_check', False)):
+            add_trace_check(self)
+
         self._ssl_verify = is_affirmative(init_config.get("ssl_verify", True))
         self.keystone_server_url = init_config.get("keystone_server_url")
         self._hypervisor_name_cache = {}
@@ -1176,6 +1179,7 @@ def ensure_auth_scope(self, instance):
 
         return instance_scope
 
+    @traced
     def check(self, instance):
         # have we been backed off
         if not self.should_run(instance):

openstack/setup.py

@@ -25,7 +25,7 @@ def get_requirements(fpath):
         return f.readlines()
 
 
-CHECKS_BASE_REQ = 'datadog_checks_base'
+CHECKS_BASE_REQ = 'datadog_checks_base >= 3.0.0,<4.0.0'
 
 setup(
     name='datadog-openstack',