最近公司要发展海外项目,所以要将现有的系统全部平移过去,另外数据库也要从原来的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; }
这里面完成为参数赋初始值,和执行的逻辑。
自动测试Mapper除了传参为List和Set,其余都能测到。在xml中所有的if条件都会拼接到。
mybatis-config.xml内容如下
<!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="POOLED"> <property name="driver" value="com.mysql.jdbc.Driver"/> <property name="url" value="连接地址"/> <property name="username" value="账号"/> <property name="password" value="密码"/> </dataSource> </environment> </environments> </configuration>
compile files('../lib/mybatis-3.5.0-hupengfeiTest.jar')此处路径填写相对路径
如果目录结构如下,那么就compile files('lib/mybatis-3.5.0-hupengfeiTest.jar')
mybatis-3.5.0-hupengfeiTest.jar在github下面的lib目录中
-lib -- mybatis-3.5.0-hupengfeiTest.jar -build.gradle
如果目录结构如下,那么就compile files('../lib/mybatis-3.5.0-hupengfeiTest.jar')
-lib -- mybatis-3.5.0-hupengfeiTest.jar -service -- build.gradle
@RunWith(SpringJUnit4Cla***unner.class) @SpringBootTest(classes = { AirApplication.class })//此处AirApplication.class为项目中的启动类,自行修改 public class TestMapper { @Test public void testCeshi() throws IllegalAccessException, IntrospectionException, InvocationTargetException, NoSuchMethodException, InstantiationException, IOException, ClassNotFoundException { //读取Mybatis配置 Reader resourceAsReader = Resources.getResourceAsReader("mybatis-config.xml"); //生成SqlSessionFactory SqlSessionFactory sqlSessionFactory = new SqlSessionFactoryBuilder().build(resourceAsReader); resourceAsReader.close(); AutoTestMapper autoTestMapper = new AutoTestMapper(存放Mapper的Java文件夹的全路径名); //执行测试方法 autoTestMapper.openSqlSession(sqlSessionFactory); } }
就会在控制台中打印出执行失败的Mapper以及其原因。如下图所示