在Sentinel中,限流和降级的提示词都是一样的。我们可以进行优化
Sentinel1.8.6中,Spring Cloud Alibaba Sentinel 不再依赖 sentinel-web-servlet 实现,而是依赖 sentinel-spring-webmvc-dapter , 基于 Spring 的 Interceptor 拦截资源,而不再是 CommonFilter。
低版本 实现 com.alibaba.csp.sentinel.adapter.servlet.callback.UrlBlockHandler的blocked方法
高版本 实现 com.alibaba.csp.sentinel.adapter.spring.webmvc.callback.BlockExceptionHandler的handle方法
这里我们实现BlockExceptionHandler的handle方法
- package com.itmuch.contentcenter.sentineltest;
-
- import com.alibaba.csp.sentinel.adapter.spring.webmvc.callback.BlockExceptionHandler;
- import com.alibaba.csp.sentinel.slots.block.BlockException;
- import com.alibaba.csp.sentinel.slots.block.authority.AuthorityException;
- import com.alibaba.csp.sentinel.slots.block.degrade.DegradeException;
- import com.alibaba.csp.sentinel.slots.block.flow.FlowException;
- import com.alibaba.csp.sentinel.slots.block.flow.param.ParamFlowException;
- import com.alibaba.csp.sentinel.slots.system.SystemBlockException;
- import com.fasterxml.jackson.databind.ObjectMapper;
- import lombok.AllArgsConstructor;
- import lombok.Builder;
- import lombok.Data;
- import lombok.NoArgsConstructor;
- import org.springframework.stereotype.Component;
-
- import javax.servlet.http.HttpServletRequest;
- import javax.servlet.http.HttpServletResponse;
-
- @Component
- public class MyUrlBlockHandler implements BlockExceptionHandler {
-
- //实现 com.alibaba.csp.sentinel.adapter.spring.webmvc.callback.BlockExceptionHandler的handle方法
-
- @Override
- public void handle(HttpServletRequest request, HttpServletResponse response, BlockException ex) throws Exception {
- ErrorMsg msg = null;
- if(ex instanceof FlowException){
- msg = ErrorMsg.builder()
- .status(100)
- .msg("限流了")
- .build();
-
- }else if(ex instanceof DegradeException){
- msg = ErrorMsg.builder()
- .status(101)
- .msg("降级了")
- .build();
-
- }else if(ex instanceof ParamFlowException){
- msg = ErrorMsg.builder()
- .status(102)
- .msg("热点参数限流")
- .build();
-
- }else if(ex instanceof SystemBlockException){
- msg = ErrorMsg.builder()
- .status(103)
- .msg("系统规则(负载/...不满足要求)")
- .build();
-
- }else if(ex instanceof AuthorityException){
- msg = ErrorMsg.builder()
- .status(104)
- .msg("授权规则不通过")
- .build();
-
- }
- response.setStatus(500);
- response.setCharacterEncoding("utf-8");
- response.setHeader("ContentType","application/json;charset=utf-8");
- response.setContentType("application/json;charset=utf-8");
- new ObjectMapper()
- .writeValue(
- response.getWriter(),
- msg
- );
-
- }
-
-
- }
- @Data
- @Builder
- @AllArgsConstructor
- @NoArgsConstructor
- class ErrorMsg{
- private Integer status;
- private String msg;
- }
实现完后,别忘了加上@Component,这样可以实现不同类型返回不同的提示。
同理RequestOriginParser、UrlCleaner也移动到了com.alibaba.csp.sentinel.adapter.spring.webmvc.callback包里,可以自己实现扩展功能。