首页 > 其他分享 >手写MyBatis 重要基本原理框架

手写MyBatis 重要基本原理框架

时间：2024-07-31 14:38:57浏览次数：12  
标签：return String java 基本原理 sql MyBatis import 手写 public

1. 手写MyBatis 重要基本原理框架

@

目录

这里我们手写MyBatis 就仅仅只实现 insert() 和 selectOne(单条记录的查询)操作，并且有些局限就是，数据库的所有字段类型，都必须是 varchar 类型才行。

手写框架之前，如果没有思路，可以先参考一下mybatis的客户端程序，通过客户端程序来逆推需要的类，参考代码：

@Test
public void testInsert(){
    SqlSession sqlSession = null;
    try {
        // 1.创建SqlSessionFactoryBuilder对象
        SqlSessionFactoryBuilder sqlSessionFactoryBuilder = new SqlSessionFactoryBuilder();
        // 2.创建SqlSessionFactory对象
        SqlSessionFactory sqlSessionFactory = sqlSessionFactoryBuilder.build(Resources.getResourceAsStream("mybatis-config.xml"));
        // 3.创建SqlSession对象
        sqlSession = sqlSessionFactory.openSession();
        // 4.执行SQL
        Car car = new Car(null, "111", "宝马X7", "70.3", "2010-10-11", "燃油车");
        int count = sqlSession.insert("insertCar",car);
        System.out.println("更新了几条记录：" + count);
        // 5.提交
        sqlSession.commit();
    } catch (Exception e) {
        // 回滚
        if (sqlSession != null) {
            sqlSession.rollback();
        }
        e.printStackTrace();
    } finally {
        // 6.关闭
        if (sqlSession != null) {
            sqlSession.close();
        }
    }
}

@Test
public void testSelectOne(){
    SqlSession sqlSession = null;
    try {
        // 1.创建SqlSessionFactoryBuilder对象
        SqlSessionFactoryBuilder sqlSessionFactoryBuilder = new SqlSessionFactoryBuilder();
        // 2.创建SqlSessionFactory对象
        SqlSessionFactory sqlSessionFactory = sqlSessionFactoryBuilder.build(Resources.getResourceAsStream("mybatis-config.xml"));
        // 3.创建SqlSession对象
        sqlSession = sqlSessionFactory.openSession();
        // 4.执行SQL
        Car car = (Car)sqlSession.selectOne("selectCarByCarNum", "111");
        System.out.println(car);
        // 5.提交
        sqlSession.commit();
    } catch (Exception e) {
        // 回滚
        if (sqlSession != null) {
            sqlSession.rollback();
        }
        e.printStackTrace();
    } finally {
        // 6.关闭
        if (sqlSession != null) {
            sqlSession.close();
        }
    }
}

1.1 第一步：IDEA中创建模块

模块：godbatis（创建普通的Java Maven模块，打包方式jar），引入相关依赖

在这里插入图片描述

<?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>org.god.ibatis</groupId>
    <artifactId>godbatis</artifactId>
    <version>1.0-SNAPSHOT</version>
    <packaging>jar</packaging>

    <properties>
        <maven.compiler.source>17</maven.compiler.source>
        <maven.compiler.target>17</maven.compiler.target>
    </properties>

    <dependencies>
        <!--        dom4j 依赖-->
        <dependency>
            <groupId>org.dom4j</groupId>
            <artifactId>dom4j</artifactId>
            <version>2.1.3</version>
        </dependency>

        <!--        jaxen 依赖-->
        <dependency>
            <groupId>jaxen</groupId>
            <artifactId>jaxen</artifactId>
            <version>1.2.0</version>
        </dependency>
        <dependency>
            <groupId>junit</groupId>
            <artifactId>junit</artifactId>
            <version>4.13.2</version>
            <scope>test</scope>
        </dependency>

        <!--        mysql驱动依赖-->
        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
            <version>8.0.30</version>
        </dependency>
        <dependency>
            <groupId>org.mybatis</groupId>
            <artifactId>mybatis</artifactId>
            <version>3.5.10</version>
            <scope>compile</scope>
        </dependency>
    </dependencies>

</project>

1.2 第二步：资源工具类，方便获取指向配置文件的输入流

和MyBatis 框架一样，这里我们创建一个 Resoures 工具类，用来获取配置文件的输入流对象。

工具类的构造方法都是建议私有化的
因为工具类中的方法都是静态的，不需要创建对象就能调用
为了避免new对象，所有构造方法私有化
这只是一种编程习惯

在这里插入图片描述

package org.god.ibatis.utils;


import java.io.InputStream;

/**
 * godbatis 框架提供的一个工具类
 * 这个工具类专门完成“类路径” 中资源的加载
 */
public class Resources {

    /**
     * 工具类的构造方法都是建议私有化的
     * 因为工具类中的方法都是静态的，不需要创建对象就能调用
     * 为了避免new对象，所有构造方法私有化
     * 这只是一种编程习惯
     */

    private Resources() {}


    /**
     * 从类路径当中加载资源
     * @param resource 放在类路径当中的资源文件
     * @return 指向资源文件的一个输入流
     */
    public static InputStream getResourceAsStream(String resource) {
        InputStream resourceAsStream = ClassLoader.getSystemClassLoader().getResourceAsStream(resource);
        return resourceAsStream;
    }
}

1.3 第三步：定义SqlSessionFactoryBuilder类

提供一个无参数构造方法，再提供一个build方法，该build方法要返回SqlSessionFactory对象

package org.god.core;

import java.io.InputStream;


public class SqlSessionFactoryBuilder {

    /**
     * 创建构建器对象
     */
    public SqlSessionFactoryBuilder() {
    }


    /**
     * 获取SqlSessionFactory对象
     * 该方法主要功能是：读取godbatis核心配置文件，并构建SqlSessionFactory对象
     * @param inputStream 指向核心配置文件的输入流
     * @return SqlSessionFactory对象
     */
    public SqlSessionFactory build(InputStream inputStream){
        // 解析配置文件，创建数据源对象
        // 解析配置文件，创建事务管理器对象
        // 解析配置文件，获取所有的SQL映射对象
        // 将以上信息封装到SqlSessionFactory对象中
        // 返回
        return null;
    }
}

1.4 第四步：分析SqlSessionFactory类中有哪些属性

  • 事务管理器

    • GodJDBCTransaction

  • SQL映射对象集合

    • Map<String, GodMappedStatement>

1.5 第五步：定义JDBCTransaction

事务管理器最好是定义一个接口，然后每一个具体的事务管理器都实现这个接口。

在这里插入图片描述

package org.god.ibatis.core;


import java.sql.Connection;

/**
 * 事务管理接口
 * 所有的事务管理器都应该遵循该规范
 * JDBC 事务管理器，MANAGED 事务管理器都应该实现这个接口
 * Transaction事务管理器，提供管理事务方法。
 */
public interface Transaction {

    /**
     * 提交事务
     */
    void commit();


    /**
     * 回滚事务
     */
    void rollback();


    /**
     * 关闭事务
     */
    void close();

    /**
     * 真正的开启数据库连接
     */
    void openConnection();

    /**
     * 获取数据库连接对象的
     */
    Connection getConnection();

}

关于MyBatis 的事务，在 MyBatis 中有两种类型的事务管理器（也就是 type="[JDBC|MANAGED]"）：，但是这里我们只实现JDBC，这个值的事务。其他另外一个就不实现了。

在这里插入图片描述

package org.god.ibatis.core;


import javax.sql.DataSource;
import java.sql.Connection;
import java.sql.SQLException;

/**
 * JDBC事务管理器（godbatis 框架目前只有JdbcTransaction 进行实现）
 */
public class JdbcTransaction implements Transaction{
    /**
     * 数据源属性
     * 经典的设计：面向接口编程
     */
    private DataSource dataSource;


    /**
     * 自动提交标志
     * true 表示自动提交
     * false 表示不采用自动提交
     */
    private boolean autoCommit;


    /**
     * 连接对象
     */
    private Connection connection;

    public Connection getConnection() {
        return connection;
    }

    public void setConnection(Connection connection) {
        this.connection = connection;
    }

    /**
     * 创建管理器对象
     * @param dataSource
     * @param autoCommit
     */
    public JdbcTransaction(DataSource dataSource, boolean autoCommit) {
        this.dataSource = dataSource;
        this.autoCommit = autoCommit;
    }

    @Override
    public void commit() {
        try {
            connection.commit();
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }

    @Override
    public void rollback() {
        try {
            connection.rollback();
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }

    @Override
    public void close() {
        try {
            connection.close();
        } catch (SQLException e) {
            e.printStackTrace();
        }

    }

    public void openConnection() {
        if (connection == null) {
            try {
                this.connection = dataSource.getConnection();
                connection.setAutoCommit(autoCommit);
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }
    }
}

在这里插入图片描述

package org.god.ibatis.core;

import java.sql.Connection;

public class ManagedTransaction implements Transaction {
    @Override
    public void commit() {

    }

    @Override
    public void rollback() {

    }

    @Override
    public void close() {

    }

    @Override
    public void openConnection() {

    }

    @Override
    public Connection getConnection() {
        return null;
    }
}

1.6 第六步：事务管理器中需要数据源，定义UNPOOLEDDataSource

在MyBatis中有三种内建的数据源类型（也就是 type="[UNPOOLED|POOLED|JNDI]"），这里我们就实现UNPOOLED 这个值，另外两个值，就不实现了。

数据源是获取connection对象的
POOlED UNPOOLED JNDI
所有的数据源都要实现 JDK带的规范，javax.sql.DataSource

在这里插入图片描述

package org.god.ibatis.core;

import java.io.PrintWriter;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.sql.SQLFeatureNotSupportedException;
import java.util.logging.Logger;


/**
 * 数据源的实现类：UNPOOLED
 * 不使用连接池，每一次都新建Connection对象
 */
public class UnPooledDataSource implements javax.sql.DataSource {

    private String driver;
    private String url;
    private String username;
    private String password;


    /**
     * 创建一个数据源对象
     *
     * @param driver
     * @param url
     * @param username
     * @param password
     */
    public UnPooledDataSource(String driver, String url, String username, String password) {

        try {
            // 直接注册驱动
            Class.forName(driver);
        } catch (ClassNotFoundException e) {
            throw new RuntimeException(e);
        }
        this.driver = driver;
        this.url = url;
        this.username = username;
        this.password = password;
    }

    @Override
    public Connection getConnection() throws SQLException {
        // 这个连接池godbatis框架可以自己写一个连接池
        // 从数据库连接池当中获取Connection对象。（这个数据库练级吃是我godbatins框架内部封装好的。）
        Connection connection = DriverManager.getConnection(url, username, password);
        return connection;
    }

    @Override
    public Connection getConnection(String username, String password) throws SQLException {
        return null;
    }

    @Override
    public PrintWriter getLogWriter() throws SQLException {
        return null;
    }

    @Override
    public void setLogWriter(PrintWriter out) throws SQLException {

    }

    @Override
    public void setLoginTimeout(int seconds) throws SQLException {

    }

    @Override
    public int getLoginTimeout() throws SQLException {
        return 0;
    }

    @Override
    public Logger getParentLogger() throws SQLFeatureNotSupportedException {
        return null;
    }

    @Override
    public <T> T unwrap(Class<T> iface) throws SQLException {
        return null;
    }

    @Override
    public boolean isWrapperFor(Class<?> iface) throws SQLException {
        return false;
    }
}

在这里插入图片描述

package org.god.ibatis.core;

import java.io.PrintWriter;
import java.sql.Connection;
import java.sql.SQLException;
import java.sql.SQLFeatureNotSupportedException;
import java.util.logging.Logger;


/**
 * 数据源的实现类：POOlED
 * 使用godbatis 框架内置的数据库连接池来获取Connection对象。（这个不实现）
 */
public class PooledDataSource implements javax.sql.DataSource{
    @Override
    public Connection getConnection() throws SQLException {
        return null;
    }

    @Override
    public Connection getConnection(String username, String password) throws SQLException {
        return null;
    }

    @Override
    public PrintWriter getLogWriter() throws SQLException {
        return null;
    }

    @Override
    public void setLogWriter(PrintWriter out) throws SQLException {

    }

    @Override
    public void setLoginTimeout(int seconds) throws SQLException {

    }

    @Override
    public int getLoginTimeout() throws SQLException {
        return 0;
    }

    @Override
    public Logger getParentLogger() throws SQLFeatureNotSupportedException {
        return null;
    }

    @Override
    public <T> T unwrap(Class<T> iface) throws SQLException {
        return null;
    }

    @Override
    public boolean isWrapperFor(Class<?> iface) throws SQLException {
        return false;
    }
}

在这里插入图片描述

package org.god.ibatis.core;

import java.io.PrintWriter;
import java.sql.Connection;
import java.sql.SQLException;
import java.sql.SQLFeatureNotSupportedException;
import java.util.logging.Logger;


/**
 * 数据源的实现类：JNDI
 * 使用godbatis 框架内置的数据库连接池来获取Connection对象。（这个不实现）
 */
public class JNDIDataSource implements javax.sql.DataSource{
    @Override
    public Connection getConnection() throws SQLException {
        return null;
    }

    @Override
    public Connection getConnection(String username, String password) throws SQLException {
        return null;
    }

    @Override
    public PrintWriter getLogWriter() throws SQLException {
        return null;
    }

    @Override
    public void setLogWriter(PrintWriter out) throws SQLException {

    }

    @Override
    public void setLoginTimeout(int seconds) throws SQLException {

    }

    @Override
    public int getLoginTimeout() throws SQLException {
        return 0;
    }

    @Override
    public Logger getParentLogger() throws SQLFeatureNotSupportedException {
        return null;
    }

    @Override
    public <T> T unwrap(Class<T> iface) throws SQLException {
        return null;
    }

    @Override
    public boolean isWrapperFor(Class<?> iface) throws SQLException {
        return false;
    }
}

1.7 第七步：定义一个MappedStatement 类用于存放 SQL 标签

  • 普通的Java类，POJO，封装了一个SQL标签
  • 一个MappedStatement 对象对应一个SQL标签
  • 一个SQL标签中的所有信息封装到MappedStatement对象当中
  • 面向对象编程思想

在这里插入图片描述

package org.god.ibatis.core;


/**
 * 普通的Java类，POJO，封装了一个SQL标签
 * 一个MappedStatement 对象对应一个SQL标签
 * 一个SQL标签中的所有信息封装到MappedStatement对象当中
 * 面向对象编程思想
 */
public class MappedStatement {


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

    /**
     * 要封装的结果集类型，有的时候 resultType 是 null
     * 比如：insert，delete，update 语句的时候resultType是null
     * 只有当 sql语句 select 语句的时候 resultType 才有值。
     */
    private String resultType;

    public MappedStatement(String sql, String resultType) {
        this.sql = sql;
        this.resultType = resultType;
    }

    @Override
    public String toString() {
        return "MappedStatement{" +
                "sql='" + sql + '\'' +
                ", resultType='" + resultType + '\'' +
                '}';
    }


    public String getSql() {
        return sql;
    }

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

    public String getResultType() {
        return resultType;
    }

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

1.8 第八步：完善SqlSessionFactory类

SqlSessionFactory对象；

  • 一个数据库对应一个SqlSessionFactory对象
  • 通过SqlSessionFactory对象可以获取SqlSession对象（开启会话）
  • 一个SqlSessionFactory 对象可以开启对哦个SqlSession 会话

在这里插入图片描述

在SqlSessionFactory中添加openSession方法

package org.god.ibatis.core;



import java.util.List;
import java.util.Map;

/**
 * SqlSessionFactory对象；
 * 一个数据库对应一个SqlSessionFactory对象
 * 通过SqlSessionFactory对象可以获取SqlSession对象（开启会话）
 * 一个SqlSessionFactory 对象可以开启对哦个SqlSession 会话
 */
public class SqlSessionFactory {


    /**
     * 获取Sql会话对象
     * @return
     */
    public SqlSession openSession() {

        // 开启会话的前提是开启连接
        transaction.openConnection();
        // 创建SqlSession 对象
        SqlSession sqlSession = new SqlSession(this);
        
        return sqlSession;


    }
    public SqlSessionFactory() {
    }


    /**
     * 事务管理属性
     * 事务管理器可以灵活切换的
     * SqlSessionFactory类中的事务管理器应该是面向接口编程的
     * SqlSessionFactory类中的应该有一个事务管理器接口
     */
     private Transaction transaction;



     /**
     * 数据源属性
     */


    /**
     * 存放SqL语句的Map集合
     * key 是 sqlid
     * value 是对应的 SQL标签信息对象
     */
    private Map<String, MappedStatement> mappedStatements;


    public Map<String, MappedStatement> getMappedStatements() {
        return mappedStatements;
    }

    public void setMappedStatements(Map<String, MappedStatement> mappedStatements) {
        this.mappedStatements = this.getMappedStatements();
    }

    public SqlSessionFactory(Transaction transaction, Map<String, MappedStatement> mappedStatement) {
        this.transaction = transaction;
        this.mappedStatements = mappedStatement;
    }

    public Transaction getTransaction() {
        return transaction;
    }

    public void setTransaction(Transaction transaction) {
        this.transaction = transaction;
    }


}

1.9 第九步：完善SqlSessionFactoryBuilder中的build方法

这里我们定义一个常量类，用于方便后续的读取：，提高代码的可读性。

在这里插入图片描述

package org.god.ibatis.core;

public class Const {
    public static final String UN_POOLED_DATASOURCE = "UNPOOLED";
    public static final String POOLED_DATASOURCE = "POOLED";
    public static final String JNDI_DATASOURCE = "JNDI";


    public static final String JDBC_TRANSACTION = "JDBC";

    public static final String MANAGED_TRANSACTION = "MANAGED";

}

在这里插入图片描述

package org.god.ibatis.core;


import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.dom4j.Element;
import org.dom4j.Node;
import org.dom4j.io.SAXReader;
import org.god.ibatis.utils.Resources;

import javax.sql.DataSource;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * SqlSessionFactory 构建器对象
 * 通过SqlSessionFactoryBuilder的 build 方法来解析
 * godbatis-config.xml文件，然后创建sqlSessionFactory对象
 */

public class SqlSessionFactoryBuilder {

    /**
     *
     */
    public SqlSessionFactoryBuilder() {
    }


    /**
     * 解析 godbatis-config.xml 文件，来构建sqlSessionFactory对象
     *
     * @param in 指向godbatis-config.xml 文件的一个输入流
     * @return SqlSessionFactory
     */
    public SqlSessionFactory build(InputStream in) {

        SqlSessionFactory factory = null;
        try {
            // 解析 godbatis-config.xml 文件
            SAXReader reader = new SAXReader();
            Document document = reader.read(in);
            Element environments = (Element) document.selectSingleNode("/configuration/environments");
            String defaultId = environments.attributeValue("default");
            Element environment =
                    (Element) document.selectSingleNode("/configuration/environments/environment[@id='" + defaultId + "']");

            Element transactionElt = environment.element("transactionManager");
            Element dataSourceElt = environment.element("dataSource");
            List<String> sqlMapperXMLPathList = new ArrayList<>();
            List<Node> nodes = document.selectNodes("//mapper"); // 获取整个配置文件中所有的mapper对象
            nodes.forEach(node -> {
                Element mapper = (Element) node;
                String resource = mapper.attributeValue("resource");
                sqlMapperXMLPathList.add(resource);
            });


            // 获取数据源对象
            DataSource dataSource = getDataSource(dataSourceElt);

            // 获取事务管理器
            Transaction transaction = getTransaction(transactionElt, dataSource);

            // 获取mappedStatements
            Map<String, MappedStatement> mappedStatements = getMappedStatements(sqlMapperXMLPathList);


            // 解析完成之后，构建SqlSessionFactory对象
            factory = new SqlSessionFactory(transaction, mappedStatements);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return factory;

    }

    private Map<String, MappedStatement> getMappedStatements(List<String> sqlMapperXMLPathList) {

        Map<String, MappedStatement> mappedStatements = new HashMap<>();
        sqlMapperXMLPathList.forEach(sqlMapperXMLPath -> {
            try {
                SAXReader reader = new SAXReader();
                Document document = reader.read(Resources.getResourceAsStream(sqlMapperXMLPath));
                Element mapper = (Element) document.selectSingleNode("/mapper");
                String namespace = mapper.attributeValue("namespace");
                List<Element> elements = mapper.elements();
                elements.forEach(element -> {
                    String id = element.attributeValue("id");
                    // 这里进行了namespace和 id 的拼接，生成最终的sqlId
                    String sqlId = namespace + "." + id;

                    String resultType = element.attributeValue("resultType");
                    System.out.println(resultType);
                    String sql = element.getTextTrim();
                    System.out.println(sql);

                    MappedStatement mappedStatement = new MappedStatement(sql, resultType);
                    mappedStatements.put(sqlId,mappedStatement);
                });
            } catch (DocumentException e) {
                e.printStackTrace();
            }
        });

        return mappedStatements;
    }

    private Transaction getTransaction(Element transactionElt, DataSource dataSource) {
        Transaction transaction = null;

        String type = transactionElt.attributeValue("type").trim().toUpperCase();
        if (Const.JDBC_TRANSACTION.equals(type)) {
            transaction = new JdbcTransaction(dataSource, false); // 默认是开启事务的，将来需要手动提交的
        }

        if (Const.MANAGED_TRANSACTION.equals(type)) {
            transaction = new ManagedTransaction();
        }

        return transaction;
    }


    /**
     * 获取数据源对象
     *
     * @param dataSourceElt
     * @return
     */
    private DataSource getDataSource(Element dataSourceElt) {


        Map<String, String> map = new HashMap<>();

        // 获取所有的property
        List<Element> propertyElts = dataSourceElt.elements("property");

        propertyElts.forEach(propertyElt -> {
            String name = propertyElt.attributeValue("name");
            String value = propertyElt.attributeValue("value");
            map.put(name, value);
        });


        DataSource dataSource = null;

        //UNPOOLED POOLED JNDI
        String type = dataSourceElt.attributeValue("type").trim().toUpperCase();

        if (Const.UN_POOLED_DATASOURCE.equals(type)) {
            dataSource = new UnPooledDataSource(map.get("driver"), map.get("url"), map.get("username"), map.get(
                    "password"));
        }

        if (Const.POOLED_DATASOURCE.equals(type)) {
            dataSource = new PooledDataSource();
        }

        if (Const.JNDI_DATASOURCE.equals(type)) {
            dataSource = new JNDIDataSource();
        }
        return dataSource;
    }


}

1.10 第十步：编写SqlSession类中commit rollback close方法

在这里插入图片描述

package org.god.ibatis.core;

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;

public class SqlSession {

    private SqlSessionFactory factory;

    public SqlSession(SqlSessionFactory factory) {
        this.factory = factory;
    }

    /**
     * 提交事务
     */
    public void commit() {
        factory.getTransaction().commit();
    }


    /**
     * 回滚事务
     */
    public void rollback() {
        factory.getTransaction().rollback();
    }


    /**
     * 关闭事务
     */
    public void close() {
        factory.getTransaction().close();
    }
}

这里我们手写MyBatis 就仅仅只实现 insert() 和 selectOne(单条记录的查询)操作，并且有些局限就是，数据库的所有字段类型，都必须是 varchar 类型才行。

1.11 第十一步：编写SqlSession类中的insert方法

在这里插入图片描述

package org.god.ibatis.core;

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;

public class SqlSession {

    private SqlSessionFactory factory;

    public SqlSession(SqlSessionFactory factory) {
        this.factory = factory;
    }


    /**
     * 执行insert语句，向数据库表当中插入记录
     *
     * @param
     * @return
     */
    public int insert(String sqlId, Object pojo) {
        int count = 0;

        try {
            // JDBC代码，执行insert 语句 ，完成插入操作
            Connection connection = factory.getTransaction().getConnection();
            // insert into t_car values(#{id},#{name},#{age})
            String godbatisSql = factory.getMappedStatements().get(sqlId).getSql();
            System.out.println(factory.getMappedStatements().get(sqlId));
            System.out.println(factory.getMappedStatements().get(sqlId).getSql());
            // insert into t_user values(?,?,?)
            String sql = godbatisSql.replaceAll("#\\{[a-zA-Z0-9_$]*}", "?");
            PreparedStatement ps = connection.prepareStatement(sql);
            // 给 ？ 占位符传值
            // 难度是什么？
            // 第一: 你不知道有多少个？
            // 第二：你不知道该将Pojo对象中的那个属性赋值给哪个？
            // ps.String(第几个问号，传什么值)； // 这里都是setString，所以数据库中的字段类型要求都是varchar才行，
            // 这是godbatis 比较失败的地方
            int formIndex = 0;
            int index = 1;

            while (true) {
                int jingIndex = godbatisSql.indexOf("#", formIndex);
                if (jingIndex < 0) {
                    break;
                }

                System.out.println(index);

                int youKuoHaoIndex = godbatisSql.indexOf("}", formIndex);
                String propertyName = godbatisSql.substring(jingIndex + 2, youKuoHaoIndex).trim();
                System.out.println(propertyName);
                formIndex = youKuoHaoIndex + 1;

                // 有属性名id,怎么获取id 的属性值呢？调用 getId()方法
                String getMethodName = "get" + propertyName.toUpperCase().charAt(0) + propertyName.substring(1);
                Method getMethod = pojo.getClass().getDeclaredMethod(getMethodName);
                Object propertyValue = getMethod.invoke(pojo);
                ps.setString(index,propertyValue.toString());
                index++;

            }
            count = ps.executeUpdate();
        } catch (Exception e) {
            e.printStackTrace();
        }

        // 给？占位符传值
        return count;
    }


    /**
     * 查询一个对象
     * @param sqlId
     * @param parameterObj
     * @return
     */
    public Object selectOne(String sqlId, Object parameterObj){
        MappedStatement mappedStatement = factory.getMappedStatements().get(sqlId);
        Connection connection = factory.getTransaction().getConnection();
        // 获取sql语句
        String godbatisSql = mappedStatement.getSql();
        String sql = godbatisSql.replaceAll("#\\{[a-zA-Z0-9_\\$]*}", "?");
        // 执行sql
        PreparedStatement ps = null;
        ResultSet rs = null;
        Object obj = null;
        try {
            ps = connection.prepareStatement(sql);
            ps.setString(1, parameterObj.toString());
            rs = ps.executeQuery();
            if (rs.next()) {
                // 将结果集封装对象，通过反射
                String resultType = mappedStatement.getResultType();
                Class<?> aClass = Class.forName(resultType);
                obj = aClass.newInstance();
                // 给对象obj属性赋值
                ResultSetMetaData rsmd = rs.getMetaData();
                int columnCount = rsmd.getColumnCount();
                for (int i = 1; i <= columnCount; i++) {
                    String columnName = rsmd.getColumnName(i);
                    String setMethodName = "set" + columnName.toUpperCase().charAt(0) + columnName.substring(1);
                    Method setMethod = aClass.getDeclaredMethod(setMethodName, aClass.getDeclaredField(columnName).getType());
                    setMethod.invoke(obj, rs.getString(columnName));
                }
            }
        } catch (Exception e) {
            throw new RuntimeException(e);
        } finally {
            if (rs != null) {
                try {
                    rs.close();
                } catch (SQLException e) {
                    throw new RuntimeException(e);
                }
            }
            try {
                ps.close();
            } catch (SQLException e) {
                throw new RuntimeException(e);
            }
        }
        return obj;
    }




    /**
     * 提交事务
     */
    public void commit() {
        factory.getTransaction().commit();
    }


    /**
     * 回滚事务
     */
    public void rollback() {
        factory.getTransaction().rollback();
    }


    /**
     * 关闭事务
     */
    public void close() {
        factory.getTransaction().close();
    }
}

1.12 第十二步：编写SqlSession类中的selectOne方法

在这里插入图片描述

package org.god.ibatis.core;

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;

public class SqlSession {

    private SqlSessionFactory factory;

    public SqlSession(SqlSessionFactory factory) {
        this.factory = factory;
    }


    /**
     * 执行insert语句，向数据库表当中插入记录
     *
     * @param
     * @return
     */
    public int insert(String sqlId, Object pojo) {
        int count = 0;

        try {
            // JDBC代码，执行insert 语句 ，完成插入操作
            Connection connection = factory.getTransaction().getConnection();
            // insert into t_car values(#{id},#{name},#{age})
            String godbatisSql = factory.getMappedStatements().get(sqlId).getSql();
            System.out.println(factory.getMappedStatements().get(sqlId));
            System.out.println(factory.getMappedStatements().get(sqlId).getSql());
            // insert into t_user values(?,?,?)
            String sql = godbatisSql.replaceAll("#\\{[a-zA-Z0-9_$]*}", "?");
            PreparedStatement ps = connection.prepareStatement(sql);
            // 给 ？ 占位符传值
            // 难度是什么？
            // 第一: 你不知道有多少个？
            // 第二：你不知道该将Pojo对象中的那个属性赋值给哪个？
            // ps.String(第几个问号，传什么值)； // 这里都是setString，所以数据库中的字段类型要求都是varchar才行，
            // 这是godbatis 比较失败的地方
            int formIndex = 0;
            int index = 1;

            while (true) {
                int jingIndex = godbatisSql.indexOf("#", formIndex);
                if (jingIndex < 0) {
                    break;
                }

                System.out.println(index);

                int youKuoHaoIndex = godbatisSql.indexOf("}", formIndex);
                String propertyName = godbatisSql.substring(jingIndex + 2, youKuoHaoIndex).trim();
                System.out.println(propertyName);
                formIndex = youKuoHaoIndex + 1;

                // 有属性名id,怎么获取id 的属性值呢？调用 getId()方法
                String getMethodName = "get" + propertyName.toUpperCase().charAt(0) + propertyName.substring(1);
                Method getMethod = pojo.getClass().getDeclaredMethod(getMethodName);
                Object propertyValue = getMethod.invoke(pojo);
                ps.setString(index,propertyValue.toString());
                index++;

            }
            count = ps.executeUpdate();
        } catch (Exception e) {
            e.printStackTrace();
        }

        // 给？占位符传值
        return count;
    }


    /**
     * 查询一个对象
     * @param sqlId
     * @param parameterObj
     * @return
     */
    public Object selectOne(String sqlId, Object parameterObj){
        MappedStatement mappedStatement = factory.getMappedStatements().get(sqlId);
        Connection connection = factory.getTransaction().getConnection();
        // 获取sql语句
        String godbatisSql = mappedStatement.getSql();
        String sql = godbatisSql.replaceAll("#\\{[a-zA-Z0-9_\\$]*}", "?");
        // 执行sql
        PreparedStatement ps = null;
        ResultSet rs = null;
        Object obj = null;
        try {
            ps = connection.prepareStatement(sql);
            ps.setString(1, parameterObj.toString());
            rs = ps.executeQuery();
            if (rs.next()) {
                // 将结果集封装对象，通过反射
                String resultType = mappedStatement.getResultType();
                Class<?> aClass = Class.forName(resultType);
                obj = aClass.newInstance();
                // 给对象obj属性赋值
                ResultSetMetaData rsmd = rs.getMetaData();
                int columnCount = rsmd.getColumnCount();
                for (int i = 1; i <= columnCount; i++) {
                    String columnName = rsmd.getColumnName(i);
                    String setMethodName = "set" + columnName.toUpperCase().charAt(0) + columnName.substring(1);
                    Method setMethod = aClass.getDeclaredMethod(setMethodName, aClass.getDeclaredField(columnName).getType());
                    setMethod.invoke(obj, rs.getString(columnName));
                }
            }
        } catch (Exception e) {
            throw new RuntimeException(e);
        } finally {
            if (rs != null) {
                try {
                    rs.close();
                } catch (SQLException e) {
                    throw new RuntimeException(e);
                }
            }
            try {
                ps.close();
            } catch (SQLException e) {
                throw new RuntimeException(e);
            }
        }
        return obj;
    }

    /**
     * 执行查询语句，返回一个对象，该方法只适合返回一条记录的sql语句
     * @param sqlId
     * @param param
     * @return
     */
    public Object selectOne2(String sqlId,Object param) {
        Object obj = null;
        try {
            Connection connection = factory.getTransaction().getConnection();
            MappedStatement mappedStatement = factory.getMappedStatements().get(sqlId);

            // 这是哪个DQL查询语句
            // select id,name,age from t_user where id = #{id}
            String godbatisSql = mappedStatement.getSql();
            String sql = godbatisSql.replaceAll("#\\{[a-zA-Z0-9_$]*}", "?");
            PreparedStatement ps = connection.prepareStatement(sql);
            // 给占位符传值
            ps.setString(1,param.toString());
            // 查询返回的结果集
            ResultSet rs = ps.executeQuery();
            // 要封装的结果类型
            String resultType = mappedStatement.getResultType();
            // 从结果集中取数据，封装Java对象

            if(rs.next()) {
                // 获取 resultType 的 Class对象
                Class<?> resultTypeClass = Class.forName(resultType);
                // 调用无参数构造方法创建对象
                obj = resultTypeClass.newInstance();  // Object obj = new User();
                // 给User类的id,name,age 属性赋值
                // 给Obj 对象的哪个属性赋哪个值

                /*
                解决问题的关键：将查询结果的列名作为属性名
                列名是id，那么属性就是：id
                列名是name，那么属性名就是：name
                 */
                ResultSetMetaData rsmd = rs.getMetaData();
                int columnCount = rsmd.getColumnCount();

                for (int i = 0;i < columnCount; i++) {
                    String propertyName = rsmd.getColumnName(i+1);
                    // 拼接方法名:
                   String setMethodName =  "set" + propertyName.toUpperCase().charAt(0) + propertyName.substring(1);
                   // 获取set 方法
                   // Method setMethod = resultTypeClass.getDeclaredMethod(setMethodName,resultTypeClass.getDeclaredField(propertyName).getType());
                    Method setMethod = resultTypeClass.getDeclaredMethod(setMethodName,String.class);
                    // 调用set 方法给对象Obj属性赋值
                    setMethod.invoke(obj,rs.getString(propertyName));
                }
            }


        } catch (Exception e) {
            e.printStackTrace();
        }
        return obj;
    }















    /**
     * 提交事务
     */
    public void commit() {
        factory.getTransaction().commit();
    }


    /**
     * 回滚事务
     */
    public void rollback() {
        factory.getTransaction().rollback();
    }


    /**
     * 关闭事务
     */
    public void close() {
        factory.getTransaction().close();
    }
}

2. 将我们自己手写的MyBatiks 名为“godbatis”的框架，使用Maven打包

在这里插入图片描述

在这里插入图片描述

根据所提示的放置的路径的位置，查看本地仓库中是否已经有jar包：

在这里插入图片描述

3. 使用我们自己手写的 MyBatis 名为 “godbatis” 的框架，运行测试

使用godbatis 就和使用MyBatis是一样的。

第一步：准备数据库表t_user。这里我们的手写的MyBatis 框架有所局限，缺陷，只能识别字段类型为 varchar 的数据，所以我们这里的数据表的，字段的类型都定义为了 varchar 类型的。

在这里插入图片描述

表数据：

在这里插入图片描述

第二步：创建模块，普通的Java Maven模块：godbatis-test

第三步：引入依赖

在这里插入图片描述

<?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.rainbowsea</groupId>
    <artifactId>godbatis-test</artifactId>
    <version>1.0-SNAPSHOT</version>
    <packaging>jar</packaging>

    <properties>
        <maven.compiler.source>17</maven.compiler.source>
        <maven.compiler.target>17</maven.compiler.target>
    </properties>

    <dependencies>
        <dependency>
            <groupId>junit</groupId>
            <artifactId>junit</artifactId>
            <version>4.13.2</version>
            <scope>test</scope>
        </dependency>
        <dependency>
            <groupId>org.god.ibatis</groupId>
            <artifactId>godbatis</artifactId>
            <version>1.0-SNAPSHOT</version>
        </dependency>
    </dependencies>

</project>

第四步：编写pojo类

在这里插入图片描述

package com.rainbowsea.godbatis.pojo;

public class User {
    private String id;
    private String name;
    private String age;


    public User(String id, String name, String age) {
        this.id = id;
        this.name = name;
        this.age = age;
    }

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

    public User() {
    }

    public String getId() {
        return id;
    }

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

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public String getAge() {
        return age;
    }

    public void setAge(String age) {
        this.age = age;
    }
}

第五步：编写核心配置文件：godbatis-config.xml

在这里插入图片描述

<?xml version="1.0" encoding="UTF-8" ?>
<configuration>
    <environments default="mybatis">
        <environment id="mybatis">
            <!--            MANAGED 没有用第三框架管理的话，都是会被提交的，没有事务上的管理了。-->
            <transactionManager type="JDBC"/>
<!--            数据源是获取connection对象的 -->
<!--            POOlED UNPOOLED  JNDI -->
<!--            所有的数据源都要实现 JDK带的规范，javax.sql.DataSource-->
            <dataSource type="UNPOOLED">
                <property name="driver" value="com.mysql.cj.jdbc.Driver"/>
                <property name="url" value="jdbc:mysql://localhost:3306/mybatis"/>
                <property name="username" value="root"/>
                <property name="password" value="MySQL123"/>
            </dataSource>
        </environment>
    </environments>
    <mappers>
        <mapper resource="sqlMapper.xml"></mapper>
    </mappers>
</configuration>

第六步：编写sql映射文件：sqlMapper.xml

在这里插入图片描述

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


<!--namespace 一定要是：对应的接口的全限定类名-->
<mapper namespace="user">

    <!--	id 要是 namespace 对应接口上的方法名: -->
    <insert id="insert">
        insert into t_user values(#{id},#{name},#{age})
    </insert>

    <select id="selectAll" resultType="org.god.ibatis.pojo.User">
        select id,name,age from t_user where id = #{id}
    </select>
</mapper>

第七步：编写测试类

在这里插入图片描述

package com.rainbowsea.godbatis.test;

import org.god.ibatis.core.SqlSession;
import org.god.ibatis.core.SqlSessionFactory;
import org.god.ibatis.core.SqlSessionFactoryBuilder;
import org.god.ibatis.pojo.User;
import org.god.ibatis.utils.Resources;
import org.junit.Test;

public class UserMapperTest {

    @Test
    public void testInsertUser() {
        SqlSessionFactoryBuilder sqlSessionFactoryBuilder = new SqlSessionFactoryBuilder();
        SqlSessionFactory sqlSessionFactory = sqlSessionFactoryBuilder.build(Resources.getResourceAsStream("godbatis-config.xml"));
        SqlSession sqlSession = sqlSessionFactory.openSession();


        // 执行SQL insert
        User user = new User("99999", "张三", "20");
        int count = sqlSession.insert("user.insert", user);
        sqlSession.commit();
        sqlSession.close();

    }

    @Test
    public void testSelectOne() {
        SqlSessionFactoryBuilder sqlSessionFactoryBuilder = new SqlSessionFactoryBuilder();
        SqlSessionFactory sqlSessionFactory = sqlSessionFactoryBuilder.build(Resources.getResourceAsStream("godbatis-config.xml"));
        SqlSession sqlSession = sqlSessionFactory.openSession();


        // 执行SQL语句
        Object obj = sqlSession.selectOne2("user.selectAll", "666");
        System.out.println(obj);
        sqlSession.close();
    }
}

在这里插入图片描述

在这里插入图片描述

4. 总结：

  1. 大部分的框架的实现都是：反射机制 + 设计模式 + 注解 的方式实现的。
  2. 这里我们手写MyBatis 就仅仅只实现 insert() 和 selectOne(单条记录的查询)操作，并且有些局限就是，数据库的所有字段类型，都必须是 varchar 类型才行。
  3. 手写MyBatis 让我们更好的了解了 MyBatis 的运行机理，是如何通过反射机制读取相关信息进行设置的。

5. 最后：

“在这个最后的篇章中，我要表达我对每一位读者的感激之情。你们的关注和回复是我创作的动力源泉，我从你们身上吸取了无尽的灵感与勇气。我会将你们的鼓励留在心底，继续在其他的领域奋斗。感谢你们，我们总会在某个时刻再次相遇。”

在这里插入图片描述

标签：return,String,java,基本原理,sql,MyBatis,import,手写,public
From： https://www.cnblogs.com/TheMagicalRainbowSea/p/18250655

相关文章

  • 手写操作系统：二级引导程序
    项目简介在上篇博客，我们完成了主引导扇区的编写，在主引导扇区我们初始化了寄存器，加载了二级引导程序到内存地址0x8000处，并跳转至0x8000处执行，在本文我们将继续编写二级引导程序。在二级引导程序将完成以下任务检测内存容量打开A20地址线设置全局段描述符表进入保护模式......
  • mybatis中映射和注解在数据库进行增删改查的方法
    此两种方法都是基于maven项目的基础上运行主要步骤如下：1、在mevan中的pom文件添加依赖文件： 2、在src目录下编写编写配置文件（mybatis-cfg.xml） 3、编写实体类(Student.java) 4、编写映射文件（StudentMapper.xml） 5、编写测试文件 6、调用JUnit5进行调试核心文件含义：......
  • Mybatis批量更新数据库错误
    问题：记录一次使用Mybatis批量更新数据库的错误，错误信息，Errorupdatingdatabase.Cause:org.postgresql.util.PSQLException:错误:字段"update_time"的类型为timestampwithouttimezone,但表达式的类型为text建议：你需要重写或转换表达式位置：391如下图，说我有一......
  • 设置mybatis 是否打印日志
    @Bean(name="dbzyqsV102SqlSessionFactory")//@Qualifier表示查找Spring容器中名字为blogDataSource的对象publicSqlSessionFactorydbzyqsV102SqlSessionFactory(@Qualifier("dbzyqsV102DataSource")DataSourcedatasource)throwsEx......
  • 一款新的超强的ORM框架 mybatis-mp ！！！
    官网：https://mybatis-mp.cn1.单表查询SysUsersysUser=QueryChain.of(sysUserMapper).eq(SysUser::getId,1).eq(SysUser::getUserName,'admin').get();2.VO映射@Data@ResultEntity(SysUser.class)publicclassSysUserVo{p......
  • RocketMQ知识总结（基本原理）
    文章收录在网站：http://hardyfish.top/文章收录在网站：http://hardyfish.top/文章收录在网站：http://hardyfish.top/文章收录在网站：http://hardyfish.top/基本原理总体架构图零拷贝零拷贝技术是一个思想，指的是指计算机执行操作时，CPU不需要先将数据从某处内存复制到另......
  • MyBatis-Plus学习笔记
    使用SpringBoot创建工程并添加依赖pom.xml版本：SpringBoot2.3JDK1.8<?xmlversion="1.0"encoding="UTF-8"?><!--定义项目元数据，基于MavenPOM4.0.0模型--><projectxmlns="http://maven.apache.org/POM/4.0.0"xmlns:xsi="http://www.w3.......
  • sharding-jdbc 兼容 MybatisPlus的动态数据源
    背景：之前的项目做读写分离的时候用的MybatisPlus的动态数据做的，很多地方使用的@DS直接指定的读库或者写库实现的业务；随着表数据量越来越大，现在打算把比较大的表进行水平拆分，准备使用ShardingJDBC实现，但是发现两者配合起来并不是那么顺利，网上大部分文章都是直接把整个Sharding的......
  • springboot整合mybatis
    第一部分：创建springboot文件 第一步：打开软件，点击file，点击new然后选择module，在右侧选择springboot第二步：选择配置和JDK以及java版本①选择maven类型②选择JDK1.8版本③选择java8版本④选择jar包类型http://t.csdnimg.cn/XeplRhttp://t.csdnimg.cn/XeplR第三步：选择......
  • MyBatisPlus分页查询
     一、导入依赖<!--MyBatis-plus的依赖--><dependency><groupId>com.baomidou</groupId><artifactId>mybatis-plus-boot-starter</artifactId><version>3.5.4</version></dependency><!--mysql的依赖--&......

赞助商

阅读排行

网站主页关于我们联系我们网站地图

本网站内容转载自其他媒体，侵权联系[admin##ips99.com]。

Copyright © 2020-2023 IPS99 版权所有 IPS99