WebSocket编程：Java实现实时双向通信应用

WebSocket 简介

WebSocket 是一种在单个 TCP 连接上进行全双工通信的协议，适用于需要实时交互的应用场景（如聊天、实时数据推送）。与 HTTP 不同，WebSocket 允许服务器主动向客户端推送数据，无需频繁建立连接。

Java WebSocket 实现方式

Java 提供了两种主要方式实现 WebSocket：

Java API for WebSocket (JSR 356)：标准 API，支持注解和编程式配置。
Spring WebSocket：基于 Spring 框架的封装，简化集成（如 STOMP 协议支持）。

使用 JSR 356 实现 WebSocket

服务端实现

java 复制代码

import javax.websocket.*;
import javax.websocket.server.ServerEndpoint;

@ServerEndpoint("/ws") // 定义 WebSocket 端点路径
public class WebSocketServer {

    @OnOpen
    public void onOpen(Session session) {
        System.out.println("客户端连接: " + session.getId());
    }

    @OnMessage
    public void onMessage(String message, Session session) {
        System.out.println("收到消息: " + message);
        // 向客户端发送响应
        session.getAsyncRemote().sendText("服务器回复: " + message);
    }

    @OnClose
    public void onClose(Session session) {
        System.out.println("连接关闭: " + session.getId());
    }

    @OnError
    public void onError(Session session, Throwable error) {
        error.printStackTrace();
    }
}

客户端实现（JavaScript 示例）

javascript 复制代码

const socket = new WebSocket("ws://localhost:8080/ws");
socket.onopen = () => console.log("连接已建立");
socket.onmessage = (event) => console.log("收到消息: " + event.data);
socket.send("Hello WebSocket!");

使用 Spring WebSocket 实现

依赖配置

Maven 需添加以下依赖：

XML 复制代码

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

服务端配置

java 复制代码

import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.config.annotation.*;

@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {

    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        registry.addHandler(new MyWebSocketHandler(), "/ws").setAllowedOrigins("*");
    }
}

自定义处理器

java 复制代码

import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketSession;
import org.springframework.web.socket.handler.TextWebSocketHandler;

public class MyWebSocketHandler extends TextWebSocketHandler {

    @Override
    protected void handleTextMessage(WebSocketSession session, TextMessage message) {
        System.out.println("收到消息: " + message.getPayload());
        session.sendMessage(new TextMessage("服务器回复: " + message.getPayload()));
    }
}