百度360必应搜狗淘宝本站头条
当前位置:网站首页 > IT知识 > 正文

Java WebSocket 示例(java nio websocket)

liuian 2025-07-17 20:39 41 浏览

一、环境准备

1. 依赖配置(Maven)

在 pom.xml 中添加 WebSocket 依赖: xml


<!-- Spring Boot WebSocket -->
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-websocket</artifactId>
</dependency>
<!-- 原生 WebSocket API(备用) -->
<dependency>
    <groupId>javax.websocket</groupId>
    <artifactId>javax.websocket-api</artifactId>
    <version>1.1</version>
</dependency>



二、服务端实现(Spring Boot)

1. WebSocket 配置类


import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.server.standard.ServerEndpointExporter;

@Configuration
public class WebSocketConfig {
    @Bean
    public ServerEndpointExporter serverEndpointExporter() {
        return new ServerEndpointExporter(); // 自动注册 @ServerEndpoint
    }
}


2. WebSocket 服务端类


import org.springframework.stereotype.Component;
import javax.websocket.*;
import javax.websocket.server.ServerEndpoint;
import java.io.IOException;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.atomic.AtomicInteger;

@Component
@ServerEndpoint("/chat/{username}")
public class ChatWebSocketServer {
    // 存储所有会话
    private static final ConcurrentHashMap<String, Session> sessions = new ConcurrentHashMap<>();
    private static final AtomicInteger onlineCount = new AtomicInteger(0);

    @OnOpen
    public void onOpen(Session session, @PathParam("username") String username) {
        sessions.put(username, session);
        onlineCount.incrementAndGet();
        System.out.println("用户 " + username + " 上线,当前在线人数:" + onlineCount);
    }

    @OnMessage
    public void onMessage(String message, Session session, @PathParam("username") String username) {
        System.out.println("收到消息:" + message);
        // 广播消息给所有用户
        broadcast(username + ": " + message);
    }

    @OnClose
    public void onClose(Session session, @PathParam("username") String username) {
        sessions.remove(username);
        onlineCount.decrementAndGet();
        System.out.println("用户 " + username + " 下线,当前在线人数:" + onlineCount);
    }

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

    // 广播消息
    private void broadcast(String message) {
        sessions.values().forEach(session -> {
            try {
                session.getAsyncRemote().sendText(message);
            } catch (IOException e) {
                e.printStackTrace();
            }
        });
    }
}



三、客户端实现

1. 浏览器端(HTML + JavaScript)


<!DOCTYPE html>
<html>
<head>
    <title>WebSocket 测试客户端</title>
</head>
<body>
    <input type="text" id="username" placeholder="用户名">
    <input type="text" id="message" placeholder="消息内容">
    <button onclick="sendMessage()">发送</button>
    <div id="output"></div>

    <script>
        const ws = new WebSocket("ws://localhost:8080/chat/用户昵称");
        ws.onmessage = function(event) {
            document.getElementById("output").innerHTML += event.data + "<br>";
        };

        function sendMessage() {
            const msg = document.getElementById("message").value;
            ws.send(msg);
        }
    </script>
</body>
</html>


2. Java 客户端(控制台交互)


import javax.websocket.*;
import java.net.URI;
import java.util.Scanner;

@ClientEndpoint
public class WebSocketClient {
    private Session session;

    public WebSocketClient(URI uri) {
        try {
            WebSocketContainer container = ContainerProvider.getWebSocketContainer();
            container.connectToServer(this, uri);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    @OnOpen
    public void onOpen(Session session) {
        this.session = session;
        System.out.println("已连接到服务器");
    }

    @OnMessage
    public void onMessage(String message) {
        System.out.println("收到消息:" + message);
    }

    public void sendMessage(String message) {
        try {
            session.getBasicRemote().sendText(message);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public static void main(String[] args) {
        WebSocketClient client = new WebSocketClient(URI.create("ws://localhost:8080/chat/TestUser"));
        Scanner scanner = new Scanner(System.in);
        while (true) {
            String msg = scanner.nextLine();
            client.sendMessage(msg);
        }
    }
}



四、运行与测试

1. 启动服务端

运行 Spring Boot 应用,访问 http://localhost:8080。

2. 浏览器客户端

3. 在浏览器中打开 HTML 文件,输入用户名和消息即可发送。 Java 控制台客户端

运行 WebSocketClient 类,输入消息发送到服务端。


五、关键功能说明

1. 路径参数传递

通过 @PathParam("username") 获取 URL 中的参数(如用户昵称)

2. 消息广播

使用 ConcurrentHashMap 存储所有会话,遍历发送消息

3. 异步发送

session.getAsyncRemote().sendText() 避免阻塞主线程


六、注意事项

1. 依赖冲突

确保 javax.websocket-api 与 Spring Boot 版本兼容。

2. 线程安全

使用 ConcurrentHashMap 保证多线程环境下会话操作的原子性

3. 异常处理

在 @OnError 中捕获连接异常,避免服务端崩溃

4. 路径配置

若需携带参数(如认证 Token),需在服务端解析 session.getRequestParameterMap()

相关推荐

教你把多个视频合并成一个视频的方法

一.情况介绍当你有一个m3u8文件和一个目录,目录中有连续的视频片段,这些片段可以连成一段完整的视频。m3u8文件打开后像这样:m3u8文件,可以理解为播放列表,里面是播放视频片段的顺序。视频片段像这...

零代码编程:用kimichat合并一个文件夹下的多个文件

一个文件夹里面有很多个srt字幕文件,如何借助kimichat来自动批量合并呢?在kimichat对话框中输入提示词:你是一个Python编程专家,完成如下的编程任务:这个文件夹:D:\downloa...

Java APT_java APT 生成代码

JavaAPT(AnnotationProcessingTool)是一种在Java编译阶段处理注解的工具。APT会在编译阶段扫描源代码中的注解,并根据这些注解生成代码、资源文件或其他输出,...

Unit Runtime:一键运行 AI 生成的代码,或许将成为你的复制 + 粘贴神器

在我们构建了UnitMesh架构之后,以及对应的demo之后,便着手于实现UnitMesh架构。于是,我们就继续开始UnitRuntime,以用于直接运行AI生成的代码。PS:...

挣脱臃肿的枷锁:为什么说Vert.x是Java开发者手中的一柄利剑?

如果你是一名Java开发者,那么你的职业生涯几乎无法避开Spring。它如同一位德高望重的老国王,统治着企业级应用开发的大片疆土。SpringBoot的约定大于配置、SpringCloud的微服务...

五年后,谷歌还在全力以赴发展 Kotlin

作者|FredericLardinois译者|Sambodhi策划|Tina自2017年谷歌I/O全球开发者大会上,谷歌首次宣布将Kotlin(JetBrains开发的Ja...

kotlin和java开发哪个好,优缺点对比

Kotlin和Java都是常见的编程语言,它们有各自的优缺点。Kotlin的优点:简洁:Kotlin程序相对于Java程序更简洁,可以减少代码量。安全:Kotlin在类型系统和空值安全...

移动端架构模式全景解析:从MVC到MVVM,如何选择最佳设计方案?

掌握不同架构模式的精髓,是构建可维护、可测试且高效移动应用的关键。在移动应用开发中,选择合适的软件架构模式对项目的可维护性、可测试性和团队协作效率至关重要。随着应用复杂度的增加,一个良好的架构能够帮助...

颜值非常高的XShell替代工具Termora,不一样的使用体验!

Termora是一款面向开发者和运维人员的跨平台SSH终端与文件管理工具,支持Windows、macOS及Linux系统,通过一体化界面简化远程服务器管理流程。其核心定位是解决多平台环境下远程连接、文...

预处理的底层原理和预处理编译运行异常的解决方案

若文章对您有帮助,欢迎关注程序员小迷。助您在编程路上越走越好![Mac-10.7.1LionIntel-based]Q:预处理到底干了什么事情?A:预处理,顾名思义,预先做的处理。源代码中...

为“架构”再建个模:如何用代码描述软件架构?

在架构治理平台ArchGuard中,为了实现对架构的治理,我们需要代码+模型描述所要处理的内容和数据。所以,在ArchGuard中,我们有了代码的模型、依赖的模型、变更的模型等,剩下的两个...

深度解析:Google Gemma 3n —— 移动优先的轻量多模态大模型

2025年6月,Google正式发布了Gemma3n,这是一款能够在2GB内存环境下运行的轻量级多模态大模型。它延续了Gemma家族的开源基因,同时在架构设计上大幅优化,目标是让...

比分网开发技术栈与功能详解_比分网有哪些

一、核心功能模块一个基本的比分网通常包含以下模块:首页/总览实时比分看板:滚动展示所有正在进行的比赛,包含比分、比赛时间、红黄牌等关键信息。热门赛事/焦点战:突出显示重要的、关注度高的比赛。赛事导航...

设计模式之-生成器_一键生成设计

一、【概念定义】——“分步构建复杂对象,隐藏创建细节”生成器模式(BuilderPattern):一种“分步构建型”创建型设计模式,它将一个复杂对象的构建与其表示分离,使得同样的构建过程可以创建...

构建第一个 Kotlin Android 应用_kotlin简介

第一步:安装AndroidStudio(推荐IDE)AndroidStudio是官方推荐的Android开发集成开发环境(IDE),内置对Kotlin的完整支持。1.下载And...