1. 程式人生 > >深入理解 Java 中的 try-with-resource

深入理解 Java 中的 try-with-resource

背景

眾所周知,所有被開啟的系統資源,比如流、檔案或者Socket連線等,都需要被開發者手動關閉,否則隨著程式的不斷執行,資源洩露將會累積成重大的生產事故。

在Java的江湖中,存在著一種名為finally的功夫,它可以保證當你習武走火入魔之時,還可以做一些自救的操作。在遠古時代,處理資源關閉的程式碼通常寫在finally塊中。然而,如果你同時打開了多個資源,那麼將會出現噩夢般的場景:


    
  1. public class Demo
    {
  2. public static void main(String[] args) {
  3. BufferedInputStream bin = null;
  4. BufferedOutputStream bout = null
    ;
  5. try {
  6. bin = new BufferedInputStream( new FileInputStream( new File( ”test.txt”)));
  7. bout = new
    BufferedOutputStream( new FileOutputStream( new File( ”out.txt”)));
  8. int b;
  9. while ((b = bin.read()) != - 1) {
  10. bout.write(b);
  11. }
  12. }
  13. catch (IOException e) {
  14. e.printStackTrace();
  15. }
  16. finally {
  17. if (bin != null) {
  18. try {
  19. bin.close();
  20. }
  21. catch (IOException e) {
  22. e.printStackTrace();
  23. }
  24. finally {
  25. if (bout != null) {
  26. try {
  27. bout.close();
  28. }
  29. catch (IOException e) {
  30. e.printStackTrace();
  31. }
  32. }
  33. }
  34. }
  35. }
  36. }
  37. }

Oh My God!!!關閉資源的程式碼竟然比業務程式碼還要多!!!這是因為,我們不僅需要關閉BufferedInputStream,還需要保證如果關閉BufferedInputStream時出現了異常, BufferedOutputStream也要能被正確地關閉。所以我們不得不借助finally中巢狀finally大法。可以想到,開啟的資源越多,finally中巢狀的將會越深!!!

更為可惡的是,Python程式設計師面對這個問題,竟然微微一笑很傾城地說:“這個我們一點都不用考慮的嘞~”:

但是兄弟莫慌!我們可以利用Java 1.7中新增的try-with-resource語法糖來開啟資源,而無需碼農們自己書寫資源來關閉程式碼。媽媽再也不用擔心我把手寫斷掉了!我們用try-with-resource來改寫剛才的例子:


    
  1. public class TryWithResource {
  2. public static void main(String[] args) {
  3. try (BufferedInputStream bin = new BufferedInputStream( new FileInputStream( new File( "test.txt")));
  4. BufferedOutputStream bout = new BufferedOutputStream( new FileOutputStream( new File( "out.txt")))) {
  5. int b;
  6. while ((b = bin.read()) != - 1) {
  7. bout.write(b);
  8. }
  9. }
  10. catch (IOException e) {
  11. e.printStackTrace();
  12. }
  13. }
  14. }

是不是很簡單?是不是很刺激?再也不用被Python程式設計師鄙視了!好了,下面將會詳細講解其實現原理以及內部機制。

動手實踐

為了能夠配合try-with-resource,資源必須實現AutoClosable介面。該介面的實現類需要重寫close方法:


    
  1. public class Connection implements AutoCloseable {
  2. public void sendData() {
  3. System.out.println( "正在傳送資料");
  4. }
  5. @Override
  6. public void close() throws Exception {
  7. System.out.println( "正在關閉連線");
  8. }
  9. }

呼叫類:


    
  1. public class TryWithResource {
  2. public static void main(String[] args) {
  3. try (Connection conn = new Connection()) {
  4. conn.sendData();
  5. }
  6. catch (Exception e) {
  7. e.printStackTrace();
  8. }
  9. }
  10. }

執行後輸出結果:

正在傳送資料
正在關閉連線

通過結果我們可以看到,close方法被自動呼叫了。

原理

那麼這個是怎麼做到的呢?我相信聰明的你們一定已經猜到了,其實,這一切都是編譯器大神搞的鬼。我們反編譯剛才例子的class檔案:


    
  1. public class TryWithResource {
  2. public TryWithResource() {
  3. }
  4. public static void main(String[] args) {
  5. try {
  6. Connection e = new Connection();
  7. Throwable var2 = null;
  8. try {
  9. e.sendData();
  10. } catch (Throwable var12) {
  11. var2 = var12;
  12. throw var12;
  13. } finally {
  14. if(e != null) {
  15. if(var2 != null) {
  16. try {
  17. e.close();
  18. } catch (Throwable var11) {
  19. var2.addSuppressed(var11);
  20. }
  21. } else {
  22. e.close();
  23. }
  24. }
  25. }
  26. } catch (Exception var14) {
  27. var14.printStackTrace();
  28. }
  29. }
  30. }

看到沒,在第15~27行,編譯器自動幫我們生成了finally塊,並且在裡面呼叫了資源的close方法,所以例子中的close方法會在執行的時候被執行。

異常遮蔽

我相信,細心的你們肯定又發現了,剛才反編譯的程式碼(第21行)比遠古時代寫的程式碼多了一個addSuppressed。為了瞭解這段程式碼的用意,我們稍微修改一下剛才的例子:我們將剛才的程式碼改回遠古時代手動關閉異常的方式,並且在sendDataclose方法中丟擲異常:


    
  1. public class Connection implements AutoCloseable {
  2. public void sendData() throws Exception {
  3. throw new Exception( "send data");
  4. }
  5. @Override
  6. public void close() throws Exception {
  7. throw new MyException( "close");
  8. }
  9. }

修改main方法:


    
  1. public class TryWithResource {
  2. public static void main(String[] args) {
  3. try {
  4. test();
  5. }
  6. catch (Exception e) {
  7. e.printStackTrace();
  8. }
  9. }
  10. private static void test() throws Exception {
  11. Connection conn = null;
  12. try {
  13. conn = new Connection();
  14. conn.sendData();
  15. }
  16. finally {
  17. if (conn != null) {
  18. conn.close();
  19. }
  20. }
  21. }
  22. }

執行之後我們發現:


    
  1. basic.exception.MyException: close
  2. at basic .exception .Connection .close( Connection .java :10)
  3. at basic .exception .TryWithResource .test( TryWithResource .java :82)
  4. at basic .exception .TryWithResource .main( TryWithResource .java :7)
  5. ......

好的,問題來了,由於我們一次只能丟擲一個異常,所以在最上層看到的是最後一個丟擲的異常——也就是close方法丟擲的MyException,而sendData丟擲的Exception被忽略了。這就是所謂的異常遮蔽。由於異常資訊的丟失,異常遮蔽可能會導致某些bug變得極其難以發現,程式設計師們不得不加班加點地找bug,如此毒瘤,怎能不除!幸好,為了解決這個問題,從Java 1.7開始,大佬們為Throwable類新增了addSuppressed方法,支援將一個異常附加到另一個異常身上,從而避免異常遮蔽。那麼被遮蔽的異常資訊會通過怎樣的格式輸出呢?我們再執行一遍剛才用try-with-resource包裹的main方法:


    
  1. java.lang.Exception: send data
  2. at basic .exception .Connection .sendData( Connection .java :5)
  3. at basic .exception .TryWithResource .main( TryWithResource .java :14)
  4. ......
  5. Suppressed: basic .exception .MyException: close
  6. at basic .exception .Connection .close( Connection .java :10)
  7. at basic .exception .TryWithResource .main( TryWithResource .java :15)
  8. ... 5 more

可以看到,異常資訊中多了一個Suppressed的提示,告訴我們這個異常其實由兩個異常組成,MyException是被Suppressed的異常。可喜可賀!

一個小問題

在使用try-with-resource的過程中,一定需要了解資源的close方法內部的實現邏輯。否則還是可能會導致資源洩露。

舉個例子,在Java BIO中採用了大量的裝飾器模式。當呼叫裝飾器的close方法時,本質上是呼叫了裝飾器內部包裹的流的close方法。比如:


    
  1. public class TryWithResource {
  2. public static void main(String[] args)