天天看點

nginx 對多程序進行負載均衡(通過線程池來判斷)

線上程池中計算目前負載,做一個決定,是否運作自身程序繼續搶任務

我們重點看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) {

        ngx_accept_mutex_held = 0;

    return NGX_OK;

}

本文轉自 zhegaozhouji 51CTO部落格,原文連結:http://blog.51cto.com/1038741/1828067