Sentinel中，如何自定义限流处理逻辑？

用心去追梦

于 2024-09-03 11:06:06 发布

阅读量470

点赞数 4

文章标签： sentinel windows

本文链接：https://blog.csdn.net/qq_33240556/article/details/141856301

版权

在 Sentinel 中，自定义限流处理逻辑通常涉及到几个方面：如何定义资源、如何配置限流规则、如何处理限流触发时的异常，以及如何实现更复杂的自定义逻辑。以下是一些详细的步骤和示例代码，帮助你理解如何在 Sentinel 中实现自定义的限流处理逻辑。

1. 定义资源

首先，你需要定义需要限流的资源。这可以通过 @SentinelResource 注解来完成，或者通过 SphU.entry 方法显式地进入资源。

import com.alibaba.csp.sentinel.annotation.SentinelResource;

public class CustomLimitingExample {

    @SentinelResource(value = "customLimitedMethod", fallback = "handleFallback")
    public String customLimitedMethod() {
        // 业务逻辑
        return "Hello from customLimitedMethod";
    }

    public String handleFallback() {
        // 限流触发后的回退逻辑
        return "Too many requests, please try again later.";
    }
}

2. 配置限流规则

接下来，你需要配置限流规则。这可以通过 Sentinel 控制台直接配置，也可以通过编程的方式加载规则。

通过编程方式配置

import com.alibaba.csp.sentinel.slots.block.flow.FlowRule;
import com.alibaba.csp.sentinel.slots.block.flow.FlowRuleManager;

import java.util.ArrayList;
import java.util.List;

public class CustomLimitingConfig {

    public static void initFlowRules() {
        List<FlowRule> rules = new ArrayList<>();
        FlowRule rule = new FlowRule();
        rule.setResource("customLimitedMethod"); // 被限流的资源名
        rule.setCount(20); // 每秒允许的请求数量
        rule.setGrade(com.alibaba.csp.sentinel.slots.block.RuleConstant.FLOW_GRADE_QPS); // 限流模式: QPS
        rule.setLimitApp("default"); // 限流应用
        rules.add(rule);
        FlowRuleManager.loadRules(rules);
    }
}

3. 处理限流触发时的异常

当限流规则被触发时，Sentinel 会抛出 BlockException。你需要捕获这个异常并处理它。可以通过 @SentinelResource 注解的 blockHandler 属性来指定限流时的处理方法。

import com.alibaba.csp.sentinel.annotation.SentinelResource;
import com.alibaba.csp.sentinel.slots.block.BlockException;

public class CustomLimitingExample {

    @SentinelResource(value = "customLimitedMethod", fallback = "handleFallback", blockHandler = "handleBlock")
    public String customLimitedMethod() {
        // 业务逻辑
        return "Hello from customLimitedMethod";
    }

    public String handleFallback() {
        // 限流触发后的回退逻辑
        return "Too many requests, please try again later.";
    }

    public String handleBlock(BlockException e) {
        // 限流触发时的处理逻辑
        return "Blocked by Sentinel";
    }
}

4. 实现更复杂的自定义逻辑

如果你需要实现更复杂的自定义逻辑，可以使用 Sentinel 提供的 API 和插件机制。例如，你可以自定义限流处理器或者使用 Sentinel 的扩展点来实现特定的限流策略。

自定义限流处理器

你可以通过扩展 com.alibaba.csp.sentinel.slots.block.flow.FlowController 接口来自定义限流逻辑。不过，这通常需要深入了解 Sentinel 的内部机制。

使用 Sentinel 的扩展点

Sentinel 提供了一些扩展点，允许你在限流过程中插入自定义的逻辑。例如，你可以实现 com.alibaba.csp.sentinel.adapter.spring.webmvc.fallback.FallbackStrategy 接口来定义限流触发时的处理策略。

import com.alibaba.csp.sentinel.adapter.spring.webmvc.fallback.FallbackStrategy;
import com.alibaba.csp.sentinel.slots.block.BlockException;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.web.bind.annotation.ControllerAdvice;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

@ControllerAdvice
public class CustomFallbackStrategy implements FallbackStrategy {

    @Override
    public ModelAndView handle(HttpServletRequest request, HttpServletResponse response, Throwable ex) {
        // 自定义限流处理逻辑
        return new ModelAndView("error", HttpStatus.TOO_MANY_REQUESTS.value(), MediaType.TEXT_PLAIN);
    }

    @Override
    public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, Throwable ex, BlockException be) {
        // 限流触发后的额外处理逻辑
    }
}