Spring - Websocket詳細配置

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.servlet.config.annotation.EnableWebMvc;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurerAdapter;
import org.springframework.web.socket.WebSocketHandler;
import org.springframework.web.socket.config.annotation.EnableWebSocket;
import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;

@Configuration
@EnableWebMvc
@EnableWebSocket
public class WebSocketConfig extends WebMvcConfigurerAdapter implements
        WebSocketConfigurer {

    public WebSocketConfig() {
    }

    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        registry.addHandler(systemWebSocketHandler(), "/websck").addInterceptors(new HandshakeInterceptor()).setAllowedOrigins("http://localhost");

        System.out.println("registed!");
        registry.addHandler(systemWebSocketHandler(), "/sockjs/websck/info").addInterceptors(new HandshakeInterceptor()).setAllowedOrigins("http://localhost")
                .withSockJS();

    }

    @Bean
    public WebSocketHandler systemWebSocketHandler() {
        //return new InfoSocketEndPoint();
        return new SystemWebSocketHandler();
    }

}

 

@Component
public class SystemWebSocketHandler implements WebSocketHandler {

    @Override
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
        System.out.println("connect to the websocket success......");
        session.sendMessage(new TextMessage("Server:connected OK!"));
    }

    @Override
    public void handleMessage(WebSocketSession wss, WebSocketMessage<?> wsm) throws Exception {
        TextMessage returnMessage = new TextMessage(wsm.getPayload()
            + " received at server");
      wss.sendMessage(returnMessage);
    }

    @Override
    public void handleTransportError(WebSocketSession wss, Throwable thrwbl) throws Exception {
        if(wss.isOpen()){
            wss.close();
        }
       System.out.println("websocket connection closed......");
    }

    @Override
    public void afterConnectionClosed(WebSocketSession wss, CloseStatus cs) throws Exception {
        System.out.println("websocket connection closed......");
    }

    @Override
    public boolean supportsPartialMessages() {
        return false;
    }
    
}

import org.springframework.stereotype.Component;
import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketSession;
import org.springframework.web.socket.handler.TextWebSocketHandler;

@Component
public class InfoSocketEndPoint extends TextWebSocketHandler {

   public InfoSocketEndPoint() {
      // TODO Auto-generated constructor stub
   }

   @Override
   protected void handleTextMessage(WebSocketSession session,
         TextMessage message) throws Exception {
      // TODO Auto-generated method stub
      super.handleTextMessage(session, message);
      TextMessage returnMessage = new TextMessage(message.getPayload()
            + " received at server");
      session.sendMessage(returnMessage);
   }

/* @Override
   protected void handleBinaryMessage(WebSocketSession arg0, BinaryMessage arg1) {
      // TODO Auto-generated method stub
      super.handleBinaryMessage(arg0, arg1);
   }*/
   
   

}

 

import org.springframework.http.server.ServerHttpRequest;
import org.springframework.http.server.ServerHttpResponse;
import org.springframework.stereotype.Component;
import org.springframework.web.socket.WebSocketHandler;
import org.springframework.web.socket.server.support.HttpSessionHandshakeInterceptor;

import java.util.Map;


@Component
public class HandshakeInterceptor extends HttpSessionHandshakeInterceptor {

   @Override
   public boolean beforeHandshake(ServerHttpRequest request,
         ServerHttpResponse response, WebSocketHandler wsHandler,
         Map<String, Object> attributes) throws Exception {
      System.out.println("Before Handshake");
      return super.beforeHandshake(request, response, wsHandler, attributes);
   }

   @Override
   public void afterHandshake(ServerHttpRequest request,
         ServerHttpResponse response, WebSocketHandler wsHandler,
         Exception ex) {
      System.out.println("After Handshake");
      super.afterHandshake(request, response, wsHandler, ex);
   }

}

<%@ page language="java" contentType="text/html; charset=UTF-8"
       pageEncoding="UTF-8"%>
<!DOCTYPE html PUBLIC "-//W3C//DTD HTML 4.01 Transitional//EN" "http://www.w3.org/TR/html4/loose.dtd">
<html>
<head>
   <title>WebSocket/SockJS Echo Sample (Adapted from Tomcat's echo sample)</title>
   <style type="text/css">
      #connect-container {
         float: left;
         width: 400px
      }

      #connect-container div {
         padding: 5px;
      }

      #console-container {
         float: left;
         margin-left: 15px;
         width: 400px;
      }

      #console {
         border: 1px solid #CCCCCC;
         border-right-color: #999999;
         border-bottom-color: #999999;
         height: 170px;
         overflow-y: scroll;
         padding: 5px;
         width: 100%;
      }

      #console p {
         padding: 0;
         margin: 0;
      }
   </style>

   <script src="./sockjs-0.3.4.js"></script>

   <script type="text/javascript">
      var ws = null;
      var url = null;
      var transports = [];

      function setConnected(connected) {
         document.getElementById('connect').disabled = connected;
         document.getElementById('disconnect').disabled = !connected;
         document.getElementById('echo').disabled = !connected;
      }

      function connect() {
         if (!url) {
            alert('Select whether to use W3C WebSocket or SockJS');
            return;
         }

         //ws = (url.indexOf('sockjs') != -1) ?new SockJS(url, undefined, {protocols_whitelist: transports}) : new WebSocket(url);
         if ('WebSocket' in window) {
            ws = new WebSocket("ws://localhost:8080/ws/websck");
         }else {
            ws = new SockJS("http://localhost:8080/ws/sockjs/websck/info");
         }
         //websocket = new SockJS("http://localhost:8084/SpringWebSocketPush/sockjs/websck");
         ws.onopen = function () {
            alert('open');
            setConnected(true);
            //log('Info: connection opened.');
         };
         ws.onmessage = function (event) {
            alert('Received:' + event.data);
            log('Received: ' + event.data);
         };
         ws.onclose = function (event) {
            setConnected(false);
            log('Info: connection closed.');
            log(event);
         };
      }

      function disconnect() {
         if (ws != null) {
            ws.close();
            ws = null;
         }
         setConnected(false);
      }

      function echo() {
         if (ws != null) {
            var message = document.getElementById('message').value;
            log('Sent: ' + message);
            ws.send(message);
         } else {
            alert('connection not established, please connect.');
         }
      }

      function updateUrl(urlPath) {
         if (urlPath.indexOf('sockjs') != -1) {
            url = urlPath;
            document.getElementById('sockJsTransportSelect').style.visibility = 'visible';
         }
         else {
            if (window.location.protocol == 'http:') {
               url = 'ws://' + window.location.host + urlPath;
            } else {
               url = 'wss://' + window.location.host + urlPath;
            }
            document.getElementById('sockJsTransportSelect').style.visibility = 'hidden';
         }
      }

      function updateTransport(transport) {
         alert(transport);
         transports = (transport == 'all') ?  [] : [transport];
      }

      function log(message) {
         var console = document.getElementById('console');
         var p = document.createElement('p');
         p.style.wordWrap = 'break-word';
         p.appendChild(document.createTextNode(message));
         console.appendChild(p);
         while (console.childNodes.length > 25) {
            console.removeChild(console.firstChild);
         }
         console.scrollTop = console.scrollHeight;
      }
   </script>
</head>
<body>
<noscript><h2 style="color: #ff0000">Seems your browser doesn't support Javascript! Websockets
   rely on Javascript being enabled. Please enable
   Javascript and reload this page!</h2></noscript>
<div>
   <div id="connect-container">
      <input id="radio1" type="radio" name="group1" οnclick="updateUrl('/websocket');">
      <label for="radio1">W3C WebSocket</label>
      <br>
      <input id="radio2" type="radio" name="group1" οnclick="updateUrl('/sockjs/websocket');">
      <label for="radio2">SockJS</label>
      <div id="sockJsTransportSelect" style="visibility:hidden;">
         <span>SockJS transport:</span>
         <select οnchange="updateTransport(this.value)">
            <option value="all">all</option>
            <option value="websocket">websocket</option>
            <option value="xhr-polling">xhr-polling</option>
            <option value="jsonp-polling">jsonp-polling</option>
            <option value="xhr-streaming">xhr-streaming</option>
            <option value="iframe-eventsource">iframe-eventsource</option>
            <option value="iframe-htmlfile">iframe-htmlfile</option>
         </select>
      </div>
      <div>
         <button id="connect" οnclick="connect();">Connect</button>
         <button id="disconnect" disabled="disabled" οnclick="disconnect();">Disconnect</button>
      </div>
      <div>
         <textarea id="message" style="width: 350px">Here is a message!</textarea>
      </div>
      <div>
         <button id="echo" οnclick="echo();" disabled="disabled">Echo message</button>
      </div>
   </div>
   <div id="console-container">
      <div id="console"></div>
   </div>
</div>
</body>

</html>