造一个方形轮子文章目录：造一个方形的轮子
01、解决遗留问题
这一篇没有太多的解决上一篇遗留的问题，静态文件暂时没有处理，继续算做遗留问题，处理了一下Square框架的依赖，修改了mysql数据库驱动的scope配置：
<dependency>
<groupId>mysql</groupId>
<artifactId>mysql-connector-java</artifactId>
<version>${mysql.version}</version>
<scope>provided</scope>
</dependency>
02、集成Mybatis
集成第三方，找了一个比较常用的框架，持久层的Mybatis，起初觉得这东西用起来很方便，没有太细的考虑过，如果要集成他要怎么做，一上手写还是发现了很多问题。
集成的大体思路是：
1、添加square-mybatis项目，做成依赖，处理Mybatis的相关配置
2、demo项目添加Mybatis和square-mybatis依赖
3、添加mapper的xml配置及对应的接口
4、关联使Mybatis框架生效
这里有几个问题：
1、square-mybatis里的代码如果在square框架里加载（因为有Bean操作）
2、Mybatis的mapper.xml文件放在哪？如何加载？
3、打包后怎么保证Mybatis框架生效
这几个问题其实是做之前有大体思路，但做下来发现有些跟写之前想的确实不一样，下边我会尽量记录一下遇到的问题，及解决过程
03、square框架加载square-mybatis
最开始我在square框架里添加了一个@config 注解及一个SquareConfig.java，作用就是使用这个注解标记的 类同时继承SquareConfig接口，实现config()方法，会在square框架加载过程中处理依赖之前，执行这个config()方法，目的是把配置的Mybatis框架的Mapper注入到Bean容器中，好在接下来的initDI()方法中使用，保证依赖类的存在，但是写完后发现因为这个类在 square-mybatis项目里，是以依赖的形式提供给demo的，所以demo项目启动过程中扫描不到jar包里的文件。
所以最终将square-mybatis中的Conifg类实现成了静态方法，在demo中添加了InitConfig.java来实现调用功能，下边看代码：
square-mybatis项目的MybatisConfig.java：
package com.jisuye;
//import ...
/**
* 集成Mybatis配置类
* @author ixx
* @date 2019-09-08
*/
public class MybatisConfig {
private static Logger logger = LoggerFactory.getLogger(MybatisConfig.class);
public static void init(){
// 获取DataSource
DataSource dataSource = DbUtil.getDataSource();
// 使用代码构造Mybatis配置
TransactionFactory transactionFactory = new JdbcTransactionFactory();
Environment environment = new Environment("development", transactionFactory, dataSource);
Configuration configuration = new Configuration(environment);
URL mapperUrl = ClassLoader.getSystemResource("mapper/");
String path = mapperUrl.getPath();
logger.info("mapperUrl path :{}", path);
// 判断当前启动是在否jar包中
if(path.indexOf("!/")>0){
getXmlByJar(path, configuration);
} else {
getXmlByFile(path, configuration);
}
SqlSessionFactory sqlSessionFactory = new SqlSessionFactoryBuilder().build(configuration);
try {
SqlSession session = sqlSessionFactory.openSession();
// 循环所有Bean处理Config
for(Class clzz : BeansMap.getClassList()) {
Annotation[] annotations = clzz.getAnnotations();
for (Annotation annotaion : annotations) {
// 如果是Mybatis的Mapper则注入到容器里
if(annotaion instanceof Mapper){
Object mapper = session.getMapper(clzz);
BeanObject tmpBeanObject = new BeanObject(clzz, mapper);
tmpBeanObject.setObject(mapper);
BeansMap.put(clzz.getName(), tmpBeanObject);
}
}
}
} catch (Exception e){
logger.error("init mybatis config error!!", e);
}
}
/**
* 本地调试获取xml配置文件
* @param path
* @param configuration
*/
private static void getXmlByFile(String path, Configuration configuration){
File mapperDir = null;
try {
mapperDir = new File(path);
if(mapperDir.exists()){
File[] files = mapperDir.listFiles();
logger.info("files size :{}", files.length);
for (File file : files) {
logger.info("file path:{}", file.getPath());
logger.info("file abspath:{}", file.getAbsolutePath());
try {
XMLMapperBuilder xmlMapperBuilder = new XMLMapperBuilder(
new FileInputStream(file), configuration,
file.getPath(),
configuration.getSqlFragments());
xmlMapperBuilder.parse();
} catch (Exception e) {
e.printStackTrace(); // 出现错误抛出异常
}
}
}
} catch (Exception e) {
logger.error("load mapper xml file error!", e);
}
}
/**
* 处理jar包中的xml配置文件（打包后启动）
*/
private static void getXmlByJar(String path, Configuration configuration){
logger.info("xml get by path:{}", path);
path = path.substring(0, path.indexOf("!/")).replace("file:", "");
logger.info("xml get by jar path:{}", path);
try {
JarFile jarFile = new JarFile(path);
Enumeration<JarEntry> jarEntryEnumeration = jarFile.entries();
while (jarEntryEnumeration.hasMoreElements()){
JarEntry jarEntry = jarEntryEnumeration.nextElement();
String name = jarEntry.getName();
// logger.info("file name ====={}", name);
if(name.startsWith("mapper/") && name.endsWith(".xml")){
XMLMapperBuilder xmlMapperBuilder = new XMLMapperBuilder(
ClassLoader.getSystemResourceAsStream(name), configuration,
name,
configuration.getSqlFragments());
xmlMapperBuilder.parse();
}
}
} catch (Exception e) {
e.printStackTrace();
}
}
}
这个类处理时遇到的问题比较多，首先最开始没找到设置mapper.xml文件的地方，就用了Mybatis默认的mapper.xml跟对应的接口在同一个目录下，结果发现这样需要在pom文件中配置resource引入.xml文件 不然编译后找不到xml文件，而且还要添加一个配置，指定Mybatis扫描的package路径.后来翻看了mybatis-spring的代码找到了遍历加载xml配置文件的方法，又遇到打包后在jar包里加载不到的问题，好在之前处理打包插件，弄过jar文件，直接在做个判断如果在jar里启动的，特殊处理一下，最后就可以了。
04、square项目调整
添加Config相关注解及接口:
package com.jisuye.annotations;
/**
* Config注解
* @author ixx
* @date 2019-09-06
*/
@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
public @interface Config {
String value() default "";
}
package com.jisuye.core;
/**
* 定义config接口，使用@config注解的类需要继承
* @author ixx
* @date 2019-09-08
*/
public interface SquareConfig {
void config();
}
BeanInitUtil.java添加initConfig()方法：
//...
public static void init(Class clazz){
String path = clazz.getResource("").getPath();
// ....
// 处理aop类
initAop();
// 处理config
initConfig();
// 处理依赖注入
initDI();
}
private static void initConfig(){
// 循环所有Bean处理Config
Set<String> keySet = BeansMap.keySet();
String[] keys = new String[keySet.size()];
keySet.toArray(keys);
List<Object> list = new ArrayList<>();
for (String key : keys) {
BeanObject beanObject = BeansMap.get(key);
if(list.contains(beanObject)){
continue;
}
list.add(beanObject);
for (Annotation annotaion : beanObject.getAnnotaions()) {
// 如果是配置则调用config()方法
if(annotaion instanceof Config){
try {
Method method = beanObject.getBeanClass().getMethod("config");
method.invoke(beanObject.getObject());
} catch (Exception e) {
log.error("execute config method error!!", e);
}
}
}
}
}
//...
DbUtil原来只有自己使用，现在集成Mybatis后获取 DataSource部分可以放在这个类里，于是添加了一个dataSource 参数，及对应的getDataSource()方法，首次调用如果为空会去初始化：
package com.jisuye.util;
// import ...
/**
* 数据库操作工具
* @author ixx
* @date 2019-07-01
*/
public class DbUtil {
private static final Logger log = LoggerFactory.getLogger(DbUtil.class);
private static Connection connection;
private static DataSource dataSource;
/** 初始化方法*/
public static void init(){
try {
//...
HikariDataSource ds = new HikariDataSource();
//...
connection = ds.getConnection();
dataSource = ds;
} catch (Exception e) {
log.error("mysql connection init error..", e);
throw new SquareException("mysql connection init error....");
}
}
// ...
public static DataSource getDataSource(){
if (dataSource == null) {
init();
}
return dataSource;
}
}
05、修改square-demo项目，测试集成
首先添加pom依赖：
<dependency>
<groupId>org.mybatis</groupId>
<artifactId>mybatis</artifactId>
<version>3.5.2</version>
</dependency>
<dependency>
<groupId>com.jisuye</groupId>
<artifactId>square-mybatis</artifactId>
<version>0.1-SNAPSHOT</version>
</dependency>
添加InitConfig.java配置类，只是用来调用MybatisConfig.init()方法。
package com.jisuye.config;
// import ...
@Config
public class InitConfig implements SquareConfig {
@Override
public void config() {
MybatisConfig.init();
}
}
添加AbcMapper.java接口
package com.jisuye.mapper;
//import ...
/**
* 测试mapper
*/
@Mapper
public interface AbcMapper {
List<AbcEntity> selectAbc(int id);
}
在resource下添加mapper目录，添加AbcMapper.xml文件：
<?xml version="1.0" encoding="UTF-8" ?>
<!DOCTYPE mapper
PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
"http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="com.jisuye.mapper.AbcMapper">
<resultMap id="abc" type="com.jisuye.entity.AbcEntity">
<result column="id" javaType="Integer" property="id" />
<result column="name" javaType="java.lang.String" property="name" />
<result column="age" javaType="Integer" property="age" />
</resultMap>
<select id="selectAbc" resultMap="abc">
select * from abc where id = #{id}
</select>
</mapper>
application.yml中添加数据库相关配置
server:
port: 8765
servlet:
context-path: /square-demo
square:
datasource:
url: jdbc:mysql://localhost:3306/test?useUnicode=true&characterEncoding=utf8&useSSL=false&autoReconnect=true
username: root
password: 123456
HelloController 添加调用Mapper的方法：
package com.jisuye.controller;
//import ...
@Controller("/")
public class HelloController {
@Resource
private AbcMapper abcMapper;
@GetMapping("/id")
public String selectAbc(@RequestParam("id") int id) {
List<AbcEntity> list = abcMapper.selectAbc(id);
if(list != null && list.size()>0){
return "success! name is : " + list.get(0).getName();
} else {
return "error! select by db.";
}
}
@GetMapping("/hello")
public String hello(@RequestParam("name") String name){
return "hello "+name;
}
}
首先在IDE中启动程序测试，程序启动后访问：http://localhost:8765/square-demo/id?id=1
返回结果：success! name is : ixx
查看日志输出：
19:42:45.594 [http-nio-8765-exec-1] DEBUG org.apache.ibatis.transaction.jdbc.JdbcTransaction - Setting autocommit to false on JDBC Connection [HikariProxyConnection@2030223697 wrapping com.mysql.cj.jdbc.ConnectionImpl@d088afc]
19:42:45.604 [http-nio-8765-exec-1] DEBUG com.jisuye.mapper.AbcMapper.selectAbc - ==> Preparing: select * from abc where id = ?
19:42:45.643 [http-nio-8765-exec-1] DEBUG com.jisuye.mapper.AbcMapper.selectAbc - ==> Parameters: 1(Integer)
19:42:45.681 [http-nio-8765-exec-1] DEBUG com.jisuye.mapper.AbcMapper.selectAbc - <== Total: 1
19:42:45.681 [http-nio-8765-exec-1] INFO com.jisuye.core.DispatcherServlet - exec method ：selectAbc
19:42:45.681 [http-nio-8765-exec-1] INFO com.jisuye.core.DispatcherServlet - response:success! name is : ixx
ok !本地测试通过，再试一下打包。
在square-demo目录下执行mvn clean package 构建完毕后执行java -jar target/square-demo-1.0-SNAPSHOT.jar
等程序启动成功后访问：http://localhost:8765/square-demo/id?id=1
返回结果：success! name is : ixx
控制台也有上边同样的日志输出，说明打包启动也OK了。
06、遗留问题
集成这篇还算完事吧，至少该实现的都实现了，当然也相当于会对Mybatis插件做了些定制如果集成其它的功能应该还需要做一定的抽象工作。
上一篇遗留的最主要的问题就是静态文件的问题了，不知道下一篇会不会解决。。。
本篇代码地址: https://github.com/iuv/square/tree/square10
spring-mybatis地址: https://github.com/iuv/square-mybatis
演示项目地址: https://github.com/iuv/square-demo
本文作者： ixx
本文链接： http://jianpage.com/2019/09/09/square10
版权声明： 本作品采用 知识共享署名-非商业性使用-相同方式共享 4.0 国际许可协议 进行许可。转载请注明出处！
网友评论