手写Mybatis源码(原来真的很简朴!!!)

打印 上一主题 下一主题

主题 866|帖子 866|积分 2598


一、JDBC操纵数据库_问题分析

   JDBC使用流程
  

  • 加载数据库驱动
  • 创建数据库毗连
  • 创建编译对象
  • 设置入参实行SQL
  • 返回效果集
   代码示例
  1. public class JDBCTest {
  2.     public static void main(String[] args) throws Exception {
  3.         Connection connection = null;
  4.         PreparedStatement preparedStatement = null;
  5.         ResultSet resultSet = null;
  6.         try {
  7.             // 加载数据库驱动
  8.             Class.forName("com.mysql.jdbc.Driver");
  9.             // 通过驱动管理类获取数据库链接
  10.             connection = DriverManager.getConnection("jdbc:mysql://localhost:3306/mybatis",
  11.             "root","123456789");
  12.             // 定义sql语句?表示占位符
  13.             String sql = "select * from user where name = ?";
  14.             // 获取预处理statement
  15.             preparedStatement = connection.prepareStatement(sql);
  16.             // 设置参数,第一个参数为sql语句中参数的序号(从1开始),第二个参数为设置的参数值
  17.             preparedStatement.setString(1, "zhangsan");
  18.             // 向数据库发出sql执行查询,查询出结果集
  19.             resultSet = preparedStatement.executeQuery();
  20.             // 遍历查询结果集
  21.             while (resultSet.next()) {
  22.                 int id = resultSet.getInt("id");
  23.                 String username = resultSet.getString("name");
  24.                 // 封装User
  25.                 User user = new User();
  26.                 user.setId(id);
  27.                 user.setName(username);
  28.                 System.out.println(user);
  29.             }
  30.         } catch (Exception e) {
  31.             e.printStackTrace();
  32.         } finally {
  33.             // 释放资源
  34.             if (resultSet != null) {
  35.                 try {
  36.                     resultSet.close();
  37.                 } catch (SQLException e) {
  38.                     e.printStackTrace();
  39.                 }
  40.             }
  41.             if (preparedStatement != null) {
  42.                 try {
  43.                     preparedStatement.close();
  44.                 } catch (SQLException e) {
  45.                     e.printStackTrace();
  46.                 }
  47.             }
  48.             if (connection != null) {
  49.                 try {
  50.                     connection.close();
  51.                 } catch (SQLException e) {
  52.                     e.printStackTrace();
  53.                 }
  54.             }
  55.         }
  56.     }
  57. }
复制代码
  问题分析
  存在问题解决思绪数据库配置信息存在硬编码问题使用配置文件频仍创建、开释数据库毗连问题使用数据库毗连池SQL语句使用配置文件数据库配置信息存在硬编码问题使用配置文件 二、自界说恒久层框架_思绪分析

主要分两部分,项目使用端:平常写代码所说的背景服务;恒久层框架:即项目使用端引入的jar包

核心接口/类重点说明:
类名界说角色定位分工协作Resources资源辅助类负责读取配置文件转化为输入流Configuration数据库资源类负责存储数据库毗连信息MappedStatementSQL与效果集资源类负责存储SQL映射界说、存储效果集映射界说SqlSessionFactoryBuilder会话工厂构建者负责解析配置文件,创建会话工厂SqlSessionFactorySqlSessionFactory会话工厂负责创建会话SqlSessionSqlSession会话指派实行器ExecutorExecutor实行器负责实行SQL (共同指定资源Mapped Statement) 项目使用端:

  • 引入自界说恒久层框架的jar包
  • sqlMapConfig.xml:数据库配置信息,以及mapper.xml的全路径
  • mapper.xml:SQL配置信息,存放SQL语句、参数范例、返回值范例相关信息
    留意: sqlMapConfig.xml中引入mapper.xml是为了只读取一次配置文件,否则每个实体类会有一个mapper.xml,则需要读取很多次
  自界说框架本身:

  • 加载配置文件:根据配置文件的路径,加载配置文件成字节输入流,存储在内存中


  • 创建两个javaBean(容器对象):存放配置文件解析出来的内容


  • 解析配置文件(使用dom4j),并创建SqlSession会话对象


  • 创建SqlSessionFactory接口以及实现类DefaultSqlSessionFactory


  • 创建SqlSession接口以及实现类DefaultSqlSession

SqlSession接口界说以上方法,DefaultSqlSession来决定什么操纵调用对应的sql实行器

  • 创建Executor实行器接口以及实现类SimpleExecutor简朴实行器

三、自界说框架_编码

   项目使用端
  创建sqlMapConfig核心配置文件:
  1. <configuration>
  2.     <!--配置数据库信息-->
  3.     <dataSource>
  4.         <property name="driverClassname" value="com.mysql.jdbc.Driver" />
  5.         <property name="url" value="jdbc:mysql://localhost:3306/mybatis" />
  6.         <property name="username" value="root" />
  7.         <property name="password" value="123456789" />
  8.     </dataSource>
  9.     <!--引入映射配置文件,为了只加载一次xml-->
  10.     <mappers>
  11.         <mapper resource="mapper/UserMapper.xml"/>
  12.     </mappers>
  13. </configuration>
复制代码
创建映射配置文件:
获取某个sql语句的唯一标示statementId:namespace.id 如:user.selectList
  1. <mapper namespace="user">
  2.     <!--查询所有-->
  3.     <select id="selectList" resultType="com.xc.pojo.User" >
  4.         select * from user
  5.     </select>
  6.     <!--按条件查询-->
  7.     <select id="selectOne" resultType="com.xc.pojo.User" parameterType="com.xc.pojo.User" >
  8.         select * from user where id = #{id} and name = #{name}
  9.     </select>
  10. </mapper>
复制代码
User实体:
  1. @Data
  2. public class User {
  3.     private Integer id;
  4.     private String name;
  5. }
复制代码
pom.xml引入自界说框架
  1. <dependency>
  2.      <groupId>com.xc</groupId>
  3.      <artifactId>own-mybatis</artifactId>
  4.      <version>1.0-SNAPSHOT</version>
  5. </dependency>
复制代码
  自界说框架本身
  1、加载配置文件



  • 根据配置文件的路径,加载配置文件成字节输入流,存到内存中
  1. public class Resources {
  2.     public static InputStream getResourceAsStream(String path){
  3.         return Resources.class.getClassLoader().getResourceAsStream(path);
  4.     }
  5. }
复制代码
2、创建两个配置类对象



  • 映射配置类:mapper.xml解析出来内容
  • 每个pojo实体都会对应一个mapper.xml文件即一个MapperStatement对象
  • sqlCommandType:第四节 自界说框架_优化才会用到
  1. @Data
  2. public class MapperStatement {
  3.     //唯一标识 statementId:namespace.id
  4.     private String statementId;
  5.     //返回值类型
  6.     private String resultType;
  7.     //参数类型
  8.     private String parameterType;
  9.     //sql语句
  10.     private String sql;
  11.    
  12.     // 判断当前是什么操作的一个属性-增删改查
  13.     private String sqlCommandType;
  14. }
复制代码


  • 核心配置类:数据库配置信息以及映射配置类的map集合
  • 将多个MapperStatement对象存入Map集合,statementId(namespace.id)作为key
  • 将所有的配置文件都聚合到Configuration中,方便一次读取以及统一管理
  1. @Data
  2. public class Configuration {
  3.     //数据源对象
  4.     private DataSource dataSource;
  5.     //map.xml对象集合 key:statementId
  6.     private Map<String,MapperStatement> mapperStatementMap = new HashMap<>();
  7. }
复制代码
3、解析配置文件,填充配置类对象



  • XMLConfigBuilder类的parse方法:解析核心配置类,返回Configuration对象
  • 创建SqlSession工厂对象,以便之后创建SqlSession会话
  1. public class SqlSessionFactoryBuilder {
  2.     public SqlSessionFactory build(InputStream inputStream) throws Exception {
  3.         //1.解析配置文件,封装容器对象:专门解析核心配置文件的解析类
  4.         XMLConfigBuilder xmlConfigBuilder = new XMLConfigBuilder();
  5.         Configuration configuration = xmlConfigBuilder.parse(inputStream);
  6.         //2.创建SqlSessionFactory工厂对象
  7.         return new DefaultSqlSessionFactory(configuration);
  8.     }
  9. }
复制代码


  • XMLConfigBuilder核心配置解析类内里嵌套着XMLMapperBuilder映射配置文件解析类
  • 输入流转化为Document对象,一是根据property标签获取数据库配置信息并创建数据源添加到configuration
  • 二是根据mapper标签通过XMLMapperBuilder解析类遍历解析配置文件同样添加到configuration的map集合类
  1. public class XMLConfigBuilder {
  2.     private Configuration configuration;
  3.     public XMLConfigBuilder() {
  4.         this.configuration = new Configuration();
  5.     }
  6.         // 使用dom4j+xpath解析
  7.     public Configuration parse(InputStream inputStream) throws Exception {
  8.         //将xml转化为Document对象
  9.         Document document = new SAXReader().read(inputStream);
  10.         //获取跟节点,对于sqlMapConfig.xml来说就是<Configuration>标签
  11.         Element rootElement = document.getRootElement();
  12.         // -------------解析数据库配置文件----------------
  13.         // "//"表示从匹配选择的当前节点,而不考虑它们的位置
  14.         // 即这里获取数据源url用户密码信息
  15.         // 例:<property name="driverClassName" value="com.mysql.jdbc.Driver"></property>
  16.         List<Element> propertyList = rootElement.selectNodes("//property");
  17.         Properties properties = new Properties();
  18.         for (Element element : propertyList) {
  19.             // 获取<property>标签中,name和value属性的值
  20.             String name = element.attributeValue("name");
  21.             String value = element.attributeValue("value");
  22.             properties.setProperty(name,value);
  23.         }
  24.         // 创建数据源对象
  25.         DruidDataSource dataSource = new DruidDataSource();
  26.         dataSource.setDriverClassName(properties.getProperty("driverClassName"));
  27.         dataSource.setUrl(properties.getProperty("url"));
  28.         dataSource.setUsername(properties.getProperty("username"));
  29.         dataSource.setPassword(properties.getProperty("password"));
  30.         // 将创建好的数据源添加到Configuration对象中
  31.         configuration.setDataSource(dataSource);
  32.         // -------------解析映射配置文件----------------
  33.         /*
  34.         1.获取映射配置文件路径
  35.         2.根据路径进行映射文件的加载解析
  36.         3.封装到MapperStatement,存入configuration的map集合中
  37.         */
  38.         // 例:<mapper resource="mapper/UserMapper.xml"></mapper>
  39.         List<Element> mapperList = rootElement.selectNodes("//mapper");
  40.         for (Element element : mapperList) {
  41.             String resource = element.attributeValue("resource");
  42.             InputStream resourceAsStream = Resources.getResourceAsStream(resource);
  43.             // XMLMapperBuilder 专门解析映射配置文件的对象-最后会存入configuration的map集合对象中
  44.             XMLMapperBuilder xmlMapperBuilder = new XMLMapperBuilder(configuration);
  45.             xmlMapperBuilder.parse(resourceAsStream);
  46.         }
  47.         return configuration;
  48.     }
  49. }
复制代码


  • 与XMLConfigBuilder解析类原理一样
  • 传入configuration,并将解析好的MapperStatement对象添加到mapperStatementMap
  1. public class XMLMapperBuilder {
  2.     private Configuration configuration;
  3.     public XMLMapperBuilder(Configuration configuration) {
  4.         this.configuration = configuration;
  5.     }
  6.     public void parse(InputStream resourceAsStream) throws Exception {
  7.         // 将输入流转化为Document对象,并获取跟节点<mapper>
  8.         Document document = new SAXReader().read(resourceAsStream);
  9.         Element rootElement = document.getRootElement();
  10.         // 例:<mapper namespace="user">
  11.         String namespace = rootElement.attributeValue("namespace");
  12.         /* 例:
  13.             <select id="selectOne" resultType="com.xc.pojo.User" parameterType="com.xc.pojo.User" >
  14.                 select * from user where id = #{id} and name = #{name}
  15.             </select>
  16.         */
  17.         List<Element> selectList = rootElement.selectNodes("//select");
  18.         for (Element element : selectList) {
  19.             String id = element.attributeValue("id");
  20.             String resultType = element.attributeValue("resultType");
  21.             String parameterType = element.attributeValue("parameterType");
  22.             String sql = element.getTextTrim();
  23.             // 封装MapperStatement对象
  24.             MapperStatement mapperStatement = new MapperStatement();
  25.             String statementId = namespace + "." + id;
  26.             mapperStatement.setStatementId(statementId);
  27.             mapperStatement.setParameterType(parameterType);
  28.             mapperStatement.setResultType(resultType);
  29.             mapperStatement.setSql(sql);
  30.             //第四节 自定义框架_优化才会用到
  31.             mapperStatement.setSqlCommandType("select");
  32.             // 添加到configurations的map集合中
  33.             configuration.getMapperStatementMap().put(statementId,mapperStatement);
  34.         }
  35.     }
  36. }
复制代码
4、创建SqlSessionFactory工厂接口及DefaultSqlSessionFactory实现类



  • 为了创建SqlSession会话,调用增删改查方法
  1. public interface SqlSessionFactory {
  2.     // 创建SqlSession对象
  3.     SqlSession openSession();
  4. }
复制代码


  • 创建简朴实行器,与核心配置类共同创建SqlSession会话实现类
  • configuration提供数据配置和sql以及参数和效果集封装
  • simpleExecutor提供JDBC实行sql底层原理
  1. public class DefaultSqlSessionFactory implements SqlSessionFactory {
  2.     private Configuration configuration;
  3.     public DefaultSqlSessionFactory(Configuration configuration) {
  4.         this.configuration = configuration;
  5.     }
  6.     @Override
  7.     public SqlSession openSession() {
  8.         // 1.创建执行器对象-具体的包装jdbc的sql操作,关闭连接等
  9.         Executor simpleExecutor = new SimpleExecutor();
  10.         // 2.创建sqlSession对象-判断执行增删改查哪些操作等
  11.         return new DefaultSqlSession(configuration,simpleExecutor);
  12.     }
  13. }
复制代码
5、创建SqlSession会话接口及DefaultSqlSession实现类



  • statementId(“namespace.id”):定位具体Mapper.xml的sql语句以及入参和返回
  • param:替换sql语句中的占位符?,可能字符串、对象、Map、集合
  1. public interface SqlSession {
  2.     // 查询多个结果
  3.     <E> List<E> selectList(String statementId, Object param) throws Exception;
  4.     // 查询单个结果
  5.     <T> T selectOne(String statementId, Object param) throws Exception;
  6.     // 清理资源
  7.     void close();
  8.    
  9. }
复制代码


  • 使用聚合进来的configuration对象获取MapperStatement映射配置对象向下传给实行器
  • 另外一个聚合进来的executor简朴实行器来实行底层JDBC操纵
  • DefaultSqlSession的作用则是聚合配置类分发到不同实行器的不同方法
  • 实行器种类:简朴实行器、可重用实行器、批量实行器(这里只模仿第一种)
  1. public class DefaultSqlSession implements SqlSession {
  2.     private Configuration configuration;
  3.     private Executor executor;
  4.     public DefaultSqlSession(Configuration configuration, Executor executor) {
  5.         this.configuration = configuration;
  6.         this.executor = executor;
  7.     }
  8.     @Override
  9.     public <E> List<E> selectList(String statementId, Object param) throws Exception {
  10.         // 根据StatementId获取映射配置对象MapperStatement
  11.         MapperStatement mapperStatement = configuration.getMapperStatementMap().get(statementId);
  12.         // 然后将具体的查询操作委派给SimpleExecutor执行器
  13.         // 执行底层jdbc需要:1.数据库配置,2.sql配置信息
  14.         return executor.query(configuration,mapperStatement,param);
  15.     }
  16.     @Override
  17.     public <T> T selectOne(String statementId, Object param) throws Exception {
  18.         // 调用selectList()
  19.         List<Object> selectList = selectList(statementId, param);
  20.         if (selectList.size() == 1){
  21.             return (T) selectList.get(0);
  22.         }else if (selectList.size() > 1){
  23.             throw new Exception("返回数据不止一条!!!");
  24.         }else {
  25.             return null;
  26.         }
  27.     }
  28.     @Override
  29.     public void close() {
  30.         executor.close();
  31.     }
  32. }
复制代码
6、创建Executor实行器接口及SimpleExecutor实现类



  • 实行器接口界说增删改查方法,具体的JDBC底层操纵由它的实现类来完成
  1. public interface Executor {
  2.     <E> List<E> query(Configuration configuration, MapperStatement mapperStatement, Object param)
  3.             throws Exception;
  4.     void close();
  5. }
复制代码


  • 实行器实现类团体流程就是JDBC那一套,从加载驱动随处理效果集
  • getBoundSql方法功能:

    • 一是将 <select>标签的sql语句“#{字段名}”替换成?赋值给finalSql。(这里使用的mybatis代码,不消深究就是根据正则表达式替换,最后有gitee代码链接)
    • 二是将替换?时间的字段名取出来放到ParameterMapping对象中,有多个,根据?次序,放入parameterMappingList集合

  • 入参根据<select>标签的parameterType属性获取全限定类名,反射获取Class对象

    • 遍历parameterMappingList,获取字段名,加上Class对象获取Field属性类
    • query查询方法有个param参数,即入参对象(有可能字符串,集合这里只考虑对象),通过Field属性和param对象通过反射获取属性值

  • 效果集根据<select>标签的resultType属性获取全限定类名,反射获取实例对象

    • 通过效果集元数据获取字段名columnName,再resultSet.getObject获取数据库中对应字段值
    • 通过java实体字段名和Class对象获取对应字段的Get方法的Method
    • 这里说下1和2步骤中两个字段不是一回事,如果数据库和实体中不一样,这里需要转化成一致

  1. public class SimpleExecutor implements Executor{
  2.     private Connection connection = null;
  3.     private PreparedStatement preparedStatement = null;
  4.     private ResultSet resultSet = null;
  5.     @Override
  6.     public <E> List<E> query(Configuration configuration, MapperStatement mapperStatement, Object param)
  7.                     throws Exception {
  8.         // 1.加载驱动,获取数据库连接
  9.         connection = configuration.getDataSource().getConnection();
  10.         // 2.获取preparedStatement预编译对象
  11.         // 从mapperStatement中获取sql语句
  12.         // 例:select * from user where id = #{id} and username = #{username}
  13.         String sql = mapperStatement.getSql();
  14.         // 1)需要转化为:select * from user where id = ? and username = ?
  15.         // 2)需要将替换的值保存下来,后续?赋值需要用到
  16.         BoundSql boundSql = getBoundSql(sql);
  17.         String finalSql = boundSql.getFinalSql();
  18.         preparedStatement = connection.prepareStatement(finalSql);
  19.         // 3.根据入参赋值?
  20.         // 获取入参的全限定类名 com.xc.pojo.User
  21.         String parameterType = mapperStatement.getParameterType();
  22.         // 没有入参类型,表示sql没有参数,这里不需要?赋值
  23.         if (parameterType != null){
  24.             // 入参对象Class类
  25.             Class<?> parameterTypeClass = Class.forName(parameterType);
  26.             // 获取#{}参数字段list
  27.             List<ParameterMapping> parameterMappingList = boundSql.getParameterMappingList();
  28.             for (int i = 0; i < parameterMappingList.size(); i++) {
  29.                 // 获取字段名称
  30.                 ParameterMapping parameterMapping = parameterMappingList.get(i);
  31.                 String paramName = parameterMapping.getContent();
  32.                 // 通过反射获取入参对象的Field属性
  33.                 Field field = parameterTypeClass.getDeclaredField(paramName);
  34.                 // 禁用安全检查,不用排除private,效率提升好多倍
  35.                 field.setAccessible(true);
  36.                 // param为User对象,通过属性反射获取到user对象id和name的属性值
  37.                 Object value = field.get(param);
  38.                 // 赋值占位符,占位符?数字从1开始
  39.                 preparedStatement.setObject(i + 1, value);
  40.             }
  41.         }
  42.         // 4.执行sql,发起查询
  43.         resultSet = preparedStatement.executeQuery();
  44.         // 5.处理返回结果集
  45.         List<E> list = new ArrayList<>();
  46.         while (resultSet.next()){
  47.             // 通过<select>标签的返回值类型,创建返回对象
  48.             String resultType = mapperStatement.getResultType();
  49.             Class<?> resultTypeClass = Class.forName(resultType);
  50.             Object obj = resultTypeClass.newInstance();
  51.             // 结果集的元数据信息:字段名,字段值等
  52.             // resultSet: 一条结果集对应一张表的所有字段
  53.             ResultSetMetaData metaData = resultSet.getMetaData();
  54.             for (int i = 1; i <= metaData.getColumnCount(); i++) {
  55.                 //数据库字段名
  56.                 String columnName = metaData.getColumnName(i);
  57.                 // 字段的值
  58.                 Object value = resultSet.getObject(columnName);
  59.                 // columnName:数据库字段,而下方需要实体中的字段,如果两边不一样,则这需要有一个转化
  60.                 // 获取读写方法即get、set方法
  61.                 PropertyDescriptor propertyDescriptor = new PropertyDescriptor(columnName, resultTypeClass);
  62.                 // 获取读方法-字段的set方法
  63.                 Method writeMethod = propertyDescriptor.getWriteMethod();
  64.                 // 反射为返回对象赋值
  65.                 // 参数1:实例对象 参数2:要设置的值
  66.                 writeMethod.invoke(obj,value);
  67.             }
  68.             list.add((E) obj);
  69.         }
  70.         return list;
  71.     }
  72.     /**
  73.      * @Description 获取?占位符的sql,以及保存#{}中的字段名称
  74.      *
  75.      * 1、将<select>标签中的sql的 "#{字段名}" 整个部分替换成?,赋值 finalSql
  76.      * 2、截取#{}中的字段名称,添加到ParameterMapping对象的content属性,多个,赋值List<ParameterMapping>
  77.      *
  78.      * @Author xuchang
  79.      * @Date 2022/10/19 16:44:51
  80.      */
  81.     private BoundSql getBoundSql(String sql) {
  82.         // 1.创建标记处理器:配合标记解析器完成标记的处理解析工作
  83.         ParameterMappingTokenHandler parameterMappingTokenHandler = new ParameterMappingTokenHandler();
  84.         // 2.创建标记解析器
  85.         GenericTokenParser genericTokenParser = new GenericTokenParser("#{", "}", parameterMappingTokenHandler);
  86.         // #{}占位符替换成? 2.解析替换的过程中 将#{}里面的值保存下来 ParameterMapping
  87.         String finalSql = genericTokenParser.parse(sql);
  88.         // #{}里面的值的一个集合 id username
  89.         List<ParameterMapping> parameterMappings = parameterMappingTokenHandler.getParameterMappings();
  90.         return new BoundSql(finalSql, parameterMappings);
  91.     }
  92.     @Override
  93.     public void close() {
  94.         // 释放资源
  95.         if (resultSet != null) {
  96.             try {
  97.                 resultSet.close();
  98.             } catch (SQLException e) {
  99.                 e.printStackTrace();
  100.             }
  101.         }
  102.         if (preparedStatement != null) {
  103.             try {
  104.                 preparedStatement.close();
  105.             } catch (SQLException e) {
  106.                 e.printStackTrace();
  107.             }
  108.         }
  109.         if (connection != null) {
  110.             try {
  111.                 connection.close();
  112.             } catch (SQLException e) {
  113.                 e.printStackTrace();
  114.             }
  115.         }
  116.     }
  117. }
复制代码
  1. @Data
  2. @AllArgsConstructor
  3. public class BoundSql {
  4.     // 用?做占位符的sql语句
  5.     private String finalSql;
  6.     // 字段名称的集合
  7.     private List<ParameterMapping> parameterMappingList;
  8. }
复制代码
  1. @Data
  2. @AllArgsConstructor
  3. public class ParameterMapping {
  4.     // 保存#{}中对于的字段名称
  5.     private String content;
  6. }
复制代码
7、项目使用端使用自界说框架测试



  • 优点:完成一张表的增删改查只需要创建一个实体类。(其他类接口都是必须要有的)
  • 缺点1:sqlSession会话对象创建频仍,每次需要User的CRUD,就需要创建
  • 缺点2:调用方法需要手动添加参数statementId:namespace.id
  1. @Test
  2. public void test1() throws Exception {
  3.     // 1.根据配置文件的路径,加载成字节输入流,存到内存中 注意:配置文件还未解析
  4.     InputStream inputStream = Resources.getResourceAsStream("sqlMapConfig.xml");
  5.     // 2.解析配置文件,封装到Configuration对象,创建sqlSessionFactory工厂对象
  6.     SqlSessionFactory sqlSessionFactory = new SqlSessionFactoryBuilder().build(inputStream);
  7.     // 3.生产sqlSession 创建执行器对象
  8.     SqlSession sqlSession = sqlSessionFactory.openSession();
  9.     // 4.调用sqlSession方法
  10.     User user = new User();
  11.     user.setId(100);
  12.     user.setName("zhangsan");
  13.     User userOne = sqlSession.selectOne("user.selectOne", user);
  14.     System.out.println("查询单个用户:"+userOne);
  15.     List<User> userList = sqlSession.selectList("user.selectList",null);
  16.     System.out.println("查询所有用户:"+userList.toString());
  17.     // 5.释放资源
  18.     sqlSession.close();
  19. }
复制代码
输出效果:
  1. 查询单个用户:User(id=100, name=zhangsan)
  2. 查询所有用户:[User(id=100, name=zhangsan), User(id=120, name=Lisi)]
复制代码
四、自界说框架_优化

1、优化思绪



  • 去掉statementId硬编码:创建一个UserMapper或IUserDao接口,每个方法名、入参、返回值和mapper.xml的<select>标签的id、parameterType、resultType一一对应
  • sqlSession创建频仍:通过动态署理创建IUserDao的实现类,内容则是sqlSession.selectOne和sqlSession.selectList;这样sqlSession只需要创建一次,以后每次需要User的CRUD,则调用署理对象对应方法即可
2、优化代码

在sqlSession中添加方法
  1. public interface SqlSession {
  2.         ...
  3.     // 生成代理对象
  4.     <T> T getMapper(Class<?> mapperClass);
  5. }
复制代码
对应实现类方法


  • 动态署理对象:不需要编译,运行期间生成字节码,通过传入的类加载器加载,只存在于内存中
  • mapperClass:UserMapper或IUserDao接口的Class对象
  • 署理类调用接口中的方法,则会被拦截进入invoke方法,因为没有目标类,则具体的实现都在invoke内里了
  • 通过mapperClass和方法名获取到statementId
  • statementId和参数有了,但是DefaultSqlSession要有CRUD查询和更新操纵,所以sqlCommandType来区分
  1. public class DefaultSqlSession implements SqlSession {
  2.     private Configuration configuration;
  3.     private Executor executor;
  4.     public DefaultSqlSession(Configuration configuration, Executor executor) {
  5.         this.configuration = configuration;
  6.         this.executor = executor;
  7.     }
  8.     @Override
  9.     public <E> List<E> selectList(String statementId, Object param) throws Exception {
  10.         // 根据StatementId获取映射配置对象MapperStatement
  11.         MapperStatement mapperStatement = configuration.getMapperStatementMap().get(statementId);
  12.         // 然后将具体的查询操作委派给SimpleExecutor执行器
  13.         // 执行底层jdbc需要:1.数据库配置,2.sql配置信息
  14.         return executor.query(configuration,mapperStatement,param);
  15.     }
  16.     @Override
  17.     public <T> T selectOne(String statementId, Object param) throws Exception {
  18.         // 调用selectList()
  19.         List<Object> selectList = selectList(statementId, param);
  20.         if (selectList.size() == 1){
  21.             return (T) selectList.get(0);
  22.         }else if (selectList.size() > 1){
  23.             throw new Exception("返回数据不止一条!!!");
  24.         }else {
  25.             return null;
  26.         }
  27.     }
  28.     @Override
  29.     public void close() {
  30.         executor.close();
  31.     }
  32.     @Override
  33.     public <T> T getMapper(Class<?> mapperClass) {
  34.         // 使用JDK动态代理生成基于接口的对象
  35.         // 1、创建一个类(代理类),实现目标接口,实现所有的方法实现
  36.         // 2、动态代理类:代码运行期间生成的,而非编译期
  37.         Object proxyInstance = Proxy.newProxyInstance(DefaultSqlSession.class.getClassLoader(),
  38.                 new Class[]{mapperClass}, new InvocationHandler() {
  39.             // proxy:生成的代理对象本身,很少用
  40.             // method:调用接口中哪个方法,则执行对应代理里的对应方法
  41.             // args:调用方法的参数
  42.             @Override
  43.             public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
  44.                 // 执行底层JDBC
  45.                 // 1.获取statementId
  46.                 // ps:约定接口中的方法名要与<select>标签的id属性名一致,
  47.                 // 这样就可以通过接口获取statementId = namespace.id
  48.                 String methodName = method.getName();// 方法名 - findAll
  49.                 String className = mapperClass.getName();//接口类名 - com.xc.do.IUserDao
  50.                 String statementId = className + "." + methodName;
  51.                 // 2.判断调用sqlSession中CRUD的什么方法
  52.                 // MapperStatement类添加属性sqlCommandType-sql增删改查类型
  53.                 MapperStatement mapperStatement = configuration.getMapperStatementMap().get(statementId);
  54.                 // select  update delete insert
  55.                 String sqlCommandType = mapperStatement.getSqlCommandType();
  56.                 switch (sqlCommandType){
  57.                     case "select" :
  58.                         // 3.判断调用selectOne还是selectList
  59.                         // 获取返回值类型,
  60.                         Type genericReturnType = method.getGenericReturnType();
  61.                         // ParameterizedType是Type的子接口,表示一个有参数的类型,例如Collection<T>,Map<K,V>等
  62.                         if(genericReturnType instanceof ParameterizedType){
  63.                             if(args != null) {
  64.                                 return selectList(statementId, args[0]);
  65.                             }
  66.                             return  selectList(statementId, null);
  67.                         }
  68.                         return selectOne(statementId,args[0]);
  69.                     case "update":
  70.                         // 执行更新方法调用
  71.                         break;
  72.                     case "delete":
  73.                         // 执行delete方法调用
  74.                         break;
  75.                     case "insert":
  76.                         // 执行insert方法调用
  77.                         break;
  78.                 }
  79.                 return null;
  80.             }
  81.         });
  82.         return (T) proxyInstance;
  83.     }
  84. }
复制代码
3、优化后测试



  • 缺点:相较于优化前,需要新添加一个接口
  • 优点:不需要手动添加statementId,SqlSession不消频仍创建
  1. @Test
  2. public void test2() throws Exception {
  3.     // 1.根据配置文件的路径,加载成字节输入流,存到内存中 注意:配置文件还未解析
  4.     InputStream resourceAsSteam = Resources.getResourceAsStream("sqlMapConfig.xml");
  5.     // 2.解析了配置文件,封装了Configuration对象  2.创建sqlSessionFactory工厂对象
  6.     SqlSessionFactory sqlSessionFactory = new SqlSessionFactoryBuilder().build(resourceAsSteam);
  7.     // 3.生产sqlSession 创建了执行器对象
  8.     SqlSession sqlSession = sqlSessionFactory.openSession();
  9.     // 4.调用sqlSession方法
  10.     IUserDao userDao = sqlSession.getMapper(IUserDao.class);
  11.     List<User> all = userDao.findAll();
  12.     for (User user : all) {
  13.         System.out.println(user);
  14.     }
  15.     // 5.释放资源
  16.     sqlSession.close();
  17. }
复制代码
五、Spring整合优化



  • 与Spring整合之后,InputStream、SqlSessionFactory、SqlSession、IUserDao署理对象齐备不需要自己创建
  • 全都交给了spring容器管理,我们要做的就是@Autowired IUserDao userDao
  • 然后就可以用署理对象调用增删改查方法了
六、gitee代码



  • 手写源码框架项目: https://gitee.com/xuchang614/own-mybatis
  • 测试框架项目: https://gitee.com/xuchang614/own-mybatis-test

免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有账号?立即注册

x
回复

使用道具 举报

0 个回复

倒序浏览

快速回复

您需要登录后才可以回帖 登录 or 立即注册

本版积分规则

农妇山泉一亩田

金牌会员
这个人很懒什么都没写!
快速回复 返回顶部 返回列表