Sentinel-【來源相關(guān)】【錯(cuò)誤自定義】(七)

1.錯(cuò)誤重定向: 增加組建，通用異常捕獲處理(增加sentinel異常次數(shù)-Tracer.trace(e))

(1).錯(cuò)誤重定向

@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","熱點(diǎn)-異常");
        }else if (e instanceof SystemBlockException){
            backMap.put("code",103);
            backMap.put("msg","系統(tǒng)規(guī)則-異常");
        }else if (e instanceof AuthorityException){
            backMap.put("code",104);
            backMap.put("msg","認(rèn)證-異常");
        }

        // 設(shè)置返回json數(shù)據(jù)
        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異常次數(shù))

@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);
    }
}

來源處理，服務(wù)來源傳遞

思路：
    (1).【Client方】，設(shè)置當(dāng)前 Fegin 頭部的【source-origin】值為當(dāng)前應(yīng)用名稱: spring.application.name
        (2).【被調(diào)用方】，配置Sentinel的Origin解析操作

(1).Fegin 服務(wù)來源傳遞

@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控制臺】測試服務(wù)來源限制