Linux获取线程CPU使用情况

clock_gettime(CLOCK_THREAD_CPUTIME_ID, &currTime)

int pthread_getcpuclockid(pthread_t thread, clockid_t *clockid);

#include <iostream>
#include <atomic>
#include <cassert>
#include <condition_variable>
#include <ctime>
#include <mutex>
#include <thread>
#include <vector>

#include <pthread.h>

static inline time_t cpu_timestamp_ns(clockid_t clockid) {
  struct timespec t;
  if (-1 == clock_gettime(clockid, &t)) {
    perror("clock_gettime");
	return 0;
  }
  return t.tv_sec * 1000000000 + t.tv_nsec;
}

static void foo(
	size_t secs, std::atomic<size_t> *finished, const bool *permit_join,
	std::condition_variable *cv, std::mutex *mu
) {
	std::this_thread::sleep_for(std::chrono::seconds(secs));

	finished->fetch_add(1, std::memory_order_relaxed);
	std::unique_lock<std::mutex> lock(*mu);
	// 等到外面不再读取自己的cpu time再结束，不然clock_gettime会报错
	cv->wait(lock, [permit_join]() { return *permit_join; });
}

int main() {
	constexpr size_t num_threads = 8;
	std::atomic<size_t> finished(0);
	bool permit_join = false;
	std::condition_variable cv;
	std::mutex mu;
	std::vector<std::thread> ts;
	std::vector<clockid_t> clock_ids;
	for (size_t i = 0; i < num_threads; ++i) {
		// 让这些thread sleep不同的时间
		ts.emplace_back(foo, i, &finished, &permit_join, &cv, &mu);
		pthread_t thread_id = ts[i].native_handle();
		clockid_t clock_id;
		assert(pthread_getcpuclockid(thread_id, &clock_id) == 0);
		clock_ids.push_back(clock_id);
	}
	// 每个线程最开始的时间戳
	std::vector<time_t> ori;
	for (size_t i = 0; i < num_threads; ++i) {
		ori.push_back(cpu_timestamp_ns(clock_ids[i]));
	}
	while (finished.load(std::memory_order_relaxed) != num_threads) {
		time_t tot = 0;
		for (size_t i = 0; i < num_threads; ++i) {
			// 当前时间戳减去开始的时间戳就是中间经过的CPU time
			tot += cpu_timestamp_ns(clock_ids[i]) - ori[i];
		}
		std::cerr << tot << " ns" << std::endl;
		std::this_thread::sleep_for(std::chrono::seconds(1));
	}
	{
		std::unique_lock<std::mutex> lock(mu);
		permit_join = true;
	}
	cv.notify_all();
	for (size_t i = 0; i < num_threads; ++i) {
		ts[i].join();
	}

	return 0;
}