接口测试平台：支持SQL语句执行(Mysql、Oracle)(旧版已废弃)

叁歲伎倆 2023-06-24 11:26 10阅读 0赞

SQL的使用，主要有以下两个模块：  
1.数据库配置  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2x0MzI2MDMwNDM0_size_16_color_FFFFFF_t_70]  
2.case管理与执行  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2x0MzI2MDMwNDM0_size_16_color_FFFFFF_t_70 1]  
数据库管理这一块，无外乎简单的增删改查，主要是为了保存数据库信息，在此不做过多展开。  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2x0MzI2MDMwNDM0_size_16_color_FFFFFF_t_70 2]  
然后是case的管理：  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2x0MzI2MDMwNDM0_size_16_color_FFFFFF_t_70 3]  
case结构比较简单，主要是两块：  
1.sql的基础信息，包含所用数据库、case名称、sql语句及备注信息  
2.需要保存的变量、返回结果。需要保存的变量这一块，同样支持jsonPath和正则两种方式（后端将返回的表格转换为了JsonArray的形式，所以也支持jsonPath）。  
对应的bean文件如下：

public class SqlCase {
    
        /** 主键id */
        private int id;
    
        /** 创建人 */
        private String createrName;
    
        /** 创建人code */
        private String createrCode;
    
        /** 创建时间 */
        @JsonFormat(locale="zh", timezone="GMT+8", pattern="yyyy-MM-dd")
        private Date createDate;
    
        /** case名 */
        private String caseName;
    
        /** 数据库名 */
        private String databaseName;
    
        /** 数据库id */
        private Integer databaseId;
    
        /** sql语句 */
        private String sql;
    
        /** 描述 */
        private String description;
    
        /** 需要保存的变量 */
        private String variableListValue;
        private List<VariableSave> variableList;
    
        /** 响应结果 */
        private String result;
    }

case的管理本身也是增删改查，就不做拓展了。主要是讲解一下sql执行这一块的代码。  
Service层：

public ResponseVo excuteSqlRequest(SqlCase sqlCase) {
    
            ResponseVo responseVo = new ResponseVo();
    
    		// 获取数据库配置信息
            DataBase dataBase = apiTestConfigMapper.selectDataBaseById(sqlCase.getDatabaseId());
            dataBase.setSql(sqlCase.getSql());
    
    		// Sql执行
            String result = ApiTestUtils.doSqlRequest(dataBase);
            sqlCase.setResult(result);
    
            // 保存变量
            ApiTestUtils.saveVariable(result, sqlCase.getVariableListValue(), ApiTestConfig.GLOBAL_COLLECTION_ID);
    
            // 更新数据库保存的信息
            sqlCaseMapper.updateSqlCase(sqlCase);
    
            // 将执行后的结果返回给前端
            responseVo.setIsSuccess(Boolean.TRUE);
            responseVo.setResult(sqlCase);
    
            return responseVo;
        }

DataBase Bean

public class DataBase {
    
        private Integer id;
    
        private String dbName;
    
        private String url;
    
        private String username;
    
        private String password;
    
        /** sql语句，只做执行使用 */
        private String sql;
    
        private String description;
    
        /**
         * 数据库类型
         * 1：mysql  2：oracle
         */
        private Integer databaseType;
    }

ApiTestUtils.doSqlRequest  
这一块函数，入参是一个数据库信息（包含要执行的sql语句），然后通过databaseType字段值判断是mysql数据库还是oracle数据库。然后通过正则表达式去判断要执行的语句类型，如果是select则返回查询结果，是其他操作则返回操作状态。

public static String doSqlRequest(DataBase dataBase){
    
            String sql = dataBase.getSql();
            String url = "";
            Integer databaseType = dataBase.getDatabaseType();
            if (databaseType == 1){
                url = "jdbc:mysql://" + dataBase.getUrl() + "?characterEncoding=UTF-8";
            }else if (databaseType == 2){
                url = "jdbc:oracle:thin:@" + dataBase.getUrl();
            }
            String username = dataBase.getUsername();
            String password = dataBase.getPassword();
    
            //  储存结果集
            List<Map<String, Object>> list = new ArrayList<>();
    
            Connection con = null;
            PreparedStatement st = null;
            ResultSet rs = null;
    
            try {
                //  加载驱动
                if (databaseType == 1){
                    Class.forName("com.mysql.jdbc.Driver");
                }else if (databaseType == 2){
                    Class.forName("oracle.jdbc.driver.OracleDriver");
                }
                //  获取数据库连接
                con = DriverManager.getConnection(url,username,password);
                st = con.prepareStatement(sql);
    
                //  通过正则进行增删改查的判断
                String selectReg = "^select|SELECT";
                Pattern selectP = Pattern.compile(selectReg);
                Matcher selectM = selectP.matcher(sql);
    
                JSONArray jsonArray = new JSONArray();
    
                //  查询
                if (selectM.find()){
                    rs = st.executeQuery();
                    //  获取查询结果元数据,如果rs为空则抛错，如果不为空则转为map赋值后转为String返回给前端
                    ResultSetMetaData md = rs.getMetaData();
                    int columnCount = md.getColumnCount();
    
                    while (rs.next()){
                        Map<String,Object> rowData = new HashMap<>(16);
                        JSONObject jsonObject = new JSONObject();
                        for (int i = 1; i <= columnCount; i++) {
                            rowData.put(md.getColumnName(i), rs.getObject(i));
                            jsonObject.put(md.getColumnName(i), rs.getObject(i));
                        }
    
                        list.add(rowData);
                        jsonArray.add(jsonObject);
                    }
    
                    if (list.size()==0){
                        return "请检查sql语句";
                    }
                    return JSON.toJSONString(list);
                }
                //  改\查\删
                else {
                    Integer i = st.executeUpdate();
                    if(i > 0){
                        return "执行成功";
                    }else {
                        return "执行失败，请检查sql语句";
                    }
                }
            } catch (ClassNotFoundException e) {
                e.printStackTrace();
                return "数据库驱动加载失败";
            } catch (SQLException e) {
                e.printStackTrace();
                return e.toString();
            } finally {
                if (rs != null) {
                    try {
                        rs.close();
                    } catch (SQLException e) {
                        e.printStackTrace();
                        log.error("释放结果集失败");
                    }
                }
                if (st != null) {
                    try {
                        st.close();
                    } catch (SQLException e) {
                        e.printStackTrace();
                    }
                }
                if (con != null) {
                    try {
                        log.debug("数据库链接关闭成功！");
                        con.close();
                    } catch (SQLException e) {
                        e.printStackTrace();
                        log.error("关闭链接失败");
                    }
                }
            }
        }

有疑问的小伙伴欢迎在文章下方留言，我会根据问题不断优化文章内容！

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2x0MzI2MDMwNDM0_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/20191223083815516.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2x0MzI2MDMwNDM0,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2x0MzI2MDMwNDM0_size_16_color_FFFFFF_t_70 1]: https://img-blog.csdnimg.cn/20191223083848475.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2x0MzI2MDMwNDM0,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2x0MzI2MDMwNDM0_size_16_color_FFFFFF_t_70 2]: https://img-blog.csdnimg.cn/20191223084125742.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2x0MzI2MDMwNDM0,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2x0MzI2MDMwNDM0_size_16_color_FFFFFF_t_70 3]: https://img-blog.csdnimg.cn/2019122308434362.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2x0MzI2MDMwNDM0,size_16,color_FFFFFF,t_70