面試官:請寫一個你認為比較“完美”的單例

【空山新雨】發表於2020-05-15

單例模式是保證一個類的例項有且只有一個,在需要控制資源(如資料庫連線池),或資源共享(如有狀態的工具類)的場景中比較適用。如果讓我們寫一個單例實現,估計絕大部分人都覺得自己沒問題,但如果需要實現一個比較完美的單例,可能並沒有你想象中簡單。本文以主人公小雨的一次面試為背景,循序漸進地討論如何實現一個較為“完美”的單例。本文人物與場景皆為虛構,如有雷同,純屬捏造。

小雨計算機專業畢業三年,對設計模式略有涉獵,能寫一些簡單的實現,掌握一些基本的JVM知識。在某次面試中,面試官要求現場寫程式碼:請寫一個你認為比較“完美”的單例。

簡單的單例實現

憑藉著對單例的理解與印象,小雨寫出了下面的程式碼

public class Singleton {
    private static Singleton instance;

    private Singleton(){}

    public static final Singleton getInstance(){
        if(instance == null) {
            instance = new Singleton();
        }
        return instance;
    }
}

寫完後小雨審視了一遍,總覺得有點太簡單了,離“完美”貌似還相差甚遠。對,在多執行緒併發環境下,這個實現就玩不轉了,如果兩個執行緒同時呼叫 getInstance() 方法,同時執行到了 if 判斷,則兩邊都認為 instance 例項為空,都會例項化一個 Singleton 物件,就會導致至少產生兩個例項了,小雨心想。嗯,需要解決多執行緒併發環境下的同步問題,保證單例的執行緒安全。

執行緒安全的單例

一提到併發同步問題,小雨就想到了鎖。加個鎖還不簡單,synchronized 搞起,

public class Singleton {
    private static Singleton instance;

    private Singleton(){}

    public synchronized static final Singleton getInstance(){
        if(instance == null) {
            instance = new Singleton();
        }
        return instance;
    }
}

小雨再次審視了一遍,發現貌似每次 getInstance() 被呼叫時,其它執行緒必須等待這個執行緒呼叫完才能執行(因為有鎖鎖住了嘛),但是加鎖其實是想避免多個執行緒同時執行例項化操作導致產生多個例項,在單例被例項化後,後續呼叫 getInstance() 直接返回就行了,每次都加鎖釋放鎖造成了不必要的開銷。

經過一陣思索與回想之後,小雨記起了曾經看過一個叫 Double-Checked Locking 的東東,雙重檢查鎖,嗯,再優化一下,

public class Singleton {
    private static volatile Singleton instance;

    private Singleton(){}

    public static final Singleton getInstance(){
        if(instance == null) {
            synchronized (Singleton.class){
                if(instance == null) {
                    instance = new Singleton();
                }
            }
        }
        return instance;
    }
}

單例在完成第一次例項化,後續再呼叫 getInstance() 先判空,如果不為空則直接返回,如果為空,就算兩個執行緒同時判斷為空,在同步塊中還做了一次雙重檢查,可以確保只會例項化一次,省去了不必要的加鎖開銷,同時也保證了執行緒安全。並且令小雨感到自我滿足的是他基於對JVM的一些瞭解加上了 volatile 關鍵字來避免例項化時由於指令重排序優化可能導致的問題,真是畫龍點睛之筆啊。 簡直——完美!

Tips: volatile關鍵字的語義

  1. 保證變數對所有執行緒的可見性。對變數寫值的時候JMM(Java記憶體模型)會將當前執行緒的工作記憶體值重新整理到主記憶體,讀的時候JMM會從主記憶體讀取變數的值而不是從工作記憶體讀取,確保一個變數值被一個執行緒更新後,另一個執行緒能立即讀取到更新後的值。
  2. 禁止指令重排序優化。JVM在執行程式時為了提高效能,編譯器和處理器常常會對指令做重排序,使用 volatile 可以禁止進行指令重排序優化。

JVM建立一個新的例項時,主要需三步:

  1. 分配記憶體
  2. 初始化構造器
  3. 將物件引用指向分配的記憶體地址

如果一個執行緒在例項化時JVM做了指令重排,比如先執行了1,再執行3,最後執行2,則另一個執行緒可能獲取到一個還沒有完成初始化的物件引用,呼叫時可能導致問題,使用volatile可以禁止指令重排,避免這種問題。

小雨將答案交給面試官,面試官瞄了一眼說道:“基本可用了,但如果我用反射直接呼叫這個類的建構函式,是不是就不能保證單例了。” 小雨撓撓頭,對哦,如果使用反射就可以在執行時改變單例構造器的可見性,直接呼叫構造器來建立一個新的例項了,比如通過下面這段程式碼

 Constructor<Singleton> constructor = Singleton.class.getDeclaredConstructor();
 constructor.setAccessible(true);
 Singleton singleton = constructor.newInstance();

小雨再次陷入了思考。

反射安全的單例

怎麼避免反射破壞單例呢,或許可以加一個靜態變數來控制,讓構造器只有從 getInstance() 內部呼叫才有效,不通過 getInstance() 直接呼叫則丟擲異常,小雨按這個思路做了一番改造,

public class Singleton {
    private static volatile Singleton instance;
    private static boolean flag = false;

    private Singleton(){
        synchronized (Singleton.class) {
            if (flag) {
                flag = false;
            } else {
                throw new RuntimeException("Please use getInstance() method to get the single instance.");
            }
        }

    }

    public static final Singleton getInstance(){
        if(instance == null) {
            synchronized (Singleton.class){
                if(instance == null) {
                    flag = true;
                    instance = new Singleton();
                }
            }
        }
        return instance;
    }
}

使用靜態變數 flag 來控制,只有從 getInstance() 呼叫構造器才能正常例項化,否則丟擲異常。但馬上小雨就發現了存在的問題:既然可以通過反射來呼叫構造器,那麼也可以通過反射來改變 flag 的值,這樣苦心設定的 flag 控制邏輯不就被打破了嗎。看來也沒那麼“完美”。雖然並不那麼完美,但也一定程度上規避了使用反射直接呼叫構造器的場景,並且貌似也想不出更好的辦法了,於是小雨提交了答案。

面試官露出迷之微笑:“想法挺好,反射的問題基本解決了,但如果我序列化這個單例物件,然後再反序列化出來一個物件,這兩個物件還一樣嗎,還能保證單例嗎。如果不能,怎麼解決這個問題?”

SerializationSafeSingleton s1 = SerializationSafeSingleton.getInstance();
ByteArrayOutputStream bos = new ByteArrayOutputStream();
ObjectOutputStream oos = new ObjectOutputStream(bos);
oos.writeObject(s1);
oos.close();

ByteArrayInputStream bis = new ByteArrayInputStream(bos.toByteArray());
ObjectInputStream ois = new ObjectInputStream(bis);
SerializationSafeSingleton s2 = (SerializationSafeSingleton) ois.readObject();
ois.close();

s1 == s2 嗎? 答案是否,如何解決呢。

序列化安全的單例

小雨思考了一會,想起了曾經學習序列化知識時接觸的 readResolve() 方法,該方法在ObjectInputStream已經讀取一個物件並在準備返回前呼叫,可以用來控制反序列化時直接返回一個物件,替換從流中讀取的物件,於是在前面實現的基礎上,小雨新增了一個 readResolve() 方法,

public class Singleton {
    private static volatile Singleton instance;
    private static boolean flag = false;

    private Singleton(){
        synchronized (Singleton.class) {
            if (flag) {
                flag = false;
            } else {
                throw new RuntimeException("Please use getInstance() method to get the single instance.");
            }
        }

    }

    public static final Singleton getInstance(){
        if(instance == null) {
            synchronized (Singleton.class){
                if(instance == null) {
                    flag = true;
                    instance = new Singleton();
                }
            }
        }
        return instance;
    }

    /**
     * 該方法代替了從流中讀取物件
     * @return
     */
    private Object readResolve(){
        return getInstance();
    }
}
    

通過幾個步驟的逐步改造優化,小雨完成了一個基本具備執行緒安全、反射安全、序列化安全的單例實現,心想這下應該足夠完美了吧。面試官臉上繼續保持著迷之微笑:“這個實現看起來還是顯得有點複雜,並且也不能完全解決反射安全的問題,想想看還有其它實現方案嗎。”

其它方案

小雨反覆思考,前面的實現是通過加鎖來實現執行緒安全,除此之外,還可以通過類的載入機制來實現執行緒安全——類的靜態屬性只會在第一次載入類時初始化,並且在初始化的過程中,JVM是不允許其它執行緒來訪問的,於是又寫出了下面兩個版本

1.靜態初始化版本

public class Singleton {
    private static final Singleton instance = new Singleton();

    private Singleton(){}

    public static final Singleton getInstance() {
        return instance;
    }
}

該版本藉助JVM的類載入機制,本身執行緒安全,但只要 Singleton 類的某個靜態物件(方法或屬性)被訪問,就會造成例項的初始化,而該例項可能根本不會被用到,造成資源浪費,另一方面也存在反射與序列化的安全性問題,也需要進行相應的處理。

2.靜態內部類版本

public class Singleton {
    private Singleton(){}

    public static final Singleton getInstance() {
        return SingletonHolder.instance;
    }

    private static class SingletonHolder {
        private static final Singleton instance = new Singleton();
    }
}

該版本只有在呼叫 getInstance() 才會進行例項化,即延遲載入,避免資源浪費的問題,同時也能保障執行緒安全,但是同樣存在反射與序列化的安全性問題,需要相應處理。

這貌似跟前面版本的複雜性差不多啊,依然都需要解決反射與安全性的問題,小雨心想,有沒有一種既簡單又能避免這些問題的方案呢。

“完美”方案

一陣苦思冥想之後,小雨突然腦中靈光閃現,列舉!(這也是《Effective Java》的作者推薦的方式啊)

public enum Singleton {
    INSTANCE;

    public void func(){
        ...
    }
}

可以直接通過 Singleton.INSTANCE 來引用單例,非常簡單的實現,並且既是執行緒安全的,同時也能應對反射與序列化的問題,面試官想要的估計就是它了吧。小雨再次提交了答案,這一次,面試官臉上的迷之微笑逐漸消失了……

Tips:為什麼列舉是執行緒、反射、序列化安全的?

  1. 列舉實際是通過一個繼承自Enum的final類來實現(通過反編譯class檔案可看到具體實現),在static程式碼塊中對其成員進行初始化,因此藉助類載入機制來保障其執行緒安全
  2. 列舉是不支援通過反射例項化的,在Constructor類的newInstance方法中可看到
if ((clazz.getModifiers() & Modifier.ENUM) != 0)
    throw new IllegalArgumentException("Cannot reflectively create enum objects");
  1. 列舉在序列化的時候僅僅是將列舉物件的name屬性輸出到結果中,反序列化的時候則是通過java.lang.Enum的valueOf方法來根據名字查詢列舉物件。並且,編譯器是不允許任何對這種序列化機制的定製的,禁用了writeObject、readObject、readObjectNoData、writeReplace和readResolve等方法。列舉通過這種機制保障了序列化安全。

總結

列舉方案近乎“完美”,但實際中,大部分情況下,我們使用雙重檢查鎖方案或靜態內部類方案基本都能滿足我們的場景並能很好地執行。並且方案從來沒有“完美”,只有更好或更合適。本文只是從單例實現的不斷演進的過程中,瞭解或回顧如反射、序列化、執行緒安全、Java記憶體模型(volatile語義)、JVM類載入機制、JVM指令重排序優化等方面的知識,同時也是啟示我們在設計或實現的過程中,多從各個角度思考,儘可能全面地考慮問題。或者,在相關面試中能更好地迎合面試官的“完美”期望。


作者:雨歌,一枚仍在學習路上的IT老兵
歡迎關注作者公眾號:半路雨歌,一起學習成長
qrcode

相關文章