Java：返回结果统一转换

柔情只为你懂 2024-03-22 11:43 61阅读 0赞

### 返回结果统一转换 ###

import org.springframework.core.MethodParameter;
    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.web.bind.annotation.RestControllerAdvice;
    import org.springframework.web.servlet.mvc.method.annotation.ResponseBodyAdvice;
    
    /**
     * 返回结果统一转换
     */
    @RestControllerAdvice(basePackages = "com.wuxian")
    public class ResponseAdvisor implements ResponseBodyAdvice<Object> {
        
    
        /**
         *  判断哪些需要拦截
         */
        @Override
        public boolean supports(MethodParameter returnType, Class<? extends HttpMessageConverter<?>> converterType) {
        
            return true;
        }
    
        /**
         * 返回结果包装
         * @param body controller返回的数据
         * @param returnType
         * @param selectedContentType
         * @param selectedConverterType
         * @param request
         * @param response
         * @return
         */
        @Override
        public Object beforeBodyWrite(Object body, MethodParameter returnType, MediaType selectedContentType, Class<? extends HttpMessageConverter<?>> selectedConverterType, ServerHttpRequest request, ServerHttpResponse response) {
        
    
            if (body instanceof Result){
        
                return body;
            }
    
            if (body instanceof Boolean){
        
                boolean result = (boolean)body;
                return new BaseResponse<Boolean>(result);
            }
    
            if (body instanceof PageVO){
        
                return new BaseResponse<>(body);
            }
    
            if (body instanceof ExceptionResponse){
        
                return new BaseResponse<>(400,((ExceptionResponse)body).getMsg());
            }
    
            return new BaseResponse<>(body);
        }
    }

import com.fasterxml.jackson.annotation.JsonInclude;
    
    /**
     * 返回结果通用对象封装
     */
    @Data
    @JsonInclude(JsonInclude.Include.NON_NULL)
    public class BaseResponse<D> implements Serializable {
        
    
        private static final long serialVersionUID = -2145852753254617071L;
    
        //是否成功
        private boolean success;
    
        //异常信息
        private String message;
    
        //200表示正常,300接口返回false，400调用异常
        private int code;
    
        //泛型，用来存储原有controller返回值
        private D data;
    
        public BaseResponse() {
        
            this.code = 200;
            this.message = "success";
            this.success = true;
        }
    
    
        public BaseResponse(D data){
        
            this();
            this.data = data;
        }
    
        public BaseResponse(boolean success){
        
            this();
            this.success = success;
            if(success == false){
        
                this.message = "操作失败";
                this.code = 300;
            }
        }
    
        public BaseResponse(boolean success,String msg){
        
            this(success);
            this.message = msg;
        }
    
        public BaseResponse(int code,String msg){
        
            this(false,msg);
            this.code = code;
        }
    }

/**
     * 返回结果通用封装
     */
    @Data
    public class Result {
        
        // 返回状态
        private int status;
        // 状态描述
        private String desc;
        // 返回数据
        private Object data;
    
        private String token;
    }

@Data
    public class ExceptionResponse{
        
        private String msg;
        public ExceptionResponse(String msg){
        
            this.msg = msg;
        }
    }