【Linux】多线程安全之道：互斥、加锁技术与底层原理

news2024/10/21 7:25:13

1.线程的互斥

1.1.进程线程间的互斥相关背景概念

1.2.互斥量mutex的基本概念

所以多线程之间为什么要有互斥？

为什么抢票会抢到负数，无法获得正确结果？

为什么--操作不是原子性的呢？

解决方式：

2.三种加锁的方式

2.1全局变量（静态分布）的锁

2.2局部变量（动态分布）的锁

2.3.销毁锁（互斥量）的方式：

2.4.互斥量加锁和解锁

2.5 RAII风格的锁

代码：

3.互斥的底层实现？

1.线程的互斥

1.1.进程线程间的互斥相关背景概念

临界资源：多线程执行流共享的资源就叫做临界资源
临界区：每个线程内部，访问临界资源的代码，就叫做临界区
互斥：任何时刻，互斥保证有且只有一个执行流进入临界区，访问临界资源，通常对临界资源起保护作用
原子性（后面讨论如何实现）：不会被任何调度机制打断的操作，该操作只有两态，要么完成，要么未完成

1.2.互斥量mutex的基本概念

大部分情况，线程使用的数据都是局部变量，变量的地址空间在线程栈空间内，这种情况，变量归属单个线程，其他线程无法获得这种变量。
但有时候，很多变量都需要在线程间共享，这样的变量称为共享变量，可以通过数据的共享，完成线程之间的交互。
多个线程并发的操作共享变量，会带来一些问题

所以多线程之间为什么要有互斥？

上面概念有些抽象，我们来看一个实际的例子方便我们理解——抢票系统：

代码：

// 操作共享变量会有问题的售票系统代码
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <pthread.h>
int ticket = 100;
void *route(void *arg)
{
char *id = (char*)arg;
while ( 1 ) {
if ( ticket > 0 ) {
usleep(1000);
printf("%s sells ticket:%d\n", id, ticket);
ticket--;
} else {
break;
}
}
}
int main( void )
{
pthread_t t1, t2, t3, t4;
pthread_create(&t1, NULL, route, "thread 1");
pthread_create(&t2, NULL, route, "thread 2");
pthread_create(&t3, NULL, route, "thread 3");
pthread_create(&t4, NULL, route, "thread 4");
pthread_join(t1, NULL);
pthread_join(t2, NULL);
 pthread_join(t3, NULL);
pthread_join(t4, NULL);
}

执行结果：

这是没有加锁（互斥）的代码执行的结果，发现我们抢票抢着抢着竟然抢到了负数！这是万万不行的。

为什么抢票会抢到负数，无法获得正确结果？

共享资源被访问的时候，没有被保护，并且本身操作不是原子的！

if 语句判断条件为真以后，代码可以并发的切换到其他线程
usleep 这个模拟漫长业务的过程，在这个漫长的业务过程中，可能有很多个线程会进入该代码段
--ticket 操作本身就不是一个原子操作

前两者我们好理解，

为什么--操作不是原子性的呢？

ticket需要先从内存中读取数据放在CPU上，然后CPU进行加法或者减法操作，最后再将数据放在内存当中。因此就不是原子性的。

-- 操作并不是原子操作，而是对应三条汇编指令：

load ：将共享变量ticket从内存加载到寄存器中
update : 更新寄存器里面的值，执行-1操作
store ：将新值，从寄存器写回共享变量ticket的内存地址

解决方式：

要解决以上问题，需要做到三点：

代码必须要有互斥行为：当代码进入临界区执行时，不允许其他线程进入该临界区。
如果多个线程同时要求执行临界区的代码，并且临界区没有线程在执行，那么只能允许一个线程进入该临界区。
如果线程不在临界区中执行，那么该线程不能阻止其他线程进入临界区。

要做到这三点，本质上就是需要一把锁。Linux上提供的这把锁叫互斥量。

2.三种加锁的方式

2.1全局变量（静态分布）的锁

pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER

注意这种锁是定义在全局代码段的，这种锁也不需要销毁

2.2局部变量（动态分布）的锁

int pthread_mutex_init(pthread_mutex_t *restrict mutex, const pthread_mutexattr_t *restrict
attr);
参数：
mutex：要初始化的互斥量
attr：NULL

这种锁需要我们在局部代码段进行定义和初始化，并且也需要我们自己去手动销毁。

2.3.销毁锁（互斥量）的方式：

int pthread_mutex_destroy(pthread_mutex_t *mutex)

注意以上这两种锁的使用都是需要在指定加锁的区域进行加锁和解锁。

2.4.互斥量加锁和解锁

int pthread_mutex_lock(pthread_mutex_t *mutex);
int pthread_mutex_unlock(pthread_mutex_t *mutex);
返回值:成功返回0,失败返回错误号

2.5 RAII风格的锁

C++注重RAII的编程思想，所以我们可以将锁自己封装成为一个RAII风格的锁

我们可以将锁进行封装，定义一个LockGuard的类，里面只有一个锁的成员变量，构造函数是加锁，析构函数是解锁，所以我们可以创建一个局部的对象，让编译器自己去调用构造函数和析构函数，这样就不需要我们进行加锁和解锁

代码：

#ifndef __LOCK_GUARD_HPP__
#define __LOCK_GUARD_HPP__

#include <iostream>
#include <pthread.h>

class LockGuard
{
public:
    LockGuard(pthread_mutex_t *mutex):_mutex(mutex)
    {
        pthread_mutex_lock(_mutex); // 构造加锁
    }
    ~LockGuard()
    {
        pthread_mutex_unlock(_mutex);//析构解锁
    }
private:
    pthread_mutex_t *_mutex;
};

#endif

在我们学习了如何加锁之后，我们就可以将抢票系统进行进一步的优化：

void route(ThreadData *td)
{
    while (true)
    {
        { // 担心就用这个
            LockGuard guard(&td->_mutex); // 临时对象， RAII风格的加锁和解锁
            //std::lock_guard<std::mutex> lock(td->_mutex);
            //pthread_mutex_lock(&td->_mutex);
            if (td->_tickets > 0) // 1
            {
                usleep(1000);
                printf("%s running, get tickets: %d\n", td->_name.c_str(), td->_tickets); // 2
                td->_tickets--;                                                           // 3
                //pthread_mutex_unlock(&td->_mutex);
                td->_total++;
            }
            else
            {
                //pthread_mutex_unlock(&td->_mutex);
                //td->_mutex.unlock();
                break;
            }
        }
    }
}

执行结果：

可以看出，加锁之后就完美解决了票数会抢到负数的问题！

3.互斥的底层实现？

经过上面的例子，大家已经意识到单纯的 i++ 或者 ++i 都不是原子的，有可能会有数据一致性问题
为了实现互斥锁操作,大多数体系结构都提供了swap或exchange指令,该指令的作用是把寄存器和内存单元的数据相交换,由于只有一条指令,保证了原子性,即使是多处理器平台,访问内存的总线周期也有先后,一个处理器上的交换指令执行时另一个处理器的交换指令只能等待总线周期。