對(duì)于曾經(jīng)使用過thrift來開發(fā)RPC服務(wù)的同學(xué)來講,經(jīng)常碰到就是client去連server端的時(shí)候,程序首先要聲明連接協(xié)議,我們以socket為例。然后開始連接到服務(wù)端,調(diào)用遠(yuǎn)端的方法,關(guān)閉連接。
代碼片斷如下:

伴隨著服務(wù)越來越多,如果我們的代碼中充斥這些duplicate code,心里總是癢癢的,非得把它干掉。聯(lián)想到當(dāng)初最開始寫 jdbc常用的crud方法那段時(shí)間,代碼里面也是挺多的open,close種種。其實(shí)想想,如果能有連接池,這個(gè)問題也就好解決了。于是,首先在github上找找,有沒有現(xiàn)成的實(shí)現(xiàn)。果然,有一位前輩已經(jīng)實(shí)現(xiàn)了。于是果斷下載,地址是https://github.com/PhantomThief/thrift-pool-client。
下面是這個(gè)客戶端所實(shí)現(xiàn)的功能,感覺還是相當(dāng)不錯(cuò)的。
A Thrift Client pool for Java
raw and type safe TServiceClient pool
Multi backend servers support
Backend servers replace on the fly
Backend route by hash or random
Failover and failback support
jdk 1.8 only
不過,看了md之后,發(fā)現(xiàn)only support jdk8。心里涼了半截,難道老外都在用jdk8了?于是花了半天時(shí) 間,熟讀了新的語(yǔ)法之后,準(zhǔn)備改造前輩的代碼。通讀代碼之后,發(fā)現(xiàn)里面有好幾個(gè)類使用lambda語(yǔ)法,flatmap,collection,filter.另外就是函數(shù)是編程function.最后還有Supplier。好在guava已經(jīng)實(shí)現(xiàn)了后兩者,我只需要改改lambda的語(yǔ)法。
核心類說明:
ThriftClientImpl:它是客戶端的實(shí)現(xiàn)類,主要是業(yè)務(wù)方經(jīng)常用到的iface方法,先看圖:


里面有兩段代碼需要改動(dòng)。第一個(gè)就是設(shè)置執(zhí)行的方法過濾,第二個(gè)是代理方法里面調(diào)用,實(shí)際就是業(yè)務(wù)方真正調(diào)用的方法塊。關(guān)鍵點(diǎn)是javassist它是一種字節(jié)碼處理技術(shù),可以動(dòng)態(tài)改變方法塊。本段代碼其實(shí)類似一個(gè)代理工廠,它代理了所有業(yè)務(wù)服務(wù)的執(zhí)行片斷
DefaultThriftConnectionPoolImpl,ThriftConnectionFactory:thrift連接池實(shí)現(xiàn),基本是apache pool的通用連接池處理。
其中sokcet.open對(duì)應(yīng)著makeobject.socket.close對(duì)應(yīng)著destroyobject

ThriftClientUtils:這個(gè)類有一個(gè)獲取接口中所有方法的名稱集合的方法,把lambda的語(yǔ)法改成熟悉的方式。

其它類也是做了相關(guān)調(diào)整,這里就不一一說明了。最后講一下調(diào)用。
1連接池初始化

2 構(gòu)建單例的rpc服務(wù),因?yàn)閞pc服務(wù)可能在多臺(tái)機(jī)器上,所以會(huì)有分片

3 協(xié)議的聲明

4 調(diào)用

5結(jié)果,因?yàn)槲矣幸粋€(gè)簡(jiǎn)單的計(jì)算服務(wù),已經(jīng)發(fā)布在測(cè)試機(jī)器了,這里就是1+99的結(jié)果。

主體寫完了,改天會(huì)把github的地址貼上來,請(qǐng)隨時(shí)關(guān)注。
我的微信公眾號(hào),歡迎溝通學(xué)習(xí)。
