码迷,mamicode.com
首页 > 其他好文 > 详细

读取proc信息的可扩展实现

时间:2015-01-17 20:43:56      阅读:217      评论:0      收藏:0      [点我收藏+]

标签:

需求

1. 将内存、线程数等信息注册到zk上进行监控

2. 统计信息,为下一步做负载均衡做准备。

实现

本文只解决问题1。

从网上查询了下,这些信息可以从proc文件系统中获取,如果不知道proc的,可以Google下。

网上有读取proc信息的lib——libproc,即 procps , 据说htop等实现就是基于它的。

我下载下来了,include和lib都生成了,好不容易找到一篇教程,结果在

stackoverflow上,见有人说有内存泄露,需要如下方法做。

int main(int argc, char** argv)
{
 // fillarg  used for cmdline
 // fillstat used for cmd
 PROCTAB* proc = openproc(PROC_FILLARG | PROC_FILLSTAT);

 while (proc_t* proc_info = readproc(proc, NULL)) {
    // do something
    freeproc(proc_info)
 }
 closeproc(proc);
}


于是看proc_t的定义,充满了上世纪的风格,似乎迷失在信息各种信息中了,于是无奈放弃了,直接手撸吧。代码稍后附上,如下是几个关键的技术点

1. 即然shell 命名可以获取proc信息,在C中,我们可以通过popen建立管道,获取shell命令的输出。

2. 获取进程号,可以通过 getpid()

3. sizeof(xx)/sizeof(xx[0]) 可以获取数组的大小。

4. 设计了ZeroHelper 结构,方便了扩展

5. offsetof 宏用于获取元素的偏移

6. 函数指针的应用恰到好处。

 

技术分享
#include <time.h>
#include <stddef.h>
#include <string.h>
#include <stdint.h>
#include <ctype.h>
#include <stdio.h>
#include <stdlib.h>
#include<unistd.h>
struct StatInfo
{
    time_t   start_time;
    uint32_t duration_sec;
    uint32_t conn_cnt;
    uint32_t pmem_mb;   // peak virtual memory
    uint32_t vmem_mb;   // virtual memory
    uint32_t rmem_mb;   // real memory
    uint32_t thread_cnt;// 线程数
    uint32_t cpu_usage; // cpu
};
typedef int (*convert_fn)(const char* str, void* value_ptr);
struct ZeroHelper {
    const char* key;
    uint32_t key_len;
    uint32_t value_offset;
    convert_fn fn;
};

int proc_stat_mem_convert(const char* str, void* value_ptr)
{
    uint32_t* value = (uint32_t*)value_ptr;
    while (*str && isspace(*str))
      ++str;
    *value = atoi(str);
    return 0;
}
static const ZeroHelper proc_mem_convert_array[] = {
    {"VmPeak:", sizeof("VmPeak:") - 1, offsetof(StatInfo, pmem_mb), proc_stat_mem_convert},
    {"VmSize:", sizeof("VmSize:") - 1, offsetof(StatInfo, vmem_mb), proc_stat_mem_convert},
    {"VmRSS:", sizeof("VmRSS:") - 1, offsetof(StatInfo, rmem_mb),   proc_stat_mem_convert},
    {"Threads:", sizeof("Threads:") - 1, offsetof(StatInfo, thread_cnt), proc_stat_mem_convert},
};
int fresh_memstat_info(pid_t pid, StatInfo* info)
{
    char proc_cmd [1024];
    snprintf(proc_cmd, sizeof(proc_cmd), "cat /proc/%d/status", pid);
    FILE* fp = popen(proc_cmd, "r");
    char proc_line[1024];
    const ZeroHelper* helper = &(proc_mem_convert_array[0]);
    const int kHelperLen = sizeof(proc_mem_convert_array)/sizeof(proc_mem_convert_array[0]);
    int j = 0;
    while (fgets(proc_line, sizeof(proc_line), fp) != NULL)
    {
        if (j >= kHelperLen)
        {
            break;
        }
        // 忽略 key 头部
        if (strncmp(proc_line,
                    helper->key, helper->key_len) == 0)
        {
            helper->fn(proc_line + helper->key_len,
                       (char*)info + helper->value_offset);
            ++helper;
            ++j;
        }
    }
    pclose(fp);
    return 0;
}
int main()
{
    StatInfo info;
    pid_t pid = getpid();
    fresh_memstat_info(pid, &info);
    printf("%d\t%ukB\t%ukB\t%ukB\t%u\n",
           pid,
           info.pmem_mb, info.vmem_mb,
           info.rmem_mb, info.thread_cnt);
    sleep(1000);
    return 0;
}
View Code

 

读取proc信息的可扩展实现

标签:

原文地址:http://www.cnblogs.com/westfly/p/4230956.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!