add support for profiling threading.BoundedSemaphore objects in Python Lock profiler

Author: vlad-scherbich

ddtrace/profiling/collector/threading.py

@@ -21,6 +21,10 @@ class _ProfiledThreadingSemaphore(_lock._ProfiledLock):
     pass
 
 
+class _ProfiledThreadingBoundedSemaphore(_lock._ProfiledLock):
+    pass
+
+
 class ThreadingLockCollector(_lock.LockCollector):
     """Record threading.Lock usage."""
 
@@ -45,6 +49,14 @@ class ThreadingSemaphoreCollector(_lock.LockCollector):
     PATCH_ATTR_NAME = "Semaphore"
 
 
+class ThreadingBoundedSemaphoreCollector(_lock.LockCollector):
+    """Record threading.BoundedSemaphore usage."""
+
+    PROFILED_LOCK_CLASS = _ProfiledThreadingBoundedSemaphore
+    PATCH_MODULE = threading
+    PATCH_ATTR_NAME = "BoundedSemaphore"
+
+
 # Also patch threading.Thread so echion can track thread lifetimes
 def init_stack_v2() -> None:
     if config.stack.enabled and stack_v2.is_available:

ddtrace/profiling/profiler.py

@@ -221,6 +221,7 @@ def start_collector(collector_class: Type) -> None:
                 ("threading", lambda _: start_collector(threading.ThreadingLockCollector)),
                 ("threading", lambda _: start_collector(threading.ThreadingRLockCollector)),
                 ("threading", lambda _: start_collector(threading.ThreadingSemaphoreCollector)),
+                ("threading", lambda _: start_collector(threading.ThreadingBoundedSemaphoreCollector)),
                 ("asyncio", lambda _: start_collector(asyncio.AsyncioLockCollector)),
             ]
 

releasenotes/notes/Added-support-for-profiling-of-threading.BoundedSemaphore-objects-to-the-Python-Lock-profiler-9c54a96d93e09c46.yaml

@@ -0,0 +1,7 @@
+---
+features:
+  - |
+    profiling: Add support for ``threading.BoundedSemaphore`` locking type profiling in Python. 
+    The Lock profiler now detects and marks "internal" Lock objects, i.e. those that are part of implementation of higher-level locking types. 
+    One example of such higher-level primitive is ``threading.BoundedSemaphore``, which is implemented with ``threading.Condition``, which itself uses ``threading.Lock`` internally. 
+    Marking a locks as internal will prevent it from being logged, which means the sample will only be counted once.