add cpuinfo

winlin
Commit 15aea4d9b37d88c50d9ec50ff47bb65e35cbcb5d 15aea4d9 1 parent 951a93ab
trunk/src/app/srs_app_http_api.cpp
trunk/src/app/srs_app_server.cpp
trunk/src/kernel/srs_kernel_utility.cpp
trunk/src/kernel/srs_kernel_utility.hpp
--- a/trunk/src/app/srs_app_http_api.cpp
查看文件 @15aea4d
+++ b/trunk/src/app/srs_app_http_api.cpp
查看文件 @15aea4d
@@ -201,6 +201,7 @@ int SrsApiSummaries::do_process_request(SrsSocket* skt, SrsHttpMessage* req)
     SrsRusage* r = srs_get_system_rusage();
     SrsProcSelfStat* u = srs_get_self_proc_stat();
     SrsProcSystemStat* s = srs_get_system_proc_stat();
+    SrsCpuInfo* c = srs_get_cpuinfo();
     ss << JOBJECT_START
         << JFIELD_ERROR(ERROR_SUCCESS) << JFIELD_CONT
@@ -211,7 +212,12 @@ int SrsApiSummaries::do_process_request(SrsSocket* skt, SrsHttpMessage* req)
             << JFIELD_ORG("rusage_ok", (r->ok? "true":"false")) << JFIELD_CONT
             << JFIELD_ORG("self_cpu_stat_ok", (u->ok? "true":"false")) << JFIELD_CONT
             << JFIELD_ORG("system_cpu_stat_ok", (s->ok? "true":"false")) << JFIELD_CONT
+            << JFIELD_ORG("cpuinfo_ok", (c->ok? "true":"false")) << JFIELD_CONT
             << JFIELD_ORG("mem_kbyte", r->r.ru_maxrss) << JFIELD_CONT
+            << JFIELD_ORG("system_cpu", s->percent) << JFIELD_CONT
+            << JFIELD_ORG("self_cpu", u->percent) << JFIELD_CONT
+            << JFIELD_ORG("nb_processors", c->nb_processors) << JFIELD_CONT
+            << JFIELD_ORG("nb_processors_online", c->nb_processors_online) << JFIELD_CONT
             << JFIELD_ORG("ppid", u->ppid)
         << JOBJECT_END
         << JOBJECT_END;
@@ -241,7 +247,8 @@ int SrsApiRusages::do_process_request(SrsSocket* skt, SrsHttpMessage* req)
     ss << JOBJECT_START
         << JFIELD_ERROR(ERROR_SUCCESS) << JFIELD_CONT
         << JFIELD_ORG("data", JOBJECT_START)
-            << JFIELD_ORG("rusage_ok", (r->ok? "true":"false")) << JFIELD_CONT
+            << JFIELD_ORG("ok", (r->ok? "true":"false")) << JFIELD_CONT
+            << JFIELD_ORG("sample_time", r->sample_time) << JFIELD_CONT
             << JFIELD_ORG("ru_utime", r->r.ru_utime.tv_sec) << JFIELD_CONT
             << JFIELD_ORG("ru_stime", r->r.ru_stime.tv_sec) << JFIELD_CONT
             << JFIELD_ORG("ru_maxrss", r->r.ru_maxrss) << JFIELD_CONT
@@ -286,7 +293,9 @@ int SrsApiSelfProcStats::do_process_request(SrsSocket* skt, SrsHttpMessage* req)
     ss << JOBJECT_START
         << JFIELD_ERROR(ERROR_SUCCESS) << JFIELD_CONT
         << JFIELD_ORG("data", JOBJECT_START)
-            << JFIELD_ORG("self_cpu_stat_ok", (u->ok? "true":"false")) << JFIELD_CONT
+            << JFIELD_ORG("ok", (u->ok? "true":"false")) << JFIELD_CONT
+            << JFIELD_ORG("sample_time", u->sample_time) << JFIELD_CONT
+            << JFIELD_ORG("percent", u->percent) << JFIELD_CONT
             << JFIELD_ORG("pid", u->pid) << JFIELD_CONT
             << JFIELD_STR("comm", u->comm) << JFIELD_CONT
             << JFIELD_STR("state", u->state) << JFIELD_CONT
@@ -359,7 +368,9 @@ int SrsApiSystemProcStats::do_process_request(SrsSocket* skt, SrsHttpMessage* re
     ss << JOBJECT_START
         << JFIELD_ERROR(ERROR_SUCCESS) << JFIELD_CONT
         << JFIELD_ORG("data", JOBJECT_START)
-            << JFIELD_ORG("system_cpu_stat_ok", (s->ok? "true":"false")) << JFIELD_CONT
+            << JFIELD_ORG("ok", (s->ok? "true":"false")) << JFIELD_CONT
+            << JFIELD_ORG("sample_time", s->sample_time) << JFIELD_CONT
+            << JFIELD_ORG("percent", s->percent) << JFIELD_CONT
             << JFIELD_ORG("user", s->user) << JFIELD_CONT
             << JFIELD_ORG("nice", s->nice) << JFIELD_CONT
             << JFIELD_ORG("sys", s->sys) << JFIELD_CONT
--- a/trunk/src/app/srs_app_server.cpp
查看文件 @15aea4d
+++ b/trunk/src/app/srs_app_server.cpp
查看文件 @15aea4d
@@ -56,11 +56,11 @@ CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 // update rusage interval:
 //      SRS_SYS_CYCLE_INTERVAL * SRS_SYS_RUSAGE_RESOLUTION_TIMES
-#define SRS_SYS_RUSAGE_RESOLUTION_TIMES 15
+#define SRS_SYS_RUSAGE_RESOLUTION_TIMES 30
 // update rusage interval:
 //      SRS_SYS_CYCLE_INTERVAL * SRS_SYS_CPU_STAT_RESOLUTION_TIMES
-#define SRS_SYS_CPU_STAT_RESOLUTION_TIMES 15
+#define SRS_SYS_CPU_STAT_RESOLUTION_TIMES 30
 SrsListener::SrsListener(SrsServer* server, SrsListenerType type)
 {
--- a/trunk/src/kernel/srs_kernel_utility.cpp
查看文件 @15aea4d
+++ b/trunk/src/kernel/srs_kernel_utility.cpp
查看文件 @15aea4d
@@ -63,6 +63,7 @@ static SrsRusage _srs_system_rusage;
 SrsRusage::SrsRusage()
 {
     ok = false;
+    sample_time = 0;
     memset(&r, 0, sizeof(rusage));
 }
@@ -78,6 +79,9 @@ void srs_update_system_rusage()
         return;
     }
+    srs_update_system_time_ms();
+    _srs_system_rusage.sample_time = srs_get_system_time_ms();
+    
     _srs_system_rusage.ok = true;
 }
@@ -87,11 +91,70 @@ static SrsProcSystemStat _srs_system_cpu_system_stat;
 SrsProcSelfStat::SrsProcSelfStat()
 {
     ok = false;
+    sample_time = 0;
+    percent = 0;
+    
+    pid = 0;
+    memset(comm, 0, sizeof(comm));
+    state = 0;
+    ppid = 0;
+    pgrp = 0;
+    session = 0;
+    tty_nr = 0;
+    tpgid = 0;
+    flags = 0;
+    minflt = 0;
+    cminflt = 0;
+    majflt = 0;
+    cmajflt = 0;
+    utime = 0;
+    stime = 0;
+    cutime = 0;
+    cstime = 0;
+    priority = 0;
+    nice = 0;
+    num_threads = 0;
+    itrealvalue = 0;
+    starttime = 0;
+    vsize = 0;
+    rss = 0;
+    rsslim = 0;
+    startcode = 0;
+    endcode = 0;
+    startstack = 0;
+    kstkesp = 0;
+    kstkeip = 0;
+    signal = 0;
+    blocked = 0;
+    sigignore = 0;
+    sigcatch = 0;
+    wchan = 0;
+    nswap = 0;
+    cnswap = 0;
+    exit_signal = 0;
+    processor = 0;
+    rt_priority = 0;
+    policy = 0;
+    delayacct_blkio_ticks = 0;
+    guest_time = 0;
+    cguest_time = 0;
 }
 SrsProcSystemStat::SrsProcSystemStat()
 {
     ok = false;
+    sample_time = 0;
+    percent = 0;
+    memset(label, 0, sizeof(label));
+    user = 0;
+    nice = 0;
+    sys = 0;
+    idle = 0;
+    iowait = 0;
+    irq = 0;
+    softirq = 0;
+    steal = 0;
+    guest = 0;
 }
 SrsProcSelfStat* srs_get_self_proc_stat()
@@ -104,17 +167,14 @@ SrsProcSystemStat* srs_get_system_proc_stat()
     return &_srs_system_cpu_system_stat;
 }
-void srs_update_proc_stat()
+bool get_proc_system_stat(SrsProcSystemStat& r)
 {
-    // system cpu stat
-    if (true) {
     FILE* f = fopen("/proc/stat", "r");
     if (f == NULL) {
         srs_warn("open system cpu stat failed, ignore");
-            return;
+        return false;
     }
-        SrsProcSystemStat& r = _srs_system_cpu_system_stat;
     for (;;) {
         int ret = fscanf(f, "%4s %lu %lu %lu %lu %lu "
             "%lu %lu %lu %lu\n", 
@@ -133,17 +193,18 @@ void srs_update_proc_stat()
     }
     fclose(f);
-    }
-    // self cpu stat
-    if (true) {
+    return r.ok;
+}
+
+bool get_proc_self_stat(SrsProcSelfStat& r)
+{
     FILE* f = fopen("/proc/self/stat", "r");
     if (f == NULL) {
         srs_warn("open self cpu stat failed, ignore");
-            return;
+        return false;
     }
-        SrsProcSelfStat& r = _srs_system_cpu_self_stat;
     int ret = fscanf(f, "%d %32s %c %d %d %d %d "
         "%d %u %lu %lu %lu %lu "
         "%lu %lu %ld %ld %ld %ld "
@@ -168,5 +229,83 @@ void srs_update_proc_stat()
     }
     fclose(f);
+    
+    return r.ok;
+}
+
+void srs_update_proc_stat()
+{
+    srs_update_system_time_ms();
+    
+    // system cpu stat
+    if (true) {
+        SrsProcSystemStat r;
+        if (!get_proc_system_stat(r)) {
+            return;
+        }
+        
+        r.sample_time = srs_get_system_time_ms();
+        
+        // calc usage in percent
+        SrsProcSystemStat& o = _srs_system_cpu_system_stat;
+        
+        // @see: http://blog.csdn.net/nineday/article/details/1928847
+        int64_t total = (r.user + r.nice + r.sys + r.idle + r.iowait + r.irq + r.softirq + r.steal + r.guest) 
+            - (o.user + o.nice + o.sys + o.idle + o.iowait + o.irq + o.softirq + o.steal + o.guest);
+        int64_t idle = r.idle - o.idle;
+        if (total > 0) {
+            r.percent = (float)(1 - idle / (double)total);
+        }
+        
+        // upate cache.
+        _srs_system_cpu_system_stat = r;
+    }
+    
+    // self cpu stat
+    if (true) {
+        SrsProcSelfStat r;
+        if (!get_proc_self_stat(r)) {
+            return;
+        }
+        
+        srs_update_system_time_ms();
+        r.sample_time = srs_get_system_time_ms();
+        
+        // calc usage in percent
+        SrsProcSelfStat& o = _srs_system_cpu_self_stat;
+        
+        // @see: http://stackoverflow.com/questions/16011677/calculating-cpu-usage-using-proc-files
+        int64_t total = r.sample_time - o.sample_time;
+        int64_t usage = (r.utime + r.stime) - (o.utime + o.stime);
+        if (total > 0) {
+            r.percent = (float)(usage * 1000 / (double)total / 100);
         }
+        
+        // upate cache.
+        _srs_system_cpu_self_stat = r;
+    }
+}
+
+SrsCpuInfo::SrsCpuInfo()
+{
+    ok = false;
+    
+    nb_processors = 0;
+    nb_processors_online = 0;
+}
+
+SrsCpuInfo* srs_get_cpuinfo()
+{
+    static SrsCpuInfo* cpu = NULL;
+    if (cpu != NULL) {
+        return cpu;
+    }
+    
+    // initialize cpu info.
+    cpu = new SrsCpuInfo();
+    cpu->ok = true;
+    cpu->nb_processors = sysconf(_SC_NPROCESSORS_CONF);
+    cpu->nb_processors_online = sysconf(_SC_NPROCESSORS_ONLN);
+    
+    return cpu;
 }
--- a/trunk/src/kernel/srs_kernel_utility.hpp
查看文件 @15aea4d
+++ b/trunk/src/kernel/srs_kernel_utility.hpp
查看文件 @15aea4d
@@ -40,7 +40,11 @@ extern void srs_update_system_time_ms();
 // @see: man getrusage
 struct SrsRusage
 {
+    // whether the data is ok.
     bool ok;
+    // the time in ms when sample.
+    int64_t sample_time;
+    
     rusage r;
     SrsRusage();
@@ -56,6 +60,10 @@ struct SrsProcSelfStat
 {
     // whether the data is ok.
     bool ok;
+    // the time in ms when sample.
+    int64_t sample_time;
+    // the percent of usage. 0.153 is 15.3%.
+    float percent;
     // pid %d      The process ID.
     int pid;
@@ -203,6 +211,10 @@ struct SrsProcSystemStat
 {
     // whether the data is ok.
     bool ok;
+    // the time in ms when sample.
+    int64_t sample_time;
+    // the percent of usage. 0.153 is 15.3%.
+    float percent;
     // always be cpu
     char label[32];
@@ -248,4 +260,21 @@ extern SrsProcSystemStat* srs_get_system_proc_stat();
 // the deamon st-thread will update it.
 extern void srs_update_proc_stat();
+// @see: cat /proc/cpuinfo 
+struct SrsCpuInfo
+{
+    // whether the data is ok.
+    bool ok;
+    
+    // The number of processors configured.
+    int nb_processors;
+    // The number of processors currently online (available).
+    int nb_processors_online;
+    
+    SrsCpuInfo();
+};
+
+// get system cpu info, use cache to avoid performance problem.
+extern SrsCpuInfo* srs_get_cpuinfo();
+
 #endif