Jmeter編寫自定義函數

原創

Etyero

2020-02-21 01:47

前言：打開jmeter函數助手，我們可以看到，jmeter已經封裝好了很多實用的函數，但是有時候，這些函數也並不能完全滿足我們實際的測試所需，jmeter的魅力所在就是它的開放性，這個時候我們可以自己封裝所需的函數。

1.進到目錄src\functions\org\apache\jmeter\functions，可以看到已有的函數：

2.新增類，繼承抽象類AbstractFunction，其他不多說了，直接上代碼：

/**
 * 根據年月日，時分秒返回單位爲ms的時間戳
 * @author ljl
 */
public class DateWithTime extends AbstractFunction {
 
	private static final List<String> desc = new LinkedList<>();
	private static final String KEY = "__DateWithTime"; // 函數key，用於界面上選擇函數

	static {
		desc.add(JMeterUtils.getResString("Date:yyyy-MM-dd")); // 函數助手中顯示的參數說明，對應到參數
		desc.add(JMeterUtils.getResString("Time:HH:mm:ss"));
		desc.add(JMeterUtils.getResString("function_name_paropt"));// 保存函數返回結果的變量，用於引用
	}      
	private CompoundVariable varName, date, time;

	
	@Override
	public List<String> getArgumentDesc() {
		return desc;
	}

	public DateWithTime() {
	}

	// 函數執行，返回結果
	@Override
	public String execute(SampleResult previousResult, Sampler currentSampler) throws InvalidVariableException {
		String dataTemp = date.execute().trim();
		String timeTemp = time.execute().trim();
		String data_time = dataTemp + timeTemp;
		SimpleDateFormat sdf1 = new SimpleDateFormat("yyyy-MM-ddHH:mm:ss");
		Date date = null;
		try {
			date = sdf1.parse(data_time);
		} catch (ParseException e) {
			e.printStackTrace();
		}
		String varTime = date.getTime() + "";
		if (varName != null) {
			JMeterVariables vars = getVariables();
			final String varTrim = varName.execute().trim();
			if (vars != null && varTrim.length() > 0) {
				vars.put(varTrim, varTime);//
			}
		}
 
		return varTime;

	}

	/** {@inheritDoc} */
	// 接收參數
	@Override
	public void setParameters(Collection<CompoundVariable> parameters) throws InvalidVariableException {
		// 檢查參數數量
		checkParameterCount(parameters, 2, 3);
		Object[] values = parameters.toArray();

		date = (CompoundVariable) values[0];
		time = (CompoundVariable) values[1];
		if (values.length > 2) {
			varName = (CompoundVariable) values[2];
		} else {
			varName = null;
		}
	}

	@Override
	public String getReferenceKey() {
		return KEY;
	}	
}

3.保存後運行，打開函數助手，可以看到以我們代碼中設置的KEY爲__DateWithTime的函數，如圖：