兄弟姐妹们,我又回来了,今天带来实际开发中都需要使用的互斥锁的内容,主要聊一聊如何使用互斥锁以及都有哪几种方式实现互斥锁。实现互斥,可以有以下几种方式:互斥量(Mutex)、递归互斥量(Recursive Mutex)、读写锁(Read-Write Lock)、条件变量(Condition Variable)。
目录
一、互斥原理(mutex)
二、递归互斥量(Recursive Mutex)
三、读写锁(Read-Write Lock)
四、条件变量(Condition Variable)
五、总结
一、互斥原理(mutex)
互斥锁可以确保在任何时候只有一个线程能够进入临界区。当线程需要进入临界区时,它会尝试获取互斥锁的所有权,如果互斥锁已经被其他线程占用,那么当前线程就会进入阻塞状态,直到互斥锁被释放为止。简单说就是一块区域只能被一个线程执行。
当一个线程获取到互斥锁的所有权后,它就可以进入临界区进行操作,当操作完成后,它需要释放互斥锁,让其他线程有机会进入临界区。
下面是一个简单的互斥锁的示例代码,它演示了如何使用 std::mutex 类来保护临界区:
#include #include #include // 定义互斥锁std::mutex g_mutex;// 临界区代码void critical_section(int thread_id) {// 加锁g_mutex.lock();// 访问共享资源std::cout << "Thread " << thread_id << " enter critical section." << std::endl;// 释放锁std::this_thread::sleep_for(std::chrono::seconds(5));g_mutex.unlock();}int main() {// 创建两个线程std::thread t1(critical_section, 1);std::thread t2(critical_section, 2);// 等待两个线程执行完成t1.join();t2.join();return 0;}
main中创建两个线程去访问资源,但是其中一个需要等待另一个线程5s释放后才能访问,形成对资源的锁定。
上面的例子使用的是std::mutex实现互斥锁,需要注意这个互斥锁的声明需要相对的全局变量,也就是说对于使用锁的部分它必须是“全局的”。
二、递归互斥量(Recursive Mutex)
C++ 中的递归互斥量(Recursive Mutex)是一种特殊的互斥量,它可以被同一个线程多次锁定,而不会发生死锁。递归互斥量的实现原理是,在锁定时维护一个锁定计数器,每次解锁时将计数器减一,只有当计数器为 0 时才会释放锁。
以下是递归互斥量的示例代码:
#include #include #include std::recursive_mutex mtx;void foo(int n) {mtx.lock();std::cout << "Thread " << n << " locked the mutex." < 1) {foo(n - 1);}std::cout << "Thread " << n << " unlocked the mutex." << std::endl;mtx.unlock();}int main() {std::thread t1(foo, 3);std::thread t2(foo, 2);t1.join();t2.join();return 0;}
在上面的代码中,我们定义了一个递归函数 foo(),它接受一个整数参数 n,表示当前线程的编号。在函数中,我们首先使用递归互斥量 mtx 锁定当前线程,然后输出一条带有线程编号的信息,接着判断如果 n 大于 1,则递归调用 foo() 函数,并将参数减一。最后,我们输出一条解锁信息,并将递归互斥量解锁。
在主函数中,我们创建了两个线程 t1 和 t2,分别调用 foo() 函数,并传入不同的参数值。由于递归互斥量可以被同一个线程多次锁定,因此在 t1 线程中对 mtx 进行了两次锁定,而在 t2 线程中只进行了一次锁定。
运行结果:
可以看到,递归互斥量可以被同一个线程多次锁定,并且在解锁时必须对应减少锁定计数器。这种机制可以避免死锁的发生,但也需要注意使用时的线程安全问题。
三、读写锁(Read-Write Lock)
读写锁(Read-Write Lock)是一种特殊的互斥锁,用于在多线程环境下对共享资源进行读写操作。它允许多个线程同时读取共享资源,但只允许一个线程写入共享资源。读写锁的使用可以提高并发性能,特别是当读操作比写操作频繁时。
在 C++ 中,读写锁可以通过 std::shared_mutex 类型来实现。下面是一个简单的示例代码,演示了如何使用读写锁来保护一个共享的整型变量:
#include #include #include #include std::shared_mutex rw_lock; // 读写锁int shared_var = 0; // 共享变量// 写线程函数void writer() {for (int i = 0; i < 10; ++i) {// 独占写锁std::unique_lock lock(rw_lock);// 写共享变量++shared_var;std::cout << "Writer thread: write shared_var=" << shared_var << std::endl;// 等待一段时间std::this_thread::sleep_for(std::chrono::milliseconds(100));}}// 读线程函数void reader(int id) {for (int i = 0; i < 10; ++i) {// 共享读锁std::shared_lock lock(rw_lock);// 读共享变量int value = shared_var;std::cout << "Reader thread " << id << ": read shared_var=" << value << std::endl;// 等待一段时间std::this_thread::sleep_for(std::chrono::milliseconds(100));}}int main() {std::thread t1(writer);std::thread t2(reader, 1);std::thread t3(reader, 2);std::thread t4(reader, 3);t1.join();t2.join();t3.join();t4.join();return 0;}
在上面的代码中,我们定义了一个共享变量 shared_var 和一个读写锁 rw_lock。写线程函数 writer() 独占写锁,对 shared_var 进行自增操作,并输出当前的值。读线程函数 reader() 共享读锁,读取 shared_var 的值,并输出当前的值。所有的线程都会等待一段时间,以模拟实际的操作。
在主函数中,我们创建了一个写线程和三个读线程。由于读写锁的特性,读线程可以并发读取共享变量,而写线程会独占写锁,只有在写操作完成之后,读线程才能再次读取共享变量。因此,输出结果中读线程的顺序可能会有所不同,但是写线程的操作一定是顺序执行的。
注意,这里使用 std::unique_lockstd::shared_mutex 类型的对象来获取独占写锁,使用 std::shared_lockstd::shared_mutex 类型的对象来获取共享读锁。这些锁对象会在作用域结束时自动解锁,避免了手动解锁的问题。
四、条件变量(Condition Variable)
条件变量(Condition Variable)是一种线程间同步机制,用于在某些特定条件下阻塞或唤醒线程。在 C++ 中,条件变量是通过 std::condition_variable 类来实现的。
下面是一个使用条件变量的示例代码,其中有两个线程,一个线程不停地生产数据,另一个线程则等待数据,当有数据可用时,将数据进行消费。
#include #include #include #include #include #include std::queue data_queue; // 数据队列std::mutex data_mutex; // 互斥锁std::condition_variable data_cond; // 条件变量// 生产数据函数void producer() {for (int i = 1; i <= 10; ++i) {// 生产数据std::this_thread::sleep_for(std::chrono::milliseconds(100));std::unique_lock lock(data_mutex);data_queue.push(i);std::cout << "Producer thread: produce data " << i << std::endl;// 唤醒消费线程data_cond.notify_one();}}// 消费数据函数void consumer() {while (true) {// 等待数据std::unique_lock lock(data_mutex);data_cond.wait(lock, [] { return !data_queue.empty(); });// 消费数据int data = data_queue.front();data_queue.pop();std::cout << "Consumer thread: consume data " << data << std::endl;// 检查是否结束if (data == 10) {break;}}}int main() {std::thread t1(producer);std::thread t2(consumer);t1.join();t2.join();return 0;}
在上面的代码中,我们定义了一个数据队列 data_queue 和一个互斥锁 data_mutex,同时定义了一个条件变量 data_cond。生产数据的函数 producer() 不停地往队列中添加数据,每次添加完数据之后,通过调用 data_cond.notify_one() 唤醒等待的消费线程。消费数据的函数 consumer() 通过调用 data_cond.wait(lock, [] { return !data_queue.empty(); }) 来等待数据,当队列中有数据时,将数据从队列中取出并消费,如果取出的数据是最后一个,则退出循环。
在主函数中,我们创建了一个生产线程和一个消费线程。生产线程生产 10 个数据,消费线程从队列中消费数据,直到消费到最后一个数据为止。
注意,这里使用了 std::unique_lockstd::mutex 类型的对象来获取互斥锁,并使用 lambda 表达式 [] { return !data_queue.empty(); } 来判断条件是否满足。在调用 wait() 函数时,当前线程会阻塞,直到条件变量被其他线程唤醒或超时。当 wait() 函数返回时,当前线程会重新获取互斥。
简单一些的例子:
#include #include #include #include #include bool ready = false; // 条件变量std::mutex data_mutex; // 互斥锁std::condition_variable data_cond; // 条件变量void do_something() {// 模拟工作std::this_thread::sleep_for(std::chrono::milliseconds(500));}void waiting_thread() {// 等待条件变量std::unique_lock lock(data_mutex);data_cond.wait(lock, [] { return ready; });// 条件满足后输出一句话std::cout << "Condition satisfied, waiting thread resumes." << std::endl;do_something();}int main() {std::thread t1(waiting_thread);// 模拟条件满足后的操作std::this_thread::sleep_for(std::chrono::milliseconds(1000));{std::lock_guard lock(data_mutex);ready = true;data_cond.notify_one();}t1.join();return 0;}
在上面的代码中,我们定义了一个条件变量 ready 和一个互斥锁 data_mutex,同时定义了一个条件变量 data_cond。等待条件变量的函数 waiting_thread() 首先获取互斥锁,然后通过调用 data_cond.wait(lock, [] { return ready; }) 等待条件变量,当 ready 为 true 时,线程会被唤醒,输出一句话,并模拟一些工作的操作。在主函数中,我们创建了一个等待条件变量的线程 t1,然后模拟条件满足后的操作,即将 ready 设置为 true,然后通过调用 data_cond.notify_one() 唤醒等待的线程。
五、总结
互斥锁保证了计算机资源访问的安全,互斥锁的不当使用同时也加大了程序阻塞的风险。
提前祝大家五一前工作生活学习一切顺利。