WebSocket

dependencygroupIdorg.springframework.boot/groupIdartifactIdspring-boot-starter-websocket/artifactId

创建一个WebSocket服务类如下

org.springframework.stereotype.Component;

import

javax.websocket.server.PathParam;

import

javax.websocket.server.ServerEndpoint;

import

sid);sessionMap.remove(sid);}/***

群发**

{//服务器向客户端发送消息session.getBasicRemote().sendText(message);}

catch

org.springframework.context.annotation.Bean;

import

org.springframework.context.annotation.Configuration;

import

org.springframework.web.socket.server.standard.ServerEndpointExporter;/***

WebSocket配置类用于注册WebSocket的Bean*/

Configuration

如下创建一个服务器主动推送给客户端的任务每5秒推送消息给客户端

在这里插入代码片import

com.hezy.websocket.WebSocketServer;

import

org.springframework.beans.factory.annotation.Autowired;

import

org.springframework.scheduling.annotation.Scheduled;

import

org.springframework.stereotype.Component;

import

java.time.format.DateTimeFormatter;Component

public

通过WebSocket每隔5秒向客户端发送消息*/Scheduled(cron

0/5

{webSocketServer.sendToAllClient(这是来自服务端的消息

DateTimeFormatter.ofPattern(HH:mm:ss).format(LocalDateTime.now()));}

}Step4启动类增加注解

启动类需要增加EnableWebSocket注解EnableScheduling注解是开启定时任务的配合定时器注解Scheduled(cron

0/5

org.springframework.boot.SpringApplication;

import

org.springframework.boot.autoconfigure.SpringBootApplication;

import

org.springframework.scheduling.annotation.EnableScheduling;

import

org.springframework.web.socket.config.annotation.EnableWebSocket;/***

author

{SpringApplication.run(Start.class,

args);}

使用第三方框架GitHub地址netty-websocket-spring-boot-starter的方式整合WebSocket非常容易如下

Step1引入依赖

dependencygroupIdorg.yeauty/groupIdartifactIdnetty-websocket-spring-boot-starter/artifactIdversion0.12.0/version

import

io.netty.handler.codec.http.HttpHeaders;

import

io.netty.handler.timeout.IdleStateEvent;

import

org.springframework.util.MultiValueMap;

import

java.util.Map;ServerEndpoint(host

${ws.host},

pathMap){session.setSubprotocols(stomp);if

(!ok.equals(req)){System.out.println(Au***ntication

failed!);session.close();}}OnOpenpublic

void

pathMap){System.out.println(new

connection);System.out.println(req);}OnClosepublic

void

{throwable.printStackTrace();}OnMessagepublic

void

{System.out.println(message);session.sendText(Hello

void

{System.out.println(b);}session.sendBinary(bytes);

}OnEventpublic

READER_IDLE:System.out.println(read

idle);break;case

WRITER_IDLE:System.out.println(write

idle);break;case

ALL_IDLE:System.out.println(all

}Step3增加配置

在配置文件中新增以下配置表示websocket接收的主机和端口如下

websocket配置

填上websocket服务地址和端口记得加上一个参数reqok因为我们上面代码里有个判断

连接成功

可以看到发送到消息成功同时也接收到了服务端传递过来的消息就是说此时如果代码里需要发送消息给客户端就可以主动发送消息过来可以基于上面的代码写自己的业务逻辑。