接口测试平台：支持SQL语句执行(Mysql、Oracle)
- 2020 年 1 月 2 日
- 笔记
1、数据库配置
2、case管理与执行
数据库管理这一块，无外乎简单的增删改查，主要是为了保存数据库信息，在此不做过多展开。
然后是case的管理：
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("关闭链接失败"); } } } }
有疑问的小伙伴欢迎在公号内留言，我会根据问题不断优化文章内容！
