使用 RequestBodyAdvice 和 ResponseBodyAdvice 对controller进行增强

最新推荐文章于 2024-08-04 08:36:37 发布

原创最新推荐文章于 2024-08-04 08:36:37 发布 · 425 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#java

java 同时被 2 个专栏收录

14 篇文章

订阅专栏

controller增强

1 篇文章

订阅专栏

这篇博客介绍了如何使用RequestBodyAdvice和ResponseBodyAdvice对Controller中的请求体和响应体进行增强处理。通过这两个注解，可以在读取请求参数前和响应返回前进行特定的转换操作，例如日期格式的国际化处理。示例代码展示了如何根据当前用户的时区调整时间字符串，确保日期在不同地区显示的一致性。

RequestBodyAdvice,对所有以@RequestBody的参数的方法进行增强

/**
 * 对所有以@RequestBody的参数的方法进行增强
 */
@Slf4j
@ControllerAdvice("com.haier.hopm.biz.service.controller")
public class DateTimeRequestBodyAdvice implements RequestBodyAdvice {

    @Autowired
    private RedisUtil redisUtil;

    //此处如果返回false , 则不执行当前Advice的业务
    @Override
    public boolean supports(MethodParameter methodParameter, Type type, Class<? extends HttpMessageConverter<?>> aClass) {
        I18nTimeHandle i18nTimeHandle = methodParameter.getMethodAnnotation(I18nTimeHandle.class);
        //controller上有 I18nTimeHandle 这个注解的方法才进行处理
        if (i18nTimeHandle != null) {
            return true;
        }
        return false;
    }

    /**
     * @title 读取参数前执行
     * @description 在此做些编码 / 解密 / 封装参数为对象的操作
     */
    @Override
    public HttpInputMessage beforeBodyRead(HttpInputMessage httpInputMessage, MethodParameter methodParameter, Type type, Class<? extends HttpMessageConverter<?>> aClass) {
        return httpInputMessage;
    }

    /**
     * @title 读取参数后执行
     * @author yuchen
     */
    @Override
    public Object afterBodyRead(Object o, HttpInputMessage httpInputMessage, MethodParameter methodParameter,
                                Type type, Class<? extends HttpMessageConverter<?>> aClass) {
        String responesBody = JSONObject.toJSONStringWithDateFormat(o, JSON.DEFFAULT_DATE_FORMAT);
        //获取请求json中时间字符串
        List<String> jsonStrDateStrList = DateZoneUtil.getJsonStrDateStr(responesBody);
        if (CollectionUtil.isEmpty(jsonStrDateStrList)) {
            return o;
        }
//        log.info("RequestBody-处理前:{}", responesBody);
        //获取的当前国家所在时区,如果Redis查不到,则默认 +8
        String timeZone = redisUtil.get(Constants.COUNTRY_TIME_ZONE + UserUtil.getCountryCode());
        String sourceTimeZone = StrUtil.isEmpty(timeZone) ? "+8" : timeZone;
        for (String originDate : jsonStrDateStrList) {
            //将前端传入的时间 转换成 东八区的时间
            String targetDate = DateZoneUtil.converDateGMT(originDate, sourceTimeZone, "+8");
            responesBody = StrUtil.replace(responesBody, originDate, targetDate);
        }
//        log.info("RequestBody-处理后:{}", responesBody);
        return JSONObject.parseObject(responesBody, type);

    }

    /**
     * @title 无请求时的处理
     */

    @Override
    public Object handleEmptyBody(Object body, HttpInputMessage httpInputMessage, MethodParameter methodParameter,
                                  Type type, Class<? extends HttpMessageConverter<?>> aClass) {
        return body;
    }
}

ResponseBodyAdvice ,对controller的响应体进行增强

/**
 * 对controller的响应体进行增强
 */
@Slf4j
@ControllerAdvice("com.haier.hopm.biz.service.controller")
public class DateTimeResponseBodyAdvice implements ResponseBodyAdvice {

    @Autowired
    private RedisUtil redisUtil;

    //此处如果返回false , 则不执行当前Advice的业务
    @Override
    public boolean supports(MethodParameter methodParameter, Class aClass) {
        return true;
    }

    @Override
    public Object beforeBodyWrite(Object o, MethodParameter methodParameter, MediaType mediaType, Class aClass,
                                  ServerHttpRequest serverHttpRequest, ServerHttpResponse serverHttpResponse) {
        String responesBody = JSONObject.toJSONStringWithDateFormat(o, JSON.DEFFAULT_DATE_FORMAT);
        //获取请求json中时间字符串
        List<String> jsonStrDateStrList = DateZoneUtil.getJsonStrDateStr(responesBody);
        if (CollectionUtil.isEmpty(jsonStrDateStrList)) {
            return o;
        }
//        log.info("ResponseBody-处理前:{}", responesBody);
        //获取的当前国家所在时区,如果Redis查不到,则默认 +8
        String timeZone = redisUtil.get(Constants.COUNTRY_TIME_ZONE + UserUtil.getCountryCode());
        String sourceTimeZone = StrUtil.isEmpty(timeZone) ? "+8" : timeZone;
        for (String originDate : jsonStrDateStrList) {
            //将前端传入的时间 转换成 东八区的时间
            String targetDate = DateZoneUtil.converDateGMT(originDate, "+8", sourceTimeZone);
            responesBody = StrUtil.replace(responesBody, originDate, targetDate);
        }
//        log.info("ResponseBody-处理后:{}", responesBody);
        return JSONObject.parseObject(responesBody);
    }
}