Dubbo源码解析（二十五）远程调用——hessian协议

时间  2019-12-19
标签 dubbo 源码 解析 二十五 远程 调用 hessian 协议 栏目 Dubbo 繁體版
原文   原文链接

远程调用——hessian协议

目标：介绍远程调用中跟hessian协议相关的设计和实现，介绍dubbo-rpc-hessian的源码。html

前言

本文讲解可能是dubbo集成的第二种协议，hessian协议，Hessian 是 Caucho 开源的一个 RPC 框架，其通信效率高于 WebService 和 Java 自带的序列化。dubbo集成hessian所提供的hessian协议相关介绍能够参考官方文档，我就再也不赘述。java

文档地址：dubbo.apache.org/zh-cn/docs/…git

源码分析

（一）DubboHessianURLConnectionFactory

该类继承了HessianURLConnectionFactory类，是dubbo，用于建立与服务器的链接的内部工厂，重写了父类中open方法。github

public class DubboHessianURLConnectionFactory extends HessianURLConnectionFactory {

    /** * 打开与HTTP服务器的新链接或循环链接 * @param url * @return * @throws IOException */
    @Override
    public HessianConnection open(URL url) throws IOException {
        // 得到一个链接
        HessianConnection connection = super.open(url);
        // 得到上下文
        RpcContext context = RpcContext.getContext();
        for (String key : context.getAttachments().keySet()) {
            // 在http协议头里面加入dubbo中附加值，key为 header+key value为附加值的value
            connection.addHeader(Constants.DEFAULT_EXCHANGER + key, context.getAttachment(key));
        }

        return connection;
    }
}
复制代码

在hessian上加入dubbo本身所须要的附加值，放到协议头里面进行发送。apache

（二）HttpClientConnection

该类是基于HttpClient封装来实现HessianConnection接口，其中逻辑比较简单。服务器

public class HttpClientConnection implements HessianConnection {

    /** * http客户端对象 */
    private final HttpClient httpClient;

    /** * 字节输出流 */
    private final ByteArrayOutputStream output;

    /** * http post请求对象 */
    private final HttpPost request;

    /** * http 响应对象 */
    private volatile HttpResponse response;

    public HttpClientConnection(HttpClient httpClient, URL url) {
        this.httpClient = httpClient;
        this.output = new ByteArrayOutputStream();
        this.request = new HttpPost(url.toString());
    }

    /** * 增长协议头 * @param key * @param value */
    @Override
    public void addHeader(String key, String value) {
        request.addHeader(new BasicHeader(key, value));
    }

    @Override
    public OutputStream getOutputStream() throws IOException {
        return output;
    }

    /** * 发送请求 * @throws IOException */
    @Override
    public void sendRequest() throws IOException {
        request.setEntity(new ByteArrayEntity(output.toByteArray()));
        this.response = httpClient.execute(request);
    }

    /** * 得到请求后的状态码 * @return */
    @Override
    public int getStatusCode() {
        return response == null || response.getStatusLine() == null ? 0 : response.getStatusLine().getStatusCode();
    }

    @Override
    public String getStatusMessage() {
        return response == null || response.getStatusLine() == null ? null : response.getStatusLine().getReasonPhrase();
    }

    @Override
    public String getContentEncoding() {
        return (response == null || response.getEntity() == null || response.getEntity().getContentEncoding() == null) ? null : response.getEntity().getContentEncoding().getValue();
    }

    @Override
    public InputStream getInputStream() throws IOException {
        return response == null || response.getEntity() == null ? null : response.getEntity().getContent();
    }

    @Override
    public void close() throws IOException {
        HttpPost request = this.request;
        if (request != null) {
            request.abort();
        }
    }

    @Override
    public void destroy() throws IOException {
    }
复制代码

（三）HttpClientConnectionFactory

该类实现了HessianConnectionFactory接口，是建立HttpClientConnection的工厂类。该类的实现跟DubboHessianURLConnectionFactory类相似，可是DubboHessianURLConnectionFactory是标准的Hessian接口调用会采用的工厂类，而HttpClientConnectionFactory是Dubbo 的 Hessian 协议调用。固然Dubbo 的 Hessian 协议也是基于http的。网络

public class HttpClientConnectionFactory implements HessianConnectionFactory {

    /** * httpClient对象 */
    private final HttpClient httpClient = new DefaultHttpClient();

    @Override
    public void setHessianProxyFactory(HessianProxyFactory factory) {
        // 设置链接超时时间
        HttpConnectionParams.setConnectionTimeout(httpClient.getParams(), (int) factory.getConnectTimeout());
        // 设置读取数据时阻塞链路的超时时间
        HttpConnectionParams.setSoTimeout(httpClient.getParams(), (int) factory.getReadTimeout());
    }

    
    @Override
    public HessianConnection open(URL url) throws IOException {
        // 建立一个HttpClientConnection实例
        HttpClientConnection httpClientConnection = new HttpClientConnection(httpClient, url);
        // 得到上下文，用来得到附加值
        RpcContext context = RpcContext.getContext();
        // 遍历附加值，放入到协议头里面
        for (String key : context.getAttachments().keySet()) {
            httpClientConnection.addHeader(Constants.DEFAULT_EXCHANGER + key, context.getAttachment(key));
        }
        return httpClientConnection;
    }

}
复制代码

实现了两个方法，第一个方法是给http链接设置两个参数配置，第二个方法是建立一个链接。框架

（四）HessianProtocol

该类继承了AbstractProxyProtocol类，是hessian协议的实现类。其中实现类基于hessian协议的服务引用、服务暴露等方法。ide

1.属性

/** * http服务器集合 * key为ip：port */
private final Map<String, HttpServer> serverMap = new ConcurrentHashMap<String, HttpServer>();

/** * HessianSkeleto&emsp;集合 * key为服务名 */
private final Map<String, HessianSkeleton> skeletonMap = new ConcurrentHashMap<String, HessianSkeleton>();

/** * HttpBinder对象，默认是jetty实现 */
private HttpBinder httpBinder;
复制代码

2.doExport

@Override
protected <T> Runnable doExport(T impl, Class<T> type, URL url) throws RpcException {
    // 得到ip地址
    String addr = getAddr(url);
    // 得到http服务器对象
    HttpServer server = serverMap.get(addr);
    // 若是为空，则从新建立一个server，而后放入集合
    if (server == null) {
        server = httpBinder.bind(url, new HessianHandler());
        serverMap.put(addr, server);
    }
    // 得到服务path
    final String path = url.getAbsolutePath();
    // 建立Hessian服务端对象
    final HessianSkeleton skeleton = new HessianSkeleton(impl, type);
    // 加入集合
    skeletonMap.put(path, skeleton);

    // 得到通用的path
    final String genericPath = path + "/" + Constants.GENERIC_KEY;
    // 加入集合
    skeletonMap.put(genericPath, new HessianSkeleton(impl, GenericService.class));

    // 返回一个线程
    return new Runnable() {
        @Override
        public void run() {
            skeletonMap.remove(path);
            skeletonMap.remove(genericPath);
        }
    };
}
复制代码

该方法是服务暴露的主要逻辑实现。源码分析

3.doRefer

@Override
@SuppressWarnings("unchecked")
protected <T> T doRefer(Class<T> serviceType, URL url) throws RpcException {
    // 得到泛化的参数
    String generic = url.getParameter(Constants.GENERIC_KEY);
    // 是不是泛化调用
    boolean isGeneric = ProtocolUtils.isGeneric(generic) || serviceType.equals(GenericService.class);
    // 若是是泛化调用。则设置泛化的path和附加值
    if (isGeneric) {
        RpcContext.getContext().setAttachment(Constants.GENERIC_KEY, generic);
        url = url.setPath(url.getPath() + "/" + Constants.GENERIC_KEY);
    }

    // 建立代理工厂
    HessianProxyFactory hessianProxyFactory = new HessianProxyFactory();
    // 是不是Hessian2的请求 默认为否
    boolean isHessian2Request = url.getParameter(Constants.HESSIAN2_REQUEST_KEY, Constants.DEFAULT_HESSIAN2_REQUEST);
    // 设置是否应使用Hessian协议的版本2来解析请求
    hessianProxyFactory.setHessian2Request(isHessian2Request);
    // 是否应为远程调用启用重载方法，默认为否
    boolean isOverloadEnabled = url.getParameter(Constants.HESSIAN_OVERLOAD_METHOD_KEY, Constants.DEFAULT_HESSIAN_OVERLOAD_METHOD);
    // 设置是否应为远程调用启用重载方法。
    hessianProxyFactory.setOverloadEnabled(isOverloadEnabled);
    // 得到client实现方式，默认为jdk
    String client = url.getParameter(Constants.CLIENT_KEY, Constants.DEFAULT_HTTP_CLIENT);
    if ("httpclient".equals(client)) {
        // 用http来建立
        hessianProxyFactory.setConnectionFactory(new HttpClientConnectionFactory());
    } else if (client != null && client.length() > 0 && !Constants.DEFAULT_HTTP_CLIENT.equals(client)) {
        // 抛出不支持的协议异常
        throw new IllegalStateException("Unsupported http protocol client=\"" + client + "\"!");
    } else {
        // 建立一个HessianConnectionFactory对象
        HessianConnectionFactory factory = new DubboHessianURLConnectionFactory();
        // 设置代理工厂
        factory.setHessianProxyFactory(hessianProxyFactory);
        // 设置工厂
        hessianProxyFactory.setConnectionFactory(factory);
    }
    // 得到超时时间
    int timeout = url.getParameter(Constants.TIMEOUT_KEY, Constants.DEFAULT_TIMEOUT);
    // 设置超时时间
    hessianProxyFactory.setConnectTimeout(timeout);
    hessianProxyFactory.setReadTimeout(timeout);
    // 建立代理
    return (T) hessianProxyFactory.create(serviceType, url.setProtocol("http").toJavaURL(), Thread.currentThread().getContextClassLoader());
}
复制代码

该方法是服务引用的主要逻辑实现，根据客户端配置，来选择标准 Hessian 接口调用仍是Dubbo 的 Hessian 协议调用。

4.getErrorCode

@Override
protected int getErrorCode(Throwable e) {
    // 若是属于HessianConnectionException异常
    if (e instanceof HessianConnectionException) {
        if (e.getCause() != null) {
            Class<?> cls = e.getCause().getClass();
            // 若是属于超时异常，则返回超时异常
            if (SocketTimeoutException.class.equals(cls)) {
                return RpcException.TIMEOUT_EXCEPTION;
            }
        }
        // 不然返回网络异常
        return RpcException.NETWORK_EXCEPTION;
    } else if (e instanceof HessianMethodSerializationException) {
        // 序列化异常
        return RpcException.SERIALIZATION_EXCEPTION;
    }
    return super.getErrorCode(e);
}
复制代码

该方法是针对异常的处理。

5.HessianHandler

private class HessianHandler implements HttpHandler {

    @Override
    public void handle(HttpServletRequest request, HttpServletResponse response) throws IOException, ServletException {
        // 得到请求的uri
        String uri = request.getRequestURI();
        // 得到对应的HessianSkeleton对象
        HessianSkeleton skeleton = skeletonMap.get(uri);
        // 若是若是不是post方法
        if (!request.getMethod().equalsIgnoreCase("POST")) {
            // 返回状态设置为500
            response.setStatus(500);
        } else {
            // 设置远程地址
            RpcContext.getContext().setRemoteAddress(request.getRemoteAddr(), request.getRemotePort());

            // 得到请求头内容
            Enumeration<String> enumeration = request.getHeaderNames();
            // 遍历请求头内容
            while (enumeration.hasMoreElements()) {
                String key = enumeration.nextElement();
                // 若是key开头是deader，则把附加值取出来放入上下文
                if (key.startsWith(Constants.DEFAULT_EXCHANGER)) {
                    RpcContext.getContext().setAttachment(key.substring(Constants.DEFAULT_EXCHANGER.length()),
                            request.getHeader(key));
                }
            }

            try {
                // 执行下一个
                skeleton.invoke(request.getInputStream(), response.getOutputStream());
            } catch (Throwable e) {
                throw new ServletException(e);
            }
        }
    }

}
复制代码

该内部类是Hessian的处理器，用来处理请求中的协议头内容。

后记

该部分相关的源码解析地址：github.com/CrazyHZM/in…

该文章讲解了远程调用中关于hessian协议的部分，内容比较简单，能够参考着官方文档了解一下。接下来我将开始对rpc模块关于Http协议部分进行讲解。

相关文章
相关标签/搜索
每日一句
    每一个你不满意的现在，都有一个你没有努力的曾经。
本站公众号
   欢迎关注本站公众号,获取更多信息
联系我们 最近搜索 最新文章 沪ICP备13005482号-10 MyBatis教程 SQL 教程 MySQL教程 Java 教程 Thymeleaf 教程 Hibernate教程 Spring教程 Redis教程