dolysis
/
rocksdb
mirror of https://github.com/facebook/rocksdb.git
3
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

db_bench: dump cpu info for Mac. (#7932) 

Summary: Pull Request resolved: https://github.com/facebook/rocksdb/pull/7932

Reviewed By: jay-zhuang

Differential Revision: D26316480

Pulled By: zhichao-cao

fbshipit-source-id: 3e002e49fcb7f60bc9270550a6b3e182fe197551
This commit is contained in:
David CARLIER 2021-02-10 12:55:24 -08:00 committed by Facebook GitHub Bot
parent 7ebde3da45
commit 14fbb43f3e
1 changed files with 33 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
tools/db_bench_tool.cc
View File

@ -18,6 +18,11 @@
#include <stdio.h> #include <stdio.h>
#include <stdlib.h> #include <stdlib.h>
#include <sys/types.h> #include <sys/types.h>
#ifdef __APPLE__
#include <mach/host_info.h>
#include <mach/mach_host.h>
#include <sys/sysctl.h>
#endif
#include <atomic> #include <atomic>
#include <cinttypes> #include <cinttypes>
#include <condition_variable> #include <condition_variable>
@ -2576,7 +2581,7 @@ class Benchmark {
fprintf(stderr, "RocksDB: version %d.%d\n", fprintf(stderr, "RocksDB: version %d.%d\n",
kMajorVersion, kMinorVersion); kMajorVersion, kMinorVersion);
#if defined(__linux) #if defined(__linux) || defined(__APPLE__)
time_t now = time(nullptr); time_t now = time(nullptr);
char buf[52]; char buf[52];
// Lint complains about ctime() usage, so replace it with ctime_r(). The // Lint complains about ctime() usage, so replace it with ctime_r(). The
@ -2584,6 +2589,7 @@ class Benchmark {
fprintf(stderr, "Date: %s", fprintf(stderr, "Date: %s",
ctime_r(&now, buf)); // ctime_r() adds newline ctime_r(&now, buf)); // ctime_r() adds newline
#if defined(__linux)
FILE* cpuinfo = fopen("/proc/cpuinfo", "r"); FILE* cpuinfo = fopen("/proc/cpuinfo", "r");
if (cpuinfo != nullptr) { if (cpuinfo != nullptr) {
char line[1000]; char line[1000];
@ -2608,6 +2614,32 @@ class Benchmark {
fprintf(stderr, "CPU: %d * %s\n", num_cpus, cpu_type.c_str()); fprintf(stderr, "CPU: %d * %s\n", num_cpus, cpu_type.c_str());
fprintf(stderr, "CPUCache: %s\n", cache_size.c_str()); fprintf(stderr, "CPUCache: %s\n", cache_size.c_str());
} }
#elif defined(__APPLE__)
struct host_basic_info h;
size_t hlen = HOST_BASIC_INFO_COUNT;
if (host_info(mach_host_self(), HOST_BASIC_INFO, (host_info_t)&h,
(uint32_t*)&hlen) == KERN_SUCCESS) {
std::string cpu_type;
std::string cache_size;
size_t hcache_size;
hlen = sizeof(hcache_size);
if (sysctlbyname("hw.cachelinesize", &hcache_size, &hlen, NULL, 0) == 0) {
cache_size = std::to_string(hcache_size);
}
switch (h.cpu_type) {
case CPU_TYPE_X86_64:
cpu_type = "x86_64";
break;
case CPU_TYPE_ARM64:
cpu_type = "arm64";
break;
default:
break;
}
fprintf(stderr, "CPU: %d * %s\n", h.max_cpus, cpu_type.c_str());
fprintf(stderr, "CPUCache: %s\n", cache_size.c_str());
}
#endif
#endif #endif
} }