怎么使用Spring注解@Profile实现开发环境/测试环境/生产环境切换
环境搭建
项目结构
DruidDataSource
└── src
└── main
└── java
└── com.steakliu.druiddatasource
├── controller
│ └── DataSourceController.java
└── datasource
│ └── DataSourceConfiguration.java
└── DruidDataSourceApplication.java
└── resource
└── application.yml
└── application-dev.yml
└── application-prod.yml
└── application-test.yml
application.yml
application.yml内容如下，里面主要放的是公共的配置，比如项目的名称啊，端口啊这些，使用spring.profiles.active = dev/test/prod切换不同的环境。
server: port: 8080 spring: profiles: active: dev
application-dev.yml
spring: datasource: druid: username: root password: xiaosi520@ url: jdbc:mysql://127.0.0.1:3306/store-dev driver-class-name: com.mysql.cj.jdbc.Driver type: com.alibaba.druid.pool.DruidDataSource
application-test.yml
spring: datasource: druid: username: root password: xiaosi520@ url: jdbc:mysql://127.0.0.1:3306/store-test driver-class-name: com.mysql.cj.jdbc.Driver type: com.alibaba.druid.pool.DruidDataSource
application-prod.yml
spring: datasource: druid: username: root password: xiaosi520@ url: jdbc:mysql://127.0.0.1:3306/store-prod driver-class-name: com.mysql.cj.jdbc.Driver type: com.alibaba.druid.pool.DruidDataSource
从上面我们知道，三个yaml文件主要配置的是数据库，分为dev，test，prod三个库，我也在数据库创建了三个数据，等一下我们要根据不同的配置获取不同的数据库的数据， 三个数据库如下，其中每个库里面都有一张表，分别放了一条测试数据。
数据源配置DataSourceConfiguration
DataSourceConfiguration是一个数据源配置类，我们使用alibaba的Druid数据源，使用@Value注解来获取yaml文件的数据库配置， 定义了三个Bean，并使用@Profile来进行开发环境，测试环境，生产环境的区分，这里它会识别application.yml文件中的spring.profiles.active
package com.steakliu.druiddatasource.datasource;
import com.alibaba.druid.pool.DruidDataSource;
import org.springframework.beans.factory.annotation.Value;
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 DataSourceConfiguration {
@Value("${spring.datasource.druid.username}")
private String username;
@Value("${spring.datasource.druid.password}")
private String password;
@Value("${spring.datasource.druid.url}")
private String url;
@Value("${spring.datasource.druid.driver-class-name}")
private String driverClassName;
@Profile("dev")
@Bean
public DataSource devDataSource(){
DruidDataSource dataSource = new DruidDataSource();
dataSource.setUsername(username);
dataSource.setPassword(password);
dataSource.setUrl(url);
dataSource.setDriverClassName(driverClassName);
return dataSource;
}
@Profile("test")
@Bean
public DataSource testDataSource(){
DruidDataSource dataSource = new DruidDataSource();
dataSource.setUsername(username);
dataSource.setPassword(password);
dataSource.setUrl(url);
dataSource.setDriverClassName(driverClassName);
return dataSource;
}
@Profile("prod")
@Bean
public DataSource prodDataSource(){
DruidDataSource dataSource = new DruidDataSource();
dataSource.setUsername(username);
dataSource.setPassword(password);
dataSource.setUrl(url);
dataSource.setDriverClassName(driverClassName);
return dataSource;
}
}DataSourceController测试Controller
创建一个Controller来测试获取数据，这里使用JdbcTemplate来获取数据，
package com.steakliu.druiddatasource.controller;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;
import java.util.List;
import java.util.Map;
@RestController
public class DataSourceController {
@Autowired
private JdbcTemplate jdbcTemplate;
@GetMapping("datasource")
public List> datasource(){
List> maps = jdbcTemplate.queryForList("SELECT * FROM sys_env");
return maps;
}
} 使用Postman进行测试
spring.profiles.active = dev
spring: profiles: active: dev
spring.profiles.active = test
spring: profiles: active: test
spring.profiles.active = prod
spring: profiles: active: dev
到这里我们完成了开发环境，测试环境，生产环境的切换，其核心就是spring.profiles.active和@Profile注解，那么这时候你可能会觉得，每次打包发布前都需要 去application.yml配置文件中修改spring.profiles.active，是不是有点麻烦，没错，确实会有点麻烦，那么我们下面对其改造，使用Maven的profile 来改造，实现勾选式的配置。
使用Maven配置
我们在maven中对dev，test，prod进行定义，如下，使用dev作为默认的环境， 在application.yml 那边进行对应。
pom.xml
dev true dev test test prod prod
application.yml
application.yml里面使用@mvn.profile@这种方式来读取pom.xml中的配置。
spring: profiles: active: @mvn.profile@
定义好上面的pom.xml和application.yml后，会在我们IDEA编译器右侧的Maven栏下出现Profiles，就可以 进行选择了，所实现的效果和直接在applicaiton.yml中修改是一样的。
除了上面的这两种，我们还可以在IDEA里面进行指定，也能达到同样的效果。
