Protocol

本文讲解如何通过扩展 org.apache.dubbo.rpc.Protocol SPI，提供自定义的 RPC 协议实现。

在 通信协议 一章中，我们了解了 Dubbo 内置的几个核心 RPC 协议 dubbo、rest、和tri 以及它们的使用方式。本文讲解如何通过扩展 org.apache.dubbo.rpc.Protocol SPI，提供自定义的 RPC 协议实现。

自定义一套私有协议有两种方式，第一种是对原有的协议进行包装，添加一些特定的业务逻辑。另外一种是完全自定义一套协议。前者实现简单，在dubbo中也是有广泛的使用，比如：ProtocolFilterWrapper, QosProtocolWrapper, ProtocolListenerWrapper等。后者实现相对复杂，但却具有最大的灵活性，比如 Dubbo 框架内置的协议 dubbo、triple 协议都可以算作这种实现方式。

本示例的完整源码请参见 dubbo-samples-extensibility。除了本示例之外，Dubbo 核心仓库 apache/dubbo 以及扩展库 apache/dubbo-spi-extensions 中的众多 Protocol 实现，都可以作为扩展参考实现：

1. # Dubbo对外支持的常用协议
2. dubbo=org.apache.dubbo.rpc.protocol.dubbo.DubboProtocol
3. tri=org.apache.dubbo.rpc.protocol.tri.TripleProtocol

任务详情

基于现有的dubbo协议来实现自定义协议edubbo。

实现方式

通过对dubbo协议进行包装来实现edubbo协议。

代码结构

Common

1. src
2. |-main
3. |-java
4. |-org
5. |-apache
6. |-dubbo
7. |-samples
8. |-extensibility
9. |-protocol
10. |-common
11. |-EnhancedProtocol.java (实现Protocol接口)

Provider

1. src
2. |-main
3. |-java
4. |-org
5. |-apache
6. |-dubbo
7. |-samples
8. |-extensibility
9. |-protocol
10. |-provider
11. |-ExtensibilityProtocolProviderApplication.java
12. |-ExtensibilityProtocolServiceImpl.java
13. |-resources
14. |-META-INF
15. |-application.properties (Dubbo Provider配置文件)
16. |-dubbo
17. |-org.apache.dubbo.rpc.Protocol (纯文本文件)

Consumer

1. src
2. |-main
3. |-java
4. |-org
5. |-apache
6. |-dubbo
7. |-samples
8. |-extensibility
9. |-protocol
10. |-consumer
11. |-ExtensibilityProtocolConsumerApplication.java
12. |-ExtensibilityProtocolConsumerTask.java
13. |-resources
14. |-META-INF
15. |-application.properties (Dubbo Consumer配置文件)
16. |-dubbo
17. |-org.apache.dubbo.rpc.Protocol (纯文本文件)

代码详情

1. package org.apache.dubbo.samples.extensibility.protocol.common;

3. import org.apache.dubbo.common.URL;
4. import org.apache.dubbo.rpc.Protocol;
5. import org.apache.dubbo.rpc.Invoker;
6. import org.apache.dubbo.rpc.Exporter;
7. import org.apache.dubbo.rpc.ProtocolServer;
8. import org.apache.dubbo.rpc.RpcException;
9. import org.apache.dubbo.rpc.model.FrameworkModel;
10. import org.apache.dubbo.rpc.protocol.dubbo.DubboProtocol;

12. import java.util.List;

14. public class EnhancedProtocol implements Protocol {

16. public EnhancedProtocol(FrameworkModel frameworkModel) {
17. this.protocol = new DubboProtocol(frameworkModel);
18. }

20. private final Protocol protocol;

22. @Override
23. public int getDefaultPort() {
24. return this.protocol.getDefaultPort();
25. }

27. @Override
28. public <T> Exporter<T> export(Invoker<T> invoker) throws RpcException {
29. // do something
30. return this.protocol.export(invoker);
31. }

33. @Override
34. public <T> Invoker<T> refer(Class<T> type, URL url) throws RpcException {
35. // do something
36. return this.protocol.refer(type, url);
37. }

39. @Override
40. public void destroy() {
41. this.protocol.destroy();
42. }

44. @Override
45. public List<ProtocolServer> getServers() {
46. return protocol.getServers();
47. }
48. }

SPI配置

Provider

在resources/META-INF/dubbo/org.apache.dubbo.rpc.Protocol文件中添加如下配置：

1. edubbo=org.apache.dubbo.samples.extensibility.protocol.common.EnhancedProtocol

Consumer

在resources/META-INF/dubbo/org.apache.dubbo.rpc.Protocol文件中添加如下配置：

1. edubbo=org.apache.dubbo.samples.extensibility.protocol.common.EnhancedProtocol

配置文件

Provider

在resources/application.properties文件中添加如下配置：

1. # 自定义协议
2. dubbo.provider.protocol=edubbo

Consumer

在resources/application.properties文件中添加如下配置：

1. # 自定义协议
2. dubbo.consumer.protocol=edubbo

运行结果

以使用本地IDE的方式来运行任务，结果如下：

注册协议

输出结果

最后修改 September 13, 2024: Refactor website structure (#2860) (1a4b998f54b)