SpringBoot - @Profile注解使用详解（根据环境动态激活使用相应组件）
作者：hangge | 2021-06-02 09:10
在项目中，有时我们需要能根据当前环境，动态的激活和切换一系列组件，这个借助 Spring 提供的 @Profile 注解即可实现，下面通过样例进行演示。
1，基本介绍
（1）@Profile 注解的作用是指定组件在哪个环境的情况下才能被注册到容器中，若不指定，任何环境下都能注册这个组件。
- 加了@Profile 注解的 bean，只有这个环境被激活的时候才能注册到容器中。默认是 default 环境。
- 若 @Profile 注解写在配置类上，只有在指定的环境的时候，整个配置类里面的所有配置才能开始生效。
（2）比如下面样例，会根据当前环境的不同（dev 或 prod），自动实例化对应的 DataSource。
@Configuration
@PropertySource(value = {"classpath:/dbconfig.properties"})
public class ProfileBeanConfig implements EmbeddedValueResolverAware {
//数据库连接用户名
@Value(value = "${jdbc.username}")
private String username;
//数据库连接密码
private String password;
//开发环境数据源
@Bean(value = "dataSourceDev")
@Profile(value = "dev")
public DataSource dataSourceDev(@Value("${jdbc.driverClass}") String driverClass)
throws PropertyVetoException {
ComboPooledDataSource comboPooledDataSource = new ComboPooledDataSource();
comboPooledDataSource.setUser(this.username);
comboPooledDataSource.setPassword(this.password);
comboPooledDataSource.setDriverClass(driverClass);
comboPooledDataSource.setJdbcUrl("jdbc:mysql://localhost:3306/dev");
return comboPooledDataSource;
}
//生产环境数据源
@Bean(value = "dataSourceProduction")
@Profile("prod")
public DataSource dataSourceProduction(@Value("${jdbc.driverClass}") String driverClass)
throws PropertyVetoException {
ComboPooledDataSource comboPooledDataSource = new ComboPooledDataSource();
comboPooledDataSource.setUser(this.username);
comboPooledDataSource.setPassword(this.password);
comboPooledDataSource.setDriverClass(driverClass);
comboPooledDataSource.setJdbcUrl("jdbc:mysql://localhost:3306/production");
return comboPooledDataSource;
}
}
2，使用样例
（1）有时候我们会有这样的一个需求，在生产环境或者开发环境我们所看到的功能是不同的，这便需要我们根据配置项来激活不同的功能。
比如一些接口在开发环境中无法调用，那么在开发环境中就需要 mock 数据。因此我们可以写一个 mock 数据接口类，专门供开发环境使用。而生产环境则使用真实的接口调用类，二者通过 @Profile 注解来自动激活生效。
（2）首先我们先定义一个 Service 的接口：
public interface ProductService {
String getProductInfo(Long id);
}
（3）接着我们创建两个实现类，分别对应开发环境和生产环境：
@Service
@Profile("dev")
public class MockProductServiceImpl implements ProductService {
@Override
public String getProductInfo(Long id) {
return "这是开发环境数据：" + id;
}
}
@Service
@Profile("prod")
public class ProductServiceImpl implements ProductService {
@Override
public String getProductInfo(Long id) {
//return productResource.getProductInfo(id);
return "这是生产环境数据：" + id;
}
}
（4）最后创建一个 Controller 调用 Service 接口：
@RestController
public class TestController {
@Autowired
private ProductService productService;
@GetMapping("/hello")
public String hello(@RequestParam("id") Long id) {
return productService.getProductInfo(id);
}
}
（5）开始测试，首先我们编辑 application.properties 文件，将环境设置为 dev。
spring.profiles.active=dev
- 访问 Controller 接口可以看到调用的是 MockProductServiceImpl 这个 Servcie：
（6）接着将环境设置为 prod。
spring.profiles.active=prod
- 访问 Controller 接口可以看到调用的是 ProductServiceImpl 这个 Servcie：
全部评论（0）