C\C++实现CPU核上绑定固定线程

 参考文章：Linux中线程与CPU核的绑定_bluenet13的博客-CSDN博客

long sysconf(int name);

函数sysconf获取运行时的系统配置信息

设置线程亲和性，将线程绑定到指定CPU核

int pthread_setaffinity_np(pthread_t thread, size_t cpusetsize,const cpu_set_t *cpuset);

thread:线程id

cpusetsize:集合内存大小

cpuset:CPU核的集合

获取指定线程的CPU集合

int pthread_getaffinity_np(pthread_t thread, size_t cpusetsize, cpu_set_t *cpuset);

thread:线程id

cpusetsize:集合内存大小

cpuset:CPU核的集合

对CPU核的集合操作

void CPU_ZERO (cpu_set_t *set); //初始化，设为空

void CPU_SET (int cpu, cpu_set_t *set); //将某个cpu加入cpu集中

void CPU_CLR (int cpu, cpu_set_t *set); //将某个cpu从cpu集中移出

int CPU_ISSET (int cpu, const cpu_set_t *set); //判断某个cpu是否已在cpu集中设置了

#define _GNU_SOURCE
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <pthread.h>
#include <sched.h>

void *myfun(void *arg)
{
    cpu_set_t mask;
    cpu_set_t get;
    char buf[256];
    int i;
    int j;
    int num = sysconf(_SC_NPROCESSORS_CONF); // 在运行时获取配置信息，获取CPU核个数
    CPU_ZERO(&mask);    // 初始化，设为空
    printf("system has %d processor(s)\n", num);
    for(i = 0; i < num; i++){
        CPU_SET(i, &mask);  // 将某个cpu加入cpu集中 
    }
    // 将线程绑定到cpu集中的各个CPU核，这样做不太好，会导致线程在各CPU核中切换，导致性能下降
    // 本次只为测试
    if (pthread_setaffinity_np(pthread_self(), sizeof(mask), &mask) < 0) {
        fprintf(stderr, "set thread affinity failed\n");
    }
    CPU_ZERO(&get);
    if (pthread_getaffinity_np(pthread_self(), sizeof(get), &get) < 0) {
        fprintf(stderr, "get thread affinity failed\n");
    } 
    for (i = 0; i < num; i++) {
        if (CPU_ISSET(i, &get)) {
            printf("thread %ld is running in processor %d\n", pthread_self(), i);
        }
        j = 0;
        while (j++ < 100000000){
            memset(buf, 0, sizeof(buf));
        }
    }
    pthread_exit(NULL);
}
int main(){
    int num = sysconf(_SC_NPROCESSORS_CONF);
    printf("CPU核数:%d \n", num);
    pthread_t tid;
    if (pthread_create(&tid, NULL, myfun, NULL) != 0) {
        fprintf(stderr, "thread create failed\n");
        return -1;
    }
    pthread_join(tid, NULL);
    return 0;
}

在终端执行操作：

[root@chen02 CPUBind]# g++ -o cpu CPU_Bind.cpp -lpthread