07--SpringBoot啟動之自定義打印Banner

SpringBoot控制臺打印Banner可以在properties文件里開啟或關(guān)閉spring.main.show_banner=false
但是打印Banner功能是如何實現(xiàn)呢?如何自定義打印自己的Banner呢?
繼續(xù)run方法之前我們先來了解下Banner打印的模式
Banner接口的Mode枚舉定義了三種模式:

enum Mode {
    //關(guān)閉
    OFF,
    //打印控制臺
    CONSOLE,
    //打印到文件
    LOG
}

了解了banner的模式,繼續(xù)run方法

//打印banner
Banner printedBanner = printBanner(environment);

private Banner printBanner(ConfigurableEnvironment environment) {
    // 如果未開啟banner打印直接返回
    if (this.bannerMode == Banner.Mode.OFF) {
        return null;
    }
    // 創(chuàng)建ResourceLoader對象
    ResourceLoader resourceLoader = (this.resourceLoader != null) ? this.resourceLoader : new DefaultResourceLoader(getClassLoader());
    // 創(chuàng)建SpringApplicationBannerPrinter,該對象用來打印banner
    SpringApplicationBannerPrinter bannerPrinter = new SpringApplicationBannerPrinter(resourceLoader, this.banner);
    // 如果bannerMode模式為LOG,則將banner打印到log文件里
    if (this.bannerMode == Mode.LOG) {
        return bannerPrinter.print(environment, this.mainApplicationClass, logger);
    }
    // 打印banner到控制臺
    return bannerPrinter.print(environment, this.mainApplicationClass, System.out);
}

分析print方法,分為獲取banner和打印banner兩部分,逐步分析

// 打印banner
public Banner print(Environment environment, Class<?> sourceClass, PrintStream out) {
    // 獲取banner
    Banner banner = getBanner(environment);
    // 打印
    banner.printBanner(environment, sourceClass, out);
    // 返回PrintedBanner對象,PrintedBanner實現(xiàn)了Banner接口
    return new PrintedBanner(banner, sourceClass);
}

1. 獲取banner

// 獲取banner
private Banner getBanner(Environment environment) {
    // 創(chuàng)建Banners對象,保存獲取到的banner
    Banners banners = new Banners();
    // 如果圖片banner不為空,則加入到Banners對象的banners集合中
    banners.addIfNotNull(getImageBanner(environment));
    // 如果文字banner不為空,則加入到Banners對象的banners集合中
    banners.addIfNotNull(getTextBanner(environment));
    // 如果至少獲取到一個圖片banner或文字banner,則返回
    if (banners.hasAtLeastOneBanner()) {
        return banners;
    }
    // 如果fallbackBanner不為null,則返回(下面詳解fallbackBanner作用)
    if (this.fallbackBanner != null) {
        return this.fallbackBanner;
    }
    // 上述banner都未獲取到,返回默認banner
    return DEFAULT_BANNER;
}

可以看到,獲取的banner一共有四個步驟,逐步分析

1.1 獲取圖片banner

// 獲取圖片banner
private Banner getImageBanner(Environment environment) {
    // 如果在配置文件中自定義了spring.banner.image.location屬性,則優(yōu)先返回
    // spring.banner.image.location=classpath:myBanner.gif
    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;
    }
    // 循環(huán){ "gif", "jpg", "png" }擴展名,并從resources目錄下依次查找,按擴展名排序返回第一個
    for (String ext : IMAGE_EXTENSION) {
        Resource resource = this.resourceLoader.getResource("banner." + ext);
        if (resource.exists()) {
            return new ImageBanner(resource);
        }
    }
    return null;
}

1.2 獲取文字banner

//獲取文字banner
private Banner getTextBanner(Environment environment) {
    // getProperty()方法并不是同時返回兩個資源,而是
    // 如果用戶自定義了spring.banner.location則返回spring.banner.location的資源位置
    // 否則返回banner.txt的資源位置
    String location = environment.getProperty(BANNER_LOCATION_PROPERTY,DEFAULT_BANNER_LOCATION);
    // 從并從resources目錄下查找資源并返回
    Resource resource = this.resourceLoader.getResource(location);
    if (resource.exists()) {
        return new ResourceBanner(resource);
    }
    return null;
}

1.3 獲取fallbackBanner

// 如果fallbackBanner不為null,則返回(下面詳解fallbackBanner作用)
if (this.fallbackBanner != null) {
    return this.fallbackBanner;
}

如果上述兩個方法都沒有找到對應(yīng)的banner,那么會判斷fallbackBanner是否為null,不為null則返回,那么fallbackBanner怎么使用呢?在SpringApplication.java文件中有如下方法,大家可以猜想到,是手動設(shè)置該banner

public void setBanner(Banner banner) {
    this.banner = banner;
}
// 在BannerTest類中,有
@Test
public void testCustomBanner() {
    SpringApplication application = createSpringApplication();
    application.setBanner(new DummyBanner());
    this.context = application.run();
    assertThat(this.out.toString()).contains("My Banner");
}

即手動創(chuàng)建SpringApplication對象后,調(diào)動其setBanner方法設(shè)置fallbackBanner

1.4 獲取默認banner

如果上述三種方式都沒獲取到banner怎么辦呢?返回默認banner
private static final Banner DEFAULT_BANNER = new SpringBootBanner();
打開SpringBanner類,熟悉的banner出現(xiàn)了...

class SpringBootBanner implements Banner {

    private static final String[] BANNER = { "",
            "  .   ____          _            __ _ _",
            " /\\\\ / ___'_ __ _ _(_)_ __  __ _ \\ \\ \\ \\",
            "( ( )\\___ | '_ | '_| | '_ \\/ _` | \\ \\ \\ \\",
            " \\\\/  ___)| |_)| | | | | || (_| |  ) ) ) )",
            "  '  |____| .__|_| |_|_| |_\\__, | / / / /",
            " =========|_|==============|___/=/_/_/_/" };

    private static final String SPRING_BOOT = " :: Spring Boot :: ";

    private static final int STRAP_LINE_SIZE = 42;

    @Override
    public void printBanner(Environment environment, Class<?> sourceClass,
            PrintStream printStream) {
        for (String line : BANNER) {
            printStream.println(line);
        }
        String version = SpringBootVersion.getVersion();
        version = (version != null) ? " (v" + version + ")" : "";
        StringBuilder padding = new StringBuilder();
        while (padding.length() < STRAP_LINE_SIZE
                - (version.length() + SPRING_BOOT.length())) {
            padding.append(" ");
        }

        printStream.println(AnsiOutput.toString(AnsiColor.GREEN, SPRING_BOOT,
                AnsiColor.DEFAULT, padding.toString(), AnsiStyle.FAINT, version));
        printStream.println();
    }

}

上面就分析了SpringBoot是如何打印banner,包括如何自定義banner,大家可以打開spring-boot-sample-simple工程的resources目錄,可以看到有banner.jpg和banner.txt兩個文件,這個就是自定義的banner文件,大家可以替換成自己的banner,就能實現(xiàn)自定義banner的打印了

2. 打印banner

循環(huán)打印所有獲取到的banner

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

打印banner的代碼比較簡單,就不在粘貼分析了,大家可debug跟蹤下,打印圖片也有兩種方式

圖片banner-->ImageBanner類的printBanner方法
文字banner-->ResourceBanner類的printBanner方法
今天就分析到這里了...