前因:写测试框架的时候使用的精简测试框架不需要启动整个springboot,并不支持@Transactional测试后回滚h2数据库,而是在基础测试类里声明cleandb函数供使用,这就需要适配任意表的数据清除,不过更推荐不清理,以方法名为id使数据不重复即可
try{
JdbcTemplate jdbcTemplate = 获取DataSource结构,反射或加载Bean等都可以,不要对业务代码侵入;
Connection connection = jdbcTemplate.getDataSource().getConnection();//获取连接
ResultSet tables = connection.getMetaData().getTables(null, null, null, new String[]{"TABLE"});//获取表名
connection.close();//记得关闭连接,或者使用连接池
while(tables.next()){
//若是测试类的测试方法超过7,java.sql.Connection无法获取连接导致死循环
jdbcTemplate.update("DELETE FROM " + tables.getObject("TABLE_NAME"));
//spring-test-5.2.1.RELEASE-sources.jar!\org\springframework\test\jdbc\JdbcTestUtils.deleteFromTables如果使用了springtest可以代替使用该方法
}
} catch (SQLException e) {
e.printStackTrace();
}