springboot 根据当前环境动态的激活和切换

63 阅读 0 评论 42 点赞

我是靠谱客的博主飘逸蛋挞，最近开发中收集的这篇文章主要介绍springboot 根据当前环境动态的激活和切换，觉得挺不错的，现在分享给大家，希望可以做个参考。

概述

一、@Profile的作用：

可以根据当前环境，动态的激活和切换一系列组件的功能，指定组件在哪个环境的情况下才能被注册到容器中，不指定，任何环境下都能注册

1、加了环境标识的bean，只有这个环境被激活的时候才能注册到容器中，默认是default环境
2、加在配置类上，只有是指定的环境的时候，整个配置类里面的所有配置才能生效
3、没有标注环境标识的bean在任何环境下都加载

二、通过命令行参数设置虚拟机环境，并指定@Profile

1、创建一个实体类

public class BookService {

}

2、创建配置类，并注册bean


@Configuration
public class MainConfig {

    @Profile("test")
    @Bean
    public BookService bookService()
    {
        return new BookService();
    }
}

3、测试

@SpringBootTest
class DemoApplicationTests {
    AnnotationConfigApplicationContext applicationContext = new AnnotationConfigApplicationContext(MainConfig.class);
    @Test
    void contextLoads() {
        BookService bean = applicationContext.getBean(BookService.class);
        System.out.println(bean);
    }
}

测试结果：bean并没有被注册。因为@Profile指定了测试的环境，所以必须是测试环境下bean才能被注册

我们通过在虚拟机参数位置指定-Dspring.profiles.active=xxx来指定运行环境

结果成功被注册进来

三、在程序内设置虚拟机环境，并指定@Profile

1、创建一个需要被注册的类


public class BookService {

}

2、配置类

@Configuration
public class MainConfig {

    @Profile("test")
    @Bean
    public BookService bookService()
    {
        return new BookService();
    }
}

3、在程序内设置虚拟机环境

程序内指定：

1、创建一个applicationContext

2、设置需要激活的环境，applicationContext.getEnvironment().setActiveProfiles("");

3、注册主配置类，applicationContext.register(xxx.class)

4、启动刷新容器，applicationContext.refresh();

@SpringBootTest
class DemoApplicationTests {

    @Test
    void contextLoads() {
        AnnotationConfigApplicationContext applicationContext = new AnnotationConfigApplicationContext();
        applicationContext.getEnvironment().setActiveProfiles("test");
        applicationContext.register(MainConfig.class);
        applicationContext.refresh();
        BookService bean = applicationContext.getBean(BookService.class);
        System.out.println(bean);
    }

}

同样我们成功注册了bean