自定义简易版mybatis
说明:在自定义的mybatis框架中,使用了spring中的一些内容,例如spring-jdbc、spring-tx等等,除此之外,连接池使用了阿里巴巴的druid,由于某些帮助类是使用之前用到的,导致配置文件的读取位置与原始mybatis中配置文件的读取位置和方式不同。在自定义的mybatis中,只提供了原始mybatis的简单功能实现。
核心代码
package com.yc.mavenstudy.core;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.dom4j.Element;
import org.dom4j.XPath;
import org.dom4j.io.SAXReader;
import com.yc.mavenstudy.domain.MapperInfo;
/**
* 解析配置文件的类
* @author : 外哥
* 邮箱 : liwai2012220663@163.com
* 创建时间:2021年1月8日 下午8:25:10
*/
@SuppressWarnings("all")
public class MyBatisConfig {
private Map<String , String> dataSourceMap = new HashMap<>() ;
private List<String> mappers = new ArrayList<>() ;
private static Map<String, MapperInfo> mapperInfos = new HashMap<>() ;
public MyBatisConfig( String config) {
// 解析主配置文件
parseXML( config ) ;
// 解析映射文件
parseMapper() ;
}
/**
* 解析映射文件
*/
private void parseMapper() {
if ( mappers == null || mappers.isEmpty() ) {
// 如果没有映射文件
return ;
}
try {
SAXReader reader = null ;
Document doc = null ;
String namespace = "" ;
MapperInfo mapperInfo = null ;
String sql = null ;
String nodeName = null ;
List<Element> nodes = null ;
Pattern pattern = null ;
Matcher matcher = null ;
List<String> paramNames = null ;
for (String mapper : mappers) {
InputStream is = this.getClass().getClassLoader().getResourceAsStream(mapper);
reader = new SAXReader() ;
doc = reader.read(is) ;
// 获取命名空间
namespace = doc.getRootElement().attributeValue("namespace") ;
if ( namespace != null && !"".equals(namespace)) {
// 命名空间不为空
namespace += "." ;
}
// 获取mappers节点下的所有子节点
nodes = doc.selectNodes("/mapper/*") ;
// 遍历所有节点
for (Element el : nodes) {
mapperInfo = new MapperInfo() ;
// 获取节点名 select/insert等
nodeName = el.getName() ;
if ( "select".equalsIgnoreCase(nodeName)) {
// 表示这是一个查询语句
mapperInfo.setUpdate(false);
}
// 设置参数类型
mapperInfo.setParameterType( el.attributeValue("parameterType") );
// 设置返回值类型
mapperInfo.setResultType( el.attributeValue("resultType"));
// 获取sql语句
sql = el.getTextTrim() ;
// 创建一个正则表达式
pattern = Pattern.compile("[#][{]\\w+}") ;
// 匹配sql语句
matcher = pattern.matcher(sql) ;
paramNames = new ArrayList<>() ;
// 迭代所有匹配到的结果
while ( matcher.find() ) {
// 将匹配到的结果中的#{}符号全部替换成空
paramNames.add( matcher.group().replaceAll("[#{}]*", "")) ;
}
// 设置参数列表
mapperInfo.setParamNames(paramNames);
// 将sql语句中匹配成功的地方全部替换成?
sql = matcher.replaceAll("?") ;
// 设置sql语句
mapperInfo.setSql(sql);
// 将这个节点的信息添加到集合中
mapperInfos.put( namespace + el.attributeValue("id"), mapperInfo) ;
}
}
} catch (DocumentException e) {
e.printStackTrace();
}
}
/**
* 解析xml文件
* @param config
*/
private void parseXML(String config) {
SAXReader reader = new SAXReader() ;
InputStream is = this.getClass().getClassLoader().getResourceAsStream(config);
Document doc = null ;
try {
doc = reader.read(is) ;
List<Element> list = doc.selectNodes("//dataSource/property");
for (Element el : list) {
dataSourceMap.put(el.attributeValue("name"), el.attributeValue("value")) ;
}
list = doc.selectNodes("//mappers/mapper") ;
for (Element el : list) {
mappers.add(el.attributeValue("resource")) ;
}
} catch (DocumentException e) {
e.printStackTrace();
}
}
public Map<String, String> getDataSourceMap() {
return dataSourceMap;
}
public List<String> getMappers() {
return mappers;
}
public static MapperInfo getMapperInfo( String sqlId) {
return mapperInfos.get(sqlId);
}
}
源代码下载https://download.csdn.net/download/Simple__Code/14090580