Spring入门(七)：Spring Profile使用讲解

提示：扫一扫查出行
【扫一扫了解最新限行尾号】
复制提示

1. 使用场景

在日常的开发工作中，我们经常需要将程序部署到不同的环境，比如Dev开发环境，QA测试环境，Prod生产环境，这些环境下的一些配置肯定是不一样的，比如数据库配置，Redis配置，RabbitMQ配置。

如果每次切换发布环境，都需要修改配置重新构建的话，那对程序员来说将是噩梦，针对这种场景，Spring提供了@Profile注解来实现按照不同的环境装配不同的bean，进而实现程序只需构建一次，但可以部署到多个环境。

2. 配置profile bean

为了更好的理解，我们通过具体的代码示例来理解下Spring profile的使用方法，这里我们以数据库配置为例。

说明：本篇博客的重点是讲解@Profile注解的使用，数据库的操作只是辅助理解@Profile，因此不会讲解的太详细，不过后续会单独写博客讲解

假设我们有3套环境（Dev，QA，Prod），这3套环境的数据库都使用的是mysql，但是其地址，用户名，密码都不一样，那么在Java配置中，该如何声明这些bean呢？

2.1 Java配置中配置profile bean

首先需要了解的是，@Profile注解是从Spring 3.1版本中开始引入的，并且在这个版本中，@Profile注解只能在类级别上使用。

因此我们可以按照环境分别创建数据库配置，如下所示：

Dev环境下的数据库配置：

使用上述代码需要在pom.xml中添加如下依赖：

<dependency> <groupId>org.apache.commons</groupId> <artifactId>commons-dbcp2</artifactId> <version>2.7.0</version> </dependency>

注意事项：如果类级别上使用了@Profile("dev")，那么该类中的所有bean都会在profile为dev时创建。

QA环境下的数据库配置：

Prod环境下的数据库配置：

不过从Spring 3.2开始，@Profile注解可以与@Bean注解一起在方法级别上使用。

这也就使得我们可以将刚刚的3个配置类合并成1个配置类（推荐该方式），如下所示：

package chapter03.profile;import org.apache.commons.dbcp2.BasicDataSource; import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Configuration; import org.springframework.context.annotation.Profile;import javax.sql.DataSource;@Configuration public class DataSourceConfig { @Bean @Profile("dev") public DataSource devDataSource() { System.out.println("This is dev DataSource"); BasicDataSource basicDataSource = new BasicDataSource(); basicDataSource.setDriverClassName("com.mysql.jdbc.Driver"); basicDataSource.setUrl("jdbc:mysql://localhost:3306/mybatis_action_db"); basicDataSource.setUsername("dev"); basicDataSource.setPassword("dev"); return basicDataSource; } @Bean @Profile("qa") public DataSource qaDataSource() { System.out.println("This is qa DataSource"); BasicDataSource basicDataSource = new BasicDataSource(); basicDataSource.setDriverClassName("com.mysql.jdbc.Driver"); basicDataSource.setUrl("jdbc:mysql://localhost:3307/mybatis_action_db"); basicDataSource.setUsername("qa"); basicDataSource.setPassword("qa"); return basicDataSource; } @Bean @Profile("prod") public DataSource prodDataSource() { System.out.println("This is prod DataSource"); BasicDataSource basicDataSource = new BasicDataSource(); basicDataSource.setDriverClassName("com.mysql.jdbc.Driver"); basicDataSource.setUrl("jdbc:mysql://localhost:3308/mybatis_action_db"); basicDataSource.setUsername("prod"); basicDataSource.setPassword("prod"); return basicDataSource; } }

注意事项：没有指定profile的bean始终都会创建，与激活哪个profile无关。

2.2 xml中配置profile bean

我们也可以通过<beans>元素的profile属性，在xml中配置profile bean，如下所示：

可以参考该配置，分别创建qa和prod环境的profile xml文件。

不过还是推荐使用嵌套的<beans>元素，在一个xml文件中配置好3个环境的数据源，代码如下所示：

<?xml version="1.0" encoding="UTF-8"?> <beans xmlns="http://www.springframework.org/schema/beans" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns:p="http://www.springframework.org/schema/p" xsi:schemaLocation="http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans.xsd"> <beans profile="dev"> <bean id="dataSource" class="org.apache.commons.dbcp2.BasicDataSource" p:driverClassName="com.mysql.jdbc.Driver" p:url="jdbc:mysql://localhost:3306/mybatis_action_db" p:username="dev" p:password="dev"/> </beans> <beans profile="qa"> <bean id="dataSource" class="org.apache.commons.dbcp2.BasicDataSource" p:driverClassName="com.mysql.jdbc.Driver" p:url="jdbc:mysql://localhost:3307/mybatis_action_db" p:username="qa" p:password="qa"/> </beans> <beans profile="prod"> <bean id="dataSource" class="org.apache.commons.dbcp2.BasicDataSource" p:driverClassName="com.mysql.jdbc.Driver" p:url="jdbc:mysql://localhost:3308/mybatis_action_db" p:username="prod" p:password="prod"/> </beans> </beans>

3. 激活profile

截止目前，我们按照环境的维度创建了3个bean，但实际运行时，只会创建1个bean，具体创建哪个bean取决于处于激活状态的是哪个profile。

那么，我们该如何激活某个profile呢？

Spring在确定激活哪个profile时，需要依赖2个属性：

spring.profiles.active
spring.profiles.default

spring.profiles.active的优先级比spring.profiles.default高，即如果没有配置spring.profiles.active，就使用spring.profiles.default配置的值，如果配置了spring.profiles.active，就不会再使用spring.profiles.default配置的值。

如果两者都没有配置，就只会创建那些没有定义profile的bean。

Web应用中，在web.xml中设置spring.profiles.active的代码如下所示：

<context-param> <param-name>spring.profiles.active</param-name> <param-value>dev</param-value> </context-param>

也可以使用代码方式激活：

AnnotationConfigApplicationContext context = new AnnotationConfigApplicationContext();context.getEnvironment().setActiveProfiles("dev");

4. 单元测试

新建Main类，在其main()方法中添加如下测试代码：

package chapter03.profile;import org.springframework.context.annotation.AnnotationConfigApplicationContext;public class Main { public static void main(String[] args) { AnnotationConfigApplicationContext context = new AnnotationConfigApplicationContext(); context.getEnvironment().setActiveProfiles("dev"); context.register(DataSourceConfig.class); context.refresh(); context.close(); } }

输出结果如下所示：