使用Java API的WebSocket (JSR-356)与Spring Boot

Question

我是Spring的新手(并在stackoverflow上提问)。

我想通过Spring Boot启动一个嵌入式(Tomcat)服务器，并向其注册一个JSR-356 WebSocket端点。

这是主要方法:

@ComponentScan
@EnableAutoConfiguration
public class Server {
    public static void main(String[] args) {
        SpringApplication.run(Server.class, args);
    }
}

配置如下:

@Configuration
public class EndpointConfig {
    @Bean
    public EchoEndpoint echoEndpoint() {
        return new EchoEndpoint();
    }
    @Bean
    public ServerEndpointExporter endpointExporter() {
        return new ServerEndpointExporter();
    } 
}

EchoEndpoint的实现是直接的:

@ServerEndpoint(value = "/echo", configurator = SpringConfigurator.class)
public class EchoEndpoint {
    @OnMessage
    public void handleMessage(Session session, String message) throws IOException {
        session.getBasicRemote().sendText("echo: " + message);
    }
}

第二部分我遵循这个博客文章:https://spring.io/blog/2013/05/23/spring-framework-4-0-m1-websocket-support.

但是，当我运行应用程序时，得到:

Exception in thread "main" org.springframework.beans.factory.BeanCreationException: Error creating bean with name 'endpointExporter' defined in class path resource [hello/EndpointConfig.class]: Initialization of bean failed; nested exception is java.lang.IllegalStateException: Failed to get javax.websocket.server.ServerContainer via ServletContext attribute

该异常进一步由ServerEndpointExporter中的NullPointerException引起，因为applicationContext上的getServletContext方法此时仍返回null。

有更了解Spring的人可以帮助我吗?谢谢!

Answer 1

ServerEndpointExporter对应用程序上下文的生命周期做了一些假设，这些假设在使用Spring Boot时并不成立。具体来说，它假设当setApplicationContext被调用时，在ApplicationContext上调用getServletContext将返回一个非空值。

你可以通过替换

来解决这个问题:

@Bean
public ServerEndpointExporter endpointExporter() {
    return new ServerEndpointExporter();
} 

:

@Bean
public ServletContextAware endpointExporterInitializer(final ApplicationContext applicationContext) {
    return new ServletContextAware() {
        @Override
        public void setServletContext(ServletContext servletContext) {
            ServerEndpointExporter serverEndpointExporter = new ServerEndpointExporter();
            serverEndpointExporter.setApplicationContext(applicationContext);
            try {
                serverEndpointExporter.afterPropertiesSet();
            } catch (Exception e) {
                throw new RuntimeException(e);
            }               
        }           
    };
}

这将延迟处理，直到servlet上下文可用。

更新:您可能喜欢观看SPR-12109。一旦修复了，上面描述的解决方法就不再需要了。