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

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

(1)dubbo中接入sentinel

方法:

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

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

        
            com.alibaba.csp
            sentinel-parameter-flow-control
            1.6.2
        

**普通的热点参数限流只需要在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)的时候需要接入

        
            com.alibaba.csp
            sentinel-parameter-flow-control
            1.6.2
        

具体实现代码:

@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)限流)