仿照MyBatis手写一个持久层框架学习

首先数据准备,创建MySQL数据库mybatis,创建表并插入数据。

DROP TABLE IF EXISTS user_t;
CREATE TABLE user_t ( id INT PRIMARY KEY, username VARCHAR ( 128 ) );
INSERT INTO user_t VALUES(1,'Tom');
INSERT INTO user_t VALUES(2,'Jerry');

JDBC API允许应用程序访问任何形式的表格数据,特别是存储在关系数据库中的数据。

在这里插入图片描述

JDBC代码示例:

    @Test
    @DisplayName("JDBC模式访问数据库示例")
    public void jdbc_test() {
        Connection connection = null;
        PreparedStatement preparedStatement = null;
        ResultSet resultSet = null;
        try {
            // 加载数据库驱动
            Class.forName("com.mysql.cj.jdbc.Driver");
            // 通过驱动管理类来获取数据库连接
            connection = DriverManager.getConnection("jdbc:mysql://localhost:3306/mybatis?characterEncoding=utf-8", "root", "123456");
            // 定义SQL语句 ?表示占位符
            String sql = "select * from user_t where username = ?";
            // 获取预处理Statement
            preparedStatement = connection.prepareStatement(sql);
            // 设置参数,第一个参数为SQL语句中参数的序号(从1开始),第二个参数为设置的参数值
            preparedStatement.setString(1, "Tom");
            // 向数据库发出SQL执行查询,查询出结果集
            resultSet = preparedStatement.executeQuery();
            // 遍历查询结果集
            while (resultSet.next()) {
                int id = resultSet.getInt("id");
                String username = resultSet.getString("username");
                // 封装User
                User user = new User();
                user.setId(id);
                user.setUsername(username);
                System.out.println(user);
            }
        } catch (ClassNotFoundException e) {
            throw new RuntimeException(e);
        } catch (SQLException e) {
            throw new RuntimeException(e);
        } finally {
            if (resultSet != null) {
                try {
                    resultSet.close();
                } catch (SQLException e) {
                    throw new RuntimeException(e);
                }
            }
            if (preparedStatement != null) {
                try {
                    preparedStatement.close();
                } catch (SQLException e) {
                    throw new RuntimeException(e);
                }
            }
            if (connection != null) {
                try {
                    connection.close();
                } catch (SQLException e) {
                    throw new RuntimeException(e);
                }
            }
        }
    }

我们发现使用过程中存在以下问题:

  • 数据库配置信息硬编码
  • 频繁创建释放数据库连接,而数据库连接是宝贵的资源
  • SQL语句、参数、返回结果集获取 均存在硬编码问题
  • 需要手动封装返回结果集,较为繁琐

手写持久层框架思路分析

在这里插入图片描述

创建一个maven项目mybatis-demo,作为使用端,引入自定义持久层框架jar包

在这里插入图片描述

其中文件内容如下:

package com.mybatis.it.dao;

import com.mybatis.it.pojo.User;

import java.util.List;

public class UserDaoImpl implements IUserDao {
    @Override
    public List<User> findAll() {
        return null;
    }

    @Override
    public User findByCondition(User user) {
        return null;
    }
}
package com.mybatis.it.pojo;

public class User {
    private int id;
    private String username;

    public int getId() {
        return id;
    }

    public void setId(int id) {
        this.id = id;
    }

    public String getUsername() {
        return username;
    }

    public void setUsername(String username) {
        this.username = username;
    }

    @Override
    public String toString() {
        return "User{" +
                "id=" + id +
                ", username='" + username + '\'' +
                '}';
    }
}

创建SqlMapConfig.xml配置文件:存放数据库配置信息、存放mapper.xml路径

<?xml version="1.0" encoding="utf-8" ?>
<configuration>
    <!-- 1. 配置数据库信息-->
    <dataSource>
        <property name="driverClassName" value="com.mysql.cj.jdbc.Driver"></property>
        <property name="url" value="jdbc:mysql://localhost:3306/mybatis?characterEncoding=utf-8&amp;serverTimezone=UTC"></property>
        <property name="username" value="root"></property>
        <property name="password" value="123456"></property>
    </dataSource>

    <!-- 2. 引入映射配置文件-->
    <mappers>
        <mapper resource="mapper/UserMapper.xml"></mapper>
    </mappers>

</configuration>

创建mapper.xml配置文件:存放SQL信息、参数类型、返回值类型等

<?xml version="1.0" encoding="UTF-8" ?>
<!-- 唯一标识:namespace.id 取名字叫statementId -->
<mapper namespace="com.mybatis.it.dao.IUserDao">

    <!--
        规范:接口的全路径要和namespace的值保持一致
             接口中的方法名要和id的值保持一致
      -->
    <!-- 查询所有-->
    <select id="findAll" resultType="com.mybatis.it.pojo.User">
        select * from user_t
    </select>

    <!-- 按照条件进行查询-->
    <select id="findByCondition" resultType="com.mybatis.it.pojo.User" parameterType="com.mybatis.it.pojo.User">
        select * from user_t where id = #{id} and username = #{username}
    </select>
</mapper>

单元测试用例:

package com.mybatis.it;

import com.mybatis.it.dao.IUserDao;
import com.mybatis.it.pojo.User;
import com.mybatis.it.sdk.io.Resources;
import com.mybatis.it.sdk.session.SqlSession;
import com.mybatis.it.sdk.session.SqlSessionFactory;
import com.mybatis.it.sdk.session.SqlSessionFactoryBuilder;
import jdk.nashorn.internal.ir.annotations.Ignore;
import org.junit.jupiter.api.DisplayName;
import org.junit.jupiter.api.Test;

import java.io.InputStream;
import java.util.List;

public class MyBatisSDKTest {

    @Ignore
    @DisplayName("测试手写版本1MyBatis使用")
    public void test() {
        // 1. 根据配置文件的路径,加载成字节输入流,存到内存中 注意:配置文件还未解析
        InputStream resourceAsStream = Resources.getResourceAsStream("sqlMapConfig.xml");

        // 2. 解析了配置文件,封装了Configuration对象 ; 创建sqlSessionFactory工厂对象
        SqlSessionFactory sqlSessionFactory = new SqlSessionFactoryBuilder().build(resourceAsStream);

        // 3. 生产sqlSession 创建了执行器对象
        SqlSession sqlSession = sqlSessionFactory.openSession();

        // 4. 调用sqlSession方法
        User user = new User();
        user.setId(1);
        user.setUsername("Tom");
        User user2 = sqlSession.selectOne("user.selectOne", user);
        System.out.println(user2);

        List<Object> list = sqlSession.selectList("user.selectList", null);
        System.out.println(list);

        // 释放资源
        sqlSession.close();
    }

    @Test
    @DisplayName("测试手写版本2MyBatis使用")
    public void test2() {
        // 1. 根据配置文件的路径,加载成字节输入流,存到内存中 注意:配置文件还未解析
        InputStream resourceAsStream = Resources.getResourceAsStream("sqlMapConfig.xml");

        // 2. 解析了配置文件,封装了Configuration对象 ; 创建sqlSessionFactory工厂对象
        SqlSessionFactory sqlSessionFactory = new SqlSessionFactoryBuilder().build(resourceAsStream);

        // 3. 生产sqlSession 创建了执行器对象
        SqlSession sqlSession = sqlSessionFactory.openSession();

        // 4. 调用sqlSession方法
        User user = new User();
        user.setId(1);
        user.setUsername("Tom");
        IUserDao userDao = sqlSession.getMapper(IUserDao.class);
        User user2 = userDao.findByCondition(user);
        System.out.println(user2);

        List<User> list = userDao.findAll();
        System.out.println(list);

        // 释放资源
        sqlSession.close();
    }
}

项目pom.xml内容如下:

<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0"
         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
    <modelVersion>4.0.0</modelVersion>

    <groupId>com.mybatis.it</groupId>
    <artifactId>mybatis-demo</artifactId>
    <version>1.0-SNAPSHOT</version>

    <properties>
        <maven.compiler.source>8</maven.compiler.source>
        <maven.compiler.target>8</maven.compiler.target>
        <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
    </properties>

    <dependencies>
        <!-- 引入自定义持久层框架的jar包 -->
        <dependency>
            <groupId>com.mybatis.it.sdk</groupId>
            <artifactId>mybatis-sdk</artifactId>
            <version>1.0-SNAPSHOT</version>
        </dependency>
      
        <!-- 引入单元测试依赖 -->
        <dependency>
            <groupId>org.junit.jupiter</groupId>
            <artifactId>junit-jupiter-engine</artifactId>
            <version>5.10.1</version>
            <scope>test</scope>
        </dependency>

    </dependencies>

</project>

同时创建一个maven模块:mybatis-sdk

在这里插入图片描述

引入依赖如下(pom.xml)

<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0"
         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
    <modelVersion>4.0.0</modelVersion>

    <groupId>com.mybatis.it.sdk</groupId>
    <artifactId>mybatis-sdk</artifactId>
    <version>1.0-SNAPSHOT</version>

    <properties>
        <maven.compiler.source>8</maven.compiler.source>
        <maven.compiler.target>8</maven.compiler.target>
        <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
    </properties>

    <dependencies>

        <!-- 解析xml -->
        <dependency>
            <groupId>dom4j</groupId>
            <artifactId>dom4j</artifactId>
            <version>1.6.1</version>
        </dependency>

        <!-- xpath -->
        <dependency>
            <groupId>jaxen</groupId>
            <artifactId>jaxen</artifactId>
            <version>1.1.6</version>
        </dependency>

        <!-- mysql数据库驱动 -->
        <dependency>
            <groupId>com.mysql</groupId>
            <artifactId>mysql-connector-j</artifactId>
            <version>8.2.0</version>
        </dependency>

        <!-- 数据库连接池 -->
        <dependency>
            <groupId>com.alibaba</groupId>
            <artifactId>druid</artifactId>
            <version>1.2.20</version>
        </dependency>

        <!-- 日志 -->
        <dependency>
            <groupId>org.apache.logging.log4j</groupId>
            <artifactId>log4j-slf4j-impl</artifactId>
            <version>2.22.0</version>
            <scope>test</scope>
        </dependency>

        <!-- 单元测试 -->
        <dependency>
            <groupId>org.junit.jupiter</groupId>
            <artifactId>junit-jupiter-engine</artifactId>
            <version>5.10.1</version>
            <scope>test</scope>
        </dependency>

        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
            <version>1.18.30</version>
            <scope>provided</scope>
        </dependency>

    </dependencies>

</project>

创建Resources类:负责加载配置文件,加载成字节流,存到内存中

package com.mybatis.it.sdk.io;

import java.io.InputStream;

public class Resources {

    /**
     * 根据配置文件的路径,加载配置文件成字节输入流,存到内存中,注意配置文件还未解析
     *
     * @param path
     * @return
     */
    public static InputStream getResourceAsStream(String path) {
        InputStream inputStream = Resources.class.getClassLoader().getResourceAsStream(path);
        return inputStream;
    }
}

Configuration:全局配置类:存储SqlMapConfig.xml配置文件解析出来的内容

package com.mybatis.it.sdk.pojo;


import javax.sql.DataSource;
import java.util.HashMap;
import java.util.Map;

/**
 * 全局配置类:存放核心配置文件解析出来的内容
 */
public class Configuration {

    // 数据源对象
    private DataSource dataSource;

    /**
     * 声明一个Map集合
     * key:statementId:namespace.id
     * MappedStatement: 封装好的MappedStatement对象
     */
    private Map<String, MappedStatement> mappedStatementMap = new HashMap<>();

    public DataSource getDataSource() {
        return dataSource;
    }

    public void setDataSource(DataSource dataSource) {
        this.dataSource = dataSource;
    }

    public Map<String, MappedStatement> getMappedStatementMap() {
        return mappedStatementMap;
    }

    public void setMappedStatementMap(Map<String, MappedStatement> mappedStatementMap) {
        this.mappedStatementMap = mappedStatementMap;
    }
}

MappedStatement:映射配置类:存储mapper.xml配置文件解析出来的内容

package com.mybatis.it.sdk.pojo;

/**
 * 映射配置类:存放mapper.xml解析内容
 */
public class MappedStatement {
    // 唯一标识:statementId:namespace.id
    private String statementId;
    // 返回值类型
    private String resultType;
    // 参数值类型
    private String parameterType;
    // SQL语句
    private String sql;
    // 判断当前是什么操作的一个属性
    private String sqlCommandType;

    public String getStatementId() {
        return statementId;
    }

    public void setStatementId(String statementId) {
        this.statementId = statementId;
    }

    public String getResultType() {
        return resultType;
    }

    public void setResultType(String resultType) {
        this.resultType = resultType;
    }

    public String getParameterType() {
        return parameterType;
    }

    public void setParameterType(String parameterType) {
        this.parameterType = parameterType;
    }

    public String getSql() {
        return sql;
    }

    public void setSql(String sql) {
        this.sql = sql;
    }

    public String getSqlCommandType() {
        return sqlCommandType;
    }

    public void setSqlCommandType(String sqlCommandType) {
        this.sqlCommandType = sqlCommandType;
    }
}

解析配置文件,填充容器对象。创建SqlSessionFactoryBuilder类

提供方法:build(InputStream stream) 方法:

(1) 解析配置文件(dom4j + xpath),封装Configuration

(2) 创建SqlSessionFactory

package com.mybatis.it.sdk.session;

import com.mybatis.it.sdk.config.XMLConfigBuilder;
import com.mybatis.it.sdk.pojo.Configuration;

import java.io.InputStream;

public class SqlSessionFactoryBuilder {

    /**
     * 1.解析配置文件,封装容器对象
     * 2.创建SqlSessionFactory工厂对象
     *
     * @param inputStream
     * @return
     */
    public SqlSessionFactory build(InputStream inputStream) {
        // 1. 解析配置文件,封装容器对象 XMLConfigBuilder:专门解析核心配置文件的解析类
        XMLConfigBuilder xmlConfigBuilder = new XMLConfigBuilder();
        Configuration configuration = xmlConfigBuilder.parse(inputStream);

        // 2. 创建SqlSessionFactory工厂对象
        DefaultSqlSessionFactory defaultSqlSessionFactory = new DefaultSqlSessionFactory(configuration);
        return defaultSqlSessionFactory;
    }

}

创建SqlSessionFactory接口及DefaultSqlSessionFactory实现类

提供方法:SqlSession openSession(); 工厂模式

提供接口类:

package com.mybatis.it.sdk.session;

public interface SqlSessionFactory {
    /**
     * 1.生产sqlSession对象
     * 2. 创建执行器对象
     *
     * @return
     */
    SqlSession openSession();
}

提供实现类:

package com.mybatis.it.sdk.session;

import com.mybatis.it.sdk.executor.Executor;
import com.mybatis.it.sdk.executor.SimpleExecutor;
import com.mybatis.it.sdk.pojo.Configuration;

public class DefaultSqlSessionFactory implements SqlSessionFactory {

    private Configuration configuration;

    public DefaultSqlSessionFactory(Configuration configuration) {
        this.configuration = configuration;
    }

    @Override
    public SqlSession openSession() {
        // 1. 创建执行器对象
        Executor executor = new SimpleExecutor();

        // 2. 生产sqlSession对象
        DefaultSqlSession defaultSqlSession = new DefaultSqlSession(configuration, executor);

        return defaultSqlSession;
    }
}

创建SqlSession接口和DefaultSqlSession实现类:

提供方法:

  • selectList(); 查询所有
  • selectOne(); 查询单个
  • update(); 更新
  • delete(); 删除
  • insert(); 添加
  • close();
  • getMapper();
package com.mybatis.it.sdk.session;

import java.util.List;

public interface SqlSession {
    /**
     * 查询多个结果
     *
     * @param statementId
     * @param param       SQL参数
     * @param <E>         元素
     * @return
     */
    <E> List<E> selectList(String statementId, Object param);

    /**
     * 查询单个结果
     *
     * @param statementId
     * @param param       SQL参数
     * @param <T>         类型
     * @return
     */
    <T> T selectOne(String statementId, Object param);

    /**
     * 清除资源
     */
    void close();

    /**
     * 生成代理对象
     */
    <T> T getMapper(Class<?> mapperClass);
}

DefaultSqlSession实现类:

package com.mybatis.it.sdk.session;

import com.mybatis.it.sdk.executor.Executor;
import com.mybatis.it.sdk.pojo.Configuration;
import com.mybatis.it.sdk.pojo.MappedStatement;

import java.lang.reflect.*;
import java.util.List;

public class DefaultSqlSession implements SqlSession {

    private Configuration configuration;

    private Executor executor;

    public DefaultSqlSession(Configuration configuration, Executor executor) {
        this.configuration = configuration;
        this.executor = executor;
    }

    @Override
    public <E> List<E> selectList(String statementId, Object param) {
        // 将查询操作委托给底层的执行器
        // query():执行底层的JDBC操作:1.数据库配置信息 2.SQL配置信息
        MappedStatement mappedStatement = configuration.getMappedStatementMap().get(statementId);
        List<E> list = executor.query(configuration, mappedStatement, param);
        return list;
    }

    @Override
    public <T> T selectOne(String statementId, Object param) {
        // 去调用selectList方法
        List<Object> list = this.selectList(statementId, param);
        if (list.size() == 1) {
            return (T) list.get(0);
        } else if (list.size() > 1) {
            throw new RuntimeException("返回结果大于预期");
        } else {
            return null;
        }
    }

    @Override
    public void close() {
        executor.close();
    }

    @Override
    public <T> T getMapper(Class<?> mapperClass) {

        // 使用JDK动态代理生成基于接口的代理对象
        Object proxy = Proxy.newProxyInstance(DefaultSqlSession.class.getClassLoader(), new Class[]{mapperClass}, new InvocationHandler() {

            /**
             *
             * @param proxy 代理对象的引用,很少用
             * @param method 被调用的方法的字节码对象
             * @param args 调用的方法参数
             *
             * @return
             * @throws Throwable
             */
            @Override
            public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
                // 具体的逻辑:执行底层的JDBC
                // 通过调用sqlSession里面的方法来完成方法调用
                // 参数准备:1.statementId 2.param
                // 问题1:无法获取现有的statementId
                // 规范:接口的全路径要和namespace的值保持一致;接口中的方法名要和id的值保持一致
                String methodName = method.getName();
                String className = method.getDeclaringClass().getName();
                String statementId = className + "." + methodName;

                // 方法调用:问题2:要调用sqlSession中增删改查的什么方法?
                // 改造当前工程:sqlCommandType:判断当前是什么操作的一个属性
                MappedStatement mappedStatement = configuration.getMappedStatementMap().get(statementId);
                // sqlCommandType取值范围(insert delete update select)
                String sqlCommandType = mappedStatement.getSqlCommandType();
                switch (sqlCommandType) {
                    case "select":
                        // 执行查询方法调用
                        // 问题3:该调用selectList还是selectOne?
                        Type genericReturnType = method.getGenericReturnType();
                        // 判断是否实现了 泛型类型参数化
                        if (genericReturnType instanceof ParameterizedType) {
                            // 表示返回结果是带泛型的
                            if (args != null) {
                                return selectList(statementId, args[0]);
                            }
                            return selectList(statementId, null);
                        }
                        if (args != null) {
                            return selectOne(statementId, args[0]);
                        }
                        return selectOne(statementId, null);
                    case "update":
                        // 执行更新方法调用
                    case "delete":
                        // 执行删除方法调用
                    case "insert":
                        // 执行插入方法调用

                }
                return null;
            }
        });
        return (T) proxy;
    }
}

创建Executor接口和实现类SimpleExecutor

提供方法:query(Configuration,MappedStatement,Object parameter); 执行的就是底层JDBC代码(数据库配置信息、SQL配置信息)

package com.mybatis.it.sdk.executor;

import com.mybatis.it.sdk.pojo.Configuration;
import com.mybatis.it.sdk.pojo.MappedStatement;

import java.sql.SQLException;
import java.util.List;

public interface Executor {
    <E> List<E> query(Configuration configuration, MappedStatement mappedStatement, Object param);

    void close();
}

SimpleExecutor实现类:

package com.mybatis.it.sdk.executor;

import com.mybatis.it.sdk.config.BoundSql;
import com.mybatis.it.sdk.pojo.Configuration;
import com.mybatis.it.sdk.pojo.MappedStatement;
import com.mybatis.it.sdk.utils.GenericTokenParser;
import com.mybatis.it.sdk.utils.ParameterMapping;
import com.mybatis.it.sdk.utils.ParameterMappingTokenHandler;

import java.beans.PropertyDescriptor;
import java.lang.reflect.Field;
import java.lang.reflect.Method;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;

public class SimpleExecutor implements Executor {

    private Connection connection = null;
    private PreparedStatement preparedStatement = null;
    private ResultSet resultSet = null;

    @Override
    public <E> List<E> query(Configuration configuration, MappedStatement mappedStatement, Object param) {
        try {
            // 1. 加载驱动,获取数据源连接
            connection = configuration.getDataSource().getConnection();
            // 2. 获取preparedStatement预编译对象
            // 获取要执行的SQL语句
            /**
             * select * from user_t where id = #{id} and username = #{username}
             * 替换成
             * select * from user_t where id = ? and username = ?
             * 解析替换过程中:自定义占位符#{id}里面的值保存下来
             */
            String sql = mappedStatement.getSql();
            BoundSql boundSql = getBoundSql(sql);
            String finalSql = boundSql.getFinalSql();
            preparedStatement = connection.prepareStatement(finalSql);

            // 3.设置参数
            String parameterType = mappedStatement.getParameterType();
            if (parameterType != null) {
                Class<?> parameterTypeClass = Class.forName(parameterType);
                List<ParameterMapping> parameterMappingList = boundSql.getList();
                for (int i = 0; i < parameterMappingList.size(); i++) {
                    ParameterMapping parameterMapping = parameterMappingList.get(i);
                    // 值为#{}里面的内容
                    String paramName = parameterMapping.getContent();
                    // 反射
                    Field declaredField = parameterTypeClass.getDeclaredField(paramName);
                    // 暴力访问
                    declaredField.setAccessible(true);
                    Object value = declaredField.get(param);
                    // 赋值占位符
                    preparedStatement.setObject(i + 1, value);
                }
            }
            // 4. 执行SQL,发起查询
            resultSet = preparedStatement.executeQuery();

            // 5. 处理返回结果集
            List<E> list = new ArrayList<>();
            while (resultSet.next()) {

                // 元数据信息 包含了:字段名以及字段的值
                ResultSetMetaData metaData = resultSet.getMetaData();
                String resultType = mappedStatement.getResultType();
                Class<?> resultTypeClass = Class.forName(resultType);

                Object object = resultTypeClass.newInstance();

                for (int i = 1; i <= metaData.getColumnCount(); i++) {
                    // 字段名
                    String columnName = metaData.getColumnName(i);
                    // 字段值
                    Object value = resultSet.getObject(columnName);

                    // 封装
                    // 属性描述器:通过API方法获取某个属性的读写方法
                    PropertyDescriptor propertyDescriptor = new PropertyDescriptor(columnName, resultTypeClass);
                    Method writeMethod = propertyDescriptor.getWriteMethod();
                    // 参数1:实例对象 参数2:要设置的值
                    writeMethod.invoke(object, value);
                }
                list.add((E) object);
            }
            return list;
        } catch (Exception exception) {
            throw new RuntimeException(exception);
        }
    }

    /**
     * 1. 将#{}占位符替换成?
     * 2. 解析替换的过程中 将#{}里面保存的值保存下来
     *
     * @param sql
     * @return
     */
    private BoundSql getBoundSql(String sql) {
        // 1. 创建标记处理器:配合标记解析器完成标记的处理解析工作
        ParameterMappingTokenHandler parameterMappingTokenHandler = new ParameterMappingTokenHandler();
        // 2. 创建标记解析器
        GenericTokenParser genericTokenParser = new GenericTokenParser("#{", "}", parameterMappingTokenHandler);
        // #{}占位符替换成? 解析替换过程中 将#{}里面保存的值保存下来ParameterMapping集合中
        String finalSql = genericTokenParser.parse(sql);
        // #{}里面的值的一个集合
        List<ParameterMapping> parameterMappings = parameterMappingTokenHandler.getParameterMappings();
        BoundSql boundSql = new BoundSql(finalSql, parameterMappings);
        return boundSql;
    }

    /**
     * 释放资源
     */
    @Override
    public void close() {
        if (resultSet != null) {
            try {
                resultSet.close();
            } catch (SQLException e) {
                throw new RuntimeException(e);
            }
        }
        if (preparedStatement != null) {
            try {
                preparedStatement.close();
            } catch (SQLException e) {
                throw new RuntimeException(e);
            }
        }
        if (connection != null) {
            try {
                connection.close();
            } catch (SQLException e) {
                throw new RuntimeException(e);
            }
        }
    }
}

解析配置文件逻辑:

XMLConfigBuilder

package com.mybatis.it.sdk.config;

import com.alibaba.druid.pool.DruidDataSource;
import com.mybatis.it.sdk.io.Resources;
import com.mybatis.it.sdk.pojo.Configuration;
import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.dom4j.Element;

import org.dom4j.io.SAXReader;

import java.io.InputStream;
import java.util.List;
import java.util.Properties;


public class XMLConfigBuilder {

    private Configuration configuration;

    public XMLConfigBuilder() {
        this.configuration = new Configuration();
    }

    /**
     * 使用dom4j+xpath解析配置文件,封装Configuration对象
     *
     * @param inputStream
     * @return
     */
    public Configuration parse(InputStream inputStream) {
        try {
            Document document = new SAXReader().read(inputStream);
            Element rootElement = document.getRootElement();
            List<Element> list = rootElement.selectNodes("//property");
            Properties properties = new Properties();
            for (Element element : list) {
                String name = element.attributeValue("name");
                String value = element.attributeValue("value");
                properties.setProperty(name, value);
            }

            // 创建数据源对象
            DruidDataSource druidDataSource = new DruidDataSource();
            druidDataSource.setDriverClassName(properties.getProperty("driverClassName"));
            druidDataSource.setUrl(properties.getProperty("url"));
            druidDataSource.setUsername(properties.getProperty("username"));
            druidDataSource.setPassword(properties.getProperty("password"));

            // 创建好的数据源对象封装到Configuration对象中
            configuration.setDataSource(druidDataSource);

            /**
             * 解析映射配置文件
             * 1.获取映射配置文件的路径
             * 2.根据路径进行映射配置文件的加载解析
             * 3.封装到MappedStatement对象中 --> Configuration里面Map<String, MappedStatement>中
             */
            // 1.获取映射配置文件的路径
            List<Element> mapperList = rootElement.selectNodes("//mapper");
            for (Element element : mapperList) {
                String mapperPath = element.attributeValue("resource");
                InputStream resourceAsStream = Resources.getResourceAsStream(mapperPath);
                // 专门解析映射配置文件的对象
                XMLMapperBuilder xmlMapperBuilder = new XMLMapperBuilder(configuration);
                // 2.根据路径进行映射配置文件的加载解析
                // 3.封装到MappedStatement对象中 --> Configuration里面Map<String, MappedStatement>中
                xmlMapperBuilder.parse(resourceAsStream);
            }
            return configuration;
        } catch (DocumentException e) {
            throw new RuntimeException(e);
        }
    }
}

XMLMapperBuilder

package com.mybatis.it.sdk.config;

import com.mybatis.it.sdk.pojo.Configuration;
import com.mybatis.it.sdk.pojo.MappedStatement;
import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.dom4j.Element;
import org.dom4j.io.SAXReader;

import java.io.InputStream;
import java.util.List;

/**
 * parse:解析配置文件 --> mappedStatement --> Configuration里面Map<String, MappedStatement>中
 */
public class XMLMapperBuilder {

    private Configuration configuration;

    public XMLMapperBuilder(Configuration configuration) {
        this.configuration = configuration;
    }

    public void parse(InputStream resourceAsStream) {
        try {
            Document document = new SAXReader().read(resourceAsStream);
            Element rootElement = document.getRootElement();

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

            List<Element> selectList = rootElement.selectNodes("//select");
            for (Element element : selectList) {
                String id = element.attributeValue("id");
                String resultType = element.attributeValue("resultType");
                String parameterType = element.attributeValue("parameterType");
                String sql = element.getTextTrim();

                String statementId = namespace + "." + id;

                // 封装MappedStatement对象
                MappedStatement mappedStatement = new MappedStatement();
                mappedStatement.setStatementId(statementId);
                mappedStatement.setParameterType(parameterType);
                mappedStatement.setResultType(resultType);
                mappedStatement.setSql(sql);
                mappedStatement.setSqlCommandType("select");

                // 将封装好的MappedStatement封装到Configuration里面Map<String, MappedStatement>集合中
                this.configuration.getMappedStatementMap().put(statementId, mappedStatement);
            }

        } catch (DocumentException e) {
            throw new RuntimeException(e);
        }
    }
}

BoundSql

package com.mybatis.it.sdk.config;

import com.mybatis.it.sdk.utils.ParameterMapping;

import java.util.List;

public class BoundSql {
    private String finalSql;
    private List<ParameterMapping> list;

    public BoundSql(String finalSql, List<ParameterMapping> list) {
        this.finalSql = finalSql;
        this.list = list;
    }

    public String getFinalSql() {
        return finalSql;
    }

    public void setFinalSql(String finalSql) {
        this.finalSql = finalSql;
    }

    public List<ParameterMapping> getList() {
        return list;
    }

    public void setList(List<ParameterMapping> list) {
        this.list = list;
    }
}

解析参数工具类:

GenericTokenParser:来自mybatis源码

package com.mybatis.it.sdk.utils;

public class GenericTokenParser {

    private final String openToken; //开始标记
    private final String closeToken; //结束标记
    private final TokenHandler handler; // 标记处理器

    public GenericTokenParser(String openToken, String closeToken, TokenHandler handler) {
        this.openToken = openToken;
        this.closeToken = closeToken;
        this.handler = handler;
    }

    /**
     * 解析${}和#{}
     * 该方法主要实现了配置文件、脚本等片段中占位符的解析、处理工作,并返回最终需要的数据。
     * 其中,解析工作由该方法完成,处理工作由处理器handler的handleToken()方法来实现
     *
     * @param text
     * @return
     */
    public String parse(String text) {
        // 验证参数问题,如果是null,就返回空字符串
        if (text == null || text.isEmpty()) {
            return "";
        }
        // search open token
        // 下面继续验证是否包含开始标签,如果不包含,默认不是占位符,直接原样返回即可,否则继续执行
        int start = text.indexOf(openToken, 0);
        if (start == -1) {
            return text;
        }
        // 把text转成字符数组src,并且定义默认偏移量offset=0、存储最终需要返回字符串的变量builder
        // text变量中占位符对应的变量名是expression。判断start是否大于-1(即text中是否存在openToken),如果存在就执行
        char[] src = text.toCharArray();
        int offset = 0;
        final StringBuilder builder = new StringBuilder();
        StringBuilder expression = null;
        while (start > -1) {
            // 判断如果开始标记前如果有转义字符,就不作为openToken进行处理,否则继续处理
            if (start > 0 && src[start - 1] == '\\') {
                // this open token is escaped. remove the backslash and continue.
                builder.append(src, offset, start - offset - 1).append(openToken);
                offset = start + openToken.length();
            } else {
                // 重置expression变量,避免空指针或者老数据干扰。
                // found open token. let's search close token.
                if (expression == null) {
                    expression = new StringBuilder();
                } else {
                    expression.setLength(0);
                }
                builder.append(src, offset, start - offset);
                offset = start + openToken.length();
                int end = text.indexOf(closeToken, offset);
                while (end > -1) { // 存在结束标记时
                    if (end > offset && src[end - 1] == '\\') {// 如果结束标记前面有转义字符时
                        // this close token is escaped. remove the backslash and continue.
                        expression.append(src, offset, end - offset - 1).append(closeToken);
                        offset = end + closeToken.length();
                        end = text.indexOf(closeToken, offset);
                    } else {// 不存在转义字符,即需要作为参数进行处理
                        expression.append(src, offset, end - offset);
                        offset = end + closeToken.length();
                        break;
                    }
                }
                if (end == -1) {
                    // close token was not found.
                    builder.append(src, start, src.length - start);
                    offset = src.length;
                } else {
                    // 首先根据参数的key(即expression)进行参数处理,返回?作为占位符
                    builder.append(handler.handleToken(expression.toString()));
                    offset = end + closeToken.length();
                }
            }
            start = text.indexOf(openToken, offset);
        }
        if (offset < src.length) {
            builder.append(src, offset, src.length - offset);
        }
        return builder.toString();
    }
}

TokenHandler:来自mybatis源码

package com.mybatis.it.sdk.utils;

public interface TokenHandler {
    String handleToken(String content);
}

ParameterMapping:

package com.mybatis.it.sdk.utils;

public class ParameterMapping {
    // 值为#{}里面的内容:如:id、username
    private String content;

    public ParameterMapping(String content) {
        this.content = content;
    }

    public String getContent() {
        return content;
    }

    public void setContent(String content) {
        this.content = content;
    }
}

ParameterMappingTokenHandler:参考mybatis源码

package com.mybatis.it.sdk.utils;

import java.util.ArrayList;
import java.util.List;

public class ParameterMappingTokenHandler implements TokenHandler {

    private List<ParameterMapping> parameterMappings = new ArrayList<ParameterMapping>();

    // content是参数名称 #{id} #{username}
    @Override
    public String handleToken(String content) {
        parameterMappings.add(buildParameterMapping(content));
        return "?";
    }

    private ParameterMapping buildParameterMapping(String content) {
        ParameterMapping parameterMapping = new ParameterMapping(content);
        return parameterMapping;
    }

    public List<ParameterMapping> getParameterMappings() {
        return parameterMappings;
    }

    public void setParameterMappings(List<ParameterMapping> parameterMappings) {
        this.parameterMappings = parameterMappings;
    }
}

对应项目源码资源:https://download.csdn.net/download/liwenyang1992/88615616

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:/a/235252.html

如若内容造成侵权/违法违规/事实不符,请联系我们进行投诉反馈qq邮箱809451989@qq.com,一经查实,立即删除!

相关文章

《信息技术时代》期刊杂志论文发表投稿

《信息技术时代》期刊收稿方向&#xff1a;通信工程、大数据、计算机、办公自动化、信息或计算机教育、电子技术、系统设计、移动信息、图情信息研究、人工智能、智能技术、信息技术与网络安全等。 刊名&#xff1a;信息技术时代 主管主办单位&#xff1a;深圳湾科技发展有限…

【sgAutocomplete】自定义组件:基于elementUI的el-autocomplete组件开发的自动补全下拉框组件(带输入建议的自动补全输入框)

特性&#xff1a; 1、支持本地保存选中过的记录 2、支持动态接口获取匹配下拉框内容 3、可以指定对应的显示label和字段组件key 4、自动生成速记符字段&#xff08;包含声母和全拼两种类型&#xff09;&#xff0c;增强搜索匹配效率 sgAutocomplete源码 <template><!…

API接口并发测试:如何测试API接口的最大并发能力?

本文将深入探讨API接口并发测试&#xff0c;介绍并比较不同的API并发测试工具&#xff0c;并分享如何有效测量和提高API接口在最大并发情况下的性能。了解如何应对高并发压力是保证系统稳定性和用户满意度的关键&#xff0c;让我们一起来探索这个重要的话题。 随着互联网的迅速…

selenium库的使用

来都来了给我点个赞收藏一下再走呗&#x1f339;&#x1f339;&#x1f339;&#x1f339;&#x1f339; 目录 一、下载需要用到的python库selenium 二、selenium的基本使用 1.在python代码引入库 2.打开浏览器 3.元素定位 1&#xff09;通过id定位 2&#xff09;通过标…

计算机组成原理-指令寻址

指令寻址 指令 寻址下一条欲执行指令的地址&#xff08;始终由程序计数器PC给出) 顺序寻址 &#xff08;PC&#xff09;“1”-> PC 这里的1理解为1个指令字长&#xff0c;实际加的值会因指令长度、编址方式而不同 **跳跃寻址 **由转移指令指出 数据寻址 确定 本条指令 的…

AttributeError: module ‘lib‘ has no attribute ‘X509_V_FLAG_CB_ISSUER_CHECK‘解决方案

大家好,我是爱编程的喵喵。双985硕士毕业,现担任全栈工程师一职,热衷于将数据思维应用到工作与生活中。从事机器学习以及相关的前后端开发工作。曾在阿里云、科大讯飞、CCF等比赛获得多次Top名次。现为CSDN博客专家、人工智能领域优质创作者。喜欢通过博客创作的方式对所学的…

【Android12】Android Framework系列--AMS启动Activity分析

AMS启动Activity分析 通过ActivityManagerService(AMS)提供的方法&#xff0c;可以启动指定的Activity。比如Launcher中点击应用图标后&#xff0c;调用AMS的startActivity函数启动应用。 AMS提供的服务通过IActivityManager.aidl文件定义。 // frameworks/base/core/java/an…

智能优化算法应用:基于郊狼算法3D无线传感器网络(WSN)覆盖优化 - 附代码

智能优化算法应用&#xff1a;基于郊狼算法3D无线传感器网络(WSN)覆盖优化 - 附代码 文章目录 智能优化算法应用&#xff1a;基于郊狼算法3D无线传感器网络(WSN)覆盖优化 - 附代码1.无线传感网络节点模型2.覆盖数学模型及分析3.郊狼算法4.实验参数设定5.算法结果6.参考文献7.MA…

关于IDEA中maven的作用以及如何配置MAVEN

关于IDEA中maven的作用以及如何配置MAVEN 1、Maven是什么2、Idea中对于Maven的配置3、下载依赖时&#xff0c;Idea下方的显示3.1、Maven中央仓库的下载显示界面3.2、阿里云仓库的下载显示界面 4、Maven在Idea中的使用4.1、clean4.2、validate4.3、compile4.4、test&#xff08;…

pyqt5使用Designer实现按钮上传图片

pyqt5使用Designer实现按钮上传图片 1、ui界面 2、ui转py代码 其中uploadimg.py代码如下&#xff1a; # -*- coding: utf-8 -*-# Form implementation generated from reading ui file uploadimg.ui # # Created by: PyQt5 UI code generator 5.15.4 # # WARNING: Any manua…

git 使用记录

远程仓库为空初始化 初始化本地仓库 git init 在本地仓库书写代码&#xff08;这里可以编辑一个文本文件做测试&#xff0c;如hello.txt&#xff09; 5&#xff09;执行&#xff1a;git add 要让git管理的文件&#xff08;git add hello.txt&#xff09;>执行完此操作将我…

有趣的数学 用示例来阐述什么是初值问题一

一、初值问题简述 在多变量微积分中&#xff0c;初值问题是一个常微分方程以及一个初始条件&#xff0c;该初始条件指定域中给定点处未知函数的值。在物理学或其他科学中对系统进行建模通常相当于解决初始值问题。 通常给定的微分方程有无数个解&#xff0c;因此我们很自然地会…

Linux 基础IO

文章目录 前言基础IO定义系统IO接口文件描述符重定向原理缓冲区刷新 前言 要知道每个函数/接口的全部参数和返回值建议去官网或者直接在Linux的man手册中查&#xff0c;这不是复制粘贴函数用法的文章。 C语言文件读写介绍链接 基础IO定义 IO是Input/Output的缩写&#xff0c…

【大数据】Hudi 核心知识点详解(一)

&#x1f60a; 如果您觉得这篇文章有用 ✔️ 的话&#xff0c;请给博主一个一键三连 &#x1f680;&#x1f680;&#x1f680; 吧 &#xff08;点赞 &#x1f9e1;、关注 &#x1f49b;、收藏 &#x1f49a;&#xff09;&#xff01;&#xff01;&#xff01;您的支持 &#x…

【操作宝典】VSCode解锁指南:释放潜能的详细教程!

目录 &#x1f4d6;前言 &#x1f680; 1 配置node.js &#x1f680;2. 使用脚本测试vue项目 &#x1f680;3. VSCode运行vue &#x1f680;4. VSCode引入elementUI &#x1f31f;4.1 显示OPENSSL错误 &#x1f4d6;前言 Visual Studio Code&#xff08;简称VSCode&#x…

MySQL数据库,创建和管理表

创建数据库&#xff1a; 方式一&#xff1a;创建数据库 CREATE DATABASE 数据库名&#xff1b;&#xff08;使用的是默认的字符集&#xff09; 方式二&#xff1a;创建数据库并指定字符集 CREATE DATABASE 数据库名 CHARACTER SET 字符集&#xff1b; 方式三&#xff1a;判断数…

基于若依的ruoyi-nbcio的flowable流程管理系统增加服务任务和我的抄送功能

更多ruoyi-nbcio功能请看演示系统 gitee源代码地址 前后端代码&#xff1a; https://gitee.com/nbacheng/ruoyi-nbcio 演示地址&#xff1a;RuoYi-Nbcio后台管理系统 1、增加一个状态字段 wf_copy增加下面两个字段 就用未读已读来区分 2、前端 api接口增加如下&#xff…

城市基础设施智慧路灯改造的特点

智慧城市建设稳步有序推进。作为智慧城市的基础设施&#xff0c;智能照明是智慧城市的重要组成部分&#xff0c;而叁仟智慧路灯是智慧城市理念下的新产品。随着物联网和智能控制技术的飞速发展&#xff0c;路灯被赋予了新的任务和角色。除了使道路照明智能化和节能化外&#xf…

Oracle(2-12)User-Managed Complete Recovery

文章目录 一、基础知识1、Media Recovery 介质恢复2、Recovery Steps 恢复步骤3、恢复4、Recovery in ARCHIVELOG 在ARCHIVELOG中恢复5、Complete Recovery完全恢复6、CR in ARCHIVELOG Mode 归档日志模式下的完全恢复7、Determine Files Need Recovery确定需要恢复的文件8、Ab…

HTTP与HTTPS的区别:安全性、协议地址和默认端口等比较

目录 ​编辑 作者其他博客链接&#xff1a; 一、概述 二、HTTP与HTTPS的区别 安全性 协议地址 默认端口 性能影响 三、比较与评估 浏览器支持 部署和维护成本 隐私保护 四、最佳实践建议 作者其他博客链接&#xff1a; 深入理解HashMap&#xff1a;Java中的键值对…