一、背景
今天一个朋友问了一个问题，测试环境比线上的代码多了几行代码怎么办？
是不是要新建分支？有没有更好的办法？
其实还有类似的场景，比如测试环境和线上环境用了同一套NoSQL存储，想隔离两个环境。
下面提供一个比较简单的通用的方案，希望对大家有帮助。
二、方法
我们可以写一个环境工具类，然后再特定环境的代码上用这个工具类进行判断。
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; } }
当然也可以直接注入
@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(); } }
然后借助工具类来判断环境
@Service public class DemoServiceImpl implements DemoService { @Resource private EnvironmentHelper environmentHelper; @Override public void test() { // 其他代码1 if (environmentHelper.isQa()) { // 测试环境专用代码 } // 其他代码2 } }
当然了也可以借助Apollo来配置，比如设置一个变量表示是否开启某个功能，线上的Apollo和测试服的配置不同。
如果预发和线上服公用同一套缓存框架，通过环境工具类获取环境的名称和key组合成一个新的key字符串，即可实现环境的隔离。
三、总结
方法不难，在开发中偶尔也会用到。
另外建议大家多看Spring的官方文档，很多场景都有专门的类提供了相关的功能，而不是遇到一个问题一个问题去查，去问。
创作不易，如果觉得本文对你有帮助，欢迎点赞，欢迎关注我，如果有补充欢迎评论交流，我将努力创作更多更好的文章。