MyBatis 源码解析：SqlSession 接口设计与实现

捕风捉你2024-09-02 11:05

摘要

在 MyBatis 中，SqlSession 是执行数据库操作的核心接口。它负责处理增删改查（CRUD）等数据库操作，并管理事务的提交与回滚。你是否想了解 SqlSession 的内部设计和实现原理？本文将通过自定义实现一个基本的 SqlSession 接口，带你深入探讨 MyBatis 中 SqlSession 的设计与实现机制，并帮助你更好地掌握 SqlSession 的使用方法。

前言

SqlSession 是 MyBatis 的核心接口之一，它提供了与数据库交互的基本方法，如增删改查操作、事务管理等。理解 SqlSession 的设计与实现，有助于掌握 MyBatis 的数据库操作流程。本文将通过自定义实现一个简化版的 SqlSession 接口，并深入解析 MyBatis 中 SqlSession 的实现细节。

自定义实现：基本的增删改查操作的 SqlSession 接口

目标与功能

我们将实现一个简化版的 SqlSession 接口，该接口将支持基本的 CRUD 操作，并管理事务的提交与回滚。通过这个实现，我们可以更好地理解 MyBatis 是如何设计和管理数据库操作的。

核心流程

  1. 执行 SQL 操作：提供基本的增删改查方法，执行 SQL 语句。
  2. 事务管理：支持手动提交与回滚事务。
  3. 资源管理 ：确保 SqlSession 使用后能够正确关闭资源。

实现过程

1. 定义 SqlSession 接口

我们定义 SqlSession 接口，包含了基本的增删改查方法和事务管理方法。

java 复制代码 
/**
 * SqlSession 接口，定义了 MyBatis 中执行 SQL 的核心操作。
 * 包括基本的增删改查方法和事务管理方法。
 */
public interface SqlSession {
    <T> T selectOne(String statement);

    int insert(String statement);

    int update(String statement);

    int delete(String statement);

    void commit();

    void rollback();

    void close();
}
2. 实现 DefaultSqlSession 类

DefaultSqlSession 类是 SqlSession 接口的一个实现类，负责具体执行 SQL 操作，并管理事务和资源。

java 复制代码 
import java.sql.Connection;
import java.sql.Statement;
import java.sql.SQLException;

/**
 * DefaultSqlSession 是 SqlSession 接口的一个简化实现。
 * 它基于 Configuration 配置执行 SQL 操作，并管理事务和资源。
 */
public class DefaultSqlSession implements SqlSession {
    private final Configuration configuration;
    private final Connection connection;
    private final boolean autoCommit;

    public DefaultSqlSession(Configuration configuration, Connection connection, boolean autoCommit) {
        this.configuration = configuration;
        this.connection = connection;
        this.autoCommit = autoCommit;
    }

    @Override
    public <T> T selectOne(String statement) {
        return executeQuery(statement);
    }

    @Override
    public int insert(String statement) {
        return executeUpdate(statement);
    }

    @Override
    public int update(String statement) {
        return executeUpdate(statement);
    }

    @Override
    public int delete(String statement) {
        return executeUpdate(statement);
    }

    private <T> T executeQuery(String statement) {
        try (Statement stmt = connection.createStatement()) {
            System.out.println("Executing query: " + statement);
            stmt.execute(statement);
            // 简化的实现，实际操作中应包括结果集处理
            return null;
        } catch (SQLException e) {
            throw new RuntimeException("Error executing query", e);
        }
    }

    private int executeUpdate(String statement) {
        try (Statement stmt = connection.createStatement()) {
            System.out.println("Executing update: " + statement);
            return stmt.executeUpdate(statement);
        } catch (SQLException e) {
            throw new RuntimeException("Error executing update", e);
        }
    }

    @Override
    public void commit() {
        try {
            if (!autoCommit) {
                connection.commit();
            }
        } catch (SQLException e) {
            throw new RuntimeException("Failed to commit transaction", e);
        }
    }

    @Override
    public void rollback() {
        try {
            if (!autoCommit) {
                connection.rollback();
            }
        } catch (SQLException e) {
            throw new RuntimeException("Failed to rollback transaction", e);
        }
    }

    @Override
    public void close() {
        try {
            connection.close();
        } catch (SQLException e) {
            throw new RuntimeException("Failed to close connection", e);
        }
    }
}
  • executeQuery 方法 ：负责执行 SELECT 查询，并返回结果集（这里的实现进行了简化，未返回实际数据）。
  • executeUpdate 方法 ：负责执行 INSERT、UPDATE 和 DELETE 操作，并返回影响的行数。
  • commit 和 rollback 方法 ：用于事务管理，根据 autoCommit 的设置决定是否自动提交或回滚事务。
  • close 方法：负责关闭数据库连接，确保资源不被泄露。
3. 测试 DefaultSqlSession

我们编写一个简单的测试类来验证 DefaultSqlSession 的功能。

java 复制代码 
import java.sql.Connection;
import java.sql.DriverManager;

public class DefaultSqlSessionTest {
    public static void main(String[] args) {
        try {
            // 模拟数据库连接
            Connection connection = DriverManager.getConnection("jdbc:h2:mem:testdb", "sa", "");

            // 初始化 SqlSession
            DefaultSqlSession sqlSession = new DefaultSqlSession(new Configuration(), connection, false);

            // 执行 SQL 操作
            sqlSession.insert("INSERT INTO users (name, age) VALUES ('John', 30)");
            sqlSession.update("UPDATE users SET age = 31 WHERE name = 'John'");
            sqlSession.selectOne("SELECT * FROM users WHERE name = 'John'");
            sqlSession.delete("DELETE FROM users WHERE name = 'John'");

            // 提交事务
            sqlSession.commit();

            // 关闭资源
            sqlSession.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}

自定义实现类图

Configuration - EnvironmentConfig environmentConfig +getEnvironmentConfig() +setEnvironmentConfig(EnvironmentConfig) <<interface>> SqlSession +selectOne(String statement) +int insert(String statement) +int update(String statement) +int delete(String statement) +commit() +rollback() +close() DefaultSqlSession - Configuration configuration - Connection connection - boolean autoCommit +selectOne(String statement) +int insert(String statement) +int update(String statement) +int delete(String statement) +commit() +rollback() +close() Connection EnvironmentConfig

代码解析流程图

否 是 开始 初始化 SqlSession 执行 SQL 操作 是否自动提交? 提交或回滚事务 直接返回结果 关闭 SqlSession 并释放资源 结束

源码解析：MyBatis 中的 SqlSession 实现

1. SqlSession 的设计与实现

在 MyBatis 中，SqlSession 是数据库操作的核心接口，它为用户提供了与数据库交互的基本方法，如增删改查操作、事务管理等。MyBatis 提供了多个 SqlSession 实现，其中 DefaultSqlSession 是最常用的实现。

java 复制代码 
public class DefaultSqlSession implements SqlSession {
    private final Configuration configuration;
    private final Executor executor;
    private final boolean autoCommit;

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

    @Override
    public <T> T selectOne(String statement, Object parameter) {
        List<T> list = this.selectList(statement, parameter);
        if (list.size() == 1) {
            return list.get(0);
        } else if (list.size() > 1) {
            throw new TooManyResultsException("Expected one result (or null) to be returned by selectOne(), but found: " + list.size());
        } else {
            return null;
        }
    }

    // 其他 CRUD 方法的实现...
}
  • Executor 对象 ：MyBatis 中的 Executor 负责执行真正的 SQL 操作，并将结果映射到 Java 对象中。
  • selectOne 方法 ：执行 SELECT 操作，并返回单个结果，如果结果集包含多条数据则抛出异常。

2. 事务管理与资源管理

MyBatis 中的事务管理通过 SqlSession 的 commit 和 rollback 方法实现。在 DefaultSqlSession 中，这些方法调用数据库连接的 commit 和 rollback 方法来完成事务的提交和回滚。此外，SqlSession 还提供了 close 方法，用于关闭数据库连接和释放资源。

java 复制代码 
@Override
public void commit() {
    try {
        executor.commit(isTransactionOpen(), false);
    } catch (SQLException e) {
        throw new RuntimeException("Failed to commit transaction", e);
    }
}

@Override
public void rollback() {
    try {
        executor.rollback(isTransactionOpen());
    } catch (SQLException e) {
        throw new RuntimeException("Failed to rollback transaction", e);
    }
}

@Override
public void close() {
    try {
        executor.close(isTransactionOpen());
    } catch (SQLException e) {
        throw new RuntimeException("Failed to close session", e);
    }
}
  • commit 方法 ：提交当前事务，如果 autoCommit 为 false，则手动提交。
  • rollback 方法 ：回滚当前事务，如果 autoCommit 为 false，则手动回滚。
  • close 方法 ：关闭当前 SqlSession，释放资源，包括关闭数据库连接和清理缓存。

3. MyBatis 中 SqlSession 的完整工作流程

在 MyBatis 中，SqlSession 的工作流程可以总结为以下几个步骤：

  1. 初始化 SqlSession ：创建 SqlSession 时，初始化数据库连接和执行器，并根据配置决定是否启用自动提交。
  2. 执行 SQL 操作 ：调用 selectOne、insert、update、delete 等方法执行具体的 SQL 操作，并返回结果。
  3. 事务管理 ：根据需要调用 commit 和 rollback 方法进行事务管理，确保数据的一致性。
  4. 资源管理 ：操作完成后，调用 close 方法关闭 SqlSession，释放资源，防止资源泄漏。

总结与互动

通过本文，我们详细探讨了 MyBatis 中 SqlSession 的设计与实现，并通过自定义实现加深了对 SqlSession 的理解。掌握这些知识有助于在开发中更好地管理数据库操作，确保事务的正确性和资源的有效管理。

如果您觉得这篇文章对您有帮助，请点赞、收藏并关注！此外，欢迎在评论区留言，与我们分享您的见解或提出疑问！

相关推荐
Java小白笔记5 小时前
关于使用Mybatis-Plus 自动填充功能失效问题
spring boot·后端·mybatis
计算机学姐8 小时前
基于SpringBoot+Vue的篮球馆会员信息管理系统
java·vue.js·spring boot·后端·mysql·spring·mybatis
程序员大金9 小时前
基于SpringBoot+Vue+MySQL的智能物流管理系统
java·javascript·vue.js·spring boot·后端·mysql·mybatis
惜.己13 小时前
MyBatis中一对多关系的两种处理方法
java·开发语言·后端·sql·mysql·mybatis·idea
终末圆13 小时前
MyBatis动态SQL中的`if`标签使用【后端 19】
java·数据结构·数据库·sql·算法·spring·mybatis
飞翔的佩奇15 小时前
Java项目: 基于SpringBoot+mybatis+maven医院管理系统(含源码+数据库+任务书+开题报告+毕业论文)
java·数据库·spring boot·毕业设计·maven·mybatis·医院管理系统
这孩子叫逆16 小时前
35. MyBatis中的缓存失效机制是如何工作的？
java·spring·mybatis
ChinaRainbowSea18 小时前
十八，Spring Boot 整合 MyBatis-Plus 的详细配置
java·数据库·spring boot·spring·mybatis·web
飞翔的佩奇1 天前
Java项目: 基于SpringBoot+mybatis+maven课程答疑系统(含源码+数据库+毕业论文)
java·数据库·spring boot·毕业设计·maven·mybatis·课程答疑
OceanSky61 天前
Mybatis中sql数组为空判断
数据库·sql·mybatis·数组判空
热门推荐
01RAG 实践- Ollama+RagFlow 部署本地知识库02组基轨迹建模 GBTM的介绍与实现（Stata 或 R）032024年高教社杯数学建模国赛C题超详细解题思路分析04yolov8实战第五天——yolov8+ffmpg实时视频流检测并进行实时推流——（推流，保姆教学）05苍穹外卖面试总结06【2024数模国赛赛题思路公开】国赛B题思路丨附可运行代码丨无偿自提07浏览器插件——ModHeader 的分享和学习08Coze扣子平台完整体验和实践（附国内和国际版对比）09【2024高教社杯全国大学生数学建模竞赛】B题 生产过程中的决策问题——解题思路 代码 论文10CCF-CSP认证考试 202406-3 文本分词 100分题解