在线程池中计算当前负载,做一个决定,是否运行自身进程继续抢任务
我们重点看ngx_trylock_accept_mutex这个函数,该函数中
ngx_int_t
ngx_trylock_accept_mutex(ngx_cycle_t *cycle)
{
//在这里添加代码,如果线程池忙,不允许继续accept,则直接return
if (ngx_shmtx_trylock(&ngx_accept_mutex)) {
if (ngx_enable_accept_events(cycle) == NGX_ERROR) { //这里如果抢锁成功,那么调用ngx_enable_accept_events,将listen fd注册到自己的epoll中,从而可以抢accept事件,我这里修改下,如果本进程很忙,则直接返回(即不会抢accept,又不会阻塞程序流程)
ngx_shmtx_unlock(&ngx_accept_mutex);
return NGX_ERROR;
}
ngx_accept_mutex_held = 1;
return NGX_OK;
}
if (ngx_accept_mutex_held) {
if (ngx_disable_accept_events(cycle) == NGX_ERROR) {
return NGX_ERROR;
}
ngx_accept_mutex_held = 0;
}
return NGX_OK;
}
本文转自 zhegaozhouji 51CTO博客,原文链接:http://blog.51cto.com/1038741/1828067