mirror of
https://github.com/trapexit/mergerfs.git
synced 2025-04-22 12:32:16 +08:00
b83bd5fd14
First draft. Need to investigate why upstream patches include certain function that appear to exist in FreeBSD 14.2.
33 lines
785 B
C++
33 lines
785 B
C++
#pragma once
|
|
|
|
#include <pthread.h>
|
|
#ifdef __FreeBSD__
|
|
#include <pthread_np.h>
|
|
#endif
|
|
#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(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();
|
|
};
|