C++实现互斥锁与读写锁

互斥锁

#include <bits/stdc++.h>
using namespace std;

// 锁标志位
// true 表示以上锁，false表示未加锁
bool lock_flag;

// CPU硬件指令支持，此函数执行一气呵成
bool TestAndSet(bool *lock_flag) 
{
    bool old = *lock_flag;
    *lock_flag = true;
    return old;
}

void use()
{
    while (TestAndSet(&lock_flag));
    // 临界区代码
    printf("正在访问临界资源...\n");
    // 剩余区代码
    lock_flag = false;       // 解锁
}



int main()
{
    use();
    use();
    use();
    use();
    
    return 0;
}

读写锁

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

class readWriteLock {
private:
    std::mutex readMtx;
    std::mutex writeMtx;
    int readCnt; // 已加读锁个数
public:
    readWriteLock() : readCnt(0) {}
    void readLock()
    {
        readMtx.lock();
        if (++readCnt == 1) {
            writeMtx.lock();  // 存在线程读操作时，写加锁（只加一次）
        }
        readMtx.unlock();
    }
    void readUnlock()
    {
        readMtx.lock();
        if (--readCnt == 0) { // 没有线程读操作时，释放写锁
            writeMtx.unlock();
        }
        readMtx.unlock();
    }
    void writeLock()
    {
        writeMtx.lock();
    }
    void writeUnlock()
    {
        writeMtx.unlock();
    }
};

volatile int var = 10; // 保持变量 var 对内存可见性，防止编译器过度优化
readWriteLock rwLock; // 定义全局的读写锁变量

void Write() {
    rwLock.writeLock();
    var += 10;
    std::cout << "write var : " << var << std::endl;
    rwLock.writeUnlock();
}

void Read() {
    rwLock.readLock();
    std::cout << "read var : " << var << std::endl;
    rwLock.readUnlock();
}

int main() {
    std::vector<std::thread> writers;
    std::vector<std::thread> readers;
    for (int i = 0; i < 10; i++) {  // 10 个写线程
        writers.push_back(std::thread(Write));  // std::thread t 的写法报错
    }
    for (int i = 0; i < 100; i++) {   // 100 个读线程
        readers.push_back(std::thread(Read));
    }
    for (auto& t : writers) {   // 写线程启动
        t.join();
    }
    for (auto& t : readers) {   // 读线程启动
        t.join();
    }
    std::cin.get();
}