栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Java

路径参数绑定至bean

Java 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

路径参数绑定至bean

路径参数过多时,频繁的写@PathVariable太过繁琐,如下

自定义HandlerMethodArgumentResolver,在解析完body时,将路径参数进行解析,然后放入请求头对象中。注:以下方法仅限于post请求,且路径参数绑定至body对象中

自定义任意注解,标识该对象需要将路径参数放入,如下:

@Target(ElementType.PARAMETER)
@Retention(RetentionPolicy.RUNTIME)
@documented
public @interface SelfPathVariable {
}

接口定义如下 

@PostMapping("/{channel}/productquery/{version}/{timestamp}/{sign}")
public QueryProductResponse estimatedPrice(@SelfPathVariable @RequestBody QueryProductRequest request) {
    return new QueryProductResponse();
}

 自定义的HandlerMethodArgumentResolver

import org.springframework.core.MethodParameter;
import org.springframework.web.bind.support.WebDataBinderFactory;
import org.springframework.web.context.request.NativeWebRequest;
import org.springframework.web.context.request.RequestAttributes;
import org.springframework.web.method.support.HandlerMethodArgumentResolver;
import org.springframework.web.method.support.ModelAndViewContainer;
import org.springframework.web.servlet.mvc.method.annotation.RequestResponseBodyMethodProcessor;

import java.lang.reflect.Field;
import java.util.Map;

import static org.springframework.web.servlet.HandlerMapping.URI_TEMPLATE_VARIABLES_ATTRIBUTE;


public class SelfRequestResponseBodyMethodProcessor implements HandlerMethodArgumentResolver {

    private RequestResponseBodyMethodProcessor processor;

    public CtripRequestResponseBodyMethodProcessor(RequestResponseBodyMethodProcessor processor) {
        this.processor = processor;
    }

    @Override
    public boolean supportsParameter(MethodParameter parameter) {
        return parameter.hasParameterAnnotation(SelfPathVariable.class) && processor.supportsParameter(parameter);
    }

    @Override
    public Object resolveArgument(MethodParameter parameter, ModelAndViewContainer mavContainer, NativeWebRequest webRequest, WebDataBinderFactory binderFactory) throws Exception {
        Object o = processor.resolveArgument(parameter, mavContainer, webRequest, binderFactory);
        if (o instanceof AbstractSelfRequest) {
            Map attribute = (Map) webRequest.getAttribute(URI_TEMPLATE_VARIABLES_ATTRIBUTE, RequestAttributes.SCOPE_REQUEST);
            if (attribute != null && attribute.size() > 0) {
                for (Map.Entry entry : attribute.entrySet()) {
                    Field field = findField(o.getClass(), entry.getKey());
                    if (field != null) {
                        field.setAccessible(true);
                        field.set(o, entry.getValue());
                    }
                }
            }
        }
        return o;
    }

    private Field findField(Class aClass, String fieldName) throws NoSuchFieldException {
        Field field;
        return aClass.getSuperclass() != Object.class ? (field = findField(aClass.getSuperclass(), fieldName)) != null ? field : aClass.getDeclaredField(fieldName) : null;
    }

}
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.method.support.HandlerMethodArgumentResolver;
import org.springframework.web.method.support.HandlerMethodReturnValueHandler;
import org.springframework.web.servlet.config.annotation.ResourceHandlerRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
import org.springframework.web.servlet.mvc.method.annotation.RequestMappingHandlerAdapter;
import org.springframework.web.servlet.mvc.method.annotation.RequestResponseBodyMethodProcessor;

import javax.annotation.PostConstruct;
import java.util.ArrayList;
import java.util.List;

@Configuration
public class WebConfig implements WebMvcConfigurer {

    @Autowired
    private RequestMappingHandlerAdapter requestMappingHandlerAdapter;

    @PostConstruct
    public void init() {
        List resolvers = requestMappingHandlerAdapter.getArgumentResolvers();
        if (resolvers != null) {
            List newHandlers = new ArrayList<>(resolvers.size() + 1);
            for (HandlerMethodArgumentResolver resolver : resolvers) {
                if (resolver instanceof RequestResponseBodyMethodProcessor) {
                    newHandlers.add(new SelfRequestResponseBodyMethodProcessor((RequestResponseBodyMethodProcessor) resolver));
                }
                newHandlers.add(resolver);
            }
            requestMappingHandlerAdapter.setArgumentResolvers(newHandlers);
        }
    }

}

以上注入自定义的HandlerMethodArgumentResolver的方式太过笨拙,自行优化。

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/314960.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号