Spring Boot 获取 spring.profiles.active

wangooo

于 2021-08-26 21:48:46 发布

阅读量1.8w

点赞数 1

分类专栏： Spring 文章标签： spring java spring boot

原文链接：https://www.cnblogs.com/zhenyuyaodidiao/p/12029479.html

版权

Spring 专栏收录该内容

54 篇文章

订阅专栏

本文介绍了如何在Spring应用中通过实现ApplicationContextAware接口获取ApplicationContext，并从中读取激活的profile。同时展示了如何使用@Value注解从YAML配置文件中获取profile。示例代码包括了一个SpringContextUtil工具类，提供了获取Bean和配置属性的方法，以及一个SpringProfileService服务类，展示了如何直接在service中获取profile。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

原理

　　1、实现ApplicationContextAware（当一个类实现了ApplicationContextAware这个接口之后，这个类就可以通过setApplicationContext方法获得ApplicationContext中的上下文），获取context。通过方法：context.getEnvironment().getActiveProfiles()获取激活的profile。

　　2、通过service中成员变量上的注解：@Value("${spring.profiles.active}")，获取yaml中的profile

@Component
public class SpringContextUtil implements ApplicationContextAware {
    private static ApplicationContext applicationContext;

    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        SpringContextUtil.applicationContext = applicationContext;
    }

    /**
     * 获取applicationContext
     */
    public static ApplicationContext getApplicationContext() {
        return applicationContext;
    }

    /**
     * 通过name获取 Bean.
     */
    public static Object getBean(String name) {
        Object o = null;
        try {
            o = getApplicationContext().getBean(name);
        } catch (NoSuchBeanDefinitionException e) {
            // e.printStackTrace();
        }
        return o;
    }

    /**
     * 通过class获取Bean.
     */
    public static <T> T getBean(Class<T> clazz) {
        return getApplicationContext().getBean(clazz);
    }

    /**
     * 通过name,以及Clazz返回指定的Bean
     */
    public static <T> T getBean(String name, Class<T> clazz) {
        return getApplicationContext().getBean(name, clazz);
    }

    /**
     * 通过name获取 Bean.
     */
    public static <T> Map<String, T> getBeansOfType(Class<T> clazz) {
        return getApplicationContext().getBeansOfType(clazz);
    }

    /**
     * 获取配置文件配置项的值
     *
     * @param key 配置项key
     */
    public static String getEnvironmentProperty(String key) {
        return getApplicationContext().getEnvironment().getProperty(key);
    }

    /**
     * 获取spring.profiles.active
     */
    public static String getActiveProfile() {
        return getApplicationContext().getEnvironment().getActiveProfiles()[0];
    }
}

或通过yml文件读取

package com.test;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;

@Service
public class SpringProfileService {
    private static final Logger logger = LoggerFactory.getLogger(SpringProfileService.class);

    private static final String PRODUCTION_PROFILE = "production";
    private static final String STAGE_PROFILE = "stage";

    @Value("${spring.profiles.active}")
    private String profile;
}