cn.flood.cloud.gateway.handler.JsonErrorExceptionHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of flood-spring-cloud-starter-gateway Show documentation
There is a newer version: 2.7.14.0
Show newest version
package cn.flood.cloud.gateway.handler;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.web.ErrorProperties;
import org.springframework.boot.autoconfigure.web.ResourceProperties;
import org.springframework.boot.autoconfigure.web.reactive.error.DefaultErrorWebExceptionHandler;
import org.springframework.boot.web.reactive.error.ErrorAttributes;
import org.springframework.context.ApplicationContext;
import org.springframework.http.MediaType;
import org.springframework.web.reactive.function.BodyInserters;
import org.springframework.web.reactive.function.server.*;
import reactor.core.publisher.Mono;

import java.util.Map;

/**
 * 自定义的JsonErrorWebExceptionHandler异常处理类
 * @author xuzf
 * date: 2020-4-20
 * 参考文档：https://www.cnblogs.com/throwable/p/10848879.html
 */
public class JsonErrorExceptionHandler extends DefaultErrorWebExceptionHandler {

    private final Logger log = LoggerFactory.getLogger(this.getClass());

    @Autowired
    private ExceptionHandlerAdvice exceptionHandlerAdvice;

    public JsonErrorExceptionHandler(ErrorAttributes errorAttributes,
                                     ResourceProperties resourceProperties,
                                     ErrorProperties errorProperties,
                                     ApplicationContext applicationContext) {
        super(errorAttributes, resourceProperties, errorProperties, applicationContext);
    }

//    @Override
//    protected Map getErrorAttributes(ServerRequest request, boolean includeStackTrace) {
//        // 这里其实可以根据异常类型进行定制化逻辑
//        Throwable error = super.getError(request);
//        int code = HttpStatus.INTERNAL_SERVER_ERROR.value();
//        if (error instanceof FileNotFoundException) {
//            code = HttpStatus.NOT_FOUND.value();
//        } else if (error instanceof ResponseStatusException
//                && StringUtils.contains(error.getMessage(), HttpStatus.NOT_FOUND.toString())) {
//            code = HttpStatus.NOT_FOUND.value();
//        } else if (error instanceof TokenException) {
//            code = HttpStatus.UNAUTHORIZED.value();
//        }
//        Map errorAttributes = new HashMap<>(8);
//        errorAttributes.put("message", error.getMessage());
//        errorAttributes.put("code", code);
//        errorAttributes.put("method", request.methodName());
//        errorAttributes.put("path", request.path());
//        return errorAttributes;
//    }

    @Override
    @SuppressWarnings("all")
    protected RouterFunction getRoutingFunction(ErrorAttributes errorAttributes) {
        return RouterFunctions.route(RequestPredicates.all(), this::renderErrorResponse);
    }

//    @Override
//    protected int getHttpStatus(Map errorAttributes) {
//        return HttpStatus.INTERNAL_SERVER_ERROR.value();
//    }

    @Override
    protected Mono renderErrorResponse(ServerRequest request) {
        Map error = getErrorAttributes(request, getErrorAttributeOptions(request, MediaType.ALL));
        int errorStatus = getHttpStatus(error);
        Throwable throwable = getError(request);
        return ServerResponse.status(errorStatus)
                .contentType(MediaType.APPLICATION_JSON)
                .body(BodyInserters.fromValue(exceptionHandlerAdvice.handle(throwable)));
    }
}