SpringCloud/spring-cloud-function
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix: avoid ClassCastException for non-web application contexts (e.g. webflux, see https://github.com/aws/serverless-java-container/issues/904)

Browse Source
This commit is contained in:
Dennis Kieselhorst
2024-08-01 12:29:19 +02:00
parent 90f7565e64
commit 30d6ea9c21
1 changed files with 10 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
spring-cloud-function-adapters/spring-cloud-function-serverless-web/src/main/java/org/springframework/cloud/function/serverless/web/ServerlessAutoConfiguration.java
View File

@@ -22,7 +22,6 @@ import org.apache.commons.logging.LogFactory;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
import org.springframework.boot.web.context.ConfigurableWebServerApplicationContext;
import org.springframework.boot.web.server.WebServer;
import org.springframework.boot.web.server.WebServerException;
import org.springframework.boot.web.servlet.ServletContextInitializer;
@@ -53,7 +52,7 @@ public class ServerlessAutoConfiguration {
public static class ServerlessServletWebServerFactory
implements ServletWebServerFactory, ApplicationContextAware, InitializingBean {
private ConfigurableWebServerApplicationContext applicationContext;
private ApplicationContext applicationContext;
@Override
public WebServer getWebServer(ServletContextInitializer... initializers) {
@@ -77,28 +76,31 @@ public class ServerlessAutoConfiguration {
@Override
public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
this.applicationContext = (ConfigurableWebServerApplicationContext) applicationContext;
this.applicationContext = applicationContext;
}
@Override
public void afterPropertiesSet() throws Exception {
if (applicationContext instanceof ServletWebServerApplicationContext servletApplicationContet) {
if (applicationContext instanceof ServletWebServerApplicationContext servletApplicationContext) {
logger.info("Configuring Serverless Web Container");
ServerlessServletContext servletContext = new ServerlessServletContext();
servletApplicationContet.setServletContext(servletContext);
servletApplicationContext.setServletContext(servletContext);
DispatcherServlet dispatcher = applicationContext.getBean(DispatcherServlet.class);
try {
logger.info("Initializing DispatcherServlet");
dispatcher.init(new ProxyServletConfig(servletApplicationContet.getServletContext()));
logger.info("Initalized DispatcherServlet");
dispatcher.init(new ProxyServletConfig(servletApplicationContext.getServletContext()));
logger.info("Initialized DispatcherServlet");
}
catch (Exception e) {
throw new IllegalStateException("Faild to create Spring MVC DispatcherServlet proxy", e);
throw new IllegalStateException("Failed to create Spring MVC DispatcherServlet proxy", e);
}
for (ServletContextInitializer initializer : new ServletContextInitializerBeans(this.applicationContext)) {
initializer.onStartup(servletContext);
}
}
else {
logger.debug("Skipping Serverless configuration for " + this.applicationContext);
}
}
}
}