手写Mybatis框架

分析JDBC操作问题

https://upload-images.jianshu.io/upload_images/19453215-b93db688d71a8474.png

问题总结

  1. 数据库连接创建、释放频繁造成系统资源浪费,从而影响系统性能。
  2. Sql语句在代码中硬编码,造成代码不易维护,实际应用中sql变化的可能较大,sql变动需要改变 java代码。
  3. 使用preparedStatement向占有位符号传参数存在硬编码,因为sql语句的where条件不一定,可能 多也可能少,修改sql还要修改代码,系统不易维护。
  4. 对结果集解析存在硬编码(查询列名),sql变化导致解析代码变化,系统不易维护,如果能将数据 库 记录封装成pojo对象解析比较方便

解决方案

  1. 使用数据库连接池初始化连接资源
  2. 将数据库连接配置、sql语句抽取到xml配置文件中, 使其满足开闭原则
  3. 使用反射、内省等底层技术,自动将实体与表进行属性与字段的自动映射

自定义框架设计思路

架构设计图

https://upload-images.jianshu.io/upload_images/19453215-35a00b64f4f80aea.png

设计思路

使用端(提供配置)

  1. 使用sqlMapperConfig.xml提供全局配置信息(数据源等)
  2. 使用Mapper.xml提供sql语句文件信息
  3. 提供SqlSession接口类实现select、insert、update、delete操作

框架端(JDBC封装)

  1. 读取配置文件
    • 读取sqlMapperconfig.xml
    • 读取Mapper.xml
  2. 解析配置文件
    • 解析文件生成Configuration、MapperStatement对象
  3. 执行JDBC流程

自定义框架实现

使用端

创建配置文件

sqlMapperConfig.xml(全局配置文件,包括数据源配置,mapper路径等)

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

<configuration>

<!--数据源配置-->

<datasource>

<property name="driverClass" value="com.mysql.cj.jdbc.Driver"/>

<property name="jdbcUrl" value="jdbc:mysql://localhost:3306/test"/>

<property name="username" value="root"/>

<property name="password" value=""/>

</datasource>

<!--引入mapper.xml文件路径-->

<mapper resource="mapper/userMapper.xml"/>

</configuration>

mapper.xml(namespace命名空间、SQL语句配置等)

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

<mapper namespace="com.gigabytc.dao.UserDao">

<select id="findAll" resultType="com.gigabytc.entity.User" >

select * from user

</select>

<select id="findByCondition" resultType="com.gigabytc.entity.User" parameterType="com.gigabytc.entity.User">

select * from user where id = #{id} and username = #{username}

</select>

<insert id="addUser" parameterType="com.gigabytc.entity.User" resultType="int">

insert into user values(#{id}, #{username}, #{password}, #{birthday})

</insert>

<update id="updateUser" parameterType="com.gigabytc.entity.User" resultType="int" >

update user SET username = #{username}, password=#{password}, birthday=#{birthday} WHERE id = #{id}

</update>

<delete id="deleteUser" parameterType="com.gigabytc.entity.User" resultType="int" >

delete from user where id=#{id}

</delete>

</mapper>

使用sqlSession操作接口(读取配置文件构建SqlSession数据库操作)

InputStream resourceAsStream = Resources.getResourceAsStream("sqlMapperConfig.xml");

SqlSessionFactory sqlSessionFactory = new SqlSessionFactoryBuilder().build(resourceAsStream);

sqlSession = sqlSessionFactory.openSession();

List<User> list = sqlSession.selectList("user.findAll");

框架端

解析配置文件

Configuration(全局配置)

@Data
public class Configuration {

/**
- 数据源
*/
private DataSource dataSource;

/**
- SQL statement集合
*/
private Map<String, MapperStatement> mapperStatementMap = new HashMap<>();

}

MapperStatement(SQL语句配置)

@Data
public class MapperStatement {

/**
    statement-id标识(namespace+id)
*/
private String id;

/**
- 参数类型
*/
private String parameterType;

/**
- 返回类型
*/
private String resultType;

/**
- sql语句
*/
private String sql;

/**
- sql类型
*/
private SqlType sqlType;

}

XmlConfigBuilder(全局配置解析)

public class XMLConfigBuilder {

public Configuration parseConfiguration(InputStream inputStream) throws DocumentException, PropertyVetoException {

Configuration configuration = new Configuration();

//解析数据源配置

Document document = new SAXReader().read(inputStream);

Element rootElement = document.getRootElement();

List<Element> list = rootElement.selectNodes("//datasource/property");

Properties properties = new Properties();

for (Element element : list) {

String name = element.attributeValue("name");

String value = element.attributeValue("value");

properties.setProperty(name,value);

}

//构建连接池

ComboPooledDataSource comboPooledDataSource = new ComboPooledDataSource();

comboPooledDataSource.setDriverClass(properties.getProperty("driverClass"));

comboPooledDataSource.setJdbcUrl(properties.getProperty("jdbcUrl"));

comboPooledDataSource.setUser(properties.getProperty("username"));

comboPooledDataSource.setPassword(properties.getProperty("password"));

configuration.setDataSource(comboPooledDataSource);

//读取mapper.xml文件路径,解析mapper.xml

List<Element> mapperList = rootElement.selectNodes("//mapper");

for (Element element : mapperList) {

String resource = element.attributeValue("resource");

XMLMapperBuilder xmlMapperBuilder = new XMLMapperBuilder(configuration);

xmlMapperBuilder.parseMapper(Resources.getResourceAsStream(resource));

}
return configuration;
}
}

XmlMapperStatementBuilder(SQL语句解析)

public class XMLMapperBuilder {

private Configuration configuration;

public XMLMapperBuilder(Configuration configuration) {

this.configuration = configuration;

}

public void parseMapper(InputStream inputStream) throws DocumentException {

//解析SQL statement并封装到集合

Document document = new SAXReader().read(inputStream);

Element rootElement = document.getRootElement();

//解析sql语句封装MapperStatement

buildStatement(rootElement, new String[]{"select", "update", "insert", "delete"});

}

private void parseStatement(String namespace, Element element) {

String id = element.attributeValue("id");

String parameterType = element.attributeValue("parameterType");

String resultType = element.attributeValue("resultType");

String sql = element.getTextTrim();

String elementName = element.getName();

SqlType sqlType = SqlType.valueOf(elementName.toUpperCase());

MapperStatement mapperStatement = new MapperStatement();

mapperStatement.setId(id);

mapperStatement.setParameterType(parameterType);

mapperStatement.setResultType(resultType);

mapperStatement.setSql(sql);

mapperStatement.setSqlType(sqlType);

String statementId = namespace + "." + id;

configuration.getMapperStatementMap().put(statementId, mapperStatement);

}

private void buildStatement(Element rootElement, String[] nodes) {

String namespace = rootElement.attributeValue("namespace");

for (String node : nodes) {

List<Element> list = rootElement.selectNodes("//" + node);

for (Element element : list) {

parseStatement(namespace, element);

}

}

}

}

SqlSession相关接口实现

SqlSessionFactoryBuilder

public class SqlSessionFactoryBuilder {

public SqlSessionFactory build(InputStream inputStream) throws PropertyVetoException, DocumentException {

XMLConfigBuilder xmlConfigBuilder = new XMLConfigBuilder();

Configuration configuration = xmlConfigBuilder.parseConfiguration(inputStream);

return new DefaultSqlSessionFactory(configuration);

}

}

DefaultSqlSessionFactory

public class DefaultSqlSessionFactory implements SqlSessionFactory{

private final Configuration configuration;

public DefaultSqlSessionFactory(Configuration configuration) {

this.configuration = configuration;

}

@Override

public SqlSession openSession() {

return new DefaultSqlSession(configuration);

}

}

DefaultSqlSession

public class DefaultSqlSession implements SqlSession{

public DefaultSqlSession(Configuration configuration) {

this.configuration = configuration;

}

private Configuration configuration;

@Override

public <E> List<E> selectList(String statementId, Object... params) throws Exception {

Executor selectExecutor = new SelectExecutor();

MapperStatement mapperStatement = configuration.getMapperStatementMap().get(statementId);

return selectExecutor.query(configuration, mapperStatement, params);

}

@Override

public <T> T selectOne(String statementId, Object... params) throws Exception {

List<Object> objects = selectList(statementId, params);

if (objects.size() == 1) {

return (T) objects.get(0);

}else {

throw new RuntimeException("查询结果过多或不存在");

}

}

@Override

public int insert(String statementId, Object... params) throws Exception {

return this.update(statementId,params);

}

@Override

public int update(String statementId, Object... params) throws Exception {

Executor updateExecutor = new UpdateExecutor();

MapperStatement mapperStatement = configuration.getMapperStatementMap().get(statementId);

return updateExecutor.update(configuration,mapperStatement,params);

}

@Override

public int delete(String statementId, Object... params) throws Exception {

return this.update(statementId,params);

}

@Override

public <T> T getMapper(Class<?> c) {

Object o = Proxy

.newProxyInstance(DefaultSqlSession.class.getClassLoader(), new Class[]{c}, new InvocationHandler() {

@Override

public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {

Class<?> declaringClass = method.getDeclaringClass();

String statementId = declaringClass.getName() + "." + method.getName();

MapperStatement mapperStatement = configuration.getMapperStatementMap().get(statementId);

SqlType sqlType = mapperStatement.getSqlType();

switch (sqlType) {

case SELECT:

Type genericReturnType = method.getGenericReturnType();

// 判断是否进行了 泛型类型参数化

if(genericReturnType instanceof ParameterizedType){

return selectList(statementId, args);

}

return selectOne(statementId,args);

case INSERT:

return insert(statementId,args);

case UPDATE:

return update(statementId,args);

case DELETE:

return delete(statementId,args);

default :

break;

}

return null;

}

});

return (T) o;

}

}

参考代码

lov3r/lagou_mybatis

本文由博客一文多发平台 OpenWrite 发布!

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 214,233评论 6 495
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,357评论 3 389
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 159,831评论 0 349
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,313评论 1 288
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,417评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,470评论 1 292
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,482评论 3 412
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,265评论 0 269
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,708评论 1 307
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,997评论 2 328
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,176评论 1 342
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,827评论 4 337
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,503评论 3 322
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,150评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,391评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,034评论 2 365
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,063评论 2 352

推荐阅读更多精彩内容