MySQL JDBC StreamResult通信原理淺析

使用MySQL JDBC讀取過較大數據量的人應該清楚(例如超過1GB),在讀取的時候內存很可能會Java堆內存溢出,而我們的解決方案是statement.setFetchSize(Integer.MIN_VALUE)並確保遊標是隻讀向前滾動的即可(為遊標的默認值),也可以強制類型轉換為com.mysql.jdbc.StatementImpl,然後調用其內部方法:enableStreamingResults()這樣讀取數據內存就不會掛掉了,這兩者達到的效果是一致的。當然也可以使用useCursorFetch,但是這種方式測試結果性能要比StreamResult慢很多,為什麼?在本文會闡述其大致的原理。

我在前面的部分文章和書籍中都有介紹過其MySQL JDBC在這一塊內部處理的代碼分成三個不同的類來完成的,不過我一直沒有去深究過數據庫和JDBC之間到底是如何通信的過程。有一段時間我一直認為這都屬於服務端行為或者是客戶端與服務端配合的行為,然後並不其然,今天我們來講一下這個行為是怎麼回事。

【先回顧一下簡單的通信】:

JDBC與數據庫之間的通信是通過Socket完成的,因此我們可以把數據庫當成一個SocketServer的提供方,因此當SocketServer返回數據的時候(類似於SQL結果集的返回)其流程是:服務端程序數據(數據庫) -> 內核Socket Buffer -> 網絡 -> 客戶端Socket Buffer -> 客戶端程序(JDBC所在的JVM內存)

到目前為止,IT行業中大家所看到的JDBC無論是:MySQL JDBC、SQL Server JDBC、PG JDBC、Oracle JDBC。甚至於是NoSQL的Client:Redis Client、MongoDB Client、Memcached,數據的返回基本也是這樣一個邏輯。

MySQL JDBC StreamResult通信原理淺析

【使用MySQL JDBC默認直接讀取數據為什麼會掛?】

(1)MySQL Server方在發起的SQL結果集會全部通過OutputStream向外輸出數據,也就是向本地的Kennel對應的socket buffer中寫入數據,這是一次內存拷貝(內存拷貝這個不是本文的重點)。

(2)此時Kennel的Buffer有數據的時候就會把數據通過TCP鏈路(JDBC主動發起的Socket鏈路),回傳數據,此時數據會回傳到JDBC所在機器上,會先進入Kennel區域,同樣進入到一個Buffer區。

(3)JDBC在發起SQL操作後,Java代碼是在inputStream.read()操作上阻塞,當緩衝區有數據的時候,就會被喚醒,然後將緩衝區的數據讀取到Java內存中,這是JDBC端的一次內存拷貝。

(4)接下來MySQL JDBC會不斷讀取緩衝區數據到Java內存中,MySQL Server會不斷髮送數據。注意在數據沒有完全組裝完之前,客戶端發起的SQL操作不會響應,也就是給你的感覺MySQL服務端還沒響應,其實數據已經到本地,JDBC還沒對調用execute方法的地方返回結果集的第一條數據,而是不斷從緩衝器讀取數據。

(5)關鍵是這個傻帽就像一把這個數據讀取完,根本不管家裡放不放的下,就會將整個表的內容讀取到Java內存中,先是FULL GC,接下來就是內存溢出。

【JDBC參數上設置useCursorFetch=true可以解決】

這個方案配合FetchSize設置,確實可以解決問題,這個方案其實就是告訴MySQL服務端我要多少數據,每次要多少數據,通信過程有點像這樣:

MySQL JDBC StreamResult通信原理淺析

這樣做就像我們生活中的那樣,我需要什麼就去超市買什麼,需要多少就去買多少。不過這種交互不像現在網購,坐在家裡就可以把東西送到家裡來,它一定要走路(網絡鏈路),也就是需要網絡的時間開銷,假如數據有1億數據,將FetchSize設置成1000的話,會進行10萬次來回通信;如果網絡延遲同機房0.02ms,那麼10萬次通信會增加2秒的時間,不算大。那麼如果跨機房2ms的延遲時間會多出來200秒(也就是3分20秒),如果國內跨城市10~40ms延遲,那麼時間將會1000~4000秒,如果是跨國200~300ms呢?時間會多出十多個小時出來。

在這裡的計算中,我們還沒有包含系統調用次數增加了很多,線程等待和喚醒的上下文次數變多,網絡包重傳的情況對整體性能的影響,因此這種方案看似合理,但是性能確不怎麼樣。

另外,由於MySQL方不知道客戶端什麼時候將數據消費完,而自身的對應表可能會有DML寫入操作,此時MySQL需要建立一個臨時表空間來存放需要拿走的數據。因此對於當你啟用useCursorFetch讀取大表的時候會看到MySQL上的幾個現象:

(1)IOPS飆升,因為存在大量的IO讀取,如果是普通硬盤,此時可能會引起業務寫入的抖動

(2)磁盤空間飆升,這塊臨時空間可能比原表更大,如果這個表在整個庫內部佔用相當大的比重有可能會導致數據庫磁盤寫滿,空間會在結果集讀取完成後或者客戶端發起Result.close()時由MySQL去回收。

(3)CPU和內存會有一定比例的上升,根據CPU的能力決定。

(4)客戶端JDBC發起SQL後,長時間等待SQL響應數據,這段時間就是服務端在準備數據,這個等待與原始的JDBC不設置任何參數的方式也表現出等待,在內部原理上是不一樣的,前者是一直在讀取網絡緩衝區的數據,沒有響應給業務,現在是MySQL數據庫在準備臨時數據空間,沒有響應給JDBC。

【Stream讀取數據】

我們知道第1種方式會導致Java掛掉,第2種方式效率低而且對MySQL數據庫的影響較大,客戶端響應也較慢,僅僅能夠解決問題而已,那麼現在來看下Stream讀取方式。

前面提到當你使用statement.setFetchSize(Integer.MIN_VALUE)或com.mysql.jdbc.StatementImpl.enableStreamingResults()就可以開啟Stream讀取結果集的方式,在發起execute之前FetchSize不能再手工設置,且確保遊標是FORWARD_ONLY的。

這種方式很神奇,似乎內存也不掛了,響應也變快了,對MySQL的影響也變小了,至少IOPS不會那麼大了,磁盤佔用也沒有了。以前僅僅看到JDBC中走了單獨的代碼,認為這是MySQL和JDBC之間的另一種通信協議,殊不知,它竟然是“客戶端行為”,沒錯,你沒看錯,它就是客戶端行為。

它在發起enableStreamingResults()的時候,幾乎不會做任何與服務端的交互工作,也就是服務端會按照方式1回傳數據,那麼服務端使勁向緩衝區懟數據,客戶端是如何扛得住壓力的呢?

在JDBC當中,當你開啟Stream結果集處理的時候,它並不是一把將所有數據讀取到Java內存中的,也就是圖1中並不是一次性將數據讀取到Java緩衝區的,而是每次讀取一個package(這個package可以理解成Java中的一個byte[]數組),一次最多讀取這麼多,然後會看是否繼續向下讀取保證數據的完整性。業務代碼是按照字節解析成行也業務方使用的。

服務端剛開始使勁向緩衝區懟數據,這些數據也會懟滿客戶端的內核緩衝區,當兩邊的緩衝區都被懟滿的時候,服務端的1個Buffer嘗試通過TCP傳遞數據給接收方時,此時由於消費方的緩衝區也是滿的,因此發送方的線程會阻塞住,等待對方消費,對方消費一部分,就可以推送一部分數據過去。連起來看就是JDBC的Stream數據未來得及消費之前,緩衝區數據如果是滿的,那麼MySQL發送數據的線程就阻塞住了,這樣確保了一個平衡(關於這一點,大家可以使用Java的Socket來嘗試下是否是這樣的)。

對於JDBC客戶端,數據獲取的時候每次都在本地的內核緩衝區當中,就在小區的快遞包裹箱拿回家一個距離,那麼自然比起每次去超市的RT要小得多了,而且這個過程是準備好的數據,所以沒有IO阻塞的過程(除非MySQL服務端傳遞的數據還不如消費端處理數據來得快,那一般也只有消費端不做任何業務,拿到數據直接放棄的測試代碼,才會發生這樣的事情),這個時候不論:跨機房、跨地區、跨國家,只要服務端開始響應就會源源不斷地傳遞數據過來,而這個動作即使是第1種方式也是必然需要經歷的過程。

相對於第1種方式,JDBC使用的時候會不導致內存溢出,即使讀取大表不內存溢出也會很長時間才會響應;不過這種方式相對方式1來講對數據庫影響相對較大,在傳遞的數據的過程中,相應的數據行會被上鎖(防止被修改),使用InnoDB會分段加鎖處理,使用MyISAM會加全表鎖,可能導致業務阻塞。

【理論上可以更進一步,只要你願意】

理論上這種方式是比較好的了,但是就完美主義來講,我們可以繼續探討一下,對於懶人來講,我們連到小區樓下快遞包裹箱去拿一下的動力也是沒有的,我們心裡想的就是要是誰給我拿到家裡來送到我嘴巴里,連嘴巴都給我掰開多好。

在技術上理論上確實可以做到這樣,因為JDBC從內核拷貝內存到Java當中是需要花時間的,要是有另一個人把這個事情做了,我在家裡幹別的事情的時候它就給我送到家裡來了,我要用的時候就直接從家裡來,這個時間豈不是省掉了。每錯,對於你來講確實省掉了,不過問題就是誰來送?

在程序中一定需要加一個線程來幹這個事情,把內核的數據拷貝到應用內存,甚至於解析成行數據,應用程序直接使用,但這一定完美嗎?其實這個中間就有個協調問題了,例如家裡要炒菜,缺一包調料,原本可以自己到樓下買,但是非要讓別人送家裡,這個時候

其它的菜都下鍋了,就剩一包調料,那麼你沒別的辦法,只能等這包調料送到家裡來以後才能進行炒菜的下一道工序。所以,在理想情況下,它可以節約很多次內存拷貝時間,會增加一些協調鎖的開銷。

那麼可以不可以直接從內核緩衝區讀取數據呢?

理論上也是可以的,在解釋這個問題之前,我們先了解下除了這一次內存拷貝還有那些:

JDBC按照二進制將內核緩衝區的數據讀取後,也會進一步解析成具體的結構化數據,由於此時要給業務方返回ResultSet的具體行的結構化數據,也就是生成RowData的數據一定會有一次拷貝,而且JDBC返回某些對象類型數據的時候(例如byte []數組),在某些場景的實現,它不希望你通過結果集修改返回結果中的byte []的內容(byte[1] = 0xFF)去修改ResultSet本身內容,可能還會再做1次內存拷貝,業務代碼使用過程中還會存在拼字符串,網絡輸出等,又是一堆的內存拷貝,這些在業務層面是無法避免的,相對這點點拷貝來講,簡直微不足道,所以我們也沒去幹這事情,以為從整體上看幾乎微不足道,除非你的程序瓶頸在這裡。

因此從整體上看內存拷貝是無法避免的,多的這一次無非是系統級的調用,開銷會更大一點,從技術上來講,我們是可以做到直接從內核態直接讀取數據的;但這個時候就需要按照字節將Buffer從的數據拿走才能讓遠程更多的數據傳遞過來,沒有第三個位置存放Buffer了,否則又回到了內核到應用的內存拷貝上來了。

相對來講,服務端倒是可以優化直接將數據通過直接IO的方式傳遞(不過這種方式數據的協議就和數據的存儲格式一致了,顯然只是理論上的), 要真正做到自定義的協議,又要通過內核態數據直接發送,需要通過修改OS級別的文件系統協議,來達到轉換的目的。


分享到:


相關文章: