MyBatis 源码解析：Environment 与 DataSource 配置实现

devtools/2024/11/20 0:24:48/

前言

在 MyBatis 框架中，Environment 和 DataSource 是配置管理的核心部分。Environment 负责管理不同的运行环境（如开发、测试、生产环境），而 DataSource 则管理数据库连接的配置和管理。理解这两个组件的工作原理有助于我们更好地掌握 MyBatis 的配置管理和框架初始化过程。

本篇文章将通过自定义实现一个环境配置管理类来简化 Environment 和 DataSource 的实现，随后深入解析 MyBatis 中这两个组件的源码实现细节。希望通过这些内容，帮助读者全面理解 MyBatis 的环境和数据源配置。

自定义实现：环境与数据源配置管理

目标与功能

我们将实现一个简化版的环境配置管理类，功能包括：

  1. 支持多环境的配置管理（如开发、测试、生产环境）。
  2. 支持不同环境的事务管理器配置（如 JDBC 或 JNDI）。
  3. 支持多种数据源配置方式。
  4. 提供接口切换当前环境，并获取对应的数据库连接。

实现过程

1. 定义环境配置类

首先，我们定义一个名为 EnvironmentConfig 的类，用于存储每个环境的配置信息，包括环境名称、事务管理器类型以及数据源配置项。

import java.util.HashMap;
import java.util.Map;/*** EnvironmentConfig 类用于存储单个环境的配置，包括环境名称、事务管理器类型和数据源配置。*/
public class EnvironmentConfig {private String environmentName;  // 环境名称，如 "development", "production"private String transactionManagerType;  // 事务管理器类型，如 "JDBC", "MANAGED"private Map<String, String> dataSourceConfig = new HashMap<>();  // 数据源配置项/*** 构造函数，初始化环境名称和事务管理器类型。* @param environmentName 环境名称* @param transactionManagerType 事务管理器类型*/public EnvironmentConfig(String environmentName, String transactionManagerType) {this.environmentName = environmentName;this.transactionManagerType = transactionManagerType;}/*** 设置数据源配置项。* @param key 配置项的键（如 "url", "username", "password"）* @param value 配置项的值*/public void setDataSourceConfig(String key, String value) {dataSourceConfig.put(key, value);}/*** 获取数据源配置项的 Map。* @return 包含所有数据源配置项的 Map*/public Map<String, String> getDataSourceConfig() {return dataSourceConfig;}public String getEnvironmentName() {return environmentName;}public String getTransactionManagerType() {return transactionManagerType;}
}
2. 创建多环境配置管理类

我们需要一个 EnvironmentManager 类来管理多个 EnvironmentConfig 实例。该类应支持添加环境配置、切换当前环境，并获取当前环境的配置项。

import java.util.HashMap;
import java.util.Map;/*** EnvironmentManager 类用于管理多个环境配置，并支持切换不同环境。*/
public class EnvironmentManager {private Map<String, EnvironmentConfig> environments = new HashMap<>();  // 存储多个环境配置private EnvironmentConfig currentEnvironment;  // 当前选定的环境配置/*** 添加一个环境配置。* @param environmentConfig 要添加的环境配置对象*/public void addEnvironment(EnvironmentConfig environmentConfig) {environments.put(environmentConfig.getEnvironmentName(), environmentConfig);}/*** 根据环境名称切换当前环境配置。* @param environmentName 要切换的环境名称*/public void switchEnvironment(String environmentName) {currentEnvironment = environments.get(environmentName);if (currentEnvironment == null) {throw new IllegalArgumentException("环境配置 '" + environmentName + "' 不存在！");}}/*** 获取当前环境的配置。* @return 当前环境的 EnvironmentConfig 对象*/public EnvironmentConfig getCurrentEnvironment() {return currentEnvironment;}/*** 打印当前所有的环境配置名称。*/public void printAllEnvironments() {System.out.println("Available environments: " + environments.keySet());}
}
3. 实现数据源管理

数据源配置是环境配置的重要部分。我们实现一个 DataSourceManager 类，该类可以根据 EnvironmentConfig 中的配置项创建数据库连接。

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.util.Map;/*** DataSourceManager 类用于管理数据源，并根据环境配置创建数据库连接。*/
public class DataSourceManager {/*** 根据当前环境的配置创建数据库连接。* @param environmentConfig 当前环境配置* @return 数据库连接对象* @throws SQLException 数据库连接失败时抛出异常*/public static Connection createConnection(EnvironmentConfig environmentConfig) throws SQLException {Map<String, String> config = environmentConfig.getDataSourceConfig();String url = config.get("url");String username = config.get("username");String password = config.get("password");// 判断事务管理器类型，假设这里只支持 JDBCif ("JDBC".equalsIgnoreCase(environmentConfig.getTransactionManagerType())) {return DriverManager.getConnection(url, username, password);} else {throw new UnsupportedOperationException("当前事务管理器类型不支持: " + environmentConfig.getTransactionManagerType());}}
}
4. 使用示例

我们可以创建多个 EnvironmentConfig 实例来代表不同的环境，并为每个环境配置不同的数据源，然后通过 EnvironmentManager 进行环境切换和数据库连接管理。

public class Main {public static void main(String[] args) {// 创建开发环境的配置EnvironmentConfig devEnvironment = new EnvironmentConfig("development", "JDBC");devEnvironment.setDataSourceConfig("url", "jdbc:mysql://localhost:3306/dev_db");devEnvironment.setDataSourceConfig("username", "dev_user");devEnvironment.setDataSourceConfig("password", "dev_password");// 创建生产环境的配置EnvironmentConfig prodEnvironment = new EnvironmentConfig("production", "JDBC");prodEnvironment.setDataSourceConfig("url", "jdbc:mysql://localhost:3306/prod_db");prodEnvironment.setDataSourceConfig("username", "prod_user");prodEnvironment.setDataSourceConfig("password", "prod_password");// 创建环境管理器EnvironmentManager environmentManager = new EnvironmentManager();environmentManager.addEnvironment(devEnvironment);environmentManager.addEnvironment(prodEnvironment);// 打印所有环境environmentManager.printAllEnvironments();// 切换到开发环境并创建连接environmentManager.switchEnvironment("development");try (Connection devConnection = DataSourceManager.createConnection(environmentManager.getCurrentEnvironment())) {System.out.println("Connected to Development DB: " + devConnection.getMetaData().getURL());} catch (SQLException e) {e.printStackTrace();}// 切换到生产环境并创建连接environmentManager.switchEnvironment("production");try (Connection prodConnection = DataSourceManager.createConnection(environmentManager.getCurrentEnvironment())) {System.out.println("Connected to Production DB: " + prodConnection.getMetaData().getURL());} catch (SQLException e) {e.printStackTrace();}}
}

下面是为自定义实现部分准备的类图和流程图。

自定义实现类图

uses
manages
EnvironmentConfig
- String environmentName
- String transactionManagerType
- Map dataSourceConfig
+getEnvironmentName()
+getTransactionManagerType()
+getDataSourceConfig()
+setDataSourceConfig(String key, String value)
EnvironmentManager
- Map environments
- EnvironmentConfig currentEnvironment
+addEnvironment(EnvironmentConfig environmentConfig)
+switchEnvironment(String environmentName)
+getCurrentEnvironment()
+printAllEnvironments()
DataSourceManager
+createConnection(EnvironmentConfig environmentConfig)
类图解释：
  • EnvironmentConfig 类用于存储单个环境的配置，包括环境名称、事务管理器类型和数据源配置。
  • EnvironmentManager 类管理多个 EnvironmentConfig 实例，并支持环境切换。
  • DataSourceManager 类根据 EnvironmentConfig 提供的数据源配置创建数据库连接。

自定义实现流程图
开始
定义 EnvironmentConfig 类
实现 EnvironmentManager 类
实现 DataSourceManager 类
创建并配置多个环境
切换到特定环境
创建数据库连接
完成
流程图解释：
  1. 定义 EnvironmentConfig 类：用于存储环境的配置信息。
  2. 实现 EnvironmentManager 类：管理多个环境并支持环境切换。
  3. 实现 DataSourceManager 类：根据当前环境配置创建数据库连接。
  4. 创建并配置多个环境：为不同的环境设置特定的数据库配置。
  5. 切换到特定环境：根据需求切换到目标环境。
  6. 创建数据库连接：使用 DataSourceManager 进行数据库连接的创建。
  7. 完成：整个流程结束。
5. 自定义实现总结

以上的自定义实现展示了如何在一个应用程序中管理多种环境配置，并根据不同环境动态创建数据库连接。该实现主要涵盖了以下几点：

  1. 环境配置管理：通过 EnvironmentConfig 类存储不同环境的配置信息，包括事务管理器和数据源配置。
  2. 多环境切换：使用 EnvironmentManager 类可以轻松切换不同的环境配置。
  3. 数据源管理：通过 DataSourceManager 类可以根据当前环境配置动态创建数据库连接。

这种实现方式与 MyBatis 框架中的 Environment 和 DataSource 管理有着异曲同工之妙，下面我们将深入解析 MyBatis 源码中的相关实现。

源码解析：MyBatis 中的 Environment 与 DataSource

1. Environment 的初始化

在 MyBatis 中，Environment 类用于配置不同的运行环境。每个 Environment 都包括一个事务管理器和数据源。

public class Environment {private final String id;  // 环境IDprivate final TransactionFactory transactionFactory;  // 事务管理器工厂private final DataSource dataSource;  // 数据源public Environment(String id, TransactionFactory transactionFactory, DataSource dataSource) {this.id = id;this.transactionFactory = transactionFactory;this.dataSource = dataSource;}// Builder 模式用于构建 Environment 对象public static class Builder {private final String id;private TransactionFactory transactionFactory;private DataSource dataSource;public Builder(String id) {this.id = id;}public Builder transactionFactory(TransactionFactory transactionFactory) {this.transactionFactory = transactionFactory;return this;}public Builder dataSource(DataSource dataSource) {this.dataSource = dataSource;return this;}public Environment build() {return new Environment(id, transactionFactory, dataSource);}}
}
代码详解
  • Environment 类：表示 MyBatis 的运行环境，每个环境都包含了 id、transactionFactory 和 dataSource。
  • Builder 模式：用于创建 Environment 实例，通过链式调用设置事务管理器和数据源。

2. 解析 <environments> 节点

environmentsElement 方法用于解析 MyBatis 配置文件中的 <environments> 节点，并选择一个默认的环境进行初始化。

private void environmentsElement(XNode context) throws Exception {if (context != null) {String defaultEnvironment = context.getStringAttribute("default");  // 获取默认环境for (XNode child : context.getChildren()) {String id = child.getStringAttribute("id");if (isSpecifiedEnvironment(id, defaultEnvironment)) {TransactionFactory txFactory = transactionManagerElement(child.evalNode("transactionManager"));DataSourceFactory dsFactory = dataSourceElement(child.evalNode("dataSource"));DataSource dataSource = dsFactory.getDataSource();Environment.Builder environmentBuilder = new Environment.Builder(id).transactionFactory(txFactory).dataSource(dataSource);configuration.setEnvironment(environmentBuilder.build());}}}
}
代码详解
  • defaultEnvironment：从 <environments> 节点中获取默认的环境 ID。
  • transactionManagerElement 和 dataSourceElement：分别解析 <transactionManager> 和 <dataSource> 节点，返回对应的 TransactionFactory 和 DataSource 对象。
  • isSpecifiedEnvironment：判断当前解析的环境 ID 是否为默认环境。
  • configuration.setEnvironment：最终将解析后的 Environment 对象设置到 MyBatis 的 Configuration 中。

3. 配置 DataSource

在 MyBatis 中，DataSource 的配置支持多种类型，如 JDBC 和 JNDI。MyBatis 提供了 DataSourceFactory 接口，允许开发者自定义数据源的创建逻辑。

public interface DataSourceFactory {void setProperties(Properties props);  // 设置数据源配置属性DataSource getDataSource();  // 获取数据源实例
}

MyBatis 自带了多个 DataSourceFactory 实现，如 UnpooledDataSourceFactory 和 PooledDataSourceFactory，分别对应不带连接池和带连接池的数据源配置。

配置示例

假设我们使用 UnpooledDataSourceFactory 配置一个简单的 JDBC 数据源：

Properties props = new Properties();
props.setProperty("driver", "com.mysql.cj.jdbc.Driver");
props.setProperty("url", "jdbc:mysql://localhost:3306/mybatis");
props.setProperty("username", "root");
props.setProperty("password", "password");DataSourceFactory dsFactory = new UnpooledDataSourceFactory();
dsFactory.setProperties(props);DataSource dataSource = dsFactory.getDataSource();

4. 总结

在这篇文章中，我们通过自定义实现与 MyBatis 源码解析，深入理解了 Environment 和 DataSource 的配置及其实现原理。Environment 作为 MyBatis 的核心配置类，允许开发者灵活地管理多环境配置，而 DataSource 则是数据库连接的关键组件。掌握这些知识点，将有助于你在实际项目中更好地应用 MyBatis。

如果你觉得这篇文章对你有帮助，请点赞、收藏并关注，这样你就不会错过我之后的更多精彩内容！如果你有任何问题或见解，欢迎在评论区留言讨论！


http://www.ppmy.cn/devtools/104760.html

相关文章

线程池在接受到30个比较耗时的任务时的状态，在前面30个比较耗时的任务还没执行完成的情况下，再来多少个任务会触发拒绝策略？

线程池在接受到30个比较耗时的任务时的状态，在前面30个比较耗时的任务还没执行完成的情况下，再来多少个任务会触发拒绝策略？

目录 一、提出问题 二、解答 问题 1: 线程池在接受到30个比较耗时的任务时的状态 问题 2: 在前面30个比较耗时的任务还没执行完成的情况下&#xff0c;再来多少个任务会触发拒绝策略&#xff1f; 总结 一、提出问题 我们首先自定义一个线程池&#xff1a; new ThreadPoo…
阅读更多...
【开端】基于nginx部署的具有网关的web日志分析

【开端】基于nginx部署的具有网关的web日志分析

一、绪论 基于nginx部署的具有网关的web日志分析&#xff0c;我们可以分析的日志有nginx的access.log &#xff0c;网关的日志和应用的日志 二、日志分析 1、nginx日志 参数 说明 示例 $remote_addr 客户端地址 172.17.0.1 $remote_user 客户端用户名称 -- $time_lo…
阅读更多...
【零知识证明】Groth16

【零知识证明】Groth16

一 相关介绍 1 Groth16 Groth16是一种用于零知识证明系统中的简洁非交互式知识论证(SNARK)协议&#xff0c;是一种表示计算的方式&#xff0c;在算术电路上操作&#xff0c;使用加法和乘法门。使用配对友好的椭圆曲线来实现高效的证明生成和验证。 Groth16的主要特点包括: …
阅读更多...
机器学习之实战篇——预测二手房房价(线性回归）

机器学习之实战篇——预测二手房房价(线性回归）

机器学习之实战篇——预测二手房房价(线性回归&#xff09; 前言数据集和实验文件下载相关文章推荐实验过程导入相关模块数据预处理手动梯度下降训练使用scikit-learn随机梯度下降 前言 实验中难免有许多缺陷和错误&#xff0c;望批评指正&#xff01; 数据集和实验文件下载 …
阅读更多...
Python进阶————面向对象高级

Python进阶————面向对象高级

面向对象高级 前言一、继承1.1. 单继承1.2. 多继承1.3. 方法重写1.4. 子类调用父类方法1.4.1 父类名.父类方法名()1.4.2 super().父类方法名() 1.5. 多层继承 二、封装2.1. 私有属性2.2. 私有方法 三、多态3.1. 多态的条件3.2. 多态的定义 四、面向对象的其他特性4.1. 对象属性…
阅读更多...
【C++ Primer Plus习题】8.6

【C++ Primer Plus习题】8.6

问题: 解答: #include <iostream> using namespace std;template <typename T> T maxn(T arr[], int len)//通用 {T max 0;for (int i 0; i < len; i){if (max < arr[i]){max arr[i];}}return max; }template<> const char* maxn<const char*&g…
阅读更多...
基于 ASP.NET的教材管理信息系统的设计与实现（最新定制开发，阿龙原创设计）✅

基于 ASP.NET的教材管理信息系统的设计与实现（最新定制开发，阿龙原创设计）✅

博主介绍&#xff1a; ✌我是阿龙&#xff0c;一名专注于Java技术领域的程序员&#xff0c;全网拥有10W粉丝。作为CSDN特邀作者、博客专家、新星计划导师&#xff0c;我在计算机毕业设计开发方面积累了丰富的经验。同时&#xff0c;我也是掘金、华为云、阿里云、InfoQ等平台…
阅读更多...
谷歌浏览器http自动跳转https问题

谷歌浏览器http自动跳转https问题

1.地址栏输入&#xff1a; chrome://net-internals/#hsts 2.找到底部Delete domain security policies一栏&#xff0c;输入想处理的域名&#xff0c;点击delete。 3.搞定了&#xff0c;再次访问http域名不再自动跳转https了。
阅读更多...
最新文章

Copyright @ 2022~2023