多核多线程精确上限计数

/*************************************************************************
    > File Name: counter.c
    > Author: Tom
    > Mail: 
    > Function: 
    > Created Time: Sat 09 Apr 2022 11:56:29 PM HKT
 ************************************************************************/

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include <unistd.h>
#include <stdbool.h>
#include <pthread.h>

#define TOTAL_N 200000304
#define BASE_COUNTER_MAX 10000
#define N 10
#define MIN(x, y) ((x) > (y) ? (y) : (x))

static int s_max = TOTAL_N;

static int atomic_get(int base)
{
    bool b = 0;
    int nums = 0;;

    do {
        int cur = s_max;
        __sync_synchronize();

        nums = MIN(cur, base);
        if (nums == 0) {
            return 0;
        }
        int new = cur - nums;
        b = __sync_bool_compare_and_swap(&s_max, cur, new);
    } while (!b);

    return nums;
}

static void *thread_counter(void *arg)
{
    int total = 0;
    int nums = 0;
    int i = *(int *)&arg;

    while (nums = atomic_get(BASE_COUNTER_MAX)) {
        while (nums--) {
            total += 1;
        }
    }

    printf("thread number = %d, total = %d\n", i, total);

    pthread_exit(NULL);
}

int main(int argc, char *argv[])
{
    int i = 0;
    pthread_t thid[N];

    for (i = 0; i < N; i++) {
        pthread_create(&thid[i], NULL, thread_counter, *(void **)&i);
    }

    for (i = 0; i < N; i++) {
        pthread_join(thid[i], NULL);
    }

    return EXIT_SUCCESS;
}