javax.websocket 使用指南

除了https://blog.csdn.net/u011556070/article/details/105844786 的建立ws server方法之外这里再收录记载一种JSR356的方法javax.websocket

需要注意在这种方法中使用spring的bean会需要额外的工作

@Configuration
public class WebConfig {

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

    @Bean
    public WebSocketEndPoint webSocketEndPoint() {
        return new WebSocketEndPoint();
    }
}

@Slf4j
@ServerEndpoint(value = "/task/block/{taskId}")
public class WebSocketEndPoint {

    @OnOpen
    public void onOpen(@PathParam("taskId") String taskId, Session session) {
        log.info("taskId={}　链接websocket", taskId);
    }

    @OnClose
    public void onClose(Session session) {
        log.info("sessionId={}　关闭websocket", session.getId());
    }

    @OnMessage
    public void onMessage(Session session, String message) {
        log.info("接受信息：" + message);
    }

    @OnError
    public void onError(Session session, Throwable error) {
        log.error("客户端错误：" + error.getMessage(), error);
    }
}

上述这种方法的优势会比spring websocket使用起来略微简单一点，但是无法在endpoint中直接用@Autowired注入bean, 原因是@ServerEndpoint的JSR356和spring的注入顺序的关系

如果想在endpoint中使用spring的DI需要自己写一个SpringContextUtil

@Component
public class SpringContextUtil implements ApplicationContextAware {

    private static ApplicationContext applicationContext;

    //通过名字获取上下文中的bean
    public static Object getBean(String name) {
        return applicationContext.getBean(name);
    }

    //通过名字获取上下文中的bean
    public static <T> T getBean(String name, Class<T> tClass) {
        return applicationContext.getBean(name, tClass);
    }

    //通过类型获取上下文中的bean
    public static <T> T getBean(Class<T> requiredType) {
        return applicationContext.getBean(requiredType);
    }

    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        this.applicationContext=applicationContext;
    }

    public static ApplicationContext getApplicationContext() {
        return applicationContext;
    }
}

然后即可在使用如下方式注入所需要的bean， 例如

 private SomeHandler someHandler = SpringContextUtil.getBean(SomeHandler.class);