tio websocket 客户端 java 代码 工具类

为了更好地组织代码并提高可复用性,我们可以将WebSocket客户端封装成一个工具类。这样可以在多个地方方便地使用WebSocket客户端功能。以下是使用tio库实现的一个WebSocket客户端工具类。

  1. 添加依赖

确保项目中添加了tio的依赖。如果使用的是Maven,可以在pom.xml文件中添加以下依赖:

复制代码
<dependencies>
    <!-- 其他依赖 -->
    <dependency>
        <groupId>com.github.xiaoymin</groupId>
        <artifactId>tio-core</artifactId>
        <version>2.1.10</version>
    </dependency>
    <dependency>
        <groupId>com.github.xiaoymin</groupId>
        <artifactId>tio-websocket</artifactId>
        <version>2.1.10</version>
    </dependency>
</dependencies>
  1. 创建WebSocket客户端工具类

创建一个工具类TioWebSocketClientUtil,用于管理WebSocket客户端的连接和消息发送。

复制代码
import org.tio.client.AioClient;
import org.tio.client.intf.ClientAioListener;
import org.tio.core.ChannelContext;
import org.tio.core.intf.Packet;
import org.tio.websocket.client.WsClientConfig;
import org.tio.websocket.client.intf.ClientWsAioListener;
import org.tio.websocket.common.WsPacket;

import java.nio.charset.StandardCharsets;
import java.util.concurrent.CountDownLatch;

public class TioWebSocketClientUtil {

    private static final String SERVER_URL = "ws://your-spring-boot-server-url/websocket-endpoint";
    private static final int PORT = 8080; // 服务器端口
    private static final CountDownLatch latch = new CountDownLatch(1);
    private static AioClient aioClient;

    public static void init() throws Exception {
        WsClientConfig config = new WsClientConfig();
        config.setHeartbeatTimeoutSeconds(60); // 心跳超时时间

        ClientWsAioListener aioListener = new ClientWsAioListener() {
            @Override
            public void onAfterConnected(ChannelContext channelContext, boolean isConnected, boolean isReconnect, boolean isFromHeartbeat) throws Exception {
                super.onAfterConnected(channelContext, isConnected, isReconnect, isFromHeartbeat);
                System.out.println("WebSocket connection opened.");
                latch.countDown();
            }

            @Override
            public void onAfterClosed(ChannelContext channelContext, Throwable throwable, String s, String s1) throws Exception {
                super.onAfterClosed(channelContext, throwable, s, s1);
                System.out.println("WebSocket connection closed.");
            }

            @Override
            public void onAfterSent(ChannelContext channelContext, Packet packet, boolean isSentSuccess) throws Exception {
                super.onAfterSent(channelContext, packet, isSentSuccess);
                System.out.println("Message sent: " + packet);
            }

            @Override
            public void onAfterReceivedBytes(ChannelContext channelContext, byte[] bytes, int len) throws Exception {
                super.onAfterReceivedBytes(channelContext, bytes, len);
                System.out.println("Received bytes: " + new String(bytes, StandardCharsets.UTF_8));
            }

            @Override
            public void onAfterReceivedPacket(ChannelContext channelContext, Packet packet) throws Exception {
                super.onAfterReceivedPacket(channelContext, packet);
                if (packet instanceof WsPacket) {
                    WsPacket wsPacket = (WsPacket) packet;
                    System.out.println("Received message: " + wsPacket.getString());
                }
            }
        };

        aioClient = new AioClient(config, aioListener);
        aioClient.start();
    }

    public static void connect() throws Exception {
        aioClient.connect(SERVER_URL, "client", PORT, null, null, 5000);
        latch.await(); // 等待连接成功
    }

    public static void sendMessage(String message) throws Exception {
        WsPacket wsPacket = new WsPacket();
        wsPacket.setString(message);
        aioClient.send(wsPacket);
    }

    public static void close() throws Exception {
        aioClient.stop();
    }

    public static void main(String[] args) throws Exception {
        init();
        connect();

        // 发送消息
        sendMessage("Hello, server!");

        // 等待一段时间后关闭连接
        Thread.sleep(10000);
        close();
    }
}
  1. 解释

TioWebSocketClientUtil 类:封装了WebSocket客户端的基本功能。

init 方法:初始化WebSocket客户端配置并启动客户端。

connect 方法:连接到WebSocket服务器。

sendMessage 方法:发送消息给服务器。

close 方法:关闭客户端连接。

main 方法:演示如何使用这个工具类。

  1. 使用工具类

在其他类中可以通过调用TioWebSocketClientUtil的方法来使用WebSocket客户端功能:

复制代码
public class MainApp {
    public static void main(String[] args) {
        try {
            TioWebSocketClientUtil.init();
            TioWebSocketClientUtil.connect();
            
            // 发送消息
            TioWebSocketClientUtil.sendMessage("Hello, server!");
            
            // 等待一段时间后关闭连接
            Thread.sleep(10000);
            TioWebSocketClientUtil.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}

通过这种方式,可以将WebSocket客户端功能封装成一个工具类,方便在多个地方使用。这样不仅提高了代码的可复用性,也使得客户端的管理和维护更加方便。

相关推荐
一只小鱼儿吖38 分钟前
进程代理单窗口单IP技术:原理、应用与实现
网络·网络协议·tcp/ip
稳联技术39 分钟前
Ethernet IP与Profinet共舞:网关驱动绿色工业的智慧脉动
网络·网络协议·tcp/ip
学习3人组1 小时前
CentOS配置网络
linux·网络·centos
高兴达1 小时前
RPC框架--实现一个非常简单的RPC调用
网络协议·rpc·firefox
~山有木兮2 小时前
LiteHub中间件之限流实现
网络·http·中间件
cui_win2 小时前
【网络】Linux 内核优化实战 - net.core.flow_limit_table_len
linux·运维·网络
BD_Marathon3 小时前
虚拟机网络检查
网络
游戏开发爱好者83 小时前
iOS App首次启动请求异常调试:一次冷启动链路抓包与初始化流程修复
websocket·网络协议·tcp/ip·http·网络安全·https·udp
2501_915106323 小时前
频繁迭代下完成iOS App应用上架App Store:一次快速交付项目的完整回顾
websocket·网络协议·tcp/ip·http·网络安全·https·udp
cocologin4 小时前
RIP 技术深度解析
运维·网络·网络协议