Linux下避免程序被重复运行

07 | 08 | 2021

目标

在Linux下，有些程序同时运行多个实例时会出错。因此，需要使用一些方法避免程序被重复运行，即同一时间段内只能有一个程序实例运行。

思路

程序启动的时候打开某个锁文件并对该文件加锁，退出程序时解锁。

如果无法正常加锁，则说明程序已经在内存中运行了，当前线程return即可。

代码实现

首先，先以创建+可读写模式打开锁文件temp.lock（可自行定义文件名）

之后使用flock文件对文件加锁，如果加锁失败，检查errno是否为EWOULDBLOCK。如果是则说明已经有进程持锁，即程序已经有实例在运行。

否则持锁并继续逻辑。

检测代码如下：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
int isProcessRunning()
{
    lock_fd = open("temp.lock",O_CREAT|O_RDWR,0666);
    int rc = flock(lock_fd,LOCK_EX|LOCK_NB); //flock加锁，LOCK_EX -- 排它锁；LOCK_NB -- 非阻塞模式
    if(rc)  //返回值非0，无法正常持锁
    {
        if(EWOULDBLOCK == errno)    //尝试锁住该文件的时候，发现已经被其他服务锁住,errno==EWOULDBLOCK
        {
            printf("Already Running!\n");
            return -1;
        }
    }
    return 0;
}

在main函数中，只需检测返回值不为0退出即可。

1
2
if(isProcessRunning())
    return 0;

备注：

flock函数：

头文件 #include<sys/file.h>

函数原型 int flock(int fd,int operation);

参数 operation有下列四种情况:

LOCK_SH 建立共享锁定。多个进程可同时对同一个文件作共享锁定。

LOCK_EX 建立互斥锁定。一个文件同时只有一个互斥锁定。

LOCK_UN 解除文件锁定状态。

LOCK_NB 无法建立锁定时，此操作可不被阻断，马上返回进程。通常与LOCK_SH或LOCK_EX 做OR(|)组合。

返回值 返回0表示成功，若有错误则返回-1，错误代码存于errno。