mirror of https://github.com/trapexit/mergerfs.git
34 lines
810 B
C++
34 lines
810 B
C++
#pragma once
|
|
|
|
#ifndef _GNU_SOURCE
|
|
#define _GNU_SOURCE
|
|
#endif
|
|
|
|
#include <pthread.h>
|
|
#include <sched.h>
|
|
|
|
#include <set>
|
|
#include <unordered_map>
|
|
#include <vector>
|
|
|
|
|
|
class CPU
|
|
{
|
|
public:
|
|
typedef std::vector<pthread_t> ThreadIdVec;
|
|
typedef std::vector<int> CPUVec;
|
|
typedef std::unordered_map<int,int> CPU2CoreMap;
|
|
typedef std::unordered_map<int,std::set<int>> Core2CPUsMap;
|
|
|
|
public:
|
|
static int count();
|
|
static int getaffinity(const pthread_t thread_id, cpu_set_t *cpuset);
|
|
static int setaffinity(const pthread_t thread_id, cpu_set_t *cpuset);
|
|
static int setaffinity(const pthread_t thread_id, const int cpu);
|
|
static int setaffinity(const pthread_t thread_id, const std::set<int> cpus);
|
|
|
|
static CPU::CPUVec cpus();
|
|
static CPU::CPU2CoreMap cpu2core();
|
|
static CPU::Core2CPUsMap core2cpus();
|
|
};
|