Spring的Environment类的使用
Spring的Environment类的使用
明明如月学长
发布于 2021-08-31 15:52:49
发布于 2021-08-31 15:52:49
代码可运行
运行总次数：0
代码可运行
业务开发中可能需要当前环境的判断，比如某些接口严禁生产环境下调用，有些接口只能再预发环境下才可用等。
这个时候在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();
}
}当然了除了获取环境之外，还可以设置或者获取属性。
本文参与 腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。
原始发表：2019/04/26 ，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读
目录
腾讯云开发者
