Sentinel-来源相关错误自定义

Posted 2021-03-24

1.错误重定向: 增加组建，通用异常捕获处理(增加sentinel异常次数-Tracer.trace(e))

(1).错误重定向

@Component
public class MyUrlBlockHandler implements UrlBlockHandler {
    @Override
    public void blocked(HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse, BlockException e) throws IOException {
        Map<String,Object> backMap=new HashMap<>();
        backMap.put("error","sentinel error");
        if (e instanceof FlowException){
            backMap.put("code",100);
            backMap.put("msg","限流-异常");
        }else if (e instanceof DegradeException){
            backMap.put("code",101);
            backMap.put("msg","降级-异常");
        }else if (e instanceof ParamFlowException){
            backMap.put("code",102);
            backMap.put("msg","热点-异常");
        }else if (e instanceof SystemBlockException){
            backMap.put("code",103);
            backMap.put("msg","系统规则-异常");
        }else if (e instanceof AuthorityException){
            backMap.put("code",104);
            backMap.put("msg","认证-异常");
        }

        // 设置返回json数据
        String header= MediaType.APPLICATION_JSON_UTF8_VALUE;
        httpServletResponse.setStatus(200);
        httpServletResponse.setHeader("content-Type",header);
        httpServletResponse.setContentType(header);
        httpServletResponse.getWriter().write(JSON.toJSONString(backMap));
    }
}

(2).通用异常捕获处理(增加sentinel异常次数)

@RestControllerAdvice
@Slf4j
public class GlobalExceptionHandler {
    // 处理身份异常
    @ExceptionHandler(SecurityException.class)
    public ResponseEntity<ApiResult> handler(SecurityException e) {
        ApiResult backResult = ApiResult.builder().code(1001).message(e.getMessage()).build();
        Tracer.trace(e);
        return new ResponseEntity<ApiResult>(backResult, HttpStatus.OK);
    }

    // 处理表单字段异常
    @ExceptionHandler(FormFieldErrorException.class)
    public ResponseEntity<ApiResult> handler(FormFieldErrorException e) {
        ApiResult backResult = ApiResult.builder()
                .code(1003)
                .data(e.getFieldMessage())
                .message(e.getMessage())
                .build();
        Tracer.trace(e);
        return new ResponseEntity<ApiResult>(backResult, HttpStatus.OK);
    }
}

来源处理，服务来源传递

思路：
    (1).【Client方】，设置当前 Fegin 头部的【source-origin】值为当前应用名称: spring.application.name
        (2).【被调用方】，配置Sentinel的Origin解析操作

(1).Fegin 服务来源传递

@Configuration
@Slf4j
public class MyRequestInterceptor implements RequestInterceptor {
    @Value("${spring.application.name}")
    String applicationName;

    @Override
    public void apply(RequestTemplate requestTemplate) {
        requestTemplate.header("source-origin", applicationName);
    }
}

(2).来源处理，增加组建

@Slf4j
@Component
public class MyRequestOriginParse implements RequestOriginParser {
    @Override
    public String parseOrigin(HttpServletRequest httpServletRequest) {
        String origin=httpServletRequest.getHeader("source-origin");
        if (StringUtils.isBlank(origin)){
            origin="default";
        }
//        log.info("头部名称是: "+ JSON.toJSONString(httpServletRequest.getHeaderNames()));
//        log.info("来源是: "+origin);
        return origin;
    }
}

(3).【sentinel控制台】测试服务来源限制