Spring注解开发8 --- @Profile不同环境下动态注册

目录

1. @Profile:指明组件在那个环境下使用，不指定任何环境都能使用这个组件

2.在运行中指定环境的方式

 2.1  使用命令行动态参数的方式   -Dspring.profiles.active= test

 2.2 使用代码的方式设置环境

3.例子

---

Spring为我们提供根据当前环境，动态的激活和切换一系列组件的功能。

环境：开发环境，测试环境，生产环境

1. @Profile:指明组件在那个环境下使用，不指定任何环境都能注册这个组件

1）、加了环境表示的bean，只有这个环境被激活的时候才能加入到容器中。默认是“default”环境

2）、@Profile（“default”）表示默认的环境

3）、@Profile写在配置类上，表示只有满足当前的环境这个类里面的配置才会生效，写在方法上只有方法上满足条件才会注册bean

4）、没有标注环境的bean，在任何环境都是能够被注册的

2.在运行中指定环境的方式

 2.1  使用命令行动态参数的方式   -Dspring.profiles.active= test

          

 2.2 使用代码的方式设置环境

           AnnotationConfigApplicationContext的有参默认调用了register和refresh方法，所以要设置激活的环境就需要单独设置。

public class ProfileTest {

    public static void main(String[] args) {
        AnnotationConfigApplicationContext applicationContext = new AnnotationConfigApplicationContext();
       //有参构造器默认调用了3,4两步，所以这里需要手动触发
        //1.创建一个无参的applicationContext
       //2.设置需要激活的环境
        applicationContext.getEnvironment().setActiveProfiles("test","dev");
        //3.注册主配置类
        applicationContext.register(ProfileConfig.class);
        //4.启动刷新
        applicationContext.refresh();

        String[] names = applicationContext.getBeanDefinitionNames();
        for (String name: names) {
            System.out.println(name);
        }
        applicationContext.close();
    }
}

3.例子

     例如在下面导入三种环境的数据源

1)外部资源文件resource/db.properties：

db.user=root
db.pwd=root
db.driverclass =com.mysql.jdbc.Driver

2)配置类，使用三种方式完成属性的注入，@Value还有实现xxxAware接口来解析注入

@PropertySource("classpath:/db.properties")
@Configuration
public class ProfileConfig implements EmbeddedValueResolverAware {

    @Value("${db.user}")
    private String user;
    private String driverClass;

    @Profile("dev")    //dev环境下激活使用 ,里面的值是可以变动的，只是设置环境参数相对应即可
    @Bean("devDataSource")
    public DataSource dataSourceDev(@Value("${db.pwd}") String pwd) throws PropertyVetoException {
        ComboPooledDataSource dataSource = new ComboPooledDataSource();
        dataSource.setUser(user);
        dataSource.setPassword(pwd);
        dataSource.setDriverClass(driverClass);
        dataSource.setJdbcUrl("jdbc:mysql://localhost:3306/dev");
        return dataSource;
    }
    @Profile("test")    //test环境下激活使用,里面的值是可以变动的，只是设置环境参数相对应即可
    @Bean("testDataSource")
    public DataSource dataSourceTest(@Value("${db.pwd}") String pwd) throws PropertyVetoException {
        ComboPooledDataSource dataSource = new ComboPooledDataSource();
        dataSource.setUser(user);
        dataSource.setPassword(pwd);
        dataSource.setDriverClass(driverClass);
        dataSource.setJdbcUrl("jdbc:mysql://localhost:3306/test");
        return dataSource;
    }
    @Profile("prod")    //prod环境下激活使用,里面的值是可以变动的，只是设置环境参数相对应即可
    @Bean("prodDataSource")
    public DataSource dataSourceProd(@Value("${db.pwd}") String pwd) throws PropertyVetoException {
        ComboPooledDataSource dataSource = new ComboPooledDataSource();
        dataSource.setUser(user);
        dataSource.setPassword(pwd);
        dataSource.setDriverClass(driverClass);
        dataSource.setJdbcUrl("jdbc:mysql://localhost:3306/Pro");
        return dataSource;
    }

    @Override   //这个StringValueResolver 是专门来解析Value的类
    public void setEmbeddedValueResolver(StringValueResolver stringValueResolver) {
        driverClass = stringValueResolver.resolveStringValue("${db.driverclass}");
    }
}

3）.测试类 这里使用代码的方式测试

public class ProfileTest {

    public static void main(String[] args) {
        AnnotationConfigApplicationContext applicationContext = new AnnotationConfigApplicationContext();
       //有参构造器默认调用了3,4两步，所以这里需要手动触发
        //1.创建一个无参的applicationContext
       //2.设置需要激活的环境
        applicationContext.getEnvironment().setActiveProfiles("test","dev"); //这里可以写多个环境
        //3.注册主配置类
        applicationContext.register(ProfileConfig.class);
        //4.启动刷新
        applicationContext.refresh();

        String[] names = applicationContext.getBeanDefinitionNames();
        for (String name: names) {
            System.out.println(name);
        }
        applicationContext.close();
    }
}