All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.guanmengyuan.spring.ex.web.config.GlobalResponseAdvice Maven / Gradle / Ivy

Go to download

There is a newer version: 3.3.3.2
Show newest version
package com.guanmengyuan.spring.ex.web.config;

import static com.guanmengyuan.spring.ex.common.model.constant.GlobalResponseConstant.DEFAULT_PATH;

import java.util.Objects;
import java.util.Set;

import org.dromara.hutool.core.collection.CollUtil;
import org.dromara.hutool.core.collection.set.SetUtil;
import org.dromara.hutool.core.text.StrUtil;
import org.dromara.hutool.json.JSONUtil;
import org.springframework.core.MethodParameter;
import org.springframework.core.annotation.Order;
import org.springframework.http.MediaType;
import org.springframework.http.converter.HttpMessageConverter;
import org.springframework.http.server.ServerHttpRequest;
import org.springframework.http.server.ServerHttpResponse;
import org.springframework.lang.NonNull;
import org.springframework.lang.Nullable;
import org.springframework.util.AntPathMatcher;
import org.springframework.web.bind.annotation.RestControllerAdvice;
import org.springframework.web.servlet.mvc.method.annotation.ResponseBodyAdvice;

import com.guanmengyuan.spring.ex.common.model.dto.res.Res;

@RestControllerAdvice
@Order(-1)
public class GlobalResponseAdvice implements ResponseBodyAdvice {

    private final Set ignores;
    private final SpringWebProperties springWebProperties;
    private final AntPathMatcher antPathMatcher;

    public GlobalResponseAdvice(SpringWebProperties springWebProperties) {
        this.springWebProperties = springWebProperties;

        ignores = SetUtil.of();
        if (CollUtil.isNotEmpty(springWebProperties.getIgnores())) {
            ignores.addAll(springWebProperties.getIgnores());
        }
        ignores.addAll(DEFAULT_PATH);
        this.antPathMatcher = new AntPathMatcher();
    }

    @Override
    public boolean supports(@NonNull MethodParameter returnType,
            @NonNull Class> converterType) {
        return true;
    }

    @Override
    public Object beforeBodyWrite(@Nullable Object body, @NonNull MethodParameter returnType,
            @NonNull MediaType selectedContentType,
            @NonNull Class> selectedConverterType, @NonNull ServerHttpRequest request,
            @NonNull ServerHttpResponse response) {
        String path = request.getURI().getPath();
        if (!springWebProperties.getEnableGlobalRes() || StrUtil.equals(path, "/error") || body instanceof Res
                || ignores.stream().anyMatch(
                        ignore -> antPathMatcher.match(Objects.requireNonNull(ignore), Objects.requireNonNull(path)))) {
            return body;
        }
        return wrapperBody(body, returnType, response);
    }

    private Object wrapperBody(Object body, MethodParameter returnType, @NonNull ServerHttpResponse response) {
        if (returnType.getParameterType().isNestmateOf(String.class)) {
            response.getHeaders().setContentType(MediaType.APPLICATION_JSON);
            return JSONUtil.toJsonStr(Res.success(body));
        }
        return Res.success(body);
    }
}
    

    

    

            

    

            



    
    






    
© 2015 - 2024 Weber Informatics LLC | Privacy Policy