本篇是《自己動手寫把"鎖"》系列技術鋪墊的最后一個知識點。本篇主要講解LockSupport工具類,它用來實現線程的掛起和喚醒。
LockSupport是Java6引入的一個工具類,它簡單靈活,應用廣泛。
一、簡單
俗話說,沒有比較就沒有傷害。這里咱們還是通過對比來介紹LockSupport的簡單。
在沒有LockSupport之前,線程的掛起和喚醒咱們都是通過Object的wait和notify/notifyAll方法實現。
寫一段例子代碼,線程A執行一段業務邏輯后調用wait阻塞住自己。主線程調用notify方法喚醒線程A,線程A然后打印自己執行的結果。
public class TestObjWait { public static void main(String[] args)throws Exception { final Object obj = new Object(); Thread A = new Thread(new Runnable() { @Override public void run() { int sum = 0; for(int i=0;i<10;i++){ sum+=i; } try { obj.wait(); }catch (Exception e){ e.printStackTrace(); } System.out.println(sum); } }); A.start(); //睡眠一秒鍾,保證線程A已經計算完成,阻塞在wait方法 Thread.sleep(1000); obj.notify(); } }
執行這段代碼,不難發現這個錯誤:
Exception in thread "main" java.lang.IllegalMonitorStateException
at java.lang.Object.notify(Native Method)
原因很簡單,wait和notify/notifyAll方法只能在同步代碼塊里用(這個有的面試官也會考察)。所以將代碼修改為如下就可正常運行了:
public class TestObjWait { public static void main(String[] args)throws Exception { final Object obj = new Object(); Thread A = new Thread(new Runnable() { @Override public void run() { int sum = 0; for(int i=0;i<10;i++){ sum+=i; } try { synchronized (obj){ obj.wait(); } }catch (Exception e){ e.printStackTrace(); } System.out.println(sum); } }); A.start(); //睡眠一秒鍾,保證線程A已經計算完成,阻塞在wait方法 Thread.sleep(1000); synchronized (obj){ obj.notify(); } } }
那如果咱們換成LockSupport呢?簡單得很,看代碼:
public class TestObjWait { public static void main(String[] args)throws Exception { Thread A = new Thread(new Runnable() { @Override public void run() { int sum = 0; for(int i=0;i<10;i++){ sum+=i; } LockSupport.park(); System.out.println(sum); } }); A.start(); //睡眠一秒鍾,保證線程A已經計算完成,阻塞在wait方法 Thread.sleep(1000); LockSupport.unpark(A); } }
直接調用就可以了,沒有說非得在同步代碼塊里才能用。簡單吧。
二、靈活
如果只是LockSupport在使用起來比Object的wait/notify簡單,那還真沒必要專門講解下LockSupport。最主要的是靈活性。
上邊的例子代碼中,主線程調用了Thread.sleep(1000)方法來等待線程A計算完成進入wait狀態。如果去掉Thread.sleep()調用,代碼如下:
public class TestObjWait { public static void main(String[] args)throws Exception { final Object obj = new Object(); Thread A = new Thread(new Runnable() { @Override public void run() { int sum = 0; for(int i=0;i<10;i++){ sum+=i; } try { synchronized (obj){ obj.wait(); } }catch (Exception e){ e.printStackTrace(); } System.out.println(sum); } }); A.start(); //睡眠一秒鍾,保證線程A已經計算完成,阻塞在wait方法 //Thread.sleep(1000); synchronized (obj){ obj.notify(); } } }
多運行幾次上邊的代碼,有的時候能夠正常打印結果並退出程序,但有的時候線程無法打印結果阻塞住了。原因就在於:主線程調用完notify后,線程A才進入wait方法,導致線程A一直阻塞住。由於線程A不是后台線程,所以整個程序無法退出。
那如果換做LockSupport呢?LockSupport就支持主線程先調用unpark后,線程A再調用park而不被阻塞嗎?是的,沒錯。代碼如下:
public class TestObjWait { public static void main(String[] args)throws Exception { final Object obj = new Object(); Thread A = new Thread(new Runnable() { @Override public void run() { int sum = 0; for(int i=0;i<10;i++){ sum+=i; } LockSupport.park(); System.out.println(sum); } }); A.start(); //睡眠一秒鍾,保證線程A已經計算完成,阻塞在wait方法 //Thread.sleep(1000); LockSupport.unpark(A); } }
不管你執行多少次,這段代碼都能正常打印結果並退出。這就是LockSupport最大的靈活所在。
總結一下,LockSupport比Object的wait/notify有兩大優勢:
①LockSupport不需要在同步代碼塊里 。所以線程間也不需要維護一個共享的同步對象了,實現了線程間的解耦。
②unpark函數可以先於park調用,所以不需要擔心線程間的執行的先后順序。
三、應用廣泛
LockSupport在Java的工具類用應用很廣泛,咱們這里找幾個例子感受感受。以Java里最常用的類ThreadPoolExecutor為例。先看如下代碼:
public class TestObjWait { public static void main(String[] args)throws Exception { ArrayBlockingQueue<Runnable> queue = new ArrayBlockingQueue<Runnable>(1000); ThreadPoolExecutor poolExecutor = new ThreadPoolExecutor(5,5,1000, TimeUnit.SECONDS,queue); Future<String> future = poolExecutor.submit(new Callable<String>() { @Override public String call() throws Exception { TimeUnit.SECONDS.sleep(5); return "hello"; } }); String result = future.get(); System.out.println(result); } }
代碼中我們向線程池中扔了一個任務,然后調用Future的get方法,同步阻塞等待線程池的執行結果。
這里就要問了:get方法是如何組塞住當前線程?線程池執行完任務后又是如何喚醒線程的呢?
咱們跟着源碼一步步分析,先看線程池的submit方法的實現:
在submit方法里,線程池將我們提交的基於Callable實現的任務,封裝為基於RunnableFuture實現的任務,然后將任務提交到線程池執行,並向當前線程返回RunnableFutrue。
進入newTaskFor方法,就一句話:return new FutureTask<T>(callable);
所以,咱們主線程調用future的get方法就是FutureTask的get方法,線程池執行的任務對象也是FutureTask的實例。
接下來看看FutureTask的get方法的實現:
比較簡單,就是判斷下當前任務是否執行完畢,如果執行完畢直接返回任務結果,否則進入awaitDone方法阻塞等待。
awaitDone方法里,首先會用到上節講到的cas操作,將線程封裝為WaitNode,保持下來,以供后續喚醒線程時用。再就是調用了LockSupport的park/parkNanos組塞住當前線程。
上邊已經說完了阻塞等待任務結果的邏輯,接下來再看看線程池執行完任務,喚醒等待線程的邏輯實現。
前邊說了,咱們提交的基於Callable實現的任務,已經被封裝為FutureTask任務提交給了線程池執行,任務的執行就是FutureTask的run方法執行。如下是FutureTask的run方法:
c.call()就是執行我們提交的任務,任務執行完后調用了set方法,進入set方法發現set方法調用了finishCompletion方法,想必喚醒線程的工作就在這里邊了,看看代碼實現吧:
沒錯就在這里邊,先是通過cas操作將所有等待的線程拿出來,然后便使用LockSupport的unpark喚醒每個線程。
在使用線程池的過程中,不知道你有沒有這么一個疑問:線程池里沒有任務時,線程池里的線程在干嘛呢?
看過我的這篇文章《線程池的工作原理與源碼解讀》的讀者一定知道,線程會調用隊列的take方法阻塞等待新任務。那隊列的take方法是不是也跟Future的get方法實現一樣呢?咱們來看看源碼實現。
以ArrayBlockingQueue為例,take方法實現如下:
與想象的有點出入,他是使用了Lock的Condition的await方法實現線程阻塞。但當我們繼續追下去進入await方法,發現還是使用了LockSupport:
限於篇幅,jdk里的更多應用就不再追下去了。
四、LockSupport的實現
學習要知其然,還要知其所以然。接下來不妨看看LockSupport的實現。
進入LockSupport的park方法,可以發現它是調用了Unsafe的park方法,這是一個本地native方法,只能通過openjdk的源碼看看其本地實現了。
它調用了線程的Parker類型對象的park方法,如下是Parker類的定義:
類中定義了一個int類型的_counter變量,咱們上文中講靈活性的那一節說,可以先執行unpark后執行park,就是通過這個變量實現,看park方法的實現代碼(由於方法比較長就不整體截圖了):
park方法會調用Atomic::xchg方法,這個方法會原子性的將_counter賦值為0,並返回賦值前的值。如果調用park方法前,_counter大於0,則說明之前調用過unpark方法,所以park方法直接返回。
接着往下看:
實際上Parker類用Posix的mutex,condition來實現的阻塞喚醒。如果對mutex和condition不熟,可以簡單理解為mutex就是Java里的synchronized,condition就是Object里的wait/notify操作。
park方法里調用pthread_mutex_trylock方法,就相當於Java線程進入Java的同步代碼塊,然后再次判斷_counter是否大於零,如果大於零則將_counter設置為零。最后調用pthread_mutex_unlock解鎖,相當於Java執行完退出同步代碼塊。如果_counter不大於零,則繼續往下執行pthread_cond_wait方法,實現當前線程的阻塞。
最后再看看unpark方法的實現吧,這塊就簡單多了,直接上代碼:
圖中的1和4就相當於Java的進入synchronized和退出synchronized的加鎖解鎖操作,代碼2將_counter設置為1,同時判斷先前_counter的值是否小於1,即這段代碼:if(s<1)。如果不小於1,則就不會有線程被park,所以方法直接執行完畢,否則就會執行代碼3,來喚醒被阻塞的線程。
通過閱讀LockSupport的本地實現,我們不難發現這么個問題:多次調用unpark方法和調用一次unpark方法效果一樣,因為都是直接將_counter賦值為1,而不是加1。簡單說就是:線程A連續調用兩次LockSupport.unpark(B)方法喚醒線程B,然后線程B調用兩次LockSupport.park()方法, 線程B依舊會被阻塞。因為兩次unpark調用效果跟一次調用一樣,只能讓線程B的第一次調用park方法不被阻塞,第二次調用依舊會阻塞。
到這里,自己實現一把“鎖”用到的技術點都已經介紹完了,甚至本節還介紹了鎖的具體實現,相信即使沒有最后一篇介紹怎么實現“鎖”,大家也能動手寫個鎖了。
如果沒有看過之前的文章,可以先看之前這幾篇:
自己動手寫把”鎖”之---鎖的作用
自己動手寫把”鎖”之---JMM和volatile
自己動手寫把”鎖”---原子性操作
----------------------------------------------
有興趣的朋友,可以加入我的知識圈,一起研究討論。
我正在「JAVA互聯網技術」和朋友們討論有趣的話題,你一起來吧?