12-5 原子类型

1. atomic_int 

        12-3 中线程安全问题来源之一为非原子操作，而利用 C11 新增的 atomic_int 类型，其定义的变量的相关操作均为原子操作。修改 12-3 第 1 小节的部分代码，将 int count = 0; 改为 atomic_int count = 0。程序最终结果为 count 等于 2000000，与 12-3 第 1 小节的情况截然不同。

        但目前仅 gcc 编译器支持 <stdatomic.h>，msvc 不支持。

#include <stdio.h>
#include <tinycthread.h>
#include <stdatomic.h>

// int count = 0;
atomic_int count = 0;

int Counter(void *arg){
  for (int i = 0; i < 1000000; ++i) {
    count++;
  }
  return 0;
}

void TestCounterByThread(){
  thrd_t t_1;
  thrd_t t_2;

  // create two threads to run the Counter function
  thrd_create(&t_1, Counter, NULL);
  thrd_create(&t_2, Counter, NULL);

  // free up resources
  thrd_join(t_1, NULL);
  thrd_join(t_2, NULL);

  printf("count: %d", count);  // count: 2000000
}

2. atomic_is_lock_free()

        原子操作的实现分为两种，一是在 CPU 指令层面实现，二是通过加锁的方式实现。利用 atomic_is_lock_free() 函数可以查看相关变量的原子操作是哪种实现方式。 atomic_is_lock_free() 函数返回值为 1 时，说明是在 CPU 指令层面实现原子操作；返回值为 0 时，则是通过加锁实现原子操作。

void TestAtomicLock(){
  atomic_int a;
  int result = atomic_is_lock_free(&a);
  printf("result: %d\n", result);
}

3. atomic_flag

        atomic_flag 是通过无锁方式实现的原子操作，是原子操作中的布尔类型。故将其看成标志位使用。注意，atomic_flag 类型的变量初始值为 ATOMIC_FLAG_INIT。atomic_flag_test_and_set() 函数可以设置标志位和判断当前标志位是否被设置，标志位没有被设置时返回 NULL。atomic_flag_clear() 函数作用是清除当前标志位。

        两个注意点：一， struct timespec 的用法。传入的进程睡眠时间 = 当前系统时间的秒数 + 进程需要睡眠的秒数。并非线程需要睡眠 10s，仅传入 10s 这种形式。而是当前时间 + 10s。二，注意子线程函数的返回值通过 thrd_join() 函数获取。

#include <stdio.h>
#include <tinycthread.h>
#include <stdatomic.h>
#include <sys/timeb.h>

atomic_flag resume_flag = ATOMIC_FLAG_INIT;  // 定义一个 atomic_flag 类型变量

int TestAtomicFlag(void *arg){

  int current = 0;

  // 对 atomic_flag 标志位判断并赋值
  // 标志位不为空，该该循环每一秒执行一次
  while(atomic_flag_test_and_set(&resume_flag)){
    current++;
    printf("current: %d\n", current);

    struct timeb tb;
    ftime(&tb);
    thrd_sleep(&(struct timespec){.tv_sec=(time_t)tb.time + 1}, NULL);
  }

  return current;
}

int main(){

  atomic_flag_test_and_set(&resume_flag);  // 设定标志位

  thrd_t t;
  thrd_create(&t, TestAtomicFlag, NULL);

  // 主线程睡五秒，此时子线程可执行五次
  struct timeb tb;
  ftime(&tb);
  thrd_sleep(&(struct timespec){.tv_sec=(time_t)tb.time + 5}, NULL);

  atomic_flag_clear(&resume_flag);  // 清除标志位，子线程停止运行

  // 获取线程函数返回值
  int last_number;
  thrd_join(t, &last_number);

  printf("last_number: %d\n", last_number);

  return 0;
}

        试验结果有两种可能，原因是主线程和子线程开始的时间不同。虽然主线程睡眠 5s，子线程睡眠 1s，但睡眠的起始点不同，可能子线程执行了一次后才和主线程一起执行，此时子线程便会执行 6 次。