Jmeter自定義函數，取csv檔案中指定行列的資料

public class GetCSVValue extends AbstractFunction {

private static final List<String> desc = new LinkedList<String>();

private static final String KEY = "__MyCSVRead";

private CompoundVariable varName;

private CompoundVariable fileUrl;

private CompoundVariable row;

private CompoundVariable col;

private BufferedReader reader;

static {

desc.add("fileUrl");

desc.add("row");

desc.add("col");

desc.add(JMeterUtils.getResString("function_name_paropt"));

}

@Override

public String execute(SampleResult previousResult, Sampler currentSampler) throws InvalidVariableException {

String fileUrlV = String.valueOf(fileUrl.execute().trim());

int rowv = Integer.valueOf(row.execute().trim());

int colv = Integer.valueOf(col.execute().trim());

String last = null;

try {

reader = new BufferedReader(new FileReader(fileUrlV));

// reader.readLine();// 第一行資訊，為标題資訊，不用,如果需要，注釋掉

String line = null;

int index = 0;

while ((line = reader.readLine()) != null) {

String item[] = line.split(",");

if (index == rowv - 1) {

if (item.length >= colv - 1) {

last = item[colv - 1];

}

}

index++;

}

} catch (Exception e) {

e.printStackTrace();

}

if (varName != null) {

JMeterVariables vars = getVariables();

final String varTrim = varName.execute().trim();

if (vars != null && varTrim.length() > 0) {// vars will be null on TestPlan

vars.put(varTrim, last);

}

}

return last;

}

@Override

public void setParameters(Collection<CompoundVariable> parameters) throws InvalidVariableException {

checkParameterCount(parameters, 3, 4);

Object[] values = parameters.toArray();

fileUrl = (CompoundVariable) values[0];

row = (CompoundVariable) values[1];

col = (CompoundVariable) values[2];

if (values.length > 3) {

varName = (CompoundVariable) values[3];

} else {

varName = null;

}

}

@Override

public String getReferenceKey() {

return KEY;

}

public List<String> getArgumentDesc() {

return desc;

}

}