美文网首页
sentinel结合dubbo进行热点参数(用户ID)限流

sentinel结合dubbo进行热点参数(用户ID)限流

作者: 加油_汤姆叔叔 | 来源:发表于2019-07-29 14:50 被阅读0次

    sentinel结合dubbo进行热点参数限流

    (1)dubbo中接入sentinel

    方法:

    (1)只需要在pom中假如三行配置即可接入sentinel

    (2)进行热点参数限流时候,需要接入热点限流模块。

            <dependency>
                <groupId>com.alibaba.csp</groupId>
                <artifactId>sentinel-parameter-flow-control</artifactId>
                <version>1.6.2</version>
            </dependency>
    

    **普通的热点参数限流只需要在entry中传入相应的参数就可以,但是由于我们需要对dubbo的服务根据企业id来进行限流,dubbo在接入sentinel的时候将entry封装进了adapter里面,故此:

    1. 需要改写官方提供的adaper,将企业的id传入到entry中
    2. 通过dubbo的rpccontext拿到从consumer传入的企业id 和用户id **

    接入原理:

    Dubbo

    通过dubbo的拦截器将服务实例进行拦截,同时接入sentinel core对服务实例的调用进行统计。

    (1)SPI扩展原理
    ExtensionLoader:1. 通过接口类名和key值获取一个实现 2. 利用IOC依赖注入功能(实现一个代理类) 3. 采用装饰器模式进行功能增强,自动实现包装
    1. ExtensionLoader 实现初始化
    2. 扫描配置文件,根据配置文件中的类名进行类加载
    3. 扩展适配器
    (2)Filter链原理
    1. 构造Filter链

      Dubbo的Filter实现入口是 在ProtocolFilterWrapper,因为ProtocolFilterWrapper是Protocol的包装类,所以会在加载的Extension的时候被自动包装进来,该封装器实现了Protocol接口,并提供了一个参数类型为Protocol的构造方法。Dubbo依据这个构造方法识别出封装器,并将该封装器作为其他Protocol接口实现的代理

    Sentinel

    (1) 基本原理

    限流入口是entry(),在创建entry()的时候一些列插槽会被创建。

    (1)插槽组成:StatisticSlot,systemslot,autheityslot,NodeSelectorSlot等

    ​ NodeSelectorSlot负责收集资源的路径,并将路径以树状结构存储起来。

    (2)本质上是基于信号量对资源进行监控(类似于AQS)

    (3)每一个插槽都有一个入口entry(),还有一个出口(也是进入下一个插槽的起点)fullentry(),在经过NodeSelectorSlot收集资源路径,ClusterBuilderSlot记录监控信息之后,后面每过一次entry(即经历过一个插槽就会进行一次checkrule,判断是否符合规定,如果不符合则block,符合则pass)【总的来说,每经过一次slot的entry就是一次检查,检查当前对服务的访问是否符合规则】

    Sentinel接入Dubbo(同时实现企业id的流控)

    (1)利用dubbo的filter将provider的服务进行拦截

    (2)通过rpcontext获取consumer传递过来的企业id和用户id

    (3)将企业id、用户id传入entry,entry经过StatisticSlot时候会对其进行统计,判断是否是热点数据,若是则触发流控规则。

    (4)在实现步骤(3)的时候需要接入

            <dependency>
                <groupId>com.alibaba.csp</groupId>
                <artifactId>sentinel-parameter-flow-control</artifactId>
                <version>1.6.2</version>
            </dependency>
    

    具体实现代码:

    @Activate(group = "provider")
    public class SentinelDubboProviderFilter extends AbstractDubboFilter implements Filter {
    
        public SentinelDubboProviderFilter() {
            RecordLog.info("Sentinel Dubbo provider filter initialized");
        }
    
        private static final String ENTERPRISE_ID = "enterpriseId";
        private static final String USER_ID = "userId";
        private static final String DEFAULT_ENTERPRISE_ID = "0";
        private static final String DEFAULT_USER_ID = "0";
    
        public Result invoke(Invoker<?> invoker, Invocation invocation) throws RpcException {
            // Get origin caller.
            String application = DubboUtils.getApplication(invocation, "");
    
            Entry interfaceEntry = null;
            Entry methodEntry = null;
            String enterpriseId = null;
            String userId = null;
            try {
                enterpriseId = RpcContext.getContext().getAttachment(ENTERPRISE_ID);
                userId = RpcContext.getContext().getAttachment(USER_ID);
                if (userId == null){
                    userId = DEFAULT_USER_ID;
                }
                if (enterpriseId == null){
                    enterpriseId = DEFAULT_ENTERPRISE_ID;
                }
    
                String resourceName = getResourceName(invoker, invocation);
                String interfaceName = invoker.getInterface().getName();
                ContextUtil.enter(resourceName, application);
                interfaceEntry = SphU.entry(interfaceName, EntryType.IN,1,enterpriseId);
                methodEntry = SphU.entry(resourceName, EntryType.IN, 1, enterpriseId,userId);
    
                Result result = invoker.invoke(invocation);
                if (result.hasException()) {
                    Tracer.trace(result.getException());
                }
                return result;
            } catch (BlockException e) {
                return DubboFallbackRegistry.getProviderFallback().handle(invoker, invocation, e);
            } catch (RpcException e) {
                Tracer.trace(e);
                throw e;
            } finally {
                if (methodEntry != null) {
                    methodEntry.exit(1,enterpriseId,userId);
                }
                if (interfaceEntry != null) {
                    interfaceEntry.exit(1,enterpriseId);
                }
                ContextUtil.exit();
            }
        }
    }
    

    总结:只需要将dubbo与sentinel的适配器进行适配,将rpccontext中的企业id、用户id传入entry即可通过控制台对接口、方法进行流控。

    相关文章

      网友评论

          本文标题:sentinel结合dubbo进行热点参数(用户ID)限流

          本文链接:https://www.haomeiwen.com/subject/huyerctx.html