获取系统CPU的核数、软件CPU使用数、内存使用情况、IO读写情况

#ifndef PROCESS_STAT_H_YYG_20091119  
#define PROCESS_STAT_H_YYG_20091119  

typedef long           int64_t;  
typedef unsigned long  uint64_t;  

class CProcessState
{
public:	
	CProcessState(){}
	~CProcessState(){}
	/// 获取当前进程的cpu使用率,返回-1失败  
	int get_cpu_usage();  
		
	/// 获取当前进程内存和虚拟内存使用量,返回-1失败,0成功  
	int get_memory_usage(uint64_t* mem, uint64_t* vmem);  
		
	/// 获取当前进程总共读和写的IO字节数,返回-1失败,0成功  
	int get_io_bytes(uint64_t* read_bytes, uint64_t* write_bytes); 
private:
	 int  get_processor_number();
	 uint64_t file_time_2_utc(const FILETIME* ftime);
		
};
#endif/*PROCESS_STAT_H*/


实现

/** @file 
* @brief 进程统计信息函数的实现 
* @author baibingbing
* 
* 部分代码来自MSDN的例子 
* 
* 需要连接到psapi.lib 
*/  

#include <windows.h>    
#include <assert.h>  
#include "psapi.h"
#include "processState.h" 
  
/// 时间转换  
uint64_t CProcessState::file_time_2_utc(const FILETIME* ftime)  
{  
    LARGE_INTEGER li;  
  
    assert(ftime);  
    li.LowPart = ftime->dwLowDateTime;  
    li.HighPart = ftime->dwHighDateTime;  
    return li.QuadPart;  
}  

/// 获得CPU的核数  
int CProcessState::get_processor_number()  
{  
    SYSTEM_INFO info;  
    GetSystemInfo(&info);  
    return (int)info.dwNumberOfProcessors;  
}  

int CProcessState::get_cpu_usage()  
{  
    //cpu数量  
    static int processor_count_ = -1;  
    //上一次的时间  
    static int64_t last_time_ = 0;  
    static int64_t last_system_time_ = 0;  
  
    FILETIME now;  
    FILETIME creation_time;  
    FILETIME exit_time;  
    FILETIME kernel_time;  
    FILETIME user_time;  
    int64_t system_time;  
    int64_t time;  
    int64_t system_time_delta;  
    int64_t time_delta;  
  
    int cpu = -1;  
  
    if(processor_count_ == -1)  
    {  
        processor_count_ = get_processor_number();  
    }  
  
    GetSystemTimeAsFileTime(&now);  
  
    if (!GetProcessTimes(GetCurrentProcess(), &creation_time, &exit_time,  
        &kernel_time, &user_time))  
    {  
        // We don't assert here because in some cases (such as in the Task Manager)  
        // we may call this function on a process that has just exited but we have  
        // not yet received the notification.  
        return -1;  
    }  
    system_time = (file_time_2_utc(&kernel_time) + file_time_2_utc(&user_time)) / processor_count_;  
    time = file_time_2_utc(&now);  
  
    if ((last_system_time_ == 0) || (last_time_ == 0))  
    {  
        // First call, just set the last values.  
        last_system_time_ = system_time;  
        last_time_ = time;  
        return -1;  
    }  
  
    system_time_delta = system_time - last_system_time_;  
    time_delta = time - last_time_;  
  
    assert(time_delta != 0);  
  
    if (time_delta == 0)  
        return -1;  
  
    // We add time_delta / 2 so the result is rounded.  
    cpu = (int)((system_time_delta * 100 + time_delta / 2) / time_delta);  
    last_system_time_ = system_time;  
    last_time_ = time; 

    return cpu;  
}  

int CProcessState::get_memory_usage(uint64_t* mem, uint64_t* vmem)  
{  
    PROCESS_MEMORY_COUNTERS pmc;  
    if(GetProcessMemoryInfo(GetCurrentProcess(), &pmc, sizeof(pmc)))  
    {  
        if(mem) *mem = pmc.WorkingSetSize;  
        if(vmem) *vmem = pmc.PagefileUsage;  
        return 0;  
    }  
    return -1;  
} 

int CProcessState::get_io_bytes(uint64_t* read_bytes, uint64_t* write_bytes)  
{ 
    IO_COUNTERS io_counter;  
    if(GetProcessIoCounters(GetCurrentProcess(), &io_counter))  
    {  
        if(read_bytes) *read_bytes = io_counter.ReadTransferCount;  
        if(write_bytes) *write_bytes = io_counter.WriteTransferCount;  
        return 0;  
    } 
    return 0;  
}


猜你喜欢

转载自blog.csdn.net/bai2010bingbing/article/details/53408946