本文版權(quán)歸xmeter.net所有凿叠。歡迎轉(zhuǎn)載俯逾，轉(zhuǎn)載請(qǐng)注明出處程腹。原文請(qǐng)參見這里匣吊。
摘要
在利用JMeter實(shí)施性能測(cè)試的時(shí)候，除了下列JMeter支持的標(biāo)準(zhǔn)協(xié)議之外（HTTP/HTTPS寸潦，F(xiàn)TP色鸳，JDBC，JMS见转，SOAP和TCP等）命雀，可能需要支持一些別的協(xié)議，本文介紹了如何利用JMeter提供的Java Sampler擴(kuò)展機(jī)制來方便地?cái)U(kuò)展對(duì)新協(xié)議的支持斩箫。除了本文介紹的Java Sampler擴(kuò)展方式之外吏砂，更加標(biāo)準(zhǔn)的方式是擴(kuò)展標(biāo)準(zhǔn)的Sampler撵儿，讀者可以參見xmeter君寫的這篇文章來了解如何擴(kuò)展Kafka協(xié)議的支持。
簡介
JMeter跟商業(yè)化的軟件相比狐血，有些表達(dá)方式比較拗口淀歇、難以理解，比如我們這篇文章里的主角“Java Sampler”匈织。如果只是看文字的表面意思讀者可能會(huì)一頭霧水浪默，不知道這是個(gè)什么東東。Sampler的中文字面翻譯是“采樣機(jī)”或者“取樣器”报亩，在JMeter的中文版本里對(duì)這個(gè)Sampler沒有做任何翻譯浴鸿，可能當(dāng)時(shí)中文版本的翻譯者對(duì)這個(gè)如何翻譯也不太確信井氢，干脆直接在菜單里就不翻譯了弦追。Sampler在別的類似的產(chǎn)品中，更為普通的表達(dá)方式是“協(xié)議”花竞。不過Java Sampler的意思不是指支持所謂的“Java協(xié)議”劲件，或者說“Java取樣器”，比較準(zhǔn)確的意思是利用自定義的Java類來擴(kuò)展對(duì)新協(xié)議的支持约急，這些擴(kuò)展的協(xié)議通過“Java請(qǐng)求”加入到測(cè)試腳本中零远，如下面兩個(gè)步驟所示。
下面xmeter君以MQTT的pub為例介紹具體使用Java Sampler來擴(kuò)展步驟厌蔽。
準(zhǔn)備開發(fā)環(huán)境
在給JMeter開發(fā)插件的時(shí)候大部分情況不需要把JMeter的源代碼下載牵辣，只需要對(duì)相關(guān)的JMeter庫進(jìn)行引用就可以了。請(qǐng)參見文章JMeter擴(kuò)展插件實(shí)現(xiàn)對(duì)自定義協(xié)議進(jìn)行支持中的“JMeter插件實(shí)現(xiàn)步驟1 － 準(zhǔn)備開發(fā)環(huán)境”來準(zhǔn)備開發(fā)環(huán)境奴饮。需要注意的是除了引用“ApacheJMeter_core”和“ApacheJMeter_java”之外纬向，還需要引入訪問MQTT服務(wù)器所需的Java類庫，在此例中使用的是fusesource的mqtt-client庫戴卜，這些額外的類庫的引用完全取決于被擴(kuò)展的協(xié)議和協(xié)議擴(kuò)展者的偏好逾条。
讀者可以參考該鏈接的pom.xml來了解本文所需的Maven依賴。
開發(fā)Java Sampler
開發(fā)一個(gè)自己的Java Sampler包括下面幾個(gè)步驟：
1）擴(kuò)展JMeter的類org.apache.jmeter.protocol.java.sampler.AbstractJavaSamplerClient
2）實(shí)現(xiàn)下面的4個(gè)方法:
/**必需實(shí)現(xiàn)的方法**/
publicSampleResult runTest(JavaSamplerContext context)/**可選實(shí)現(xiàn)的方法**/
public Arguments getDefaultParameters()
public void setupTest(JavaSamplerContext context)
public void teardownTest(JavaSamplerContext context)
runTest方法
runTest方法的返回結(jié)果為SampleResult投剥，就是每次調(diào)用返回的結(jié)果师脂。這個(gè)方法的實(shí)現(xiàn)邏輯一般如下：
SampleResult result=newSampleResult();
result.sampleStart();
try{
...
//發(fā)出請(qǐng)求
result.sampleEnd();
//請(qǐng)求成功，設(shè)置測(cè)試結(jié)果為成功
result.setSuccessful(true);
result.setResponseData("data...");
result.setResponseMessage("message..)";
result.setResponseCodeOK();
}catch(Exceptione){
//請(qǐng)求失敗江锨，設(shè)置測(cè)試結(jié)果為失敗
result.sampleEnd();
result.setSuccessful(false);
result.setResponseCode("500");
...
}
return result;
如上述代碼所示吃警，代碼邏輯主要是：
1）對(duì)目標(biāo)系統(tǒng)發(fā)送正確的協(xié)議數(shù)據(jù)。
2）根據(jù)目標(biāo)系統(tǒng)返回的數(shù)據(jù)啄育，給SampleResult設(shè)置正確的開始酌心、結(jié)束時(shí)間，這樣JMeter引擎知道測(cè)試成功與否灸撰，并正確地顯示到JMeter的報(bào)告結(jié)果中谒府。
getDefaultParameters方法
在運(yùn)行測(cè)試的時(shí)候拼坎，需要用戶提供一些輸入，這些輸入甚至可能是JMeter的變量完疫，這個(gè)方法的用處就在于允許用戶在測(cè)試開始泰鸡，或者運(yùn)行期間指定變量的值。如下圖所示的sampler例子就暴露了一些參數(shù)壳鹤，允許測(cè)試人員在測(cè)試編輯和執(zhí)行期間更改參數(shù)值盛龄。
setupTest方法
跟寫JUnit測(cè)試的setup方法類似，這里主要運(yùn)行一些針對(duì)一個(gè)虛擬用戶的一次性起始芳誓、準(zhǔn)備性的操作余舶，比如建立連接的過程在測(cè)試階段可能只需要一次，那么這段建立連接的邏輯可以放在該方法里锹淌。
teardownTest方法
與JUnit測(cè)試的teardown方法類似匿值，這里主要運(yùn)行一些針對(duì)一個(gè)虛擬用戶的收尾的操作，比如清除連接等赂摆。這里需要注意的是挟憔，該方法的調(diào)用不是在單個(gè)虛擬用戶的線程里執(zhí)行的操作，而是所有虛擬用戶在一個(gè)線程里順序執(zhí)行的烟号。
編譯绊谭、部署與使用
完成了代碼的編寫，需要將代碼進(jìn)行編譯和部署汪拥。右擊項(xiàng)目》Run As 》Maven install达传，在target目錄下會(huì)生成一個(gè)jar包。請(qǐng)注意在本文所提供的pom.xml里迫筑，編譯出來的jar包里包含了所需的第三方j(luò)ar包宪赶，比如說fusesource的mqtt-client庫，這樣避免在JMeter運(yùn)行的時(shí)候單獨(dú)去尋找別的jar包铣焊，使得部署過程更加方便逊朽。
將編譯好的target/mqtt-jmeter-jar-with-dependencies.jar拷貝到$JMETER_HOME/lib/ext目錄下，如果開著JMeter則重啟曲伊，啟動(dòng)完畢叽讳，加入一個(gè)Java請(qǐng)求，在“類名稱”下拉列表框中應(yīng)該能看到新擴(kuò)展的類了坟募。如果不能發(fā)現(xiàn)岛蚤，請(qǐng)查看一下lib/ext目錄下是否拷貝了jar包，再者也可以看一下JMeter的日志懈糯，確認(rèn)沒有報(bào)出異常涤妒。
具體的該Java Sampler的使用可以參考這篇文章，這里就不再贅述赚哗。所有的源代碼請(qǐng)參見這里她紫。
總結(jié)
至此硅堆，本文已經(jīng)介紹完如何通過JMeter提供的擴(kuò)展機(jī)制來支持全新協(xié)議的性能測(cè)試支持，讀者可以了解到JMeter對(duì)新協(xié)議的定制擴(kuò)展還是比較簡單的贿讹。當(dāng)然渐逃，通過Java Sampler方式擴(kuò)展的支持在界面友好性上跟JMeter提供的標(biāo)準(zhǔn)的協(xié)議支持提供的功能上面還是差一些，如果想對(duì)JMeter提供更加強(qiáng)大的擴(kuò)展方式民褂，請(qǐng)參考這篇文章茄菊。
另外，本文所提到的MQTT Java Sampler的實(shí)現(xiàn)有一個(gè)更加完整的功能實(shí)現(xiàn)赊堪，XMeter已經(jīng)將其開源面殖，有興趣的讀者可以了解一下如何將Java Sampler改造成更加易用的JMeter插件。
關(guān)于我們
XMeter成立于2016年哭廉，核心團(tuán)隊(duì)都來自于IBM脊僚，是一家領(lǐng)先技術(shù)的性能測(cè)試持續(xù)集成咨詢與服務(wù)提供商。我們致力于提供給客戶可靠,簡單,低成本的性能測(cè)試解決方案群叶。
