springboot 之对controller返回值再处理

谁借莪１个温暖的怀抱￠ 2022-12-23 14:23 151阅读 0赞

通常，我们在业务处理完之后，需要统一格式，之后返回到调用端。

#### Controller： ####

@PostMapping(value = "test")
    @CustomResponse
    public Object test(){ 
        System.out.println("testing......");
        Map<String,Integer> map = new HashMap<>();
        map.put("hello",2);
        return map;
    }

#### 自定义注解 ####

//在需要处理返回结果的方法头添加该注解即可
    
    import java.lang.annotation.*;
    
    @Target({ ElementType.METHOD, ElementType.TYPE})
    @Retention(RetentionPolicy.RUNTIME)
    @Documented
    public @interface CustomResponse { 
    }

#### 统一处理: ####

import com.test.nfts.webapi.Entity.DTO.Test;
    import org.springframework.core.MethodParameter;
    import org.springframework.http.MediaType;
    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;
    
    import java.util.Map;
    
    @RestControllerAdvice()
    public class CustomResponseAdvice implements ResponseBodyAdvice { 
        @Override
        public boolean supports(MethodParameter methodParameter, Class aClass) { 
            System.out.println("Is supporting method");
    
            //true 支持; false 不支持
            return methodParameter.hasMethodAnnotation(CustomResponse.class);
        }
    
        @Override
        public Object beforeBodyWrite(Object o, MethodParameter methodParameter, MediaType mediaType, Class aClass, ServerHttpRequest serverHttpRequest, ServerHttpResponse serverHttpResponse) { 
            System.out.println("In beforeBodyWrite() method of " + getClass().getSimpleName());
            if (o instanceof Test) { 
                return o;
            }
    
            //这里可以针对返回结果进行处理，比如加密等
            ((Map<String,Integer>)o).put("word",3);
            return o;
        }
    }

#### 测试 ####

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3UwMTM5MTkxNTM_size_16_color_FFFFFF_t_70_pic_center]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3UwMTM5MTkxNTM_size_16_color_FFFFFF_t_70_pic_center]: /images/20221120/ae05be064e594d2d8d2024ec580d4c47.png