从零开始手写 dubbo rpc 框架-14-interceptor-拦截器

爱被打了一巴掌 2023-05-29 02:59 96阅读 0赞

# 拦截器 #

## 说明 ##

有时候我们需要统计方法的耗时，需要知道统计出参/入参的相关信息。

通过拦截器都可以非常方便的实现。

## 设计思路 ##

无论是客户端还是服务端，只需要在方法执行前后，加入拦截器相关的方法调用，加入对应的上下文信息即可。

# 客户端实现 #

## 个人理解 ##

目前主要在客户端添加拦截器，其实服务端是类似的。

## 接口 ##

public interface Interceptor { 
    
        /** * 开始 * @param context 上下文 * @since 0.1.4 */
        void before(final InterceptorContext context);
    
        /** * 结束 * @param context 上下文 * @since 0.1.4 */
        void after(final InterceptorContext context);
    
        /** * 异常处理 * @param context 上下文 * @since 0.1.4 */
        void exception(final InterceptorContext context);
    
    }

## 实现 ##

@Override
        public Object invoke(Object proxy, Method method, Object[] args) throws Throwable { 
            // 状态判断
            //...
    
            //1. 拦截器
            final Interceptor interceptor = proxyContext.interceptor();
            final InterceptorContext interceptorContext = DefaultInterceptorContext.newInstance()
                    .traceId(traceId);
            interceptor.before(interceptorContext);
    
            // 构建基本调用参数
            //...
    
            //proxyContext 中应该是属于当前 service 的对应信息。
            // 每一次调用，对应的 invoke 信息应该是不通的，需要创建新的对象去传递信息
            // rpcRequest 因为要涉及到网络间传输，尽可能保证其简洁性。
            //...
    
            //3. 执行远程调用
            //...
    
            interceptor.after(interceptorContext);
            return result;
        }

本期为了简单，暂时没有处理异常相关的拦截处理。

# 测试代码 #

## register ##

启动

## server ##

启动

## client ##

*  测试代码

客户端配置指定耗时拦截器。

config.interceptor(new CostTimeInterceptor());

*  日志

[INFO] [2019-11-01 23:26:10.337] [main] [c.g.h.r.c.s.i.i.CostTimeInterceptor.after] - [Interceptor] cost time 49 mills for traceId: 6de5c8da8c784801921a2d2887f6e543