在线程池中计算当前负载,做一个决定,是否运行自身进程继续抢任务



我们重点看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;

}