访问 std::recursive_mutex 使用的所有者计数器

Question

我有一个案例，我的算法的决定是基于共享 std::recursive_mutex 的深度。

#include <iostream>
#include <mutex>
#include <thread>

int g_i = 0;
std::recursive_mutex g_i_mutex;

void bar() {
  std::lock_guard<std::recursive_mutex> lock(g_i_mutex);
  switch (get_counter(g_i_mutex)) { // some way to find the number of owners
    case 1: std::cout << "depth 1\n"; break;
    case 2: std::cout << "depth 2\n"; break;
    default:;
  }
}

void foo() {
   std::lock_guard<std::recursive_mutex> lock(g_i_mutex);
   std::cout << "hello\n";
   bar();
}

int main() {
  foo(); //print hello then depth 2
  bar(); //print depth 1
}

我读到递归互斥体保存某种使用计数，并且每次调用 lock/unlock 都会增加和减少它，有没有办法访问该信息？

Answer 1

没有you cannot.

这是不可能的，因为你说的计数器是一个实现方案，它可能存在也可能不存在。如果您知道标准库的特定实现使用了计数器，您可以使用一些魔法（指针算术和强制转换）来获取它，但那将是未定义的行为。

话虽这么说，但没有什么会阻止您定义自己的 recursive_mutex:

#include <iostream>
#include <mutex>
#include <atomic>

class recursive_mutex
{
    std::recursive_mutex _mutex;
    std::atomic<unsigned> _counter;
public:
    recursive_mutex() : _mutex(), _counter(0) {}
    recursive_mutex(recursive_mutex&) = delete;
    void operator=(recursive_mutex&) = delete;
    void lock() { _mutex.lock(); ++_counter; }
    bool try_lock() { bool result = _mutex.try_lock(); _counter += result; return result; }
    void unlock() { --_counter; _mutex.unlock(); }
    unsigned counter() { return _counter; }
};

int main() {
  recursive_mutex m;
  m.lock();
  m.lock();
  std::cout << m.counter() << "\n";
  m.unlock();
  std::cout << m.counter() << "\n";
  m.unlock();
}

2
1

demo

访问 std::recursive_mutex 使用的所有者计数器

Access the owners counter used by std::recursive_mutex

c++

multithreading

stl

recursive-mutex