进程单例模式

1.问题

不同于线程单例模式,很多时候我们希望一个程序只有一个实例在运行。
常规的做法有创建一个命名文件,通过判断这个文件存在不存在来确定是否已经有进程实例启动
更进一步,可以通过对文件加锁的方式,从而避免前面方法中需要在程序退出时删除文件的操作,但是文件加锁
需要文件系统的支持。fat32等文件系统不一定支持该操作

2.socket方式

可以利用socket端口作为锁,而且程序退出后,操作系统会自动释放对该端口的绑定,即自动释放锁,不管程序是如何退出。方式如下:

     SingletonProcess single(8977);
     if (!single())
     {
       std::cerr << "process running already. See " << single.GetLockFileName() << std::endl;
       return 1;
     }
class SingletonProcess
{
public:
    SingletonProcess(uint16_t port0)
            : socket_fd(-1)
              , rc(1)
              , port(port0)
    {
    }

    ~SingletonProcess()
    {
        if (socket_fd != -1)
        {
            close(socket_fd);
        }
    }

    bool operator()()
    {
        if (socket_fd == -1 || rc)
        {
            socket_fd = -1;
            rc = 1;

            if ((socket_fd = socket(AF_INET, SOCK_DGRAM, 0)) < 0)
            {
                throw std::runtime_error(std::string("Could not create socket: ") +  strerror(errno));
            }
            else
            {
                struct sockaddr_in name;
                name.sin_family = AF_INET;
                name.sin_port = htons (port);
                name.sin_addr.s_addr = htonl (INADDR_ANY);
                rc = bind (socket_fd, (struct sockaddr *) &name, sizeof (name));
            }
        }
        return (socket_fd != -1 && rc == 0);
    }

    std::string GetLockFileName()
    {
        return "port " + std::to_string(port);
    }

private:
    int socket_fd = -1;
    int rc;
    uint16_t port;
};

猜你喜欢

转载自blog.csdn.net/werweqg/article/details/82623716