Linux 控制CPU使用率

曾經看過《編程之美》上提到說使 CPU的使用率固定在百分之多少。然后這次剛好要用到這個東西，下面是一個簡單的實現。基於多線程：

Linux 版本：

#include <iostream>
#include <pthread.h>
#include <time.h>
#include <math.h>

using namespace std;

typedef long long int int64;
const int NUM_THREADS = 24; // CPU 核數
int eachTime = 100;
int cpuinfo = 0; // 占用率

int64 GetTickCount()
{
    timespec now;
    clock_gettime(CLOCK_MONOTONIC, &now);
    int64 sec = now.tv_sec;
    int64 nsec = now.tv_nsec;
    return sec*1000 + nsec/1000000;
}


void* CPUCost(void *args)
{
    std::cout << "XXXX CPUCost" << std::endl;
    int busyTime = eachTime * cpuinfo / 100;
    //std::cout << "XXXX cpuinfo = " << cpuinfo << std::endl;
    int idleTime = eachTime - busyTime;
    int64 startTime = 0;
    while (true)
    {
        startTime = GetTickCount();
        while((GetTickCount() - startTime) <= busyTime)
        {
            ;
        }

        usleep(100);
    }
}



int main(int argc, char **argv)
{
    std::cin >> cpuinfo;
    pthread_t t[NUM_THREADS];
    for(int i = 0; i < NUM_THREADS; i++)
    {
        int ret = pthread_create(&t[i], NULL, CPUCost, NULL);
        if(ret)
         {
             std::cout << "XXXX create err" << std::endl;
         }
    }

    pthread_exit(NULL);


    return 0;
}

編譯方式： g++ testCPU.cc -lpthread -lrt -o testCPU

    注：因為只是用來測試，所以寫的很粗糙。大致的原理是對的，細節部分請大家自行優化了。

 

Windows 版本：

#include<iostream>
#include<Windows.h>
#include<cmath>

using namespace std;

int eachTime = 100;
void CPU(int busy,int idle);

int main()
{    
    int level;    
    cin >> level;    
    int busyTime = eachTime*level / 100;    
    int idleTime = eachTime - busyTime;    
    CPU(busyTime,idleTime);    
    return 0;
}
void CPU(int busy, int idle)
{    
    INT64 startTime = 0;    
    while (true)
    {        
        startTime = GetTickCount();
        while ((GetTickCount()-startTime)<=busy)        
        {            
            ;        
        }        
        Sleep(idle);    
    }
}

注： 沒有實現多線程。