RPC框架Pigeon简析(二)-- 服务注册与发布

首先服务提供方需要定义接口,

EchoService.java
package com.xx.pigeon.demo;
    public interface EchoService {
        public String echo(String name);
    }

同时,服务提供方同时需要实现该接口,然后服务提供方就可以注册服务,传统的spring注册方法如下:

<bean class="com.dianping.dpsf.spring.ServiceRegistry"
    init-method="init">
      <property name="services">
        <map>
            <entry key="http://service.xx.com/demoService/echoService_1.0.0"
    value-ref="echoServiceImpl" />
        </map>
      </property>
    </bean>

OK,然后我们启动spring上下文的时候,spring会帮助我们初始化这个bean然后调用init方法,这个方法调用了

ServiceFactory.addServices(providerConfigList);

ServiceFactory这个类是个工厂类,它会帮我们初始化服务。剩下就是RPC框架帮我们做的一些事情了。其核心的方法是:

ServicePublisher.addService(providerConfig);
ServerConfig serverConfig = ProviderBootStrap.startup(providerConfig);
ServicePublisher.publishService(providerConfig, false);

其中ServicePublisher.addService(providerConfig) 这个方法主要处理了serviceName,methodName,parameters,同时对服务进行缓存。

ServerConfig serverConfig = ProviderBootStrap.startup(providerConfig)则是启动了“容器”。pigeon目前支持两层,http和tcp。http方面,pigeon会启动内置的jetty,提供了一些服务控制的方法,比如publish、unpublish、online、offline等,同时支持用http来进行RPC。tcp方面,pigeon底层依赖netty进行,所谓启动容器,这里是启动对特定端口的监听。另外,这个方法还会对请求处理器进行注册,这个后面会聊到。

ServicePublisher.publishService(providerConfig, false);该方法为核心方法,主要�用于服务发布。

   if (existingService) {
            boolean autoPublishEnable = ConfigManagerLoader.getConfigManager().getBooleanValue(
                    Constants.KEY_AUTOPUBLISH_ENABLE, true);
            if (autoPublishEnable || forcePublish) {
                List<Server> servers = ProviderBootStrap.getServers(providerConfig);
                int registerCount = 0;
                for (Server server : servers) {
                    publishService(url, server.getRegistryUrl(url), server.getPort(), providerConfig.getServerConfig()
                            .getGroup(), providerConfig.isSupported());//注册服务到zk
                    registerCount++;
                }
                if (registerCount > 0) {
                    boolean isHeartbeatEnable = configManager.getBooleanValue(Constants.KEY_HEARTBEAT_ENABLE,
                            DEFAULT_HEARTBEAT_ENABLE);
                    if (isHeartbeatEnable) {
                        HeartBeatListener.registerHeartBeat(providerConfig);//注册心跳上报
                    }

                    boolean isNotify = configManager
                            .getBooleanValue(Constants.KEY_NOTIFY_ENABLE, DEFAULT_NOTIFY_ENABLE);
                    if (isNotify && serviceChangeListener != null) {
                        serviceChangeListener.notifyServicePublished(providerConfig);//向pigeon的管理中心上报发布情况
                    }

                    boolean autoRegisterEnable = ConfigManagerLoader.getConfigManager().getBooleanValue(
                            Constants.KEY_AUTOREGISTER_ENABLE, true);
                    if (autoRegisterEnable) {
                        ServiceOnlineTask.start();//上线服务
                    } else {
                        logger.info("auto register is disabled");
                    }

                    providerConfig.setPublished(true);
                }
            } else {
                logger.info("auto publish is disabled");
            }
        }

首先会调用方法publishService(url, server.getRegistryUrl(url), server.getPort(), providerConfig.getServerConfig()
.getGroup(), providerConfig.isSupported()),该方法会先计算weight(权重),一般会初始化为0,然后再进行服务注册,

RegistryManager.getInstance().registerService(registryUrl, group, serverAddress, weight);//注册服务地址和服务名
RegistryManager.getInstance().registerSupportNewProtocol(serverAddress, registryUrl, support);//注册服务协议

        if (weight >= 0) {
            if (!serverWeightCache.containsKey(serverAddress)) {
                RegistryManager.getInstance().setServerApp(serverAddress, configManager.getAppName());//注册应用名,一个实例注册一次
                RegistryManager.getInstance().setServerVersion(serverAddress, VersionUtils.VERSION);//注册服务版本,一个实例注册一次
            }
            serverWeightCache.put(serverAddress, weight);
        }

这几个registry其实大同小异,都是将特定的值写入zk,以注册服务为例,

void registerPersistentNode(String serviceName, String group, String serviceAddress, int weight)
            throws RegistryException {
        String weightPath = Utils.getWeightPath(serviceAddress);
        String servicePath = Utils.getServicePath(serviceName, group);
        try {
            if (client.exists(servicePath, false)) {
                Stat stat = new Stat();
                String addressValue = client.get(servicePath, stat);
                String[] addressArray = addressValue.split(",");
                List<String> addressList = new ArrayList<String>();
                for (String addr : addressArray) {
                    addr = addr.trim();
                    if (addr.length() > 0 && !addressList.contains(addr)) {
                        addressList.add(addr.trim());
                    }
                }
                if (!addressList.contains(serviceAddress)) {
                    addressList.add(serviceAddress);
                    Collections.sort(addressList);
                    client.set(servicePath, StringUtils.join(addressList.iterator(), ","), stat.getVersion());
                }
            } else {
                client.create(servicePath, serviceAddress);
            }
            if (weight >= 0) {
                client.set(weightPath, "" + weight);
            }
            if (logger.isInfoEnabled()) {
                logger.info("registered service to persistent node: " + servicePath);
            }
        } catch (Throwable e) {
            if(e instanceof BadVersionException || e instanceof NodeExistsException) {
                try {
                    Thread.sleep(500);
                } catch (InterruptedException ie) {
                    //ignore
                }
                registerPersistentNode(serviceName, group, serviceAddress, weight);
            } else {
                logger.error("failed to register service to " + servicePath, e);
                throw new RegistryException(e);
            }

        }
    }

该方法就是将服务地址和服务名写入zk。其中这里有一个Group的概念,其实就是将同一个服务不同的实例可以分到不同的swimlane里面。
当然RegistryManager可以有不同的实现,你也可以选择将服务信息写到别的地方,比如db。默认实现是写入zk,这也是大型分布式系统比较常用的方法。
然后是注册心跳上报,这个后面会具体谈到。接着,向pigeon的管理中心上报发布情况,这个其实在整个服务中,不是必须的。
最后就是服务上线,从前面的代码我们可以知道服务的权重一般会被初始化为0,而客户端在调用服务的时候会利用权重来负载均衡,也就是说权重是0的服务基本不会被调用到,那么就可以理解为通过修改权重来进行上线或者下线服务。

public void run() {
    logger.info("Service online task start");
    try {
        Thread.sleep(delay);
        if (!isStop) {
            ServiceFactory.online();
        }
    } catch (Throwable e) {
        logger.error(e.getMessage(), e);
    }
}

启动是利用另外一个线程进行的,可以设置一个时延。

public static void online() throws RegistryException {
    logger.info("online");
    ServicePublisher.setServerWeight(Constants.WEIGHT_DEFAULT);
    /*ServiceProviderFactory.notifyServiceOnline();*/
}

修改服务权重,用来上线服务。

至此,就完成了全部的服务注册和发布的过程。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 194,761评论 5 460
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 81,953评论 2 371
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 141,998评论 0 320
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 52,248评论 1 263
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 61,130评论 4 356
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 46,145评论 1 272
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 36,550评论 3 381
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 35,236评论 0 253
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 39,510评论 1 291
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 34,601评论 2 310
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 36,376评论 1 326
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,247评论 3 313
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 37,613评论 3 299
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 28,911评论 0 17
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,191评论 1 250
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 41,532评论 2 342
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 40,739评论 2 335

推荐阅读更多精彩内容

  • Spring Cloud为开发人员提供了快速构建分布式系统中一些常见模式的工具(例如配置管理,服务发现,断路器,智...
    卡卡罗2017阅读 134,497评论 18 139
  • Spring Boot 参考指南 介绍 转载自:https://www.gitbook.com/book/qbgb...
    毛宇鹏阅读 46,679评论 6 342
  • 从三月份找实习到现在,面了一些公司,挂了不少,但最终还是拿到小米、百度、阿里、京东、新浪、CVTE、乐视家的研发岗...
    时芥蓝阅读 42,139评论 11 349
  • application的配置属性。 这些属性是否生效取决于对应的组件是否声明为Spring应用程序上下文里的Bea...
    新签名阅读 5,342评论 1 27
  • 这些属性是否生效取决于对应的组件是否声明为 Spring 应用程序上下文里的 Bean(基本是自动配置的),为一个...
    发光的鱼阅读 1,411评论 0 14