JMeter 判断字符串是否存在该集合（List 或数组）

在学习 JMeter 的 JDBC 模块中，遇到了 ArrayList 和 String 的之间需要对比的问题，折腾我一周时间了。这个问题如果用代码解决的话，一个 for 循环就能搞掂，但在 JMeter 中就比较麻烦了。因为在 JMeter 界面上，一定得数字或字符串，才能进行对比，其他类型是不能对比的。以下的2种解决方案是分别使用 BeanShell 和函数解决的。

问题：

如何在 JMeter 中判断字符串是否在集合（List）或数组里面？集合和数组的大小不定，匹配的字符串位置也不定或者没有。

例子：

在测试前，我们需要在一个数据表（table）准备一些测试数据，但如果这个表已存在，那就无需创建了，直接对表进行修改，增减数据即可。

问题分析：

首先，我们不知道整个数据库的表有多少个？也不知道我们要创建的数据表是否存在，如果存在，那在哪个位置呢？那怎么完成ArrayList 和 String的对比呢，使用JMeter 组件还是代码解决？

初步解决方案：

先使用SHOW TABLES;查询数据库所有数据表，将每一个表的名称分解成一个字符串，这样比较就容易多了。

BeanShell 解决方案

1、查询当前数据库的表

在 JMeter 创建一个 JDBC Request，查询数据库当前所有的表名，并将其保存到tables这个变量中。

JDBC Request.png

2、添加 BeanShell 后置处理器，使用代码进行比较

对上述的 JDBC Request 添加一个 BeanShell 的后置处理器，进行代码处理，最后对isTableExist变量进行判断即可（true 为表已创建，false 为表未创建）。

BeanShell PostProcessor.png

一并贴上 BeanShell 代码

//将数据库的表名与创建的表名一一比较，如存在，不创建，如不存在，则创建数据表

//获取当前数据库的表数目
int tablesNum = Integer.parseInt(vars.get("tables_#"));
//默认为 false
boolean isTableExist = false;
log.info("数据表数目:" + tablesNum.toString());
if(tablesNum != 0) {
    String buildTable = vars.get("testTable");
    //对整个数据库的表名与需创建的表名一一比较
    for(int i=1; i<=tablesNum; i++) {
        String table = vars.get("tables_"+i);
        //如果数据库存在该表，则为 true
        if(buildTable.equals(table)) {
            isTableExist = true;
        }
    }
}
String isExist = Boolean.toString(isTableExist);
//isTableExist 是false，则需创建，如为 true，则无需创建
vars.put("isTableExist",isExist);
log.info("【" + vars.get("testTable") + "】数据表是否存在数据库中：" + isExist);

3、调用isTableExist值

在创建数据表时，可根据isTableExist的值来判断是否需要创建表。可在 If Controler 控制器上使用，如下图所示：

if_Controler.png

函数解决方案

这个解决方案其实就是将上述代码整理成一个函数，这样在 JMeter 调用就方便多了，不用每次都需要复制编写 BeanShell 代码，增加错误率，使用也更便捷一些。

1、查询当前数据库的表，并将表的结果保存到变量中

第一步与 BeanShell 解决方案大同小异，主要是保存的变量不一样而已。当然也可以跟第一步一样，只不过需要函数做出相应的改动，这一块就留给大家练练手。

JDBC Request2.png

2、编写函数

编写函数的过程就不在这里赘述了，大家对 JMeter 函数开发不熟悉的话，可以百度一下，资料大把。
按照惯例，还是贴上代码给大家吧。

package com.dc.functions.functions;

import org.apache.jmeter.engine.util.CompoundVariable;
import org.apache.jmeter.functions.AbstractFunction;
import org.apache.jmeter.functions.InvalidVariableException;
import org.apache.jmeter.samplers.SampleResult;
import org.apache.jmeter.samplers.Sampler;
import org.apache.jmeter.threads.JMeterVariables;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.*;

public class FindTableInDataBase extends AbstractFunction {

    private static final Logger log = LoggerFactory.getLogger(FindTableInDataBase.class);
    //显示的参数名字
    private static final List<String> desc = new LinkedList<>();
    static {
        desc.add("DataBase Result Variable Name");
        desc.add("DataBase Name");
        desc.add("Table Name");
    }
    //显示的函数名字
    private static final String KEY = "__FindTableInDataBase";
    //参数值
    private Object[] values;

    @Override
    public String execute(SampleResult sampleResult, Sampler sampler) throws InvalidVariableException {
        JMeterVariables variabless = getVariables();

        if (values.length != desc.size()) {
            log.warn("输入的参数不是" + desc.size() + "个，请检查参数。");
            return null;
        }

        String dbResultVName = ((CompoundVariable)this.values[0]).execute().trim();
        String dbName = ((CompoundVariable)this.values[1]).execute().trim();
        String testTableName = ((CompoundVariable)this.values[2]).execute().trim();
        String mapKey = "Tables_in_" + dbName;
        boolean isExist = false;
        List<Map<String, Object>> tablesList = (ArrayList<Map<String, Object>>) variabless.getObject(dbResultVName);
        log.info("数据库【" + dbName + "】的数据表：" + tablesList.toString());
        log.info("----------------------------");
        for (int i = 0; i < tablesList.size(); i ++) {
            Map<String, Object> tableMap = tablesList.get(i);
            if (tableMap.get(mapKey).equals(testTableName)) {
                isExist = true;
                log.info("数据表【" + testTableName + "】存在于数据库【" + dbName + "】");
            }
        }
        log.info("----------------------------");
        if (isExist == false) {
            log.info("在数据库【" + dbName + "】查找不到数据表【" + testTableName + "】");
        }
        return Boolean.toString(isExist);
    }

    //设置参数值
    @Override
    public void setParameters(Collection<CompoundVariable> collection) throws InvalidVariableException {
        checkMinParameterCount(collection, 3);
        checkParameterCount(collection, 3);
        this.values = collection.toArray();
    }

    //返回函数名字
    @Override
    public String getReferenceKey() {
        return KEY;
    }

    //返回参数名字
    @Override
    public List<String> getArgumentDesc() {
        return desc;
    }
}

在你的函数开发工程里，创建一个名为FindTableInDataBase的类，并将上述代码粘贴到该java 文件，然后编译工程，将 jar 包放到${JMeter_Home}/lib/ext路径上，重启 JMeter 即可。

3、调用函数

在判断是否需要需要创建数据表时，在 If Controler 控制器上直接调用该函数，根据返回的结果与预期值比对，即可得出是否需要创建数据表。

if_Controler2.png

总结

以上2种方案是我目前能想到解决方案，也是根据我最近学的知识所总结出来的，可能还有其他更加好的方法，期待与大家一同讨论进步。。