# C 语言标准库函数 mtx_init

/*********************************************
 * @brief 创建互斥量
 * @param[out] mutex 返回互斥量 ID
 * @param type 互斥量类型
 * @return 是否成功
 ********************************************/
int mtx_init(mtx_t* mutex, int type);

#include <stdio.h>
#include <threads.h>

// 共享数据和互斥量
struct Args
{
    int n;
    mtx_t mutex;
};

// 线程函数1，不加锁，结果错误
int func1(void* data)
{
    struct Args* args = (struct Args*)data;

    for (int i = 0; i < 10; i++)
    {
        int value = args->n + 1;
        for (int i = 0; i < 30000; i += 1); // 延长执行时间，提高不加锁时的出错概率
        args->n = value;
    }
    
    return 0;
}

// 线程函数2，加锁，结果正确
int func2(void* data)
{
    struct Args* args = (struct Args*)data;

    for (int i = 0; i < 10; i++)
    {
        mtx_lock(&args->mutex); // 锁定 mutex
        int value = args->n + 1;
        for (int i = 0; i < 30000; i += 1); // 延长执行时间，提高不加锁时的出错概率
        args->n = value;
        mtx_unlock(&args->mutex); // 解锁 mutex
    }
    
    return 0;
}

int main(void)
{
    struct Args data1, data2;
    mtx_init(&data2.mutex, mtx_plain); // 创建互斥量

    data1.n = 0;
    data2.n = 0;

    // 创建线程
    thrd_t th1, th2, th3, th4;
    thrd_create(&th1, func1, &data1); // 通过参数传递数据和互斥量
    thrd_create(&th2, func1, &data1);
    thrd_create(&th3, func2, &data2);
    thrd_create(&th4, func2, &data2);

    // 等待线程结束
    thrd_join(th1, NULL);
    thrd_join(th2, NULL);
    thrd_join(th3, NULL);
    thrd_join(th4, NULL);

    printf("data1 的最终值为 %d \n", data1.n);
    printf("data2 的最终值为 %d \n", data2.n);

    // 清除互斥量
    mtx_destroy(&data2.mutex);
    
    return 0;
}

$ ./a.out
data1 的最终值为 20
data2 的最终值为 20
$ ./a.out
data1 的最终值为 14
data2 的最终值为 20
$ ./a.out
data1 的最终值为 20
data2 的最终值为 20
$ ./a.out
data1 的最终值为 14
data2 的最终值为 20
$ ./a.out
data1 的最终值为 15
data2 的最终值为 20
$ ./a.out
data1 的最终值为 13
data2 的最终值为 20
$ ./a.out
data1 的最终值为 10
data2 的最终值为 20
$ ./a.out
data1 的最终值为 10
data2 的最终值为 20
$ ./a.out
data1 的最终值为 11
data2 的最终值为 20