springMVC spring websSocket 通讯协议 的使用

1、maven

<dependency>

<groupId>org.springframework</groupId>

<artifactId>spring-websocket</artifactId>

<version>4.3.5.RELEASE</version>

</dependency>

2、web.xml 要加一个webSocketServer 的映射

<!-- spring mvc -->
<servlet>
<servlet-name>dispatcher</servlet-name>
<servlet-class>org.springframework.web.servlet.DispatcherServlet</servlet-class>
<init-param> 
            <param-name>contextConfigLocation</param-name>
            <param-value>classpath:spring-mvc.xml</param-value>
        </init-param>  
<load-on-startup>1</load-on-startup>
</servlet>
<servlet-mapping>
<servlet-name>dispatcher</servlet-name>
<url-pattern>/webSocketServer/*</url-pattern>

</servlet-mapping>

3、java类

/**
 * WebScoket配置处理器
 */
@Component
@EnableWebSocket
public class WebSocketConfig extends WebMvcConfigurerAdapter implements WebSocketConfigurer {
public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
registry.addHandler(systemWebSocketHandler(), "/webSocketServer").addInterceptors(new HandShake())
.setAllowedOrigins("*");
registry.addHandler(systemWebSocketHandler(),"/webSocketServer/sockjs").addInterceptors(new                                 HandShake()).setAllowedOrigins("*").withSockJS();
}

@Bean
public WebSocketHandler systemWebSocketHandler() {
return new MyWebSocketHandler();
}
}

/**
 * Socket处理器
 */
@Component
public class MyWebSocketHandler implements WebSocketHandler {
private static Log log = LogFactory.getLog(MyWebSocketHandler.class);
public static final Map<String, WebSocketSession> userSocketSessionMap;
static {
userSocketSessionMap = new HashMap<String, WebSocketSession>();
}


/**
* 连接成功时候，会触发客户端UI页面上onopen方法 
*/
public void afterConnectionEstablished(WebSocketSession session) {
String query = session.getUri().getQuery();
String uid = query.split("=")[1];
if (userSocketSessionMap.get(uid) == null) {
userSocketSessionMap.put(uid, session);

}

// log.info("用户：【" + session.getId() + "】打开Websocket连接-->【" + uid + "】，打开数量："+userSocketSessionMap.size());

}


/**
* 消息处理，客户端UI页面在用js调用websocket.send()时候，会调用该方法 
*/
public void handleMessage(WebSocketSession session, WebSocketMessage<?> message) {
if (message.getPayloadLength() == 0) {
return;
}
Message msg = new Gson().fromJson(message.getPayload().toString(), Message.class);
// msg.setDate(new Date());
try {
// sendMessageToUser(msg.getTo(),new TextMessage(new GsonBuilder().setDateFormat("yyyy-MM-dd                                 HH:mm:ss").create().toJson(msg)));
sendMessageToUser(msg.getTo(),new TextMessage(new GsonBuilder().create().toJson(msg)));
} catch (IOException e) {
log.error(e.getMessage());
}
}


/**
* 给某个用户发送消息
* 
* @param userName
* @param message
* @throws IOException
*/
public void sendMessageToUser(String uid, TextMessage message) throws IOException {
WebSocketSession session = userSocketSessionMap.get(uid);
// log.info("用户：【" + session.getId() + "】，通过打开的Websocket连接-->【" + uid + "】，" + "发送消息:【" +                     message.getPayload().toString()+"】");
if (session != null && session.isOpen()) {
session.sendMessage(message);
}
}

/**
* 给所有在线用户发送消息
* 
* @param message
* @throws IOException
*/
public void broadcast(final TextMessage message) {
// log.info("给所有用户发送消息：" + message);
Iterator<Entry<String, WebSocketSession>> it = userSocketSessionMap.entrySet().iterator();

// 多线程群发
while (it.hasNext()) {
final Entry<String, WebSocketSession> entry = it.next();
if (entry.getValue().isOpen()) {
// entry.getValue().sendMessage(message);
new Thread(new Runnable() {
public void run() {
try {
if (entry.getValue().isOpen()) {
entry.getValue().sendMessage(message);
}
} catch (IOException e) {
e.printStackTrace();
}
}
}).start();
}
}
}



/**
* 消息传输错误处理
* 
* @throws IOException
*/
public void handleTransportError(WebSocketSession session, Throwable exception) throws IOException {
if (session.isOpen()) {
session.close();
}
Iterator<Entry<String, WebSocketSession>> it = userSocketSessionMap.entrySet().iterator();
// 移除Socket会话
while (it.hasNext()) {
Entry<String, WebSocketSession> entry = it.next();
if (entry.getValue().getId().equals(session.getId())) {
userSocketSessionMap.remove(entry.getKey());
// log.info("用户：【" + session.getId() + "】关闭Websocket连接-->【" + entry.getKey()+"】,");
break;
}
}
// log.info("当前数量：" + userSocketSessionMap.size());
}


/**
* 关闭连接后
*/
public void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus) {
Iterator<Entry<String, WebSocketSession>> it = userSocketSessionMap.entrySet().iterator();
// 移除Socket会话
while (it.hasNext()) {
Entry<String, WebSocketSession> entry = it.next();
if (entry.getValue().getId().equals(session.getId())) {
userSocketSessionMap.remove(entry.getKey());
// log.info("用户：【" + session.getId() + "】关闭Websocket连接-->【" + entry.getKey()+"】,");
break;
}
}
// log.info("当前数量：" + userSocketSessionMap.size());
}


/**
* 是否支持拆分消息
*/
public boolean supportsPartialMessages() {
return false;
}

}

4、页面js

$(function() {

var websocket;
var from = "MWEB_"+webuuid;

var wurl = $("#websocketUrl").val();
if ('WebSocket' in window) 
{
websocket = new WebSocket("ws://"+wurl+"/webSocketServer?uid="+from);
} 
else if ('MozWebSocket' in window) 
{
websocket = new MozWebSocket("ws://"+wurl+"/webSocketServer?uid="+from);
} 
else 
{
websocket = new SockJS("http://"+wurl+"/webSocketServer/sockjs?uid="+from);
}
websocket.onopen = function(event) 
{
console.log("WebSocket:已连接");
console.log(event);
};
websocket.onmessage = function(event) 
{
var data=JSON.parse(event.data);

console.log("WebSocket:收到一条消息",data);

                //接收到其他页面的信息

                getData(data);

};
websocket.onerror = function(event) 
{
console.log("WebSocket:发生错误 ");
console.log(event);
};
websocket.onclose = function(event) 
{
console.log("WebSocket:已关闭");
console.log(event);
};

        //发送信息

        $("#sendMag").click(function(){

            var id = $("#id").val();

             var name= $("#name").val();

             var data={};
var response="{\"id\":\""+id+"\",\"name\":\""+name+"\",\"height\":\""+1000
+"\",\"width\":\""+100+"\"}\r\n";
data["from"]=from;
data["text"]=response;
data["to"]="PC_"+webuuid;
websocket.send(JSON.stringify(data));

        });

        //接收信息

        getData(data){

            var dataText = JSON.parse(data.text);

            var id = dataText .id;

            var name = dataText.name;

        }

);

注：本文是自己总结，如有错误敬请谅解，评论指出错误，共同进步，共勉，谢谢。