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

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

package VertxTCPIdTest;

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, NetSocket> idMap = 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 id = jsonObject.getString("id");
                if (id != null) {
                    if(idMap.get(id) != null){
                        socket.write("id已经存在，当前客户端id有：" + idMap.keySet().toString());
                    }else{
                        idMap.put(id, socket);
                        System.out.println("客户端" + id + "已加入，当前客户端id有：" + idMap.keySet().toString());
                    }
                } else{
                    String messageHead = jsonObject.getString("messageHead");
                    String messageBody = jsonObject.getString("messageBody");
                    if(messageHead != null){
                        NetSocket netSocket = idMap.get(messageHead);
                        if(netSocket == null){
                            socket.write("客户端" + messageHead +"未连接或者已经退出连接");
                        }else{
                            netSocket.write(messageBody);
                        }
                    }
                }
            });

            // 监听客户端的退出连接
            socket.closeHandler(close -> {
                for(Map.Entry mapEntry : idMap.entrySet()){
                    if(mapEntry.getValue() == socket){
                        String id = (String)mapEntry.getKey();
                        idMap.remove(id);
                        System.out.println("客户端" + id +"退出连接，当前客户端id有：" + idMap.keySet().toString());
                    }
                }
            });
        });

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

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

package VertxTCPIdTest;

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 ID = "3";
    private static NetSocket netSocket;

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

        // 连接服务器
        client.connect(33323, "localhost", conn -> {
            if (conn.succeeded()) {
                System.out.println("客户端" + ID + "连接服务端成功");
                netSocket = conn.result();
                // 向服务器写数据
                JSONObject idJsonObject = new JSONObject();
                idJsonObject.put("id", ID);
                netSocket.write(Buffer.buffer(idJsonObject.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()));
        }
    }
}