基于zookeeper的daemon框架方案——支持容灾和心跳监控

void CDaemonMaster::RunForever()
{
    while (!g_master_need_quit)
    {
        sleep(1);

        // 无论是否打开锁开关都要加锁，这是为了监控daemon存活与否
        string zk_cur_dis_node;
        int ret = ZK_PROXY_INSTANCE->distributed_lock(_zookeeper_lock_root_path + _process_name + "/",
                zk_cur_dis_node);
        if (ret == EN_ZKPROXY_LOCKED)
        {
            // 加锁成功，可以执行主任务
        }
        else if (ret == EN_ZKPROXY_LOCK_RETRY)
        {
            // 如果必须加锁成功才能执行，则未加锁成功关闭所有子进程
            if (_zookeeper_lock_switch)
            {
                StopWorkers();
                continue;
            }
            // 如果不要求加锁成功，则可以执行主任务
        }
        else
        {
            ERROR_LOG("distributed_lock failed, ret:%d, msg:%s",
                    ret, ZK_PROXY_INSTANCE->get_err_msg());
            if (!_keep_working_when_zk_error)
            {
                // zk发生错误，杀掉所有子进程，继续下一轮
                StopWorkers();
                continue;
            }
            else
            {
                // zk出错仍然工作
                ret = 0;
            }
        }

        // fork进程，开始工作
        if(_worker_pids.size() < _worker_num)
        {
            StartWorkers();
        }

        while(true)
        {
            int status = 0;
            pid_t pid = waitpid(-1, &status, WNOHANG);
            if(pid < 0)
            {
                if(errno != ECHILD)
                {
                    ERROR_LOG("waitpid error, msg=%s", strerror(errno));
                }
                break;
            }
            else if(pid == 0)
            {
                //DEBUG_LOG("no process exit");
                break;
            }
            else
            {
                Alarm("[%s] child process dead", _process_name.c_str());
                ERROR_LOG("child pid=%d exit unexpected! status=%d, WIFEXITED(status)=%d, WEXITSTATUS(status)=%d, WIFSIGNALED(status)=%d, WTERMSIG(status)=%d",
                        pid,
                        status,
                        WIFEXITED(status),
                        WEXITSTATUS(status),
                        WIFSIGNALED(status),
                        WTERMSIG(status));

                _worker_pids.erase(pid);
            }
        }
    }

    // 停止工作
    StopWorkers();

    // 释放锁
    ZK_PROXY_INSTANCE->distributed_lock_clean();

}