steven/TinySTL
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

增加在linux下查询程序内存使用量的接口

Browse Source
This commit is contained in:
邹晓航
2014-10-23 13:31:32 +08:00
parent 78a59a83b1
commit 995d7230b2
2 changed files with 16 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
TinySTL/Profiler/Profiler.cpp
View File

@@ -29,12 +29,19 @@ namespace TinySTL{
} }
size_t ProfilerInstance::memory(MemoryUnit mu){ size_t ProfilerInstance::memory(MemoryUnit mu){
size_t memory = 0;
#ifdef WIN32 #ifdef WIN32
PROCESS_MEMORY_COUNTERS pmc; PROCESS_MEMORY_COUNTERS pmc;
HANDLE hProcess = GetCurrentProcess(); HANDLE hProcess = GetCurrentProcess();
if (!GetProcessMemoryInfo(hProcess, &pmc, sizeof(pmc))) if (!GetProcessMemoryInfo(hProcess, &pmc, sizeof(pmc)))
throw std::runtime_error("GetProcessMemoryInfo failed"); throw std::runtime_error("GetProcessMemoryInfo failed");
size_t memory = 0; CloseHandle(hProcess);
#else
struct rusage usage;
if(getrusage(RUSAGE_SELF, &usage) == -1)
throw std::runtime_error("getrusage failed");
memory = usage.ru_maxrss / 1024;//<2F><><EFBFBD><EFBFBD>ijЩlinuxƽ̨<C6BD><CCA8><EFBFBD><EFBFBD>ȫ֧<C8AB><D6A7>getrusage<67><65>ru_maxrss<73><73><EFBFBD>Ƿ<EFBFBD><C7B7><EFBFBD>0
#endif
switch (mu){ switch (mu){
case MemoryUnit::KB_: case MemoryUnit::KB_:
memory = pmc.WorkingSetSize KB; memory = pmc.WorkingSetSize KB;
@@ -46,10 +53,7 @@ namespace TinySTL{
memory = pmc.WorkingSetSize GB; memory = pmc.WorkingSetSize GB;
break; break;
} }
CloseHandle(hProcess);
return memory; return memory;
#else
#endif
} }
} }
} }

14
TinySTL/Profiler/Profiler.h
View File

@@ -14,6 +14,8 @@
#include <Psapi.h> #include <Psapi.h>
#pragma comment(lib, "psapi.lib") #pragma comment(lib, "psapi.lib")
#else #else
#include <sys/resource>
#include <sys/time.h>
#endif #endif
namespace TinySTL{ namespace TinySTL{
@@ -34,14 +36,14 @@ namespace TinySTL{
static TimePoint startTime; static TimePoint startTime;
static TimePoint finishTime; static TimePoint finishTime;
public: public:
static void start(); static void start();//<2F><>ʼ<EFBFBD><CABC>ʱ
static void finish(); static void finish();//<2F><><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ʱ
static void dumpDuringTime(std::ostream& os = std::cout); static void dumpDuringTime(std::ostream& os = std::cout);//<2F><>ӡʱ<D3A1><CAB1>
static double second(); static double second();//<2F><><EFBFBD><EFBFBD>Ϊ<EFBFBD><CEAA>λ<EFBFBD><CEBB><EFBFBD><EFBFBD>ʱ<EFBFBD><CAB1>
static double millisecond(); static double millisecond();//<2F>Ժ<EFBFBD><D4BA><EFBFBD>Ϊ<EFBFBD><CEAA>λ<EFBFBD><CEBB><EFBFBD><EFBFBD>ʱ<EFBFBD><CAB1>
static size_t memory(MemoryUnit mu = MemoryUnit::KB_); static size_t memory(MemoryUnit mu = MemoryUnit::KB_);//<2F><>ѯ<EFBFBD><D1AF>ǰ<EFBFBD><C7B0><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ڴ<EFBFBD>ʹ<EFBFBD><CAB9><EFBFBD><EFBFBD>
}; };
} }
} }