引言

在現代Web應用中，實時通信技術已經成為提升用戶體驗的關鍵因素之一。WebSocket協議作為一種全雙工的通信協議，允許客戶端與服務器之間建立持久連接，並進行雙向數據傳輸。在Spring Boot 2框架下，我們可以輕松地實現WebSocket功能並將其用於發送圖片等二進制數據。

一、WebSocket簡介

WebSocket是一種在單個TCP連接上進行全雙工通信的協議，它允許服務端主動向客戶端推送數據，無需客戶端發起請求。這使得WebSocket非常適合實現實時聊天、股票報價更新、遊戲狀態同步以及本文將要探討的圖片流式傳輸等場景。

二、Spring Boot 2中的WebSocket支持

Spring Framework 4.0引入了對WebSocket的支持，並且Spring Boot 2對其進行了進一步優化和封裝，使開發者可以更便捷地創建WebSocket應用。

2.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.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)方法
   }

2.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發送圖片是一項強大的功能，它極大地提升了應用程序的實時交互能力。然而，在實踐中還需要根據具體需求設計合適的解決方案，以確保性能、穩定性和可擴展性。

基於Spring Boot 2 使用WebSocket發送圖片

引言