【高併發】面試官:講講高併發場景下如何優化加鎖方式?

冰河團隊發表於2020-10-15

寫在前面

很多時候,我們在併發程式設計中,涉及到加鎖操作時,對程式碼塊的加鎖操作真的合理嗎?還有沒有需要優化的地方呢?

前言

在《【高併發】優化加鎖方式時竟然死鎖了!!》一文中,我們介紹了產生死鎖時的四個必要條件,只有四個條件同時具備時才能發生死鎖。其中,我們在阻止請求與保持條件時,採用了一次性申請所有的資源的方式。例如在我們完成轉賬操作的過程中,我們一次性申請賬戶A和賬戶B,兩個賬戶都申請成功後,再執行轉賬的操作。其中,在我們實現的轉賬方法中,使用了死迴圈來迴圈獲取資源,直到同時獲取到賬戶A和賬戶B為止,核心程式碼如下所示。

//一次申請轉出賬戶和轉入賬戶,直到成功
while(!requester.applyResources(this, target)){
    //迴圈體為空
    ;
}

如果ResourcesRequester類的applyResources()方法執行的時間非常短,並且程式併發帶來的衝突不大,程式迴圈幾次到幾十次就可以同時獲取到轉出賬戶和轉入賬戶,這種方案就是可行的。

但是,如果ResourcesRequester類的applyResources()方法執行的時間比較長,或者說,程式併發帶來的衝突比較大,此時,可能需要迴圈成千上萬次才能同時獲取到轉出賬戶和轉入賬戶。這樣就太消耗CPU資源了,此時,這種方案就是不可行的了。

那麼,有沒有什麼方式對這種方案進行優化呢?

問題分析

既然使用死迴圈一直獲取資源這種方案存在問題,那我們換位思考一下。當執行緒執行時,發現條件不滿足,是不是可以讓執行緒進入等待狀態?當條件滿足的時候,通知等待的執行緒重新執行?

也就是說,如果執行緒需要的條件不滿足,我們就讓執行緒進入等待狀態;如果執行緒需要的條件滿足時,我們再通知等待的執行緒重新執行。這樣,就能夠避免程式進行迴圈等待進而消耗CPU的問題。

那麼,問題又來了!當條件不滿足時,如何實現讓執行緒等待?當條件滿足時,又如何喚醒執行緒呢?

不錯,這是個問題!不過這個問題解決起來也非常簡單。簡單的說,就是使用執行緒的等待與通知機制。

執行緒的等待與通知機制

我們可以使用執行緒的等待與通知機制來優化阻止請求與保持條件時,迴圈獲取賬戶資源的問題。具體的等待與通知機制如下所示。

執行的執行緒首先獲取互斥鎖,如果執行緒繼續執行時,需要的條件不滿足,則釋放互斥鎖,並進入等待狀態;當執行緒繼續執行需要的條件滿足時,就通知等待的執行緒,重新獲取互斥鎖。

那麼,說了這麼多,Java支援這種執行緒的等待與通知機制嗎?其實,這個問題問的就有點廢話了,Java這麼優秀(牛逼)的語言肯定支援啊,而且實現起來也比較簡單。

用Java實現執行緒的等待與通知機制

實現方式

其實,使用Java語言實現執行緒的等待與通知機制有多種方式,這裡我就簡單的列舉一種方式,其他的方式大家可以自行思考和實現,有不懂的地方也可以問我!

在Java語言中,實現執行緒的等待與通知機制,一種簡單的方式就是使用synchronized並結合wait()、notify()和notifyAll()方法來實現。

實現原理

我們使用synchronized加鎖時,只允許一個執行緒進入synchronized保護的程式碼塊,也就是臨界區。如果一個執行緒進入了臨界區,則其他的執行緒會進入阻塞佇列裡等待,這個阻塞佇列和synchronized互斥鎖是一對一的關係,也就是說,一把互斥鎖對應著一個獨立的阻塞佇列。

在併發程式設計中,如果一個執行緒獲得了synchronized互斥鎖,但是不滿足繼續向下執行的條件,則需要進入等待狀態。此時,可以使用Java中的wait()方法來實現。當呼叫wait()方法後,當前執行緒就會被阻塞,並且會進入一個等待佇列中進行等待,這個由於呼叫wait()方法而進入的等待佇列也是互斥鎖的等待佇列。而且,執行緒在進入等待佇列的同時,會釋放自身獲得的互斥鎖,這樣,其他執行緒就有機會獲得互斥鎖,進而進入臨界區了。整個過程可以表示成下圖所示。

當執行緒執行的條件滿足時,可以使用Java提供的notify()和notifyAll()方法來通知互斥鎖等待佇列中的執行緒,我們可以使用下圖來簡單的表示這個過程。

這裡,需要注意如下事項:

(1)使用notify()和notifyAll()方法通知執行緒時,呼叫notify()和notifyAll()方法時,滿足執行緒的執行條件,但是當執行緒真正執行的時候,條件可能已經不再滿足了,可能有其他執行緒已經進入臨界區執行。

(2)被通知的執行緒繼續執行時,需要先獲取互斥鎖,因為在呼叫wait()方法等待時已經釋放了互斥鎖。

(3)wait()、notify()和notifyAll()方法操作的佇列是互斥鎖的等待佇列,如果synchronized鎖定的是this物件,則一定要使用this.wait()、this.notify()和this.notifyAll()方法;如果synchronized鎖定的是target物件,則一定要使用target.wait()、target.notify()和target.notifyAll()方法。

(4)wait()、notify()和notifyAll()方法呼叫的前提是已經獲取了相應的互斥鎖,也就是說,wait()、notify()和notifyAll()方法都是在synchronized方法中或程式碼塊中呼叫的。如果在synchronized方法外或程式碼塊外呼叫了三個方法,或者鎖定的物件是this,使用target物件呼叫三個方法的話,JVM會丟擲java.lang.IllegalMonitorStateException異常。

具體實現

實現邏輯

在實現之前,我們還需要考慮以下幾個問題:

  • 選擇哪個互斥鎖

在之前的程式中,我們在TansferAccount類中,存在一個ResourcesRequester 類的單例物件,所以,我們是可以使用this作為互斥鎖的。這一點大家需要重點理解。

  • 執行緒執行轉賬操作的條件

轉出賬戶和轉入賬戶都沒有被分配過。

  • 執行緒什麼時候進入等待狀態

執行緒繼續執行需要的條件不滿足的時候,進入等待狀態。

  • 什麼時候通知等待的執行緒執行

當存線上程釋放賬戶的資源時,通知等待的執行緒繼續執行。

綜上,我們可以得出以下核心程式碼。

while(不滿足條件){
    wait();
}

那麼,問題來了!為何是在while迴圈中呼叫wait()方法呢?因為當wait()方法返回時,有可能執行緒執行的條件已經改變,也就是說,之前條件是滿足的,但是現在已經不滿足了,所以要重新檢驗條件是否滿足。

實現程式碼

我們優化後的ResourcesRequester類的程式碼如下所示。

public class ResourcesRequester{
    //存放申請資源的集合
    private List<Object> resources = new ArrayList<Object>();
    //一次申請所有的資源
    public synchronized void applyResources(Object source, Object target){
        while(resources.contains(source) || resources.contains(target)){
            try{
                wait();
            }catch(Exception e){
                e.printStackTrace();
            }
        }
        resources.add(source);
        resources.add(targer);
    }
    
    //釋放資源
    public synchronized void releaseResources(Object source, Object target){
        resources.remove(source);
        resources.remove(target);
        notifyAll();
    }
}

生成ResourcesRequester單例物件的Holder類ResourcesRequesterHolder的程式碼如下所示。

public class ResourcesRequesterHolder{
    private ResourcesRequesterHolder(){}
    
    public static ResourcesRequester getInstance(){
        return Singleton.INSTANCE.getInstance();
    }
    private enum Singleton{
        INSTANCE;
        private ResourcesRequester singleton;
        Singleton(){
            singleton = new ResourcesRequester();
        }
        public ResourcesRequester getInstance(){
            return singleton;
        }
    }
}

執行轉賬操作的類的程式碼如下所示。

public class TansferAccount{
    //賬戶的餘額
    private Integer balance;
    //ResourcesRequester類的單例物件
    private ResourcesRequester requester;
   
    public TansferAccount(Integer balance){
        this.balance = balance;
        this.requester = ResourcesRequesterHolder.getInstance();
    }
    //轉賬操作
    public void transfer(TansferAccount target, Integer transferMoney){
        //一次申請轉出賬戶和轉入賬戶,直到成功
        requester.applyResources(this, target))
        try{
            //對轉出賬戶加鎖
            synchronized(this){
                //對轉入賬戶加鎖
                synchronized(target){
                    if(this.balance >= transferMoney){
                        this.balance -= transferMoney;
                        target.balance += transferMoney;
                    }   
                }
            }
        }finally{
            //最後釋放賬戶資源
            requester.releaseResources(this, target);
        }
    }
}

可以看到,我們在程式中通知處於等待狀態的執行緒時,使用的是notifyAll()方法而不是notify()方法。那notify()方法和notifyAll()方法兩者有什麼區別呢?

notify()和notifyAll()的區別

  • notify()方法

隨機通知等待佇列中的一個執行緒。

  • notifyAll()方法

通知等待佇列中的所有執行緒。

在實際工作過程中,如果沒有特殊的要求,儘量使用notifyAll()方法。因為使用notify()方法是有風險的,可能會導致某些執行緒永久不會被通知到!

重磅福利

微信搜一搜【冰河技術】微信公眾號,關注這個有深度的程式設計師,每天閱讀超硬核技術乾貨,公眾號內回覆【PDF】有我準備的一線大廠面試資料和我原創的超硬核PDF技術文件,以及我為大家精心準備的多套簡歷模板(不斷更新中),希望大家都能找到心儀的工作,學習是一條時而鬱鬱寡歡,時而開懷大笑的路,加油。如果你通過努力成功進入到了心儀的公司,一定不要懈怠放鬆,職場成長和新技術學習一樣,不進則退。如果有幸我們江湖再見!

另外,我開源的各個PDF,後續我都會持續更新和維護,感謝大家長期以來對冰河的支援!!

相關文章