diff --git a/src/main/java/im/zhaojun/zfile/config/WebMvcConfig.java b/src/main/java/im/zhaojun/zfile/config/WebMvcConfig.java index 84de067..08aa2f0 100644 --- a/src/main/java/im/zhaojun/zfile/config/WebMvcConfig.java +++ b/src/main/java/im/zhaojun/zfile/config/WebMvcConfig.java @@ -2,12 +2,19 @@ package im.zhaojun.zfile.config; import im.zhaojun.zfile.model.enums.StorageTypeEnumDeSerializerConvert; import org.springframework.boot.web.embedded.tomcat.TomcatServletWebServerFactory; +import org.springframework.boot.web.server.ConfigurableWebServerFactory; +import org.springframework.boot.web.server.ErrorPage; +import org.springframework.boot.web.server.WebServerFactoryCustomizer; import org.springframework.boot.web.servlet.server.ServletWebServerFactory; import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Configuration; import org.springframework.format.FormatterRegistry; +import org.springframework.http.HttpStatus; import org.springframework.web.servlet.config.annotation.WebMvcConfigurer; +import java.util.HashSet; +import java.util.Set; + /** * @author zhaojun */ @@ -31,4 +38,14 @@ public class WebMvcConfig implements WebMvcConfigurer { return webServerFactory; } + @Bean + public WebServerFactoryCustomizer webServerFactoryCustomizer(){ + return factory -> { + ErrorPage error404Page = new ErrorPage(HttpStatus.NOT_FOUND, "/index.html"); + Set errorPages = new HashSet<>(); + errorPages.add(error404Page); + factory.setErrorPages(errorPages); + }; + } + } \ No newline at end of file