spring boot源碼學習 SpringApplication(二)-Banner

先說一下 banner相關的配置屬性

#spring.main.bannerMode=off # 關閉banner打印
#banner.location=coffee.banne # 設置banner文本文件的位置 、相對於src/main/resources
#banner.image.location=11111.jpg #設置圖片、打印的時候會亂碼

代碼中的用法

核心類 org.springframework.bootSpringApplicationBannerPrinter
核心代碼
Banner printedBanner = printBanner(environment);
//如果沒定義banner.location 會
private static final Banner DEFAULT_BANNER = new SpringBootBanner();

/**
* 有時候可能會返回多個banner,如果按照我的想法 我可能會返回一個list 
* 但是 人家缺返回了一個banners噢
**/
private Banner getBanner(Environment environment, Banner definedBanner) {
        Banners banners = new Banners();
        //
        banners.addIfNotNull(getImageBanner(environment));
        banners.addIfNotNull(getTextBanner(environment));
        if (banners.hasAtLeastOneBanner()) {
            return banners;
        }
        if (this.fallbackBanner != null) {
            return this.fallbackBanner;
        }
        return DEFAULT_BANNER;
    }

    private Banner getTextBanner(Environment environment) {
    //默認加載 banner.txt文件,可以在resources文件夾下新建一個 banner.txt
        String location = environment.getProperty(BANNER_LOCATION_PROPERTY,
                DEFAULT_BANNER_LOCATION);
        Resource resource = this.resourceLoader.getResource(location);
        if (resource.exists()) {
            return new ResourceBanner(resource);
        }
        return null;
    }

    private Banner getImageBanner(Environment environment) {
        String location = environment.getProperty(BANNER_IMAGE_LOCATION_PROPERTY);
        if (StringUtils.hasLength(location)) {
            Resource resource = this.resourceLoader.getResource(location);
            return (resource.exists() ? new ImageBanner(resource) : null);
        }
        for (String ext : IMAGE_EXTENSION) {
            Resource resource = this.resourceLoader.getResource("banner." + ext);
            if (resource.exists()) {
                return new ImageBanner(resource);
            }
        }
        return null;
    }

看一下banners的實現,真的很巧妙

    private static class Banners implements Banner {

        private final List<Banner> banners = new ArrayList<Banner>();

        public void addIfNotNull(Banner banner) {
            if (banner != null) {
                this.banners.add(banner);
            }
        }

        public boolean hasAtLeastOneBanner() {
            return !this.banners.isEmpty();
        }

        @Override
        public void printBanner(Environment environment, Class<?> sourceClass,
                PrintStream out) {
            for (Banner banner : this.banners) {
                banner.printBanner(environment, sourceClass, out);
            }
        }

    }
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章