springboot 中使用websocket简单例子.docx

springboot 中使用websocket简单例子.docx

《springboot 中使用websocket简单例子.docx》由会员分享，可在线阅读，更多相关《springboot 中使用websocket简单例子.docx（6页珍藏版）》请在冰豆网上搜索。

springboot中使用websocket简单例子

springboot中使用websocket简单例子

gradle中添加依赖，引入websocket支持

compile（"org.springframework.boot:

spring-boot-starter-websocket:

${springBootVersion}"）

启用websocket

packagecn.xiaojf.today.ws.configuration;

importcn.xiaojf.today.ws.handler.CountWebSocketHandler;

importcn.xiaojf.today.ws.interceptor.HandshakeInterceptor;

importorg.springframework.context.ApplicationContext;

importorg.springframework.context.annotation.Bean;

importorg.springframework.context.annotation.Configuration;

importorg.springframework.web.socket.config.annotation.EnableWebSocket;

importorg.springframework.web.socket.config.annotation.WebSocketConfigurer;

importorg.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;

importorg.springframework.web.socket.server.standard.ServerEndpointExporter;

/**

*websocket配置

*@authorxiaojf2017/3/29:

50.

*/

@Configuration

@EnableWebSocket

publicclassWebSocketConfigimplementsWebSocketConfigurer{

@Bean

publicServerEndpointExporterserverEndpointExporter（ApplicationContextcontext）{

returnnewServerEndpointExporter（）;

}

@Override

publicvoidregisterWebSocketHandlers（WebSocketHandlerRegistryregistry）{

registry.addHandler（newCountWebSocketHandler（）,"/web/count"）.addInterceptors（newHandshakeInterceptor（））;

}

}

消息拦截处理

packagecn.xiaojf.today.ws.interceptor;

importorg.springframework.http.server.ServerHttpRequest;

importorg.springframework.http.server.ServerHttpResponse;

importorg.springframework.web.socket.WebSocketHandler;

importorg.springframework.web.socket.server.support.HttpSessionHandshakeInterceptor;

importjava.util.Map;

/**

*消息拦截处理类

*@authorxiaojf2017/3/210:

36.

*/

publicclassHandshakeInterceptorextendsHttpSessionHandshakeInterceptor{

@Override

publicbooleanbeforeHandshake（ServerHttpRequestrequest,ServerHttpResponseresponse,WebSocketHandlerwsHandler,

Map<String,Object>attributes）throwsException{

//解决Theextension[x-webkit-deflate-frame]isnotsupported问题

if（request.getHeaders（）.containsKey（"Sec-WebSocket-Extensions"））{

request.getHeaders（）.set（"Sec-WebSocket-Extensions","permessage-deflate"）;

}

System.out.println（"BeforeHandshake"）;

returnsuper.beforeHandshake（request,response,wsHandler,attributes）;

}

@Override

publicvoidafterHandshake（ServerHttpRequestrequest,ServerHttpResponseresponse,WebSocketHandlerwsHandler,

Exceptionex）{

System.out.println（"AfterHandshake"）;

super.afterHandshake（request,response,wsHandler,ex）;

}

}

消息处理相应类

packagecn.xiaojf.today.ws.handler;

importcom.alibaba.druid.support.json.JSONUtils;

importmons.collections.map.HashedMap;

importorg.springframework.stereotype.Component;

importorg.springframework.web.socket.CloseStatus;

importorg.springframework.web.socket.TextMessage;

importorg.springframework.web.socket.WebSocketSession;

importorg.springframework.web.socket.handler.TextWebSocketHandler;

importjava.io.IOException;

importjava.util.*;

/**

*@authorxiaojf2017/3/29:

55.

*/

@Component

publicclassCountWebSocketHandlerextendsTextWebSocketHandler{

privatestaticlongcount=0;

privatestaticMap<String,WebSocketSession>sessionMap=newHashedMap（）;

@Override

protectedvoidhandleTextMessage（WebSocketSessionsession,TextMessagemessage）throwsException{

//session.sendMessage（newTextMessage（session.getPrincipal（）.getName（）+",你是第"+（sessionMap.size（））+"位访客"））;//p2p

Objectparse=JSONUtils.parse（message.getPayload（））;

Collection<WebSocketSession>sessions=sessionMap.values（）;

for（WebSocketSessionws:

sessions）{//广播

ws.sendMessage（message）;

}

sendMessage（sessionMap.keySet（）,"你好"）;

}

@Override

publicvoidafterConnectionEstablished（WebSocketSessionsession）throwsException{

sessionMap.put（session.getPrincipal（）.getName（）,session）;

super.afterConnectionEstablished（session）;

}

@Override

publicvoidafterConnectionClosed（WebSocketSessionsession,CloseStatusstatus）throwsException{

sessionMap.remove（session.getPrincipal（）.getName（））;

super.afterConnectionClosed（session,status）;

}

/**

*发送消息

*@authorxiaojf2017/3/211:

43

*/

publicstaticvoidsendMessage（Stringusername,Stringmessage）throwsIOException{

sendMessage（Arrays.asList（username）,Arrays.asList（message））;

}

/**

*发送消息

*@authorxiaojf2017/3/211:

43

*/

publicstaticvoidsendMessage（Collection<String>acceptorList,Stringmessage）throwsIOException{

sendMessage（acceptorList,Arrays.asList（message））;

}

/**

*发送消息，p2p群发都支持

*@authorxiaojf2017/3/211:

43

*/

publicstaticvoidsendMessage（Collection<String>acceptorList,Collection<String>msgList）throwsIOException{

if（acceptorList!

=null&&msgList!

=null）{

for（Stringacceptor:

acceptorList）{

WebSocketSessionsession=sessionMap.get（acceptor）;

if（session!

=null）{

for