【RPC注册发现框架实战】一个简易的RPC注册发现框架

Java实现 服务端起10个线程ID监听40-49这10个端口，这10个端口注册到注册中心，提供同一个服务，发个A，响应B，客户端起10个线程去注册中心请求
好的，我们可以通过实现一个简单的服务端、注册中心和客户端来达到这个目的。以下是详细的代码示例：

服务端代码

import java.io.*;
import java.net.*;
import java.util.concurrent.*;public class MultiPortServer {private static final int PORT_START = 40;private static final int PORT_END = 49;private static final ExecutorService executor = Executors.newFixedThreadPool(PORT_END - PORT_START + 1);public static void main(String[] args) {for (int port = PORT_START; port <= PORT_END; port++) {int finalPort = port;executor.submit(() -> startServer(finalPort));RegistryCenter.registerService(finalPort, "localhost");}}private static void startServer(int port) {try (ServerSocket serverSocket = new ServerSocket(port)) {System.out.println("Server listening on port " + port);while (true) {Socket clientSocket = serverSocket.accept();handleClient(clientSocket);}} catch (IOException e) {e.printStackTrace();}}private static void handleClient(Socket clientSocket) {try (BufferedReader in = new BufferedReader(new InputStreamReader(clientSocket.getInputStream()));PrintWriter out = new PrintWriter(clientSocket.getOutputStream(), true)) {String message = in.readLine();if ("A".equals(message)) {out.println("B");} else {out.println("Unknown message");}} catch (IOException e) {e.printStackTrace();}}
}

注册中心代码

import java.util.*;
import java.util.concurrent.*;public class RegistryCenter {private static final Map<Integer, String> serviceRegistry = new ConcurrentHashMap<>();public static void registerService(int port, String address) {serviceRegistry.put(port, address);System.out.println("Service registered on port " + port + " at address " + address);}public static String lookupService(int port) {return serviceRegistry.get(port);}public static List<Integer> getAllServicePorts() {return new ArrayList<>(serviceRegistry.keySet());}
}

客户端代码

import java.io.*;
import java.net.*;
import java.util.List;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;public class Client {private static final int CLIENT_COUNT = 10;private static final ExecutorService executor = Executors.newFixedThreadPool(CLIENT_COUNT);public static void main(String[] args) {List<Integer> servicePorts = RegistryCenter.getAllServicePorts();for (int i = 0; i < CLIENT_COUNT; i++) {executor.submit(() -> {int portIndex = (int) (Math.random() * servicePorts.size());int servicePort = servicePorts.get(portIndex);queryService(servicePort);});}executor.shutdown();}private static void queryService(int servicePort) {String serviceAddress = RegistryCenter.lookupService(servicePort);if (serviceAddress != null) {try (Socket socket = new Socket(serviceAddress, servicePort);PrintWriter out = new PrintWriter(socket.getOutputStream(), true);BufferedReader in = new BufferedReader(new InputStreamReader(socket.getInputStream()))) {out.println("A");String response = in.readLine();System.out.println("Response from server on port " + servicePort + ": " + response);} catch (IOException e) {e.printStackTrace();}} else {System.out.println("No service found on port " + servicePort);}}
}

运行步骤

1. 启动服务端：

   • 运行 MultiPortServer 类。它将启动10个线程，每个线程监听一个端口（40-49），并将这些服务注册到注册中心。

2. 启动客户端：

   • 运行 Client 类。它将启动10个客户端线程，每个线程随机选择一个服务端口，从注册中心获取该端口的服务地址，并发送消息 “A”。

说明

1. 服务端：

   • MultiPortServer 类创建一个线程池，每个线程监听一个特定端口，并将该端口的服务注册到 RegistryCenter。
   • 每个服务端口接收到消息 “A” 后，响应消息 “B”。

2. 注册中心：

   • RegistryCenter 类负责管理服务的注册和查询。
   • 提供了 registerService 方法用于注册服务，lookupService 方法用于查询服务地址，getAllServicePorts 方法用于获取所有注册的服务端口。

3. 客户端：

   • Client 类创建一个线程池，启动10个客户端线程，每个线程随机选择一个服务端口，向服务端发送消息 “A”，并打印服务端的响应。