java

关注公众号 jb51net

关闭
首页 > 软件编程 > java > Spring Environment类的使用

Spring之Environment类的使用方式

作者:明明如月学长

这篇文章主要介绍了Spring之Environment类的使用方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教

业务开发中可能需要当前环境的判断,比如某些接口严禁生产环境下调用,有些接口只能再预发环境下才可用等。

这个时候在Spring中可以使用Environment类。

通常有两种方式, 一种是基于接口,一种是直接注入。

一、实现EnvironmentAware接口

代码如下:

package com.chujianyun.web.util;
 
import org.springframework.context.EnvironmentAware;
import org.springframework.core.env.Environment;
import org.springframework.stereotype.Component;
 
/**
 * 环境帮助类
 * author: 明明如月
 * email: liuwangyangedu@163.com
 */
@Component
public class EnvironmentHelper implements EnvironmentAware {
 
    private Environment environment;
 
    private static final String PROD = "prod";
    private static final String PRE = "pre";
    private static final String QA = "qa";
 
    public boolean isProd() {
        return environment.acceptsProfiles(PROD);
    }
 
    public boolean isPre() {
        return environment.acceptsProfiles(PRE);
    }
 
    public boolean isQa() {
        return environment.acceptsProfiles(QA);
    }
 
    public String[] getActiveProfiles() {
        return environment.getActiveProfiles();
    }
 
    @Override
    public void setEnvironment(Environment environment) {
        this.environment = environment;
    }
}

二、直接注入

直接通过注解将环境注入进来。

package com.chujianyun.web.util;
 
import org.springframework.core.env.Environment;
import org.springframework.stereotype.Component;
 
import javax.annotation.Resource;
 
/**
 * 环境帮助类
 * author: 明明如月
 * email: liuwangyangedu@163.com
 */
@Component
public class EnvironmentHelper{
 
    @Autowired
    private Environment environment;
 
    private static final String PROD = "prod";
    private static final String PRE = "pre";
    private static final String QA = "qa";
 
    public boolean isProd() {
        return environment.acceptsProfiles(PROD);
    }
 
    public boolean isPre() {
        return environment.acceptsProfiles(PRE);
    }
 
    public boolean isQa() {
        return environment.acceptsProfiles(QA);
    }
 
    public String[] getActiveProfiles() {
        return environment.getActiveProfiles();
    }
 
}

当然了除了获取环境之外,还可以设置或者获取属性。

总结

以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

您可能感兴趣的文章:
阅读全文