SpringBoot中实现不同的环境扫描不同的包

香酥蟹

已于 2022-02-08 12:17:32 修改

阅读量799

点赞数 1

分类专栏： springboot 文章标签： spring boot spring

于 2022-02-06 17:13:21 首次发布

本文链接：https://blog.csdn.net/zyt807/article/details/122799599

版权

springboot 专栏收录该内容

15 篇文章 1 订阅

订阅专栏

本文介绍如何在Spring Boot应用中，通过环境变量智能配置@ComponentScan，避免类名冲突，实现根据不同环境加载不同包下的类。作者提供了一个自定义过滤器，根据活跃的环境配置排除特定包，以实现灵活的类加载策略。

摘要由CSDN通过智能技术生成

前言

需求很简单，我不想在idea新建个项目，但是功能同样的类，我需要多份，但是还不想改变名称，原因是太多。我就想新建个包，把他们复制过去，改改配置。但是不改类名，容器扫描会有冲突（除非给其加上别名），因为我的包名和环境名相同，如果能根据环境的不同，容器扫描不同的类最好不过。

需求解决

需求修改对ComponentScan的配置

@ComponentScan(
        value = "com.delicacy",
        excludeFilters = {@ComponentScan.Filter(
                type = FilterType.CUSTOM,
                classes = {ExcludeScanPackagesFilter.class}
        )}
)

如果环境是如下:
application-dev-aa.yml
application-dev-bb.yml
application-dev-cc.yml
包名如下：
com.xxx.aa
com.xxx.bb
com.xxx.cc

public class ExcludeScanPackagesFilter implements TypeFilter, EnvironmentAware {
    private List<String> excludePackages;

    @Override
    public boolean match(MetadataReader metadataReader, MetadataReaderFactory metadataReaderFactory) {
        ClassMetadata classMetadata = metadataReader.getClassMetadata();
        String[] var1 = classMetadata.getClassName().split("\\.");
        List<String> var2 = Arrays.asList(var1);
        if (var2.stream().anyMatch(e -> excludePackages.contains(e))) {
            log.info("排除的类的类名{}", classMetadata.getClassName());
            return true;
        }
        return false;
    }

    @Override
    public void setEnvironment(Environment environment) {
        String[] activeProfiles = environment.getActiveProfiles();
        String scanPackage = "";
        if (activeProfiles.length > 0) {
            String activeProfile = activeProfiles[0];
            scanPackage = activeProfile.split("-")[1];
        }
        switch (scanPackage) {
            case "aa": {
                this.excludePackages = Arrays.asList("bb", "cc");
                return;
            }
            default:
                this.excludePackages = Arrays.asList("aa", "bb", "cc");
        }
    }
}