dubbo实现动态负载均衡

引言

本文续写上一篇博客 dubbo框架，这里主要简单演示dubbo如何实现动态负载均衡的。

模块构建

新建子模块 springboot_dubbo_load

该模块 pom文件中引入ZK客户端依赖：

ZkServerSocket Socket服务端：

package com.baba.wlb.server;

import org.I0Itec.zkclient.ZkClient;

import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;

/**
 * @Author wulongbo
 * @Date 2021/1/5 9:56
 * @Version 1.0
 */

// ServerSocket服务端
public class ZkServerSocket implements Runnable {

    private static int port = 18081;

    private String parentService = "/service";


    /**
     * 服务器端：

     * 1.服务器启动的时候，会将当前服务器信息注册到注册中心。首先先创建一个父节点为service,在父节点下面再创建一个子节点，
     * 每个子节点都存放当前服务接口地址。
     * ## 子节点结构
     * /service 持久节点
     * ##/8080 value 39.102.56.91:18080 临时节点
     * ##/8081 value 39.102.56.91:18081 临时节点
     *
     * @param args
     */
    private ZkClient zkClient = new ZkClient("39.102.56.91:2181");

    public static void main(String[] args) {
        ZkServerSocket server = new ZkServerSocket(port);
        Thread thread = new Thread(server);
        thread.start();
    }

    public ZkServerSocket(int port) {
        this.port = port;
    }

    public void regServer() {
        // 1.先创建父节点server 为持久节点
        if (!zkClient.exists(parentService)) {
//            // 删除旧节点
//            zkClient.delete(parentService);
            // 2.创建父节点
            // 持久节点
            zkClient.createPersistent(parentService);
        }


        String serverKey = parentService + "/server_" + port;
        if (!zkClient.exists(serverKey)) {
//            // 删除旧节点
//            zkClient.delete(serverKey);
            // 3.创建子节点value为服务接口地址
            // 临时节点
            zkClient.createEphemeral(serverKey, "127.0.0.1:" + port);
        }

    }

    @Override
    public void run() {
        ServerSocket serverSocket = null;
        try {
            serverSocket = new ServerSocket(port);
            System.out.println("Server start port:" + port);
            regServer();
            Socket socket = null;
            while (true) {
                socket = serverSocket.accept();
                new Thread(new ServerHandler(socket)).start();
            }
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            try {
                if (serverSocket != null) {
                    serverSocket.close();
                }
            } catch (Exception e2) {

            }
        }
    }
}

ZkServerSocket2 Socket服务端用于模拟集群：

package com.baba.wlb.server;

import org.I0Itec.zkclient.ZkClient;

import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;

/**
 * @Author wulongbo
 * @Date 2021/1/5 9:56
 * @Version 1.0
 */

// ServerSocket服务端
public class ZkServerSocket2 implements Runnable {

    private static int port = 18080;

    private String parentService = "/service";


    /**
     * 服务器端：

     * 1.服务器启动的时候，会将当前服务器信息注册到注册中心。首先先创建一个父节点为service,在父节点下面再创建一个子节点，
     * 每个子节点都存放当前服务接口地址。
     * ## 子节点结构
     * /service 持久节点
     * ##/8080 value 39.102.56.91:18080 临时节点
     * ##/8081 value 39.102.56.91:18081 临时节点
     *
     * @param args
     */
    private ZkClient zkClient = new ZkClient("39.102.56.91:2181");

    public static void main(String[] args) {
        ZkServerSocket2 server = new ZkServerSocket2(port);
        Thread thread = new Thread(server);
        thread.start();
    }

    public ZkServerSocket2(int port) {
        this.port = port;
    }

    public void regServer() {
        // 1.先创建父节点server 为持久节点
        if (!zkClient.exists(parentService)) {
//            // 删除旧节点
//            zkClient.delete(parentService);
            // 2.创建父节点
            // 持久节点
            zkClient.createPersistent(parentService);
        }


        String serverKey = parentService + "/server_" + port;
        if (!zkClient.exists(serverKey)) {
//            // 删除旧节点
//            zkClient.delete(serverKey);
            // 3.创建子节点value为服务接口地址
            // 临时节点
            zkClient.createEphemeral(serverKey, "127.0.0.1:" + port);
        }

    }

    @Override
    public void run() {
        ServerSocket serverSocket = null;
        try {
            serverSocket = new ServerSocket(port);
            System.out.println("Server start port:" + port);
            regServer();
            Socket socket = null;
            while (true) {
                socket = serverSocket.accept();
                new Thread(new ServerHandler(socket)).start();
            }
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            try {
                if (serverSocket != null) {
                    serverSocket.close();
                }
            } catch (Exception e2) {

            }
        }
    }
}

ZkServerClient Socket客户端：

package com.baba.wlb.client;

import org.I0Itec.zkclient.IZkChildListener;
import org.I0Itec.zkclient.ZkClient;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.net.Socket;
import java.util.ArrayList;
import java.util.List;

/**
 * @Author wulongbo
 * @Date 2021/1/5 10:24
 * @Version 1.0
 */
public class ZkServerClient {

    // 存放服务列表信息
    public static List listServer = new ArrayList();

    // 客户端：读取service节点，获取下面的子节点value值，本地实现远程调用。
    private static String parentService = "/service";

    private static ZkClient zkClient = new ZkClient("39.102.56.91:2181");

    public static void main(String[] args) {
        initServer();
        ZkServerClient client = new ZkServerClient();
        BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(System.in));
        while (true) {
            String name;
            try {
                name = bufferedReader.readLine();
                if ("exit".equals(name)) {
                    System.exit(0);
                }
                client.send(name);
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }

    private void send(String name) {
        String server = ZkServerClient.getServer();
        String[] cfg = server.split(":");
        Socket socket = null;
        BufferedReader in = null;
        PrintWriter out = null;
        try {
            socket = new Socket(cfg[0], Integer.parseInt(cfg[1]));
            in = new BufferedReader(new InputStreamReader(socket.getInputStream()));
            out = new PrintWriter(socket.getOutputStream(), true);
            out.println(name);
            while (true) {
                String resp = in.readLine();
                if (resp == null)
                    break;
                else if (resp.length() > 0) {
                    System.out.println("Receive:" + resp);
                    break;
                }
            }
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            if (out != null) {
                out.close();
            }
            if (in != null) {
                try {
                    in.close();
                } catch (IOException e1) {
                    e1.printStackTrace();
                }

            }
            if (socket != null) {
                try {
                    socket.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
    }

    // 注册所有server
    private static void initServer() {
//        listServer.clear();
//        listServer.add("39.102.56.91:18080");

        // 从zookeeper上获取服务列表信息
        List children = zkClient.getChildren(parentService);
        getChildData(zkClient, children);

        // 使用Zk时间通知获取最新服务列表信息
        zkClient.subscribeChildChanges(parentService, new IZkChildListener() {
            @Override
            public void handleChildChange(String parentPath, List currentChildren) throws Exception {
                System.out.println("注册中心服务里列表信息发生变化...");
                getChildData(zkClient, currentChildren);
            }
        });
    }

    public static void getChildData(ZkClient zkClient, List children) {
        listServer.clear();
        children.stream().forEach(
                node -> {
                    String serverAddress = zkClient.readData(parentService + "/" + node);
                    listServer.add(serverAddress);
                }
        );
        System.out.println("服务接口地址：" + listServer.toString());
    }

    // 请求总数
    private static int reqCount = 1;

    // 获取当前server信息
    public static String getServer() {
//        return listServer.get(0);
        int index = reqCount % listServer.size();
        String address = listServer.get(index);
        System.out.println("客户端请求服务器端：" + address);
        reqCount++;
        return address;
    }


}

启动Socket服务端

分别启动 ZkServerSocket，ZkServerSocket2
我们可以看到zookeeper上注册了一个持久节点 server和两个临时节点 server_18080、server_18081，他们对应的value值分别是：127.0.0.1:18080、127.0.0.1:18081

启动Socket客户端

可以发现服务接口地址：[127.0.0.1:18080, 127.0.0.1:18081]

我们在控制台输入内容： 这是第一次传输，通过取模算法，负载到18081这台服务上

在端口为：18081 的Socket服务端可以收到消息

我们在控制台输入内容： 这是第二次传输

在端口为：18080 的Socket服务端可以收到消息

现在强制停掉 18081 的服务，只剩下 18080 这台，需等待约15s 由于节点发生变化，通过事件通知的方式给到订阅方，订阅方通过事件监听的方式来动态获取服务端节点。

我们在控制台输入内容： 这是第三次传输、这是第四次传输，都只会转发给 18080

再次启动18081，复活服务，又可实现动态负载。

总结

本文演示的是dubbo如何实现动态负载均衡的，只要我们对注册中心zookeeper 有一定的了解，便很容易明白！

dubbo实现动态负载均衡

引言

模块构建

启动Socket服务端

启动Socket客户端

总结

你可能感兴趣的:(java,springboot,后端)