当前位置: 首页 > news >正文

C++锁: 读锁,递归锁,超时锁

面试聊到锁,只知道数据库的读锁,不了解编程中的读锁,也没谈,遂败北。

代码分 C++ 和 POSIX 版本

文章目录

      • 互斥锁
      • 读写锁
      • 自旋锁
      • 递归锁
      • 超时锁
      • 原子操作
      • 条件变量
      • 信号量

互斥锁

互斥锁是最基本的线程同步工具,确保同一时刻只有一个线程可以访问共享资源。

如果加锁失败,锁被别人持有,该线程会阻塞,即不占用CPU资源,直至拿到锁。

#include <iostream>
#include <thread>
#include <mutex>std::mutex mtx;void worker() {// 使用 lock() 方法,若锁被占用则阻塞mtx.lock();std::cout << "Thread " << std::this_thread::get_id() << " has the lock." << std::endl;std::this_thread::sleep_for(std::chrono::seconds(1));std::cout << "Thread " << std::this_thread::get_id() << " is releasing the lock." << std::endl;mtx.unlock();// 使用 try_lock() 方法,若锁被占用则返回 falseif (mtx.try_lock()) {std::cout << "Thread " << std::this_thread::get_id() << " acquired the lock using try_lock." << std::endl;mtx.unlock();} else {std::cout << "Thread " << std::this_thread::get_id() << " failed to acquire the lock using try_lock." << std::endl;}
}int main() {std::thread t1(worker);std::thread t2(worker);t1.join();t2.join();return 0;
}
#include <pthread.h>pthread_mutex_t lock = PTHREAD_MUTEX_INITIALIZER;void *thread_func(void *arg) {pthread_mutex_lock(&lock);   // 加锁// 访问共享资源pthread_mutex_unlock(&lock); // 解锁return NULL;
}pthread_mutex_trylock()的方式可以立即返回~

读写锁

读写锁允许多个线程同时读取共享资源,但写操作需要独占。

C++17 是对 shared_mutex 进行 std::shared_lock<> 和 std::unique_ptr<> 这两个lock_guard实现的。

#include <iostream>
#include <thread>
#include <shared_mutex>std::shared_mutex rwMutex;
int sharedData = 0;void reader() {std::shared_lock<std::shared_mutex> lock(rwMutex);std::cout << "Reader reads: " << sharedData << std::endl;
}void writer() {std::unique_lock<std::shared_mutex> lock(rwMutex);++sharedData;std::cout << "Writer writes: " << sharedData << std::endl;
}int main() {std::thread r1(reader);std::thread w1(writer);r1.join();w1.join();return 0;
}    
#include <pthread.h>pthread_rwlock_t rwlock = PTHREAD_RWLOCK_INITIALIZER;void *reader(void *arg) {pthread_rwlock_rdlock(&rwlock); // 读锁// 读取共享资源pthread_rwlock_unlock(&rwlock); // 解锁return NULL;
}void *writer(void *arg) {pthread_rwlock_wrlock(&rwlock); // 写锁// 修改共享资源pthread_rwlock_unlock(&rwlock); // 解锁return NULL;
}

自旋锁

线程在获取锁失败时不会进入休眠,而是持续尝试获取锁。
避免了线程上下文切换的开销,适合锁持有时间极短(如几十到几百个 CPU 周期)的场景。多核 CPU 环境且锁竞争不激烈时效果最佳

像互斥锁,会让出CPU,这会交换上下文。最后还要换回来,如果等待时间小于两次交换上下文的时间,那等待就是更高效的。

C++ 模拟 【自旋锁的"忙等待"与"混合模式"的权衡】

#include <iostream>
#include <thread>
#include <atomic>class SpinLock {
private:std::atomic<bool> flag = false;
public:void lock() {while (flag.exchange(true, std::memory_order_acquire)) {while (flag.load(std::memory_order_relaxed)) {std::this_thread::yield();}}}void unlock() {flag.store(false, std::memory_order_release);}
};SpinLock spinLock;
void spinWorker() {spinLock.lock();std::cout << "Spin lock acquired by thread: " << std::this_thread::get_id() << std::endl;std::this_thread::sleep_for(std::chrono::milliseconds(100));spinLock.unlock();
}int main() {std::thread s1(spinWorker);std::thread s2(spinWorker);s1.join();s2.join();return 0;
}    
  • exchange是原子类的替换值操作
    std::memory_order_acquire:这是内存序的一种,用于保证在 exchange 操作之后的所有读写操作不会被重排到 exchange 操作之前,从而确保在获取锁之后对共享资源的访问是安全的。(不会出现先访问共享资源再获取锁的情况,从而确保了对共享资源的安全访问。)【如果学过计算机系统的MIPS,就应该了解流水线,一个道理】

  • load 函数:std::atomic 类的成员函数,用于获取 flag 的当前值。
    std::memory_order_relaxed:这是一种宽松的内存序,只保证原子性,不保证任何顺序性。在这里使用宽松内存序是因为在这个内层循环中,主要目的是快速检查锁的状态,对内存顺序的要求不高,使用宽松内存序可以提高性能。
    std::this_thread::yield() 让出 CPU 时间片

#include <pthread.h>pthread_spinlock_t spinlock;void *worker(void *arg) {pthread_spin_lock(&spinlock);   // 自旋加锁// 访问共享资源pthread_spin_unlock(&spinlock); // 解锁return NULL;
}

递归锁

初始化的时候设置。
自己调用自己,加同一个锁。

#include <iostream>
#include <thread>
#include <recursive_mutex>std::recursive_mutex recursiveMutex;void recursiveWorker(int depth) {if (depth == 0) return;std::lock_guard<std::recursive_mutex> lock(recursiveMutex);std::cout << "Recursive lock acquired at depth " << depth << " by thread: " << std::this_thread::get_id() << std::endl;recursiveWorker(depth - 1);
}int main() {std::thread rw(recursiveWorker, 3);rw.join();return 0;
}    

POSIX 初始化普通锁:

#include <pthread.h>
#include <stdio.h>pthread_mutex_t lock;void functionA() {pthread_mutex_lock(&lock);printf("Function A: Locked\n");functionA();  // ✅ 不会死锁,因为递归锁允许同一线程重复加锁pthread_mutex_unlock(&lock);printf("Function A: Unlocked\n");
}int main() {pthread_mutexattr_t attr;pthread_mutexattr_init(&attr);pthread_mutexattr_settype(&attr, PTHREAD_MUTEX_RECURSIVE);pthread_mutex_init(&lock, &attr);functionA();  // 正常执行return 0;
}

超时锁

在尝试获取互斥锁时,如果锁没有在指定的 超时时间内 被获取到,线程就会放弃等待并继续执行。

超时(超过 2 秒)仍未获取到锁,它会返回非 0 值

#include <iostream>
#include <thread>
#include <mutex>
#include <chrono>std::timed_mutex tmtx;void worker() {// 尝试在 2 秒内获取锁if (tmtx.try_lock_for(std::chrono::seconds(2))) {std::cout << "Thread " << std::this_thread::get_id() << " acquired the lock using try_lock_for." << std::endl;std::this_thread::sleep_for(std::chrono::seconds(1));tmtx.unlock();} else {std::cout << "Thread " << std::this_thread::get_id() << " failed to acquire the lock using try_lock_for." << std::endl;}
}int main() {std::thread t1(worker);std::thread t2(worker);t1.join();t2.join();return 0;
}
#include <pthread.h>
#include <stdio.h>
#include <unistd.h>
#include <time.h>pthread_mutex_t lock;void* threadFunc(void* arg) {printf("Thread: Trying to lock...\n");struct timespec timeout;clock_gettime(CLOCK_REALTIME, &timeout);timeout.tv_sec += 2;  // 设置超时时间(2 秒)if (pthread_mutex_timedlock(&lock, &timeout) == 0) {printf("Thread: Locked!\n");pthread_mutex_unlock(&lock);} else {printf("Thread: Lock timeout! Giving up.\n");}return NULL;
}int main() {pthread_mutex_init(&lock, NULL);pthread_mutex_lock(&lock);  // 主线程加锁pthread_t thread;pthread_create(&thread, NULL, threadFunc, NULL);sleep(5);  // 模拟主线程长时间占有锁pthread_mutex_unlock(&lock);pthread_join(thread, NULL);return 0;
}

原子操作

一定要谈是操作该变量时

#include <iostream>
#include <thread>
#include <atomic>std::atomic<int> atomic_counter(0);
void atomic_worker() {for (int i = 0; i < 1000; ++i) {atomic_counter.fetch_add(1, std::memory_order_relaxed);}
}int main() {std::thread a1(atomic_worker);std::thread a2(atomic_worker);a1.join();a2.join();std::cout << "Atomic counter value: " << atomic_counter.load() << std::endl;return 0;
}    
#include <stdatomic.h>atomic_int counter = 0;void *worker(void *arg) {atomic_fetch_add(&counter, 1); // 原子递增return NULL;
}

条件变量

#include <iostream>
#include <thread>
#include <mutex>
#include <condition_variable>std::mutex cvMutex;
std::condition_variable cv;
bool ready = false;void waiter() {std::unique_lock<std::mutex> lock(cvMutex);cv.wait(lock, [] { return ready; });std::cout << "Waiter thread is notified." << std::endl;
}void notifier() {std::this_thread::sleep_for(std::chrono::milliseconds(200));{std::lock_guard<std::mutex> lock(cvMutex);ready = true;}cv.notify_one();std::cout << "Notifier thread sent notification." << std::endl;
}int main() {std::thread w(waiter);std::thread n(notifier);w.join();n.join();return 0;
}    
#include <pthread.h>pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
pthread_cond_t cond = PTHREAD_COND_INITIALIZER;
int ready = 0; // 共享状态void *consumer(void *arg) {pthread_mutex_lock(&mutex);while (ready == 0) {pthread_cond_wait(&cond, &mutex); // 等待条件满足}// 处理数据pthread_mutex_unlock(&mutex);return NULL;
}void *producer(void *arg) {pthread_mutex_lock(&mutex);ready = 1;pthread_cond_signal(&cond); // 唤醒等待线程pthread_mutex_unlock(&mutex);return NULL;
}

信号量

信号量可以控制多个线程同时访问资源的数量,比如限制访问数据库连接池的线程数。

#include <iostream>
#include <thread>
#include <semaphore>std::counting_semaphore<1> semaphore(0);void producer() {std::this_thread::sleep_for(std::chrono::milliseconds(200));std::cout << "Producer is releasing semaphore." << std::endl;semaphore.release();
}void consumer() {std::cout << "Consumer is waiting for semaphore." << std::endl;semaphore.acquire();std::cout << "Consumer acquired semaphore." << std::endl;
}int main() {std::thread p(producer);std::thread c(consumer);p.join();c.join();return 0;
}    
#include <semaphore.h>sem_t sem;void *worker(void *arg) {sem_wait(&sem);  // 获取信号量// 访问资源sem_post(&sem);  // 释放信号量return NULL;
}int main() {sem_init(&sem, 0, 3); // 允许最多3个线程同时访问return 0;
}

http://www.mrgr.cn/news/95985.html

相关文章:

  • 2017年计算机真题
  • STM32 - 在机器人、自动化领域,LL库相比HAL优势明显
  • 【解决】Mybatis-plus2.x升级到3.x
  • mac vim命令快捷键
  • MapStruct动态生成实现
  • 【AI】Orin Nano+ubuntu22.04上移植YoloV11,并使用DeepStream测试成功
  • mysql高级,mysql体系结构,mysql引擎,存储过程,索引,锁
  • (UI自动化测试web端)第二篇:元素定位的方法_css定位之ID选择器
  • uniapp动态循环表单校验失败:初始值校验
  • 解决Spring Cloud OpenFeign端点未暴露问题
  • Modbus RTU ---> Modbus TCP透传技术实现(Modbus透传、RS485透传、RTU透传)分站代码实现、协议转换器
  • 反序列化漏洞
  • Docker+Ollama+Xinference+RAGFlow+Dify部署及踩坑问题
  • msyql--基本操作之运维篇
  • 【JavaSE】抽象类和接口
  • uniapp页面列表,详情返回不刷新,新增或编辑后返回刷新
  • mysql中show命令的使用
  • NodeJs之fs模块
  • 【408--复习笔记】计算机组成原理
  • 【模型压缩+推理加速】知识蒸馏综述解读