spingboot中创建简单的WebSocket服务和使用OKHttp创建socket客户端接收数据

背景

springboot 中使用okhttp3创建webSocket服务端 server1 和客户端
client1，客户端clinet1调用server1用于发送图片，创建客户端client2接收此图片.

在Spring
Boot中使用OkHttp3创建WebSocket服务端和客户端，涉及到两个不同的操作：建立WebSocket服务端和创建WebSocket客户端。OkHttp3库主要用于客户端操作，因此对于服务端，我们将使用Spring的WebSocket支持。

步骤 1: 创建WebSocket服务端（Server1）

1. 添加Spring WebSocket依赖

在pom.xml中添加Spring Boot的WebSocket依赖：

<dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-websocket</artifactId>
</dependency>

2. 配置WebSocket

创建一个配置类来启用WebSocket并注册一个endpoint：

import org.springframework.context.annotation.Configuration;
import org.springframework.messaging.simp.config.MessageBrokerRegistry;
import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker;
import org.springframework.web.socket.config.annotation.StompEndpointRegistry;
import org.springframework.web.socket.config.annotation.WebSocketMessageBrokerConfigurer;@Configuration
@EnableWebSocketMessageBroker
public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {@Overridepublic void registerStompEndpoints(StompEndpointRegistry registry) {registry.addEndpoint("/ws").setAllowedOrigins("*").withSockJS();}@Overridepublic void configureMessageBroker(MessageBrokerRegistry registry) {registry.enableSimpleBroker("/topic");registry.setApplicationDestinationPrefixes("/app");}
}

3. 创建消息处理器

创建一个控制器来处理发送到服务端的消息：

import org.springframework.messaging.handler.annotation.MessageMapping;
import org.springframework.messaging.handler.annotation.SendTo;
import org.springframework.stereotype.Controller;@Controller
public class ImageMessageController {@MessageMapping("/receive-image")@SendTo("/topic/image")public byte[] receiveImage(byte[] imageData) {// 处理接收到的图片数据return imageData; // 将图片数据转发到订阅了 /topic/image 的客户端}
}

步骤 2: 创建WebSocket客户端（Client1）

1. 添加OkHttp依赖

在pom.xml中添加OkHttp3依赖：

<dependency><groupId>com.squareup.okhttp3</groupId><artifactId>okhttp</artifactId><version>4.9.0</version> <!-- 请使用最新版本 -->
</dependency>

2. 创建WebSocket客户端

创建一个客户端来连接到WebSocket服务端并发送图片：

import okhttp3.OkHttpClient;
import okhttp3.Request;
import okhttp3.WebSocket;
import okhttp3.WebSocketListener;
import okio.ByteString;public class ImageWebSocketClient extends WebSocketListener {private final OkHttpClient client = new OkHttpClient();private WebSocket webSocket;public void connect(String url) {Request request = new Request.Builder().url(url).build();webSocket = client.newWebSocket(request, this);}public void sendImage(byte[] imageData) {if (webSocket != null) {webSocket.send(ByteString.of(imageData));}}@Overridepublic void onOpen(WebSocket webSocket, okhttp3.Response response) {super.onOpen(webSocket, response);// WebSocket连接成功建立}@Overridepublic void onMessage(WebSocket webSocket, ByteString bytes) {super.onMessage(webSocket, bytes);// 处理接收到的二进制消息}@Overridepublic void onClosed(WebSocket webSocket, int code, String reason) {super.onClosed(webSocket, code, reason);// 处理WebSocket关闭事件}@Overridepublic void onFailure(WebSocket webSocket, Throwable t, okhttp3.Response response) {super.onFailure(webSocket, t, response);// 处理WebSocket连接失败事件}public void close() {if (webSocket != null) {webSocket.close(1000, "End of session");}}
}

步骤 3: 创建另一个客户端（Client2）

创建另一个客户端来接收由Client1发送的图片：

public class ImageReceivingClient extends WebSocketListener {private final OkHttpClient client = new OkHttpClient();private WebSocket webSocket;public void connect(String url) {Request request = new Request.Builder().url(url).build();webSocket = client.newWebSocket(request, this);}@Overridepublic void onMessage(WebSocket webSocket, ByteString bytes) {super.onMessage(webSocket, bytes);// 处理接收到的图片数据byte[] imageData = bytes.toByteArray();// 可以在这里将图片数据保存到文件或进行其他处理}public void close() {if (webSocket != null) {webSocket.close(1000, "End of session");}}
}

步骤 4: 使用客户端

在你的服务中，使用上面创建的ImageWebSocketClient和ImageReceivingClient来发送和接收图片。

public class WebSocketService {public void sendImage(String url, byte[] imageData) {ImageWebSocketClient client = new ImageWebSocketClient();client.connect(url);client.sendImage(imageData);client.close();}public void receiveImage(String url) {ImageReceivingClient client = new ImageReceivingClient();client.connect(url);// 保持连接，直到需要断开}
}

注意事项

• 确保WebSocket服务端和客户端使用相同的协议（ws://或wss://）。
• 在生产环境中，确保使用wss://来加密WebSocket连接。
• 处理WebSocket连接的异常和错误，确保应用的健壮性。