介绍
通常后台服务器程序都必须有且只有一个进程,那么如何单进程呢?
本例子是通过 flock
函数对 /var/run/myserver.pid
记录 pid 文件的进行加锁
- 若加锁不正常,说明后台服务进程已经在运行了,这时则直接报错退出
- 若加锁成功,说明后台服务进程没有在运行,这时可以正常启用进程
后台服务程序单进程控制
详细不多说,直接看代码
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <errno.h>
#include <fcntl.h>
#define PID_BUF_LEN (20)
#define RUN_PID_FILE "/var/run/myserver.pid"
// 服务进程单实例运行
// 返回值: 1-- 正在运行,0-- 未运行,-1-- 出错
int server_is_running()
{int fd = open(RUN_PID_FILE, O_WRONLY|O_CREAT);
if(fd < 0)
{printf("open run pid err(%d)! %s\n", errno, RUN_PID_FILE);
return -1;
}
// 加锁
// LOCK_SH 建立共享锁定。多个进程可同时对同一个文件作共享锁定。// LOCK_EX 建立互斥锁定。一个文件同时只有一个互斥锁定。if(flock(fd, LOCK_EX|LOCK_NB) == -1)
{
// 加不上锁,则是服务正在运行,已上锁了
printf("server is runing now! errno=%d\n", errno);
close(fd);
return 1;
}
// 加锁成功,证明服务没有运行
// 文件句柄不要关,也不要解锁
// 进程退出,自动就解锁了
printf("myserver is not running! begin to run..... pid=%ld\n", (long)getpid());
char pid_buf[PID_BUF_LEN] = {0};
snprintf(pid_buf, sizeof(pid_buf)-1, "%ld\n", (long)getpid());
// 把进程 pid 写入到 /var/run/myserver.pid 文件
write(fd, pid_buf, strlen(pid_buf));
return 0;
}
int main(void)
{
// 进程单实例运行检测
if(0 != server_is_running())
{printf("myserver process is running!!!!! Current process will exit !\n");
return -1;
}
while(1)
{printf("myserver doing ... \n");
sleep(2);
}
return 0;
}
运行结果
运行程序,可知进程 pid 是6965
[root@lincoding singleprocess]# ./myserver
server is not running! begin to run..... pid=6965
myserver doing ...
myserver doing ...
myserver doing ...
myserver doing ...
myserver doing ...
myserver doing ...
myserver doing ...
myserver doing ...
/var/run/myserver.pid
也记录此进程的 pid 号,ps auxf | grep myserver
可知 mysever
进程一直运行着
[root@lincoding singleprocess]# cat /var/run/myserver.pid
6965
[root@lincoding singleprocess]#
[root@lincoding singleprocess]# ps auxf | grep myserver
root 6965 0.0 0.0 3924 460 pts/0 S+ 00:32 0:00 | \_ ./myserver
root 9976 0.0 0.0 103256 856 pts/1 S+ 00:35 0:00 \_ grep myserver
[root@lincoding singleprocess]#
此时,再运行 myserver
程序,这时会报错退出,因为检测到 myserver
程序已经在运行中,不可以起另外一个进程,从而达到了后台服务程序单进程控制
[root@lincoding singleprocess]# ./myserver
server is runing now! errno=11
myserver process is running!!!!! Current process will exit !