使用Sentinel自定義返回和實現(xiàn)區(qū)分來源方式

更新時間：2025年04月02日 16:43:50 作者：Gangbb

這篇文章主要介紹了使用Sentinel自定義返回和實現(xiàn)區(qū)分來源方式,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教

Sentinel自定義返回和實現(xiàn)區(qū)分來源

1. 自定義錯誤返回

因為流控和降級頁面報的異常都是相同的，所以可以優(yōu)化一下。

使用官方的接口BlockExceptionHandler（2.2.7.RELEASE）、UrlBlockHandler（0.9.0.RELEASE）

定義處理類：

@Component
public class MyUrlBlockHandler implements BlockExceptionHandler {

    @Override
    public void handle(HttpServletRequest httpServletRequest, 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("熱點參數(shù)限流")
                    .build();
        } else if (ex instanceof SystemBlockException) {
            msg = ErrorMsg.builder()
                    .status(103)
                    .msg("系統(tǒng)規(guī)則（負載/...不滿足要求）")
                    .build();
        } else if (ex instanceof AuthorityException) {
            msg = ErrorMsg.builder()
                    .status(104)
                    .msg("授權規(guī)則不通過")
                    .build();
        }
        // http狀態(tài)碼
        response.setStatus(500);
        response.setCharacterEncoding("utf-8");
        response.setHeader("Content-Type", "application/json;charset=utf-8");
        response.setContentType("application/json;charset=utf-8");
        // spring mvc自帶的json操作工具，叫jackson
        new ObjectMapper()
                .writeValue(
                        response.getWriter(),
                        msg
                );
    }
}

@Data
@Builder
@AllArgsConstructor
@NoArgsConstructor
class ErrorMsg {
    private Integer status;
    private String msg;
}

效果：

2. 實現(xiàn)區(qū)分來源

使用RequestOriginParser接口

@Component
public class MyRequestOriginParser implements RequestOriginParser {
    @Override
    public String parseOrigin(HttpServletRequest request) {
        // 從請求參數(shù)中獲取名為 origin 的參數(shù)并返回
        // 如果獲取不到origin參數(shù)，那么就拋異常

        String origin = request.getParameter("origin");
        if (StringUtils.isBlank(origin)) {
            throw new IllegalArgumentException("origin must be specified");
        }
        return origin;
    }
}

訪問：http://127.0.0.1:8010/shares/1