SpringBoot中获取spring.profiles.active的值

分手后的思念是犯贱 2022-02-21 09:36 288阅读 0赞

转载：[https://www.cnblogs.com/linzhanfly/p/9056722.html][https_www.cnblogs.com_linzhanfly_p_9056722.html]

一、网上很多采用@Profile("dev")的方式获取，但是这个是类级别的

二、开发中可能需要代码级别

1、刚开始我想通过classpath下的文件读取方式，麻烦死了，于是换了个思路。

2、SpringBoot启动日志中有下面这句：

<table> 
 <tbody> 
 <tr> 
 <td> 1 </td> 
 <td> <code>15:57:56.128 [restartedMain] INFO&nbsp; c.d.o.OptplatformApplication - The following profiles are active: test</code> </td> 
 </tr> 
 </tbody> 
</table>

（1）跟踪代码：SpringApplication.run方法

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 </td> 
 <td> <code>public ConfigurableApplicationContext run(String... args) { </code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>StopWatch stopWatch = new StopWatch();</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>stopWatch.start();</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>ConfigurableApplicationContext context = null;</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>FailureAnalyzers analyzers = null;</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>configureHeadlessProperty();</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>SpringApplicationRunListeners listeners = getRunListeners(args);</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>listeners.starting();</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>try { </code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>ApplicationArguments applicationArguments = new DefaultApplicationArguments(</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>args);</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>ConfigurableEnvironment environment = prepareEnvironment(listeners,</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>applicationArguments);</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>Banner printedBanner = printBanner(environment);</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>context = createApplicationContext();</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>analyzers = new FailureAnalyzers(context);</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>prepareContext(context, environment, listeners, applicationArguments,</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>printedBanner);&nbsp; // 在这里打印了，跟踪进去</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>refreshContext(context);</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>afterRefresh(context, applicationArguments);</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>listeners.finished(context, null);</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>stopWatch.stop();</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>if (this.logStartupInfo) { </code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>new StartupInfoLogger(this.mainApplicationClass)</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>.logStarted(getApplicationLog(), stopWatch);</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>}</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>return context;</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>}</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>catch (Throwable ex) { </code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>handleRunFailure(context, listeners, analyzers, ex);</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>throw new IllegalStateException(ex);</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>}</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>}</code> </td> 
 </tr> 
 </tbody> 
</table>

（2）跟踪代码：SpringApplication.prepareContext方法

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 4 5 6 7 8 9 10 11 12 13 </td> 
 <td> <code>private void prepareContext(ConfigurableApplicationContext context,</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>ConfigurableEnvironment environment, SpringApplicationRunListeners listeners,</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>ApplicationArguments applicationArguments, Banner printedBanner) { </code> <code>&nbsp;&nbsp;&nbsp;</code><code>context.setEnvironment(environment);</code> <code>&nbsp;&nbsp;&nbsp;</code><code>postProcessApplicationContext(context);</code> <code>&nbsp;&nbsp;&nbsp;</code><code>applyInitializers(context);</code> <code>&nbsp;&nbsp;&nbsp;</code><code>listeners.contextPrepared(context);</code> <code>&nbsp;&nbsp;&nbsp;</code><code>if (this.logStartupInfo) { </code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>logStartupInfo(context.getParent() == null);</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>logStartupProfileInfo(context);&nbsp; // 名称很明显，继续跟踪进去</code> <code>&nbsp;&nbsp;&nbsp;</code><code>}</code> <code>&nbsp;&nbsp;&nbsp;</code><code>......</code> <code>}</code> </td> 
 </tr> 
 </tbody> 
</table>

（3）跟踪代码：SpringApplication.logStartupProfileInfo方法

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 </td> 
 <td> <code>protected</code>&nbsp;<code>void</code>&nbsp;<code>logStartupProfileInfo(ConfigurableApplicationContext context) {&nbsp;</code> <code>&nbsp;&nbsp;&nbsp;</code><code>Log log = getApplicationLog();</code> <code>&nbsp;&nbsp;&nbsp;</code><code>if</code>&nbsp;<code>(log.isInfoEnabled()) { </code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>String[] activeProfiles = context.getEnvironment().getActiveProfiles();</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>if</code>&nbsp;<code>(ObjectUtils.isEmpty(activeProfiles)) { </code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>String[] defaultProfiles = context.getEnvironment().getDefaultProfiles();</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>log.info(</code><code>"No active profile set, falling back to default profiles: "</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>+ StringUtils.arrayToCommaDelimitedString(defaultProfiles));&nbsp;</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>}</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>else</code>&nbsp;<code>{ </code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>log.info(</code><code>"The following profiles are active: "</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>+ StringUtils.arrayToCommaDelimitedString(activeProfiles));&nbsp;&nbsp;</code><code>//找到了，很明显用了ApplicationContxt容器，接下来就是写个工具类来获取Application就行啦。</code> &nbsp; <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>}</code> <code>&nbsp;&nbsp;&nbsp;</code><code>}</code> <code>}</code> </td> 
 </tr> 
 </tbody> 
</table>

（4）编写SpringContxtUtil工具类

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 </td> 
 <td> <code>/**</code> <code>&nbsp;</code><code>* 项目名称:</code> <code>&nbsp;</code><code>* 类名: SpringContextUtil</code> <code>&nbsp;</code><code>* 描述： 获取bean的工具类，可用于在线程里面获取bean</code> <code>&nbsp;</code><code>* 创建人: awsm</code> <code>&nbsp;</code><code>* 创建时间: Dec 17, 2015 10:46:44 PM</code> <code>&nbsp;</code><code>* 修改人：little evil</code> <code>&nbsp;</code><code>* 修改时间：May 18, 2018 04:01:34 PM</code> <code>&nbsp;</code><code>* 修改备注：添加getActiveProfile方法,获取当前环境</code> <code>&nbsp;</code><code>* 版本：1.1</code> <code>&nbsp;</code><code>*/</code> <code>@Component</code> <code>public class SpringContextUtil implements ApplicationContextAware { </code> &nbsp; <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>private static ApplicationContext context = null;</code> &nbsp; <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>/* (non Javadoc)</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>* @Title: setApplicationContext</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>* @Description: spring获取bean工具类</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>* @param applicationContext</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>* @throws BeansException</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>* @see org.springframework.context.ApplicationContextAware#setApplicationContext(org.springframework.context.ApplicationContext)</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>*/</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>@Override</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>public void setApplicationContext(ApplicationContext applicationContext)</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>throws BeansException { </code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>this.context = applicationContext;</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>}</code> &nbsp; <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>// 传入线程中</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>public static &lt;</code><code>T</code><code>&gt; T getBean(String beanName) { </code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>return (T) context.getBean(beanName);</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>}</code> &nbsp; <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>// 国际化使用</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>public static String getMessage(String key) { </code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>return context.getMessage(key, null, Locale.getDefault());</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>}</code> &nbsp; <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>/// 获取当前环境</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>public static String getActiveProfile() { </code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>return context.getEnvironment().getActiveProfiles()[0];</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>}</code> <code>}</code> <code>// 该工具类从网上抄来的，最后添加个获取方法就完成了，这样就能在代码级别通过环境条件来控制方法行为了。</code> </td> 
 </tr> 
 </tbody> 
</table>

[https_www.cnblogs.com_linzhanfly_p_9056722.html]: https://www.cnblogs.com/linzhanfly/p/9056722.html