Vert.x实战六：TCP客户端之间以功能名通过服务端转接通信

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

TCP发送功能名称，负责执行功能的客户端收到消息，功能和负责执行功能的客户端可配置。

package VertxTCPFunctionTest;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
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.HashMap;
import java.util.List;
import java.util.Map;

public class VertxTCPServer extends AbstractVerticle {
    private Map<String, NetSocket> idMap = new HashMap<>();
    private Map<String, List<String>> functionMap = FunctionReader.getFunctionData();

    @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 connectClientID = jsonObject.getString("id");
                if (connectClientID != null) {
                    if(idMap.get(connectClientID) != null){
                        socket.write("id已经存在，当前id有：" + idMap.keySet().toString());
                    }else{
                        idMap.put(connectClientID, socket);
                        System.out.println("客户端" + connectClientID + "已加入，当前客户端id有：" + idMap.keySet().toString());
                    }
                } else{
                    String messageHead = jsonObject.getString("messageHead");
                    String messageBody = jsonObject.getString("messageBody");
                    if(messageHead != null){
                        List<String> idList = functionMap.get(messageHead);
                        JSONObject responseJsonObject = new JSONObject();
                        JSONObject messageJsonObject = new JSONObject();
                        JSONArray infoJsonArray = new JSONArray();
                        if(idList == null){
                            infoJsonArray.add("未配置功能" + messageHead);
                            responseJsonObject.put("info", infoJsonArray);
                            socket.write(responseJsonObject.toJSONString());
                        }else{
                            for(String functionID : idList){
                                NetSocket netSocket = idMap.get(functionID);
                                if(netSocket == null){
                                    String infoPart =  "负责接收功能" + messageHead + "通知的客户端" + functionID + "不在线";
                                    infoJsonArray.add(infoPart);
                                }else{
                                    if(netSocket == socket){
                                        messageJsonObject.put("message", messageBody);
                                    }else{
                                        JSONObject otherMessageJsonObject = new JSONObject();
                                        otherMessageJsonObject.put("message", messageBody);
                                        netSocket.write(otherMessageJsonObject.toString());
                                    }

                                    String infoPart = "已发送" + messageHead + "给客户端" + functionID;
                                    infoJsonArray.add(infoPart);
                                }
                            }

                            responseJsonObject.put("info", infoJsonArray);
                            if(messageJsonObject != null){
                                responseJsonObject.put("message", messageJsonObject);
                            }

                            socket.write(responseJsonObject.toJSONString());
                        }

                    }
                }
            });

            // 监听客户端的退出连接
            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 VertxTCPFunctionTest;

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 = "2";
    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()));
        }
    }
}

package VertxTCPFunctionTest;

import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.dom4j.Element;
import org.dom4j.io.SAXReader;

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

public class FunctionReader {
    public static void main(String[] args){
        getFunctionData();
    }

    public static Map<String, List<String>> getFunctionData(){
        Document document;
        try {
            SAXReader reader = new SAXReader();
            String filePath = new File("").getAbsolutePath() + "/src/VertxTCPFunctionTest/function.xml";
            File file = new File(filePath);
            if(file == null){
                return null;
            }

            document = reader.read(file);

            Map<String, List<String>> functionMap = new HashMap<>();
            for(Element functionElement : document.getRootElement().elements("Function")){
                String functionName = functionElement.attributeValue("functionName");
                List<String> idList = functionMap.get(functionName);
                if(idList == null){
                    idList = new ArrayList<>();
                }

                for(Element idElement : functionElement.elements("ID")){
                    String id = idElement.getTextTrim();
                    if(!idList.contains(id)){
                        idList.add(id);
                    }
                }

                functionMap.put(functionName, idList);
            }

            return functionMap;
        } catch (DocumentException e) {
            e.printStackTrace();
            return null;
        }
    }
}

<Functions>
    <Function functionName = "A">
        <ID>1</ID>
        <ID>2</ID>
    </Function>
    <Function functionName = "B">
        <ID>1</ID>
        <ID>3</ID>
        <ID>4</ID>
    </Function>
</Functions>