java 自定义 header_java – RequestHeader Spring Boot上的自定义验证

执行此操作的最佳方法IMO是使用自定义注释@Segment创建自定义HandlerMethodArgumentResolver,看起来像这样：

public class SegmentHandlerMethodArgumentResolver implements HandlerMethodArgumentResolver {

@Override

public boolean supportsParameter(MethodParameter parameter) {

return parameter.getParameterType().equals(String.class)

&& parameter.getParameterAnnotation(Segment.class);

}

@Override

public Object resolveArgument(MethodParameter parameter, ModelAndViewContainer mavContainer, NativeWebRequest webRequest, WebDataBinderFactory binderFactory) throws Exception {

String apiKey = webRequest.getHeader("X-API-Key");

if (apiKey != null) {

if (!API_KEY_LIST.contains(apiKey)) {

throw new InvalidApiKeyException();

}

return apiKey;

} else {

return WebArgumentResolver.UNRESOLVED;

}

}

}

然后您的控制器签名如下所示：

@RequestMapping(value = "/example/{id}", method = GET)

public Response getExample(

@PathVariable("id") String id,

@RequestParam(value = "myParam", required = true) @Valid @Pattern(regexp = MY_REGEX) String myParamRequest,

@RequestParam(value = "callback", required = false) String callback,

@Segment String apiKeyHeader) {

// Stuff here...

}

您将在WebMvcConfigurationAdapter中注册处理程序方法参数解析程序：

@Configuration

public class MvcConfig extends WebMvcConfigurerAdapter {

@Override

public void addArgumentResolvers(List argumentResolvers) {

argumentResolvers.add(segmentHandler());

}

@Bean

public SegmentHandlerMethodArgumentResolver segmentHandler() {

return new SegmentHandlerMethodArgumentResolver();

}

}