Vert.x实战三：TCP客户端之间以角色通过服务端转接通信

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/haoranhaoshi/article/details/89296606

TCP客户端对应自己的角色，同一角色可对应多个客户端，客户端可对特定角色客户端发送信息。

package VertxTCPRoleTest;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import io.vertx.core.AbstractVerticle;
import io.vertx.core.Vertx;
import io.vertx.core.net.NetServer;
import io.vertx.core.net.NetSocket;


import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class VertxTCPServer extends AbstractVerticle {
    private Map<String, List<NetSocket>> roleMap = new HashMap<>();

    @Override
    public void start() {

        // 创建TCP服务器
        NetServer server = vertx.createNetServer();

        // 处理连接请求
        server.connectHandler(socket -> {
            socket.handler(buffer -> {
                // 在这里应该解析报文，封装为协议对象，并找到响应的处理类，得到处理结果，并响应
                String message = buffer.toString();
                System.out.println("接收到的数据为：" + message);
                JSONObject jsonObject = JSON.parseObject(message);
                String role = jsonObject.getString("role");
                if (role != null) {
                    List<NetSocket> netSocketList = roleMap.get(role);
                    if (netSocketList == null) {
                        netSocketList = new ArrayList<>();
                    }

                    netSocketList.add(socket);
                    roleMap.put(role, netSocketList);
                } else{
                    String messageHead = jsonObject.getString("messageHead");
                    String messageBody = jsonObject.getString("messageBody");
                    if(messageHead != null){
                        for(NetSocket netSocket : roleMap.get(messageHead)){
                            netSocket.write(messageBody);
                        }
                    }
                }
            });

            // 监听客户端的退出连接
            socket.closeHandler(close -> {
                for(Map.Entry mapEntry : roleMap.entrySet()){
                    List<NetSocket> netSocketList = (List<NetSocket>)mapEntry.getValue();
                    String sourceRole = (String)mapEntry.getKey();
                    for(NetSocket netSocket : netSocketList){
                        if(netSocket == socket){
                            netSocketList.remove(netSocket);
                            System.out.println("客户端（角色为" + sourceRole + "）退出连接");
                            // 删除后角色对应的Socket连接列表无子项，则删除此角色与对应Socket连接列表的映射
                            if(netSocketList.size() == 0){
                                roleMap.remove(mapEntry.getKey());
                            }

                            System.out.println("客户端角色和Socket连接列表的映射集合已更新");
                        }
                    }
                }
            });

        });

        // 监听端口
        server.listen(33323, res -> {
            if (res.succeeded()) {
                System.out.println("服务器启动成功");
            }
        });
    }

    public static void main(String[] args) {
        Vertx.vertx().deployVerticle(new VertxTCPServer());
    }
}

package VertxTCPRoleTest;

import com.alibaba.fastjson.JSONObject;
import io.vertx.core.AbstractVerticle;
import io.vertx.core.Vertx;
import io.vertx.core.buffer.Buffer;
import io.vertx.core.net.NetClient;
import io.vertx.core.net.NetSocket;

import java.util.Scanner;

public class VertxTCPClient extends AbstractVerticle {
    private static String ROLE = "A";
    private static NetSocket netSocket;

    @Override
    public void start() {
        // 创建一个TCP客户端
        NetClient client = vertx.createNetClient();

        // 连接服务器
        client.connect(33323, "localhost", conn -> {
            if (conn.succeeded()) {
                System.out.println("客户端连接服务端成功");
                netSocket = conn.result();
                // 向服务器写数据
                JSONObject roleJsonObject = new JSONObject();
                roleJsonObject.put("role", ROLE);
                netSocket.write(Buffer.buffer(roleJsonObject.toJSONString()));

                // 读取服务器的响应数据
                netSocket.handler(buffer -> System.out.println("接收到的数据为：" + buffer.toString()));
            } else {
                System.out.println("连接服务器异常");
            }
        });

    }

    public static void main(String[] args) {
        Vertx.vertx().deployVerticle(new VertxTCPClient());
        // 向服务端发送消息
        Scanner scanner = new Scanner(System.in);
        while (scanner.hasNext()) {
            String message = scanner.next();
            JSONObject messageJsonObject = new JSONObject();
            messageJsonObject.put("messageHead", message.split(":")[0]);
            messageJsonObject.put("messageBody", message.split(":")[1]);
            netSocket.write(Buffer.buffer(messageJsonObject.toJSONString()));
        }
    }
}