自定义实现一个RPC框架（三）

Bertha 。 2023-02-13 11:51 29阅读 0赞

### 文章目录 ###

*  自定义实现一个RPC框架（三）：
 *   *   *  地址：https://gitee.com/liyunfengfengfeng/rpc
         *  注册中心
         *   *  文件实现
             *  Zookeeper实现注册中心

# 自定义实现一个RPC框架（三）： #

### 地址：https://gitee.com/liyunfengfengfeng/rpc ###

### 注册中心 ###

两种方式实现

#### 文件实现 ####

package com.luban.register;
    
    import com.luban.framework.URL;
    
    import java.io.*;
    import java.util.*;
    
    /** * 不同的进程同一个Map 不能共享数据 * 所以存储到redis中 */
    public class RemoteMapRegister { 
    
        private static Map<String, List<URL>> REGISTER = new HashMap<>();
    
    
        public static void regist(String interfaceName, URL url){ 
    
            List<URL> list = REGISTER.get(interfaceName);
            if (list == null) { 
                list = new ArrayList<>();
    
            }
            list.add(url);
    
            REGISTER.put(interfaceName, list);
    
            saveFile();
        }
    
        public static List<URL> get(String interfaceName) { 
            REGISTER = getFile();
    
            List<URL> list = REGISTER.get(interfaceName);
            return list;
        }
    
    
        private static void saveFile() { 
            try { 
                FileOutputStream fileOutputStream = new FileOutputStream("/temp.txt");
                ObjectOutputStream objectOutputStream = new ObjectOutputStream(fileOutputStream);
                objectOutputStream.writeObject(REGISTER);
            } catch (IOException e) { 
                e.printStackTrace();
            }
        }
    
        private static Map<String, List<URL>> getFile() { 
            try { 
                FileInputStream fileInputStream = new FileInputStream("/temp.txt");
                ObjectInputStream objectInputStream = new ObjectInputStream(fileInputStream);
                return (Map<String, List<URL>>) objectInputStream.readObject();
            } catch (IOException | ClassNotFoundException e) { 
                e.printStackTrace();
            }
            return null;
        }
    }

#### Zookeeper实现注册中心 ####

package com.luban.register;
    
    import com.alibaba.fastjson.JSONObject;
    import com.luban.framework.URL;
    import org.apache.curator.framework.CuratorFramework;
    import org.apache.curator.framework.CuratorFrameworkFactory;
    import org.apache.curator.retry.RetryNTimes;
    import org.apache.zookeeper.CreateMode;
    
    import java.util.ArrayList;
    import java.util.HashMap;
    import java.util.List;
    import java.util.Map;
    
    /** * *************书山有路勤为径*************** * 鲁班学院 * 往期资料加木兰老师 QQ: 2746251334 * VIP课程加安其拉老师 QQ: 3164703201 * 讲师：周瑜老师 * *************学海无涯苦作舟*************** */
    public class ZookeeperRegister { 
    
        static CuratorFramework client;
    
        static { 
            client = CuratorFrameworkFactory
                    .newClient("localhost:2181", new RetryNTimes(3, 1000));
            client.start();
    
        }
    
        private static Map<String, List<URL>> REGISTER = new HashMap<>();
    
        public static void regist(String interfaceName, URL url) { 
            try { 
                /** * zookeeper节点类型 * PERSISTENT(0, false, false), * PERSISTENT_SEQUENTIAL(2, false, true), * EPHEMERAL(1, true, false), * EPHEMERAL_SEQUENTIAL(3, true, true); * * creatingParentsIfNeeded() //递归创建父目录 */
                String result = client.create().creatingParentsIfNeeded().withMode(CreateMode.EPHEMERAL).forPath(String.format("/dubbo/service/%s/%s", interfaceName, JSONObject.toJSONString(url)), null);
                System.out.println(result);
            } catch (Exception e) { 
                e.printStackTrace();
            }
    
        }
    
        public static List<URL> get(String interfaceName) { 
            List<URL> urlList = new ArrayList<>();
    
            try { 
                List<String> result = client.getChildren().forPath(String.format("/dubbo/service/%s", interfaceName));
                for (String urlstr : result) { 
                    urlList.add(JSONObject.parseObject(urlstr, URL.class));
                }
    
                REGISTER.put(interfaceName, urlList);
            } catch (Exception e) { 
                e.printStackTrace();
            }
    
            return urlList;
        }
    }