static void cpFind_restart_worker(int pid, sigset_t *block_alarm, int worker_exit_code)
{
int i, j, new_pid;
for (j = 0; j < CPGS->group_num; j++)
{
cpGroup *G = &CPGS->G[j];
for (i = G->worker_num; i >= 0; i--)
{
if (pid != G->workers[i].pid || G->workers_status[i] == CP_WORKER_DEL)
{//对比pid||回收的不拉起
continue;
}
else
{
cpLog("worker exit!worker index %d,worker id %d,exit code %d\n", i, pid, WEXITSTATUS(worker_exit_code));
cpShareMemory *sm_obj = &(G->workers[i].sm_obj);
sm_obj->mem = NULL;
pid = 0;
new_pid = cpFork_one_worker(i, j);
if (new_pid < 0)
{
cpLog("Fork worker process failed. Error: %s [%d]", strerror(errno), errno);
sigprocmask(SIG_UNBLOCK, block_alarm, NULL);
}
else
{
G->workers[i].pid = new_pid;
}
}
}
}
}
static void cpManagerAdd(int sig)
{
int i, j;
for (j = 0; j < CPGS->group_num; j++)
{
cpGroup *G = &CPGS->G[j];
for (i = G->worker_num - 1; i >= G->worker_min; i--)
{
if (G->workers[i].pid == 0)
{//只创建刚分配并且pid为0的
int new_pid = cpFork_one_worker(i, j);
if (new_pid < 0)
{
cpLog("Fork worker process failed. Error: %s [%d]", strerror(errno), errno);
}
else
{
G->workers[i].pid = new_pid;
}
}
}
}
}
static void cpManagerReload(int sig)
{
zval *group_conf = NULL;
group_conf = cpGetConfig(CPGC.ini_file);
if (!Z_BVAL_P(group_conf)) {
cpLog("parse ini file[%s] error,%s reload error!", CPGC.ini_file, CPGC.title);
} else {
zval **v, **conf;
if (zend_hash_find(Z_ARRVAL_P(group_conf), CPGC.title, strlen(CPGC.title) + 1, (void **) &conf) == SUCCESS) {
if (pthread_mutex_lock(CPGS->mutex_lock) == 0) {
if (zend_hash_find(Z_ARRVAL_PP(conf), ZEND_STRS("pool_max"), (void **) &v) == SUCCESS) {
convert_to_long(*v);
CPGS->worker_max = (int) Z_LVAL_PP(v);
}
if (zend_hash_find(Z_ARRVAL_PP(conf), ZEND_STRS("pool_min"), (void **) &v) == SUCCESS) {
convert_to_long(*v);
int new_min = (int) Z_LVAL_PP(v);
if (new_min > CPGC.worker_min) {//增加最小
while (CPGS->worker_num < new_min) {
cpCreate_worker_mem(CPGS->worker_num);
CPGS->workers_status[CPGS->worker_num] = CP_WORKER_IDLE;
CPGS->worker_num++; //先加 线程安全
int new_pid = cpFork_one_worker(CPGS->worker_num - 1);
if (new_pid < 0) {
cpLog("Fork worker process failed. Error: %s [%d]", strerror(errno), errno);
} else {
CPGS->workers[CPGS->worker_num - 1].pid = new_pid;
}
}
}
CPGC.worker_min = new_min;
}
if (zend_hash_find(Z_ARRVAL_PP(conf), ZEND_STRS("recycle_num"), (void **) &v) == SUCCESS) {
convert_to_long(*v);
CPGC.recycle_num = (int) Z_LVAL_PP(v);
}
if (zend_hash_find(Z_ARRVAL_PP(conf), ZEND_STRS("idel_time"), (void **) &v) == SUCCESS) {
convert_to_long(*v);
CPGC.idel_time = (int) Z_LVAL_PP(v);
}
if (pthread_mutex_unlock(CPGS->mutex_lock) != 0) {
cpLog("pthread_mutex_unlock. Error: %s [%d]", strerror(errno), errno);
}
}
} else {
cpLog("find %s failed,The reload can only modify 'pool_min','pool_max','recycle_num' and 'idel_time',if you want modify other options please restart pool", CPGC.title);
}
zval_ptr_dtor(&group_conf);
}
}
static void cpManagerAdd(int sig)
{
int i;
for (i = CPGS->worker_num - 1; i >= CPGC.worker_min; i--) {
if (CPGS->workers[i].pid == 0) {//只创建刚分配并且pid为0的
int new_pid = cpFork_one_worker(i);
if (new_pid < 0) {
// CPGS->workers[i].pid = -1;//todo fork失敗的處理
cpLog("Fork worker process failed. Error: %s [%d]", strerror(errno), errno);
} else {
CPGS->workers[i].pid = new_pid;
}
}
}
}
static void cpManagerReload(int sig)
{
zval *group_conf = NULL, **v;
group_conf = cpGetConfig(CPGC.ini_file);
int gid = 0;
zval **gid_ptr = NULL;
cpGroup *G = NULL;
if (!Z_BVAL_P(group_conf))
{
cpLog("parse ini file[%s] reload error!", CPGC.ini_file);
}
else
{
for (zend_hash_internal_pointer_reset(Z_ARRVAL_P(group_conf)); zend_hash_has_more_elements(Z_ARRVAL_P(group_conf)) == SUCCESS; zend_hash_move_forward(Z_ARRVAL_P(group_conf)))
{
zval **config;
zend_hash_get_current_data(Z_ARRVAL_P(group_conf), (void**) &config);
char *name;
uint keylen;
zend_hash_get_current_key_ex(Z_ARRVAL_P(group_conf), &name, &keylen, NULL, 0, NULL);
if (strcmp(name, "common") != 0)
{
if (zend_hash_find(Z_ARRVAL_P(CPGS->group), name, strlen(name) + 1, (void **) &gid_ptr) == SUCCESS)
{
gid = Z_LVAL_PP(gid_ptr);
G = &CPGS->G[gid];
}
else
{
cpLog("can not add datasource when the server runing,if you want add it please restart");
return;
}
if (pthread_mutex_lock(G->mutex_lock) == 0)
{
if (zend_hash_find(Z_ARRVAL_PP(config), ZEND_STRS("pool_max"), (void **) &v) == SUCCESS)
{
convert_to_long(*v);
G->worker_max = (int) Z_LVAL_PP(v);
}
if (zend_hash_find(Z_ARRVAL_PP(config), ZEND_STRS("pool_min"), (void **) &v) == SUCCESS)
{
convert_to_long(*v);
int new_min = (int) Z_LVAL_PP(v);
if (new_min > G->worker_min)
{//增加最小
while (G->worker_num < new_min)
{
cpCreate_worker_mem(G->worker_num, gid);
G->workers_status[G->worker_num] = CP_WORKER_IDLE;
G->worker_num++; //先加 线程安全
int new_pid = cpFork_one_worker(G->worker_num - 1, gid);
if (new_pid < 0)
{
cpLog("Fork worker process failed. Error: %s [%d]", strerror(errno), errno);
}
else
{
G->workers[G->worker_num - 1].pid = new_pid;
}
}
}
G->worker_min = new_min;
}
if (pthread_mutex_unlock(G->mutex_lock) != 0)
{
cpLog("pthread_mutex_unlock. Error: %s [%d]", strerror(errno), errno);
}
}
}
else
{
if (zend_hash_find(Z_ARRVAL_PP(config), ZEND_STRS("recycle_num"), (void **) &v) == SUCCESS)
{
convert_to_long(*v);
CPGC.recycle_num = (int) Z_LVAL_PP(v);
}
if (zend_hash_find(Z_ARRVAL_PP(config), ZEND_STRS("idel_time"), (void **) &v) == SUCCESS)
{
convert_to_long(*v);
CPGC.idel_time = (int) Z_LVAL_PP(v);
}
}
}
zval_ptr_dtor(&group_conf);
}
}
int cpServer_start()
{
int i, pid, ret, ping_pid, sock;
if (CPGC.daemonize > 0)
{
if (daemon(0, 0) < 0)
{
return FAILURE;
}
}
if ((sock = cpListen()) < 0)
{
cpLog("listen[1] fail");
return FAILURE;
}
CPGS->master_pid = getpid();
CPGL.process_type = CP_PROCESS_MASTER;
cpList_create();
pid = fork();
switch (pid)
{
//创建manager进程
case 0:
for (i = 0; i < CPGC.worker_min; i++)
{
//alloc了max个 但是只启动min个
ret = cpCreate_worker_mem(i);
pid = cpFork_one_worker(i);
if (pid < 0 || ret < 0)
{
cpLog("Fork worker process fail");
return FAILURE;
}
else
{
CPGS->workers[i].pid = pid;
CPGS->workers_status[i] = CP_WORKER_IDLE;
}
}
//数据库坏连接检测恢复进程
ret = cpCreate_ping_worker_mem();
ping_pid = cpFork_ping_worker();
if (ping_pid < 0 || ret < 0)
{
cpLog("Fork ping process fail");
return FAILURE;
}
CPGS->ping_workers->pid = ping_pid;
//标识为管理进程
CPGL.process_type = CP_PROCESS_MANAGER;
CPGS->worker_num = CPGC.worker_min; //初始为min个worker
ret = cpWorker_manager_loop();
exit(ret);
break;
//主进程
default:
CPGS->manager_pid = pid;
break;
case -1:
{
cpLog("fork manager process fail");
return FAILURE;
}
}
cpSignalInit();
if (cpReactor_start(sock) < 0)
{
cpLog("Reactor_start[1] fail");
return FAILURE;
}
return SUCCESS;
}
int cpServer_start()
{
int w, pid, ret, sock, g;
if (CPGC.daemonize > 0)
{
if (daemon(0, 0) < 0)
{
return FAILURE;
}
}
if ((sock = cpListen()) < 0)
{
cpLog("listen[1] fail");
return FAILURE;
}
CPGS->master_pid = getpid();
CPGL.process_type = CP_PROCESS_MASTER;
pid = fork();
switch (pid)
{
//创建manager进程
case 0:
for (g = 0; g < CPGS->group_num; g++)
{
for (w = 0; w < CPGS->G[g].worker_min; w++)
{
//alloc了max个 但是只启动min个
ret = cpCreate_worker_mem(w, g);
pid = cpFork_one_worker(w, g);
if (pid < 0 || ret < 0)
{
cpLog("Fork worker process fail");
return FAILURE;
}
else
{
CPGS->G[g].workers[w].pid = pid;
CPGS->G[g].workers_status[w] = CP_WORKER_IDLE;
}
}
}
//数据库坏连接检测恢复进程
// ret = cpCreate_ping_worker_mem();
// ping_pid = cpFork_ping_worker();
// if (ping_pid < 0 || ret < 0)
// {
// cpLog("Fork ping process fail");
// return FAILURE;
// }
// CPGS->ping_workers->pid = ping_pid;
//标识为管理进程
CPGL.process_type = CP_PROCESS_MANAGER;
ret = cpWorker_manager_loop();
exit(ret);
break;
//主进程
default:
CPGS->manager_pid = pid;
break;
case -1:
{
cpLog("fork manager process fail");
return FAILURE;
}
}
cpSignalInit();
if (cpReactor_start(sock) < 0)
{
cpLog("Reactor_start[1] fail");
return FAILURE;
}
return SUCCESS;
}
int cpWorker_manager_loop()
{
int pid, new_pid;
int i;
int worker_exit_code;
//reload config
cpSignalSet(SIGUSR1, cpManagerReload, 1, 0);
//close worker
cpSignalSet(SIGALRM, cpManagerRecycle, 1, 0);
//add one worker
cpSignalSet(SIGRTMIN, cpManagerAdd, 1, 0);
alarm(2);
sigset_t block_alarm;
sigemptyset(&block_alarm);
sigaddset(&block_alarm, SIGALRM);
sigaddset(&block_alarm, SIGRTMIN);
sigaddset(&block_alarm, SIGUSR1);
while (CPGS->running == 1) {
pid = wait(&worker_exit_code);
sigprocmask(SIG_BLOCK, &block_alarm, NULL);
if (CPGS->running == 1 && pid > 0) {
if (pid == CPGS->ping_workers->pid) {
cpLog("ping worker exit");
int ping_pid = cpFork_ping_worker();
if (ping_pid < 0) {
cpLog("Fork ping process fail");
} else {
CPGS->ping_workers->pid = ping_pid;
}
}
for (i = CPGS->worker_num; i >= 0; i--) {
if (pid != CPGS->workers[i].pid || CPGS->workers_status[i] == CP_WORKER_DEL) {//对比pid||回收的不拉起
continue;
} else {
if (CPGS->workers[i].run == 0) {
cpLog("restart worker!worker index %d,worker id %d,exit code %d\n", i, pid, WEXITSTATUS(worker_exit_code));
} else {
cpLog("worker exit!worker index %d,worker id %d,exit code %d,times %d,pre_len %d\n", i, pid, WEXITSTATUS(worker_exit_code), CPGS->workers[i].request, CPGS->workers[i].pre_len);
}
cpShareMemory *sm_obj = &(CPGS->workers[i].sm_obj);
sm_obj->mem = NULL;
pid = 0;
new_pid = cpFork_one_worker(i);
if (new_pid < 0) {
cpLog("Fork worker process failed. Error: %s [%d]", strerror(errno), errno);
sigprocmask(SIG_UNBLOCK, &block_alarm, NULL);
return FAILURE;
} else {
CPGS->workers[i].pid = new_pid;
}
}
}
}
sigprocmask(SIG_UNBLOCK, &block_alarm, NULL);
}
return SUCCESS;
}
