做性能測(cè)試時(shí)仑性，我們通常都是需要自己編寫(xiě)測(cè)試腳本惶楼，才能滿(mǎn)足測(cè)試用例的需求，Jmeter也是支持我們自定義編寫(xiě)測(cè)試腳本的，而且支持Java歼捐、Python等多種語(yǔ)言何陆。在這里，我們選擇使用Java語(yǔ)言開(kāi)發(fā)測(cè)試腳本豹储，是因?yàn)镴meter本身也是用Java實(shí)現(xiàn)的贷盲，所以使用Java開(kāi)發(fā)腳本會(huì)更好。

首先剥扣，使用Eclipse或者IntelliJ IDEA來(lái)新建一個(gè)項(xiàng)目巩剖，然后新建一個(gè)包，再?gòu)腏meter的lib/ext目錄下中拷貝ApacheJMeter_java.jar和ApacheJMeter_core.jar兩個(gè)文件钠怯，然后引入這兩個(gè)JAR包到項(xiàng)目中佳魔。在IntelliJ IDEA中的操作如下：

1. 點(diǎn)擊File->Project Structure
2. 點(diǎn)擊Project Structure界面左側(cè)的Modules
3. 在Dependencies標(biāo)簽界面下
4. 點(diǎn)擊底部的+號(hào)，
5. 選擇第一個(gè)選項(xiàng)JARs or directories...
6. 選擇相應(yīng)的jar包晦炊，點(diǎn)擊OK

IntelliJ IDEA下的導(dǎo)入操作

然后鞠鲜，我們?cè)诎行陆ㄒ粋€(gè)TestNum類(lèi)，該類(lèi)繼承ApacheJMeter_java.jar包中的AbstractJavaSamplerClient類(lèi)断国，引用它就可以調(diào)用了镊尺。在繼承AbstractJavaSamplerClient類(lèi)的時(shí)候，我們需要實(shí)現(xiàn)4個(gè)方法：

• setupTest()：初始化方法并思，用于初始化性能測(cè)試時(shí)的每個(gè)線(xiàn)程，實(shí)際運(yùn)行時(shí)每個(gè)線(xiàn)程僅執(zhí)行一次
• getDefaultParameters()：用于設(shè)置傳入的參數(shù)语稠，可以設(shè)置多個(gè)宋彼，已設(shè)置的參數(shù)會(huì)顯示到Jmeter參數(shù)列表中
• runTest()：性能測(cè)試時(shí)的線(xiàn)程運(yùn)行體，即測(cè)試執(zhí)行的循環(huán)體仙畦，根據(jù)線(xiàn)程數(shù)和循環(huán)次數(shù)的不同可執(zhí)行多次
• teardownTest()：測(cè)試結(jié)束方法输涕，用于結(jié)束性能測(cè)試中的每個(gè)線(xiàn)程，實(shí)際運(yùn)行時(shí)慨畸，每個(gè)線(xiàn)程僅執(zhí)行一次莱坎，在測(cè)試方法運(yùn)行結(jié)束后執(zhí)行

具體的代碼實(shí)現(xiàn)如下：

import org.apache.jmeter.config.Arguments;
import org.apache.jmeter.protocol.java.sampler.AbstractJavaSamplerClient;
import org.apache.jmeter.protocol.java.sampler.JavaSamplerContext;
import org.apache.jmeter.samplers.SampleResult;

public class TestNum extends AbstractJavaSamplerClient {
    // SampleResult類(lèi)封裝了從入口樣本返回的各種信息
    private SampleResult results;
    // 輸入的數(shù)字
    private String inNum;
    // 需要匹配的數(shù)字
    private String resultNum;

    // 用于設(shè)置傳入的參數(shù)，可以設(shè)置多個(gè)寸士，已設(shè)置的參數(shù)會(huì)顯示到Jmeter參數(shù)列表中
    // Arguments類(lèi)表示一組參數(shù)對(duì)象
    public Arguments getDefaultParameters() {
        // 聲明定義Arguments類(lèi)
        Arguments params = new Arguments();
        // 添加一個(gè)新參數(shù)
        params.addArgument("inNum", "");
        params.addArgument("resultNum", "66");
        return params;
    }

    // 初始化方法檐什，用于初始化性能測(cè)試時(shí)的每個(gè)線(xiàn)程，實(shí)際運(yùn)行時(shí)每個(gè)線(xiàn)程僅執(zhí)行一次
    //
    // JavaSamplerContext類(lèi)用于向JavaSamplerClient實(shí)現(xiàn)提供上下文信息
    // 這當(dāng)前由在GUI中指定的初始化參數(shù)組成弱卡，其他數(shù)據(jù)將來(lái)可以通過(guò)JavaSamplerContext訪(fǎng)問(wèn)
    public void setupTest(JavaSamplerContext jsc) {
        // 定義SampleResult類(lèi)
        results = new SampleResult();
        // 以String形式獲取指定參數(shù)的值乃正，或者如果未指定該值，則返回指定的默認(rèn)值
        inNum = jsc.getParameter("inNum", "");
        resultNum = jsc.getParameter("resultNum", "");
        if (inNum != null && inNum.length() > 0) {
            // 設(shè)置采樣器數(shù)據(jù)
            results.setSamplerData(inNum);
        }
        if (resultNum != null && resultNum.length() > 0) {
            results.setSamplerData(resultNum);
        }
    }

    // 性能測(cè)試時(shí)的線(xiàn)程運(yùn)行體婶博，即測(cè)試執(zhí)行的循環(huán)體瓮具，根據(jù)線(xiàn)程數(shù)和循環(huán)次數(shù)的不同可執(zhí)行多次
    public SampleResult runTest(JavaSamplerContext arg0) {
        boolean flag = false;
        // 定義一個(gè)事務(wù)，表示這是事務(wù)的起始點(diǎn)
        results.sampleStart();
        for (int i = inNum.length(); --i >= 0; ) {
            // 使用isDigit判斷是否為數(shù)字，charAt方法用于返回指定索引處的字符
            if (!Character.isDigit(inNum.charAt(i))) {
                flag = false;
            } else {
                flag = true;
            }
        }
        for (int j = resultNum.length(); --j >= 0; ) {
            if (!Character.isDigit(resultNum.charAt(j))) {
                flag = false;
            } else {
                flag = true;
            }
        }
        // 定義一個(gè)事務(wù)名党，表示這是事務(wù)的結(jié)束點(diǎn)
        results.sampleEnd();
        if (flag) {
            Integer num = Integer.parseInt(inNum);
            Integer rsNum = Integer.parseInt(resultNum);
            if (num == rsNum) {
                // 設(shè)置SampleResult對(duì)象的編碼和響應(yīng)數(shù)據(jù)
                results.setResponseData("恭喜你叹阔，答對(duì)了O(∩_∩)O~\n答案是【" + resultNum + "】", "UTF-8");
                // 設(shè)置SampleResult對(duì)象的成功屬性，即返回正確還是錯(cuò)誤
                results.setSuccessful(true);
            } else if (num > rsNum) {
                results.setResponseData("好像大了點(diǎn)~~~~(>_<)~~~~ \n您輸入的是【" + inNum + "】", "UTF-8");
                results.setSuccessful(false);
            } else {
                results.setResponseData("好像小了點(diǎn)~~~~(>_<)~~~~ \n您輸入的是【" + inNum + "】", "UTF-8");
                results.setSuccessful(false);
            }
        } else {
            results.setResponseData("請(qǐng)輸入數(shù)字：~~~~(>_<)~~~~ \n您輸入的inNum是【" + inNum + "】传睹，resultNum是【" + resultNum + "】", "UTF-8");
            results.setSuccessful(false);
        }
        return results;
    }

    // 測(cè)試結(jié)束方法耳幢，用于結(jié)束性能測(cè)試中的每個(gè)線(xiàn)程，實(shí)際運(yùn)行時(shí)蒋歌，每個(gè)線(xiàn)程僅執(zhí)行一次帅掘，在測(cè)試方法運(yùn)行結(jié)束后執(zhí)行
    public void teardownTest(JavaSamplerContext arg0) {
    }
}

然后將上面的代碼打包成jar包，生成的包名稱(chēng)為T(mén)estNum.jar堂油，在IntelliJ IDEA中的操作如下：

1. 點(diǎn)擊File->Project Structure
2. 點(diǎn)擊Project Structure界面左側(cè)的Artifact
3. 依次點(diǎn)擊底部的+>JAR>Empty
4. 輸入這個(gè)jar的Name修档，下面有這個(gè)jar包生成的路徑Output directory
5. 在Output Layout中點(diǎn)擊+>Module Output給這個(gè)jar包添加Module Output
6. 點(diǎn)擊OK退出Project Structure面板
7. 最后依次點(diǎn)擊Build>Build Artifacts...>TestNum>Build/Rebuild

生成jar包后，將jar包拷貝到Jmeter的lib/ext目錄下府框。之后啟動(dòng)Jmeter吱窝，添加線(xiàn)程組和Java請(qǐng)求，同時(shí)為了方便迫靖，我們用隨機(jī)數(shù)函數(shù)來(lái)隨機(jī)抽取數(shù)字院峡，驗(yàn)證與resultNum匹配的結(jié)果，再添加察看結(jié)果樹(shù)和聚合報(bào)告系宜。

配置自定義Java腳本

設(shè)置線(xiàn)程數(shù)為10照激，執(zhí)行后可以看到，10個(gè)請(qǐng)求中盹牧，有3個(gè)請(qǐng)求回答正確俩垃，響應(yīng)數(shù)據(jù)對(duì)應(yīng)了我們的代碼，表示執(zhí)行成功汰寓。

自定義Java腳本執(zhí)行結(jié)果

通過(guò)使用Jmeter自定義Java測(cè)試代碼口柳，再配合Jmeter自帶的各種函數(shù)，就可以實(shí)現(xiàn)絕大多數(shù)的性能測(cè)試需求有滑。