今天就跟大家聊聊有关SpringBoot中怎么利用WebSocket实现即时消息，可能很多人都不太了解，为了让大家更加了解，小编给大家总结了以下内容，希望大家根据这篇文章可以有所收获。
成都创新互联从2013年创立，是专业互联网技术服务公司，拥有项目网站建设、网站制作网站策划，项目实施与项目整合能力。我们以让每一个梦想脱颖而出为使命，1280元湘乡做网站,已为上家服务,为湘乡各地企业和个人服务,联系电话:028-86922220
依赖
org.springframework.boot spring-boot-starter-web org.springframework.boot spring-boot-starter-websocket
定义消息类型
抽象消息对象
public class AbstractMessage { /** * 消息类型 */ protected String type ; /** * 消息内容 */ protected String content ; /** * 消息日期 */ protected String date ; }消息对象子类
1、Ping检查消息
public class PingMessage extends AbstractMessage { public PingMessage() {} public PingMessage(String type) { this.type = type ; } }2、系统消息
public class SystemMessage extends AbstractMessage { public SystemMessage() {} public SystemMessage(String type, String content) { this.type = type ; this.content = content ; } }3、点对点消息
public class PersonMessage extends AbstractMessage { private String fromName ; private String toName ; }消息类型定义
public enum MessageType { /** * 系统消息 0000;心跳检查消息 0001;点对点消息2001 */ SYSTEM("0000"), PING("0001"), PERSON("2001") ; private String type ; private MessageType(String type) { this.type = type ; } public String getType() { return type; } public void setType(String type) { this.type = type; } } WebSocket服务端点
该类作用就是定义客户端连接的地址
@ServerEndpoint(value = "/message/{username}", encoders = {WsMessageEncoder.class}, decoders = {WsMessageDecoder.class}, subprotocols = {"gmsg"}, configurator = MessageConfigurator.class) @Component public class GMessageListener { public static ConcurrentMap sessions = new ConcurrentHashMap<>(); private static Logger logger = LoggerFactory.getLogger(GMessageListener.class) ; private String username ; @OnOpen public void onOpen(Session session, EndpointConfig config, @PathParam("username") String username){ UserSession userSession = new UserSession(session.getId(), username, session) ; this.username = username ; sessions.put(username, userSession) ; logger.info("【{}】用户进入, 当前连接数：{}", username, sessions.size()) ; } @OnClose public void onClose(Session session, CloseReason reason){ UserSession userSession = sessions.remove(this.username) ; if (userSession != null) { logger.info("用户【{}】, 断开连接, 当前连接数：{}", username, sessions.size()) ; } } @OnMessage public void pongMessage(Session session, PongMessage message) { ByteBuffer buffer = message.getApplicationData() ; logger.debug("接受到Pong帧【这是由浏览器发送】：" + buffer.toString()); } @OnMessage public void onMessage(Session session, AbstractMessage message) { if (message instanceof PingMessage) { logger.debug("这里是ping消息"); return ; } if (message instanceof PersonMessage) { PersonMessage personMessage = (PersonMessage) message ; if (this.username.equals(personMessage.getToName())) { logger.info("【{}】收到消息：{}", this.username, personMessage.getContent()); } else { UserSession userSession = sessions.get(personMessage.getToName()) ; if (userSession != null) { try { userSession.getSession().getAsyncRemote().sendText(new ObjectMapper().writeValueAsString(message)) ; } catch (JsonProcessingException e) { e.printStackTrace(); } } } return ; } if (message instanceof SystemMessage) { logger.info("接受到消息类型为【系统消息】") ; return ; } } @OnError public void onError(Session session, Throwable error) { logger.error(error.getMessage()) ; } }WsMessageEncoder.java类
该类的主要作用是，当发送的消息是对象时，该如何转换
public class WsMessageEncoder implements Encoder.Text { private static Logger logger = LoggerFactory.getLogger(WsMessageDecoder.class) ; @Override public void init(EndpointConfig endpointConfig) { } @Override public void destroy() { } @Override public String encode(AbstractMessage tm) throws EncodeException { String message = null ; try { message = new ObjectMapper().writeValueAsString(tm); } catch (JsonProcessingException e) { logger.error("JSON处理错误：{}", e) ; } return message; } }
WsMessageDecoder.java类
该类的作用是，当接收到消息时如何转换成对象。
public class WsMessageDecoder implements Decoder.Text { private static Logger logger = LoggerFactory.getLogger(WsMessageDecoder.class) ; private static Set msgTypes = new HashSet<>() ; static { msgTypes.add(MessageType.PING.getType()) ; msgTypes.add(MessageType.SYSTEM.getType()) ; msgTypes.add(MessageType.PERSON.getType()) ; } @Override @SuppressWarnings("unchecked") public AbstractMessage decode(String s) throws DecodeException { AbstractMessage message = null ; try { ObjectMapper mapper = new ObjectMapper() ; Map map = mapper.readValue(s, Map.class) ; String type = map.get("type") ; switch(type) { case "0000": message = mapper.readValue(s, SystemMessage.class) ; break; case "0001": message = mapper.readValue(s, PingMessage.class) ; break; case "2001": message = mapper.readValue(s, PersonMessage.class) ; break; } } catch (JsonProcessingException e) { logger.error("JSON处理错误：{}", e) ; } return message ; } // 该方法判断消息是否可以被解码（转换） @Override @SuppressWarnings("unchecked") public boolean willDecode(String s) { Map map = new HashMap<>() ; try { map = new ObjectMapper().readValue(s, Map.class); } catch (JsonProcessingException e) { e.printStackTrace(); } logger.debug("检查消息：【" + s + "】是否可以解码") ; String type = map.get("type") ; if (StringUtils.isEmpty(type) || !msgTypes.contains(type)) { return false ; } return true ; } @Override public void init(EndpointConfig endpointConfig) { } @Override public void destroy() { } }
MessageConfigurator.java类
该类的作用是配置服务端点，比如配置握手信息
public class MessageConfigurator extends ServerEndpointConfig.Configurator { private static Logger logger = LoggerFactory.getLogger(MessageConfigurator.class) ; @Override public void modifyHandshake(ServerEndpointConfig sec, HandshakeRequest request, HandshakeResponse response) { logger.debug("握手请求头信息：" + request.getHeaders()); logger.debug("握手响应头信息：" + response.getHeaders()); super.modifyHandshake(sec, request, response); } }WebSocke配置类
@Configuration public class WebSocketConfig { @Bean public ServerEndpointExporter serverEndpointExporter (){ return new ServerEndpointExporter(); } }当以jar包形式运行时需要配置该bean，暴露我们配置的@ServerEndpoint;当我们以war独立tomcat运行时不能配置该bean。
前端页面
WebSocket 进入
