引言
一、WebSocket简介
WebSocket是一种在单个TCP连接上进行全双工通信的协议，它允许服务端主动向客户端推送数据，无需客户端发起请求。这使得WebSocket非常适合实现实时聊天、股票报价更新、游戏状态同步以及本文将要探讨的图片流式传输等场景。
二、Spring Boot 2中的WebSocket支持
Spring Framework 4.0引入了对WebSocket的支持，并且Spring Boot 2对其进行了进一步优化和封装，使开发者可以更便捷地创建WebSocket应用。
1.配置WebSocket Endpoint
在Spring Boot项目中，首先需要创建一个@ServerEndpoint注解标记的类来定义WebSocket的处理逻辑：
import org.springframework.stereotype.Component;
import javax.websocket.OnClose;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.server.ServerEndpoint;
@Component
@ServerEndpoint("/image-ws")
public class ImageWebSocketHandler {
// ... 省略其他方法 ...
@OnOpen
public void onOpen(Session session) {
// 连接打开后的处理逻辑
}
@OnMessage
public void onMessage(String message, Session session) {
// 处理文本消息（这里我们主要关注图片数据）
}
@OnClose
public void onClose(Session session, CloseReason reason) {
// 连接关闭后的处理逻辑
}
}2.发送图片数据
要通过WebSocket发送图片，通常我们需要将图片转换为字节数组或者Base64字符串。然后使用Session对象的getBasicRemote().sendBinary()方法发送二进制数据。
@OnMessage
public void sendImage(byte[] imageData, Session session) {
try {
session.getBasicRemote().sendBinary(ByteBuffer.wrap(imageData));
} catch (IOException e) {
// 处理IO异常
}
}或者，如果图片已经编码为Base64字符串，也可以先解码再发送：
@OnMessage
public void sendImage(String base64ImageString, Session session) {
byte[] imageData = Base64.getDecoder().decode(base64ImageString);
// 然后调用上述的sendImage(byte[], Session)方法
}3.客户端接收图片
客户端（例如基于Vue或React的前端应用）需要监听WebSocket连接上的binary事件，并将接收到的二进制数据渲染成图片：
var socket = new WebSocket('ws://localhost:8080/image-ws');
socket.binaryType = 'arraybuffer'; // 设置接收类型为ArrayBuffer
socket.onmessage = function(event) {
if (event.data instanceof ArrayBuffer) {
var blob = new Blob([event.data], {type: "image/jpeg"}); // 根据实际图片类型设置MIME类型
var imageUrl = URL.createObjectURL(blob);
var imgElement = document.createElement('img');
imgElement.src = imageUrl;
document.body.appendChild(imgElement);
}
};三、注意事项
- 发送大体积图片时需注意性能优化，如分块传输、压缩图片等。
- 对于多用户并发场景，需要考虑如何高效地管理和分发图片到各个已连接的客户端。
- 如果需要在浏览器限制之外提供更多的控制，可以考虑使用STOMP over WebSocket，通过Spring Messaging进行更高级的消息传递。
四、总结
综上所述，在Spring Boot 2中利用WebSocket发送图片是一项强大的功能，它极大地提升了应用程序的实时交互能力。然而，在实践中还需要根据具体需求设计合适的解决方案，以确保性能、稳定性和可扩展性。
