top.zenyoung.controller.global.BaseGlobalErrorExceptionHandler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of zenyoung-webflux Show documentation

webflux-控制器模块

There is a newer version: 2.0.2

package top.zenyoung.controller.global;

import com.google.common.collect.Maps;
import lombok.extern.slf4j.Slf4j;
import org.springframework.boot.autoconfigure.web.ResourceProperties;
import org.springframework.boot.autoconfigure.web.reactive.error.AbstractErrorWebExceptionHandler;
import org.springframework.boot.web.reactive.error.ErrorAttributes;
import org.springframework.context.ApplicationContext;
import org.springframework.http.MediaType;
import org.springframework.http.codec.ServerCodecConfigurer;
import org.springframework.web.reactive.function.BodyInserters;
import org.springframework.web.reactive.function.server.*;
import reactor.core.publisher.Mono;
import top.zenyoung.common.model.RespResult;

import javax.annotation.Nonnull;
import java.util.Map;

/**
 * 全局错误处理
 *
 * @author yangyong
 * @version 1.0
 *  2020/4/1 2:29 下午
 **/
@Slf4j
public abstract class BaseGlobalErrorExceptionHandler extends AbstractErrorWebExceptionHandler {

    public BaseGlobalErrorExceptionHandler(final ErrorAttributes errorAttributes,
                                           final ResourceProperties resourceProperties,
                                           final ServerCodecConfigurer serverCodecConfigurer,
                                           final ApplicationContext applicationContext) {
        super(errorAttributes, resourceProperties, applicationContext);
        if (serverCodecConfigurer != null) {
            setMessageWriters(serverCodecConfigurer.getWriters());
            setMessageReaders(serverCodecConfigurer.getReaders());
        }
    }

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

    @Nonnull
    protected Mono renderErrorResponse(final ServerRequest request) {
        final Map errors = getErrorAttributes(request, false);
        log.error("renderErrorResponse=>\n {}", errors);
        //返回数据处理
        return ServerResponse.ok()
                .contentType(MediaType.APPLICATION_JSON)
                .body(BodyInserters.fromValue(RespResult.builder()
                        .code(getHttpStatus(errors))
                        .msg(getMessage(errors))
                        .data(Maps.newLinkedHashMap(errors))
                        .build()
                ));
    }

    protected int getHttpStatus(final Map errorAttributes) {
        return (int) errorAttributes.get("status");
    }

    protected String getMessage(final Map errorAttributes) {
        return (String) errorAttributes.get("message");
    }
}