手撸MyBatis（二）模拟实现

DAO接口：

public interface UserDAO {

List findAll();

User findById(Integer id);

void addUser(User user);

void updateUser(User user);

void deleteUser(Integer id);

}

Mapper文件：

<?xml version="1.0" encoding="UTF-8" ?>

SELECT LAST_INSERT_ID()

INSERT INTO USER (USERNAME,BIRTHDAY,SEX,ADDRESS) VALUES (#{username},#{birthday},#{sex},#{address})

UPDATE USER set username=#{username},birthday=#{birthday},sex=#{sex},address=#{address} where id=#{id}

DELETE FROM USER where id=#{id}

SELECT * FROM USER where id=#{id}

select * from user

定义一个解析类，完成对数据源解析工作：

/**

配置文件的解析类

*/

public class ConfigParser {

//保存数据源信息

private Map<String,String> dataSource = new HashMap<>();

//保存映射文件路径

private List resourceMappers = new ArrayList<>();

//保存class映射类名

private List classMappers = new ArrayList<>();

/*配置解析/

public void parseConfig(InputStream inputStream) throws DocumentException {

SAXReader reader = new SAXReader();

reader.setEntityResolver(new EntityResolver() {

@Override

public InputSource resolveEntity(String publicId, String systemId) throws SAXException, IOException {

return new InputSource(new ByteArrayInputStream(“”.getBytes()));

}

});

Document doc = reader.read(inputStream);

Element rootElement = doc.getRootElement();

List properties = rootElement.element(“environments”).element(“environment”).element(“dataSource”).elements();

for(Element property : properties){

if(“driver”.equals(property.attributeValue(“name”))){

dataSource.put(“driver”,property.attributeValue(“value”));

}

if(“url”.equals(property.attributeValue(“name”))){

dataSource.put(“url”,property.attributeValue(“value”));

}

if(“username”.equals(property.attributeValue(“name”))){

dataSource.put(“username”,property.attributeValue(“value”));

}

if(“password”.equals(property.attributeValue(“name”))){

dataSource.put(“password”,property.attributeValue(“value”));

}

}

List mapperEles = rootElement.element(“mappers”).elements();

for(Element mapper : mapperEles){

if(mapper.attribute(“resource”) != null){

resourceMappers.add(mapper.attributeValue(“resource”));

}

if(mapper.attribute(“class”) != null){

classMappers.add(mapper.attributeValue(“class”));

}

}

}

public Map<String, String> getDataSource() {

return dataSource;

}

public List getResourceMappers() {

return resourceMappers;

}

public List getClassMappers() {

return classMappers;

}

}

测试：

@Test

public void testParse(){

ConfigParser parser = new ConfigParser();

try {

parser.parseConfig(Test2.class.getClassLoader().getResourceAsStream(“mock-config.xml”));

System.out.println(parser.getDataSource());

System.out.println(parser.getClassMappers());

System.out.println(parser.getResourceMappers());

} catch (DocumentException e) {

e.printStackTrace();

}

}

编写一个类，保存Mapper方法相关的信息

/**

• SQL映射类

*/

public class Mapper {

//sql类型 insert update delete select

private String sqlType;

//Mapper方法名

private String methodName;

//sql语句

private String sql;

//参数类型

private String paramType;

//返回类型

private String returnType;

…

}

定义一个解析类，完成对映射文件的解析工作：

/**

映射文件的解析类

*/

public class MapperParser {

private Map<String,Mapper> mappers = new HashMap<>();

public void parseMappers(InputStream inputStream) throws DocumentException {

SAXReader reader = new SAXReader();

reader.setEntityResolver(new EntityResolver() {

@Override

public InputSource resolveEntity(String publicId, String systemId) throws SAXException, IOException {

return new InputSource(new ByteArrayInputStream(“”.getBytes()));

}

});

Document doc = reader.read(inputStream);

Element root = doc.getRootElement();

List elements = root.elements();

String namespace = root.attributeValue(“namespace”);

for(Element e : elements){

Mapper mapper = new Mapper();

mapper.setSqlType(e.getName());

mapper.setMethodName(e.attributeValue(“id”));

mapper.setParamType(e.attributeValue(“parameterType”));

mapper.setReturnType(e.attributeValue(“resultType”));

mapper.setSql(e.getText());

mappers.put(namespace+“.”+e.attributeValue(“id”),mapper);

}

}

public Map<String, Mapper> getMappers() {

return mappers;

}

}

测试：

@Test

public void testParse(){

ConfigParser parser = new ConfigParser();

MapperParser mParser = new MapperParser();

try {

parser.parseConfig(Test2.class.getClassLoader().getResourceAsStream(“mock-config.xml”));

System.out.println(parser.getDataSource());

System.out.println(parser.getClassMappers());

System.out.println(parser.getResourceMappers());

for(String resource : parser.getResourceMappers()){

System.out.println(“resourc:”+resource);

mParser.parseMappers(Test2.class.getClassLoader().getResourceAsStream(resource));

System.out.println(mParser.getMappers());

}

} catch (DocumentException e) {

e.printStackTrace();

}

}

编写模拟SQLSession的类，通过动态代理获得Mapper对象

/**

• 模拟SqlSession

*/

public interface MySqlSession {

/**

• 返回Mapper对象

• @param clazz

• @param

• @return

*/

T getMapper(Class clazz);

}

public class MySqlSessionImpl implements MySqlSession {

private Map<String,String> dataSource;

private Map<String,Mapper> mappers;

public MySqlSessionImpl(Map<String, String> dataSource, Map<String, Mapper> mappers) {

this.dataSource = dataSource;

this.mappers = mappers;

}

@Override

public T getMapper(Class clazz) {

XMLMapperProxy xmlMapperProxy = new XMLMapperProxy(dataSource,mappers,clazz);

return (T) xmlMapperProxy.getProxy();

}

}

动态代理的实现类

/**

• XML映射文件的代理

*/

public class XMLMapperProxy implements InvocationHandler{

private Map<String,String> dataSource;

private Map<String,Mapper> mappers;

private Class<?> clazz;
自我介绍一下，小编13年上海交大毕业，曾经在小公司待过，也去过华为、OPPO等大厂，18年进入阿里一直到现在。

深知大多数Java工程师，想要提升技能，往往是自己摸索成长或者是报班学习，但对于培训机构动则几千的学费，着实压力不小。自己不成体系的自学效果低效又漫长，而且极易碰到天花板技术停滞不前！

因此收集整理了一份《2024年Java开发全套学习资料》，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友，同时减轻大家的负担。

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，基本涵盖了95%以上Java开发知识点，真正体系化！

由于文件比较大，这里只是将部分目录截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频，并且会持续更新！

如果你觉得这些内容对你有帮助，可以扫码获取！！（备注Java获取）

读者福利

更多笔记分享

《一线大厂Java面试题解析+核心总结学习笔记+最新讲解视频+实战项目源码》，点击传送门即可获取！
3411906)]

[外链图片转存中…(img-H2g4P4tm-1712443411906)]

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，基本涵盖了95%以上Java开发知识点，真正体系化！

由于文件比较大，这里只是将部分目录截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频，并且会持续更新！

如果你觉得这些内容对你有帮助，可以扫码获取！！（备注Java获取）

读者福利

[外链图片转存中…(img-e7FiKiHv-1712443411907)]

更多笔记分享

[外链图片转存中…(img-fNUGG9WN-1712443411907)]

《一线大厂Java面试题解析+核心总结学习笔记+最新讲解视频+实战项目源码》，点击传送门即可获取！