Java如何切换数据源
在开发Java应用程序时，经常会遇到切换数据源的需求。比如，当我们需要在不同的环境中使用不同的数据库，或者需要在运行过程中切换到另一个数据库。本文将介绍一种常见的解决方案，并提供相应的代码示例。
问题描述
假设我们正在开发一个电商系统，该系统需要连接到不同的数据库来存储商品信息。我们希望能够根据需要来切换数据源，比如在开发阶段使用本地数据库，在测试阶段使用测试数据库，在生产环境中使用生产数据库。
解决方案
为了实现数据源的切换，我们可以使用Java的连接池技术，并结合工厂模式来动态创建数据源。
第一步：配置数据源信息
首先，我们需要在配置文件中定义不同数据源的信息。这些信息包括数据库的URL、用户名、密码等。我们可以使用.properties文件来存储这些信息，也可以使用其他格式的配置文件。
示例配置文件（db.properties）内容如下：
# 数据源1
db.url1=jdbc:mysql://localhost:3306/db1
db.username1=root
db.password1=123456
# 数据源2
db.url2=jdbc:mysql://localhost:3306/db2
db.username2=root
db.password2=123456
第二步：创建数据源工厂
接下来，我们需要创建一个数据源工厂，用于根据配置文件中的信息动态创建数据源。我们可以使用Apache的[BasicDataSource](
示例代码如下：
import org.apache.commons.dbcp2.BasicDataSource;
public class DataSourceFactory {
public static BasicDataSource createDataSource(String url, String username, String password) {
BasicDataSource dataSource = new BasicDataSource();
dataSource.setDriverClassName("com.mysql.jdbc.Driver");
dataSource.setUrl(url);
dataSource.setUsername(username);
dataSource.setPassword(password);
return dataSource;
}
}
第三步：切换数据源
在需要切换数据源的地方，我们可以使用工厂模式来创建相应的数据源。
示例代码如下：
import javax.sql.DataSource;
public class ProductService {
private DataSource dataSource;
public void setDataSource(DataSource dataSource) {
this.dataSource = dataSource;
}
// 其他业务逻辑...
}
在上述示例代码中，我们通过调用setDataSource()方法来设置数据源。这样，在后续的业务逻辑中，我们就可以使用dataSource对象来获取连接并执行SQL语句了。
第四步：使用配置文件来切换数据源
最后，我们可以使用配置文件来切换数据源。
示例代码如下：
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.Properties;
public class Main {
public static void main(String[] args) {
// 加载配置文件
Properties properties = new Properties();
try (InputStream input = new FileInputStream("db.properties")) {
properties.load(input);
} catch (IOException e) {
e.printStackTrace();
}
// 获取配置信息
String url = properties.getProperty("db.url1");
String username = properties.getProperty("db.username1");
String password = properties.getProperty("db.password1");
// 创建数据源
DataSource dataSource = DataSourceFactory.createDataSource(url, username, password);
// 设置数据源
ProductService productService = new ProductService();
productService.setDataSource(dataSource);
// 其他业务逻辑...
}
}
在上述示例代码中，我们首先加载配置文件，然后根据配置文件中定义的数据源信息创建数据源，并设置给ProductService对象。这样，我们就成功地切换了数据源。
序列图
下面是一个使用数据源切换方案的简单序列图：
sequenceDiagram
participant Main
participant ProductService
participant DataSourceFactory
Main ->> ProductService: 设置数据源
ProductService ->> DataSourceFactory: 创建数据源
DataSourceFactory ->> BasicDataSource: 创建连接池
BasicDataSource -->> DataSourceFactory: 返回数据源
DataSourceFactory -->> ProductService: 返回数据源
ProductService ->> DataSource: 获取连接
DataSource ->> ProductService: 返回连接
ProductService ->> DataSource: 执行SQL语句
DataSource -->> ProductService: 返回
