YT-18571: Refactor traced (fomerly tracked) spin locks and add unittests

commit_hash:2476930524e6179b49fbf312f907bd03413cd8b8

1 files changed, 15 insertions, 10 deletions

diff --git a/library/cpp/yt/threading/spin_lock_count.cpp b/library/cpp/yt/threading/spin_lock_count.cpp
index b1670e92d4a..008a5a66ad5 100644
--- a/library/cpp/yt/threading/spin_lock_count.cpp
+++ b/library/cpp/yt/threading/spin_lock_count.cpp
@@ -7,21 +7,19 @@
 
 #include <util/system/types.h>
 
-namespace NYT::NThreading::NPrivate {
+namespace NYT::NThreading {
+
+////////////////////////////////////////////////////////////////////////////////
 
 #ifndef NDEBUG
 
-////////////////////////////////////////////////////////////////////////////////
+namespace NDetail {
 
 YT_DEFINE_THREAD_LOCAL(i64, ActiveSpinLockCount, 0);
 
-////////////////////////////////////////////////////////////////////////////////
-
-void RecordSpinLockAcquired(bool isAcquired)
+void RecordSpinLockAcquired()
 {
-    if (isAcquired) {
-        ActiveSpinLockCount()++;
-    }
+    ActiveSpinLockCount()++;
 }
 
 void RecordSpinLockReleased()
@@ -30,14 +28,21 @@ void RecordSpinLockReleased()
     ActiveSpinLockCount()--;
 }
 
+} // namespace NDetail
+
+int GetActiveSpinLockCount()
+{
+    return NDetail::ActiveSpinLockCount();
+}
+
 void VerifyNoSpinLockAffinity()
 {
-    YT_VERIFY(ActiveSpinLockCount() == 0);
+    YT_VERIFY(NDetail::ActiveSpinLockCount() == 0);
 }
 
 #endif
 
 ////////////////////////////////////////////////////////////////////////////////
 
-} // namespace NYT::NThreading::NPrivate
+} // namespace NYT::NThreading