SpringBoot集成websocket（1）|（websocket客户端实现）
章节
前言
WebSocket使得客户端和服务器之间的数据交换变得更加简单，允许服务端主动向客户端推送数据。在WebSocket API中，浏览器和服务器只需要完成一次握手，两者之间就直接可以创建持久性的连接，并进行双向数据传输。
一、websocket客户端依赖引入
pom依赖如下
<dependency>
<groupId>org.java-websocket</groupId>
<artifactId>Java-WebSocket</artifactId>
<version>1.5.3</version>
</dependency>
二、websocket代码实现
1.WSClient客户端实现
可以定义了websocket的client类，其他地方如果需要使用ws连接，直接注入即可使用
import lombok.extern.slf4j.Slf4j;
import org.java_websocket.client.WebSocketClient;
import org.java_websocket.drafts.Draft_6455;
import org.java_websocket.handshake.ServerHandshake;
import org.springframework.stereotype.Component;
import java.net.URI;
@Slf4j
@Component
public class WSClient {
public WebSocketClient webSocketClient(String url) {
try {
WebSocketClient webSocketClient = new WebSocketClient(new URI(url),new Draft_6455()) {
@Override
public void onOpen(ServerHandshake handshakedata) {
log.info("[websocket] 连接成功");
}
@Override
public void onMessage(String message) {
log.info("[websocket] 收到消息={}",message);
}
@Override
public void onClose(int code, String reason, boolean remote) {
log.info("[websocket] 退出连接");
}
@Override
public void onError(Exception ex) {
log.info("[websocket] 连接错误={}",ex.getMessage());
}
};
webSocketClient.connect();
return webSocketClient;
} catch (Exception e) {
e.printStackTrace();
}
return null;
}
}
2.WSClient客户端使用实例
定义了怎么使用WSClient类建立一个ws连接
@Api(tags = {
"websocket测试"})
@RestController
@RequestMapping("/ws")
public class SendMessageCtrl {
@Autowired
private WSClient wsClient;
@GetMapping(value = "hello")
public void hello(String name) throws InterruptedException {
WebSocketClient client = this.wsClient.webSocketClient("ws://localhost:8905/ws/test");
if (client != null){
for (int i = 0; i < 10; i++) {
Thread.sleep(1000);
client.send("客户端发送消息"+i);
}
}
}
}
三、实际使用场景
1.多WSClient连接使用
上述是一个简单的demo，能够通过客户端连接websocket接口，但实际场景可能是很多用户访问本地的服务，改服务需要调用远程的ws接口，这个时候怎么区分用户间独立的建立websocket连接呢
思路：定义一个全局变量，用于存放websocket的客户端，客户端触发发送消息事件，在全局变量里面获取对应的client进行消息发送
@Api(tags = {
"websocket测试"})
@RestController
@RequestMapping("/ws")
public class SendMessageCtrl {
@Autowired
private WSClient wsClient;
private static ConcurrentHashMap<String, WebSocketClient> wsClientSet = new ConcurrentHashMap<>();
@GetMapping(value = "hello")
public void hello(String name) {
WebSocketClient client;
if (wsClientSet.containsKey(name)) {
client = wsClientSet.get(name);
} else {
client = this.wsClient.webSocketClient("ws://localhost:8905/ws/" + name);
wsClientSet.put(name, client);
}
if (client != null) {
client.send("请根据卖火柴的小女孩生成一个类似的故事");
}
}
}
总结
以上就是SpringBoot集成websocket客户端的内容，用于后端服务调用websocket请求。具体业务逻辑执行在代码中自己实现，这里只是给了具体的demo和实例，实际中运用场景还需要自己琢磨。