如何控制Spring Boot启动时的异常？

Question

我的Spring Boot应用程序试图在启动时加载一些证书文件。我希望能够在文件无法加载时显示一个明智的消息，而不是一个巨大的堆栈跟踪。

@SpringBootApplication
public class FileLoadTestApplication {
    public static void main(String[] args) {
        try {
            SpringApplication.run(FileLoadTestApplication .class, args);
        } catch (Exception e) {
            LOGGER.error("Application ending due to an error.");
            if (e.getClass().equals(ConfigException.class)) {
                if (e.getCause().getClass().equals(FileNotFoundException.class)) {
                    LOGGER.error("Unable to read file: " + e.getCause().getMessage());
                } else {
                    LOGGER.error("Initial configuration failed.", e.getCause());
                }
            }
        }
    }
}

如何在启动时控制一些异常，以便在关闭前显示一条信息性消息？

Answer 1

您可以捕获BeanCreationException并将其展开以获取原始原因。您可以使用以下代码片段

public static void main(String[] args) {
    try {
        SpringApplication.run(FileLoadTestApplication.class, args);
    } catch (BeanCreationException ex) {
        Throwable realCause = unwrap(ex);
        // Perform action based on real cause
    }
}

public static Throwable unwrap(Throwable ex) {
    if (ex != null && BeanCreationException.class.isAssignableFrom(ex.getClass())) {
        return unwrap(ex.getCause());
    } else {
        return ex;
    }
}

当@PostConstruct注释方法抛出异常时，Spring将其包装在BeanCreationException中并抛回。