diff --git a/datadog_checks_base/datadog_checks/base/utils/__init__.py b/datadog_checks_base/datadog_checks/base/utils/__init__.py index 4d28f6c46a830..0ee49e936298e 100644 --- a/datadog_checks_base/datadog_checks/base/utils/__init__.py +++ b/datadog_checks_base/datadog_checks/base/utils/__init__.py @@ -1,6 +1,9 @@ # (C) Datadog, Inc. 2018-present # All rights reserved # Licensed under a 3-clause BSD style license (see LICENSE) +import os + +from ..config import is_affirmative try: import datadog_agent @@ -15,6 +18,12 @@ else: patch(requests=True) + if is_affirmative(datadog_agent.get_config('integration_profiling')): + from ddtrace.profiling import Profiler + + prof = Profiler(service='datadog-agent-integrations') + prof.start() + except ImportError: - # Tracing Integrations is only available with Agent 6 + # Tracing & profiling Integrations is only available with Agent 6 pass