Android網絡編程(八)源碼解析OkHttp后篇[復用連接池]
1.引子
在了解OkHttp的復用連接池之前,我們首先要了解幾個概念。
TCP三次握手
通常我們進行HTTP連接網絡的時候我們會進行TCP的三次握手,然后傳輸數據,然后再釋放連接。
TCP三次握手的過程為:
-
第一次握手:建立連接。客戶端發送連接請求報文段,將SYN位置為1,Sequence Number為x;然后,客戶端進入SYN_SEND狀態,等待服務器的確認;
-
第二次握手:服務器收到客戶端的SYN報文段,需要對這個SYN報文段進行確認,設置Acknowledgment Number為x+1(Sequence Number+1);同時,自己自己還要發送SYN請求信息,將SYN位置為1,Sequence Number為y;服務器端將上述所有信息放到一個報文段(即SYN+ACK報文段)中,一並發送給客戶端,此時服務器進入SYN_RECV狀態;
-
第三次握手:客戶端收到服務器的SYN+ACK報文段。然后將Acknowledgment Number設置為y+1,向服務器發送ACK報文段,這個報文段發送完畢以后,客戶端和服務器端都進入ESTABLISHED狀態,完成TCP三次握手。
TCP四次分手
當客戶端和服務器通過三次握手建立了TCP連接以后,當數據傳送完畢,斷開連接就需要進行TCP四次分手:
-
第一次分手:主機1(可以使客戶端,也可以是服務器端),設置Sequence Number和Acknowledgment
Number,向主機2發送一個FIN報文段;此時,主機1進入FIN_WAIT_1狀態;這表示主機1沒有數據要發送給主機2了; -
第二次分手:主機2收到了主機1發送的FIN報文段,向主機1回一個ACK報文段,Acknowledgment Number為Sequence
-
第三次分手:主機2向主機1發送FIN報文段,請求關閉連接,同時主機2進入LAST_ACK狀態;
-
第四次分手:主機1收到主機2發送的FIN報文段,向主機2發送ACK報文段,然后主機1進入TIME_WAIT狀態;主機2收到主機1的ACK報文段以后,就關閉連接;此時,主機1等待2MSL后依然沒有收到回復,則證明Server端已正常關閉,那好,主機1也可以關閉連接了。
來看下面的圖加強下理解:
keepalive connections
當然大量的連接每次連接關閉都要三次握手四次分手的很顯然會造成性能低下,因此http有一種叫做keepalive connections的機制,它可以在傳輸數據后仍然保持連接,當客戶端需要再次獲取數據時,直接使用剛剛空閑下來的連接而不需要再次握手。
Okhttp支持5個並發KeepAlive,默認鏈路生命為5分鍾(鏈路空閑后,保持存活的時間)。
2.連接池(ConnectionPool)分析
引用計數
在okhttp中,在高層代碼的調用中,使用了類似於引用計數的方式跟蹤Socket流的調用,這里的計數對象是StreamAllocation,它被反復執行aquire與release操作,這兩個函數其實是在改變RealConnection中的List<Reference<StreamAllocation>>
的大小。(StreamAllocation.Java)
RealConnection是socket物理連接的包裝,它里面維護了List<Reference<StreamAllocation>>
的引用。List中StreamAllocation的數量也就是socket被引用的計數,如果計數為0的話,說明此連接沒有被使用就是空閑的,需要通過下文的算法實現回收;如果計數不為0,則表示上層代碼仍然引用,就不需要關閉連接。
主要變量
連接池的類位於okhttp3.ConnectionPool:
private static final Executor executor = new ThreadPoolExecutor(0 /* corePoolSize */, Integer.MAX_VALUE /* maximumPoolSize */, 60L /* keepAliveTime */, TimeUnit.SECONDS, new SynchronousQueue<Runnable>(), Util.threadFactory("OkHttp ConnectionPool", true)); /** The maximum number of idle connections for each address. */ //空閑的socket最大連接數 private final int maxIdleConnections; //socket的keepAlive時間 private final long keepAliveDurationNs; // 雙向隊列 private final Deque<RealConnection> connections = new ArrayDeque<>(); final RouteDatabase routeDatabase = new RouteDatabase(); boolean cleanupRunning;
主要的變量有必要說明一下:
- executor線程池,類似於CachedThreadPool,需要注意的是這種線程池的工作隊列采用了沒有容量的SynchronousQueue,不了解它的請查看Java並發編程(六)阻塞隊列這篇文章。
Deque<RealConnection>
,雙向隊列,雙端隊列同時具有隊列和棧性質,經常在緩存中被使用,里面維護了RealConnection也就是socket物理連接的包裝。- RouteDatabase,它用來記錄連接失敗的Route的黑名單,當連接失敗的時候就會把失敗的線路加進去。
public ConnectionPool() { //默認空閑的socket最大連接數為5個,socket的keepAlive時間為5秒 this(5, 5, TimeUnit.MINUTES); } public ConnectionPool(int maxIdleConnections, long keepAliveDuration, TimeUnit timeUnit) { this.maxIdleConnections = maxIdleConnections; this.keepAliveDurationNs = timeUnit.toNanos(keepAliveDuration); // Put a floor on the keep alive duration, otherwise cleanup will spin loop. if (keepAliveDuration <= 0) { throw new IllegalArgumentException("keepAliveDuration <= 0: " + keepAliveDuration); } }
通過構造函數可以看出ConnectionPool默認的空閑的socket最大連接數為5個,socket的keepAlive時間為5秒。
緩存操作
ConnectionPool提供對Deque<RealConnection>
進行操作的方法分別為put、get、connectionBecameIdle和evictAll幾個操作。分別對應放入連接、獲取連接、移除連接和移除所有連接操作,這里我們舉例put和get操作。
put操作
void put(RealConnection connection) { assert (Thread.holdsLock(this)); if (!cleanupRunning) { cleanupRunning = true; executor.execute(cleanupRunnable); } connections.add(connection); }
在添加到Deque<RealConnection>
之前首先要清理空閑的線程,這個后面會講到。
get操作
RealConnection get(Address address, StreamAllocation streamAllocation) { assert (Thread.holdsLock(this)); for (RealConnection connection : connections) { if (connection.allocations.size() < connection.allocationLimit && address.equals(connection.route().address) && !connection.noNewStreams) { streamAllocation.acquire(connection); return connection; } } return null; }
遍歷connections緩存列表,當某個連接計數的次數小於限制的大小並且request的地址和緩存列表中此連接的地址完全匹配。則直接復用緩存列表中的connection作為request的連接。
總結
可以看出連接池復用的核心就是用Deque<RealConnection>
來存儲連接,通過put、get、connectionBecameIdle和evictAll幾個操作來對Deque進行操作,另外通過判斷連接中的計數對象StreamAllocation來進行自動回收連接。