使用spring @ControllerAdvice 对jsonp进行封装

package cn.xxx.user.aop;

import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.annotation.*;
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.http.server.ServletServerHttpRequest;
import org.springframework.stereotype.Component;
import org.springframework.util.CollectionUtils;
import org.springframework.util.StringUtils;
import org.springframework.web.bind.annotation.ControllerAdvice;
import org.springframework.web.servlet.mvc.method.annotation.ResponseBodyAdvice;

import java.util.Map;

@Slf4j
@Aspect
@Component
@ControllerAdvice
public class JsonpInterceptor implements ResponseBodyAdvice {
    @Override
    public boolean supports(MethodParameter methodParameter, Class arg1) {
        //过滤,只有ctrl包下的类,走beforeBodyWrite方法。
        if (methodParameter.getMethod().getDeclaringClass().getName().contains("cn.xxx.user.ctrl"))
            return true;
        return false;
    }

    @Override
    public Object beforeBodyWrite(Object body, MethodParameter arg1,
                                  MediaType arg2, Class arg3, ServerHttpRequest request,
                                  ServerHttpResponse arg5) {
        String pcallback = null;
        //过滤,只对GET方式的方法,进行jsonp处理
        if (null != request.getMethod() && request.getMethod().matches("GET")) {
            Map<String, String[]> parameterMap = ((ServletServerHttpRequest) request).getServletRequest().getParameterMap();
            if (!CollectionUtils.isEmpty(parameterMap)) {
                String[] pcallbackArray = parameterMap.get("pcallback");
                if (null != pcallbackArray && pcallbackArray.length > 0) {
                    pcallback = pcallbackArray[0];
                    if(!StringUtils.isEmpty(pcallback)){
                        return pcallback + "(" + body.toString() + ")";
                    }
                }
            }
        }
        return body;
    }
}

  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值