最后更新于 .

如何保证一个程序在单台服务器上只有唯一实例呢,本着简单实用的思想写了一个实现函数:

/* 判断当前进程是否已经运行 */
static bool is_running(const char* prg)
{
    const char* pid_file = ".tmp_pid";
    const char* p = strrchr(prg, '/');
    if (p)
    {
        p++;
    }
    else
    {
        p = prg;
    }
    char cmd[128] = {0};
    sprintf(cmd, "pgrep %s >%s", p, pid_file);
    system(cmd);
    std::string s;
    FILE* fp = fopen(pid_file, "r");
    if (fp == NULL)
    {
        fprintf(stderr, "ERROR: can not open pid file: %s!\n", pid_file);
        return false;
    }
    int pid;
    while (read_line(fp, s))
    {
        pid = atoi(s.c_str());
        if ((pid_t)pid != getpid())
        {
            break;
        }
        else
        {
            pid = 0;
        }
    }
    fclose(fp);
    sprintf(cmd, "rm -f %s", pid_file);
    system(cmd);
    return pid>0;
}

该函数的参数是可执行文件名,可以在main函数中以argv[0]传入,原理很简单,就是调用SYSTEM函数获取同名的进程是否存在,这样有个不好的地方就是如果可执行文件被重命名过的话执行仍然是可以拉起来的,不过相信一般情况下程序不会无故重命名吧.

 

Pingbacks

Pingbacks已打开。

Trackbacks

引用地址

评论

  1. ltee

    ltee on #

    如果遇上kill -9 呢?

    Reply

发表评论