#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <errno.h>
#include <sys/file.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;
}
 

轉載於 小林coding