怎么批量測試Mybatis項目中的Sql是否正確

小編給大家分享一下怎么批量測試Mybatis項目中的Sql是否正確，相信大部分人都還不怎么了解，因此分享這篇文章給大家參考一下，希望大家閱讀完這篇文章后大有收獲，下面讓我們一起去了解一下吧！

去Oracle行動

最近公司要發展海外項目，所以要將現有的系統全部平移過去，另外數據庫也要從原來的Oracle變為Mysql。公司的數據庫交互層面使用的是Mybatis，而Oracle與Mysql也有一些語法上的不同。所以在項目中的Sql要改動，但是多個項目中涉及到的Sql非常多，如果僅憑人工一條一條辨別的話，工作量有點大。所以就萌發出了直接將數據源變為Mysql，利用反射批量執行Mapper中的方法，然后如果有參數的話，就設置為默認的初始值，然后記錄下來成功的數據和失敗的數據，這樣就可以根據失敗原因進行修改。

能夠節省很大的時間。

執行效果

代碼介紹

總體思路就三步

• 通過反射獲得要執行的Mapper類的所有方法

• 獲得方法中的參數，并賦值

• 執行

AutoTestMapper autoTestMapper = new AutoTestMapper("存放Mapper全路徑名");
autoTestMapper.openSqlSession(sqlSessionFactory);

在構造函數中傳入全路徑名后，進行解析，解析出包名和所有的文件名并存儲起來

 public AutoTestMapper(String path) throws IOException, ClassNotFoundException {
 String mapperContent = getFileContent(path);
 String pathPattern = "import [a-z,A-Z,/.]+;";
 String[] pathArr = matchMethod(pathPattern, mapperContent).split(";");
 for (int i = 0; i < pathArr.length; i++) {
 pathArr[i] = pathArr[i].replaceAll("import ", "");
 Class cls = Class.forName(pathArr[i]);
 if (!cls.isInterface()) {
 TYPE_ARRAY.add(cls);
 }
 }
 //獲得全路徑名的前綴
 String packPattern = "package [a-z,A-Z,/.]+;";
 String[] packPathArr = matchMethod(packPattern, mapperContent).split(";");
 String packPath = packPathArr[0].replaceAll("package ", "").replaceAll(";", "");
 this.PACK_PATH = packPath;
 }

然后調用openSqlSession的方法，傳入SqlSessionFactory參數

 List<Map<Class, Object>> list = new ArrayList<>();
 List<String> invokeSuccess = new ArrayList<>();
 List<String> invokeFail = new ArrayList<>();
 for (String fileName : FILE_NAME) {
 Class cls = Class.forName(PACK_PATH + "." + fileName);
 //添加Mapper
 if (!sqlSessionFactory.getConfiguration().hasMapper(cls)){
 sqlSessionFactory.getConfiguration().addMapper(cls);
 }
 //獲得Mapper
 Object mapper = sqlSessionFactory.openSession().getMapper(cls);
 //反射執行Mapper的方法
 Map<String, List<String>> resultMap = autoTestInvoke(cls, mapper);
 invokeSuccess.addAll(resultMap.get(SUCCESS_FLG));
 invokeFail.addAll(resultMap.get(FAIL_FLG));
 }

然后通過Mybatyis提供的方法getMapper()傳入類名獲得所要Mapper類。核心方法就是autoTestInvoke()方法了

 	private Map<String, List<String>> autoTestInvoke(Class c, Object o)
 {
 Method[] declaredMethods = c.getDeclaredMethods();
 String fileName = c.getName().substring(c.getName().lastIndexOf("."));
 List<String> invokeSuccess = new ArrayList<>();
 List<String> invokeFail = new ArrayList<>();
 Map<String, List<String>> resultMap = new HashMap<>();
 //給參數賦初始值
 for (Method method : declaredMethods) {
 List<Object> list = new ArrayList<>();
 for (Class cls : method.getParameterTypes()) {
 Object par = new Object();
 if (TYPE_ARRAY.contains(cls)) {
  if (cls.equals(String.class)) {
  par = "1";
  } else {
  try {
  par = cls.newInstance();
  assignment(cls, par);
  } catch (InstantiationException e) {
  if (cls.isPrimitive()) {
  cls = primitiveClazz.get(cls.getName());
  }
  try {
  par = cls.getDeclaredConstructor(String.class).newInstance("1");

  }catch (NoSuchMethodException e1){
  System.out.println(cls.getName()+e);
  }
  }
  }
 }else if ("java.util.Map".equals(cls.getName())){
  par = getMapData(c.getName()+"."+method.getName());
 }
 list.add(par);
 }
 try {
 method.invoke(o, list.toArray());
 invokeSuccess.add("Success: " + fileName + "." + method.getName());
 } catch (Exception e) {
 invokeFail.add("Error:" + method.getName() + " Error Info:" + e);
 }
 }
 resultMap.put(SUCCESS_FLG, invokeSuccess);
 resultMap.put(FAIL_FLG, invokeFail);
 return resultMap;
 }

這里面完成為參數賦初始值，和執行的邏輯。

使用說明

導入Jar包

Maven

<dependency>
 <groupId>com.github.modouxiansheng</groupId>
 <artifactId>convenientUtil</artifactId>
 <version>1.3-release</version>
</dependency>

Gradle

compile 'com.github.modouxiansheng:convenientUtil:1.1-release'

創建mybatis-config.xml文件

在項目的resource文件夾下創建mybatis-config.xml文件，里面內容如下，里面value值為想要測的數據庫的連接信息

<!DOCTYPE configuration
 PUBLIC "-//mybatis.org//DTD Config 3.0//EN"
 "http://mybatis.org/dtd/mybatis-3-config.dtd">
<configuration>
 <environments default="dev">
 <environment id="dev">
 <transactionManager type="JDBC"></transactionManager>
 <dataSource type="UNPOOLED">
 <property name="driver" value=""/>
 <property name="url" value=""/>
 <property name="username" value=""/>
 <property name="password" value=""/>
 </dataSource>
 </environment>
 </environments>
</configuration>

在測試類中編寫代碼

在測試類中編寫如下代碼

Reader resourceAsReader = Resources.getResourceAsReader("mybatis-config.xml");
SqlSessionFactory sqlSessionFactory = new SqlSessionFactoryBuilder().build(resourceAsReader);
resourceAsReader.close();
AutoTestMapper autoTestMapper = new AutoTestMapper("想要測試的Mapper文件夾全路徑名");
autoTestMapper.openSqlSession(sqlSessionFactory);

查看輸出的信息

然后會打印出執行成功的Sql，執行失敗的Sql。如果失敗的話會有原因。
Success: TSesSetManualMapper.updateFlgdelAutoInTimePay
Success: TSesSetManualMapper.getAutoSetManualOrdListCount
Success: TSesSetManualMapper.updateAutoSetManualOrd
Success: TSesSetManualMapper.queryAutoSetManualOrdDetail
Success: TSesSetManualMapper.querySetManualOrdListCount
Success: ShortMessageMapper.queryPayInsSmInfo
-------------------
|Error: |TSesSetManualMapper.queryAutoSetManualOrdList| Every derived table must have its own alias|
|Error: |TSesSetManualMapper.querySetManualOrdList| Every derived table must have its own alias|

這樣就能夠根據錯誤信息進行更改了。

以上是“怎么批量測試Mybatis項目中的Sql是否正確”這篇文章的所有內容，感謝各位的閱讀！相信大家都有了一定的了解，希望分享的內容對大家有所幫助，如果還想學習更多知識，歡迎關注億速云行業資訊頻道！