Spring 内容协商 - 一种 mime 类型扩展另一种

问题描述

我将 Spring Boot 与基于标头的内容协商一起使用。 我有一组仅对一部分客户端可用的端点，其余的都是公共的。 我的公共端点注释如下：

@PostMapping(consumes = "application/vnd.com.foo+json",produces = "application/vnd.com.foo+json")

和我的私人像这样：

@PostMapping(consumes = "application/vnd.com.foo.private+json",produces = "application/vnd.com.foo.private+json")

我希望公共端点也使用和生成私有 mime 类型，以便我的私有客户端可以在他们的所有请求中设置该 mime 类型。显然，我可以通过在所有公共端点上明确指定它来做到这一点：

@PostMapping(consumes = {"application/vnd.com.foo+json","application/vnd.com.foo.private+json"},produces = {"application/vnd.com.foo+json","application/vnd.com.foo.private+json"})

但我想要一种更简洁的方法来做到这一点。

是否有某种方法可以配置 Spring 以将私有 mime 类型视为“扩展”公共 mime 类型？

解决方法

我可以在我的 WebMvcConfigurer bean 中配置内容协商：

@Override
public void configureContentNegotiation(ContentNegotiationConfigurer configurer) {
    configurer.strategies(List.of(new HeaderContentNegotiationStrategy() {
        @Override
        public List<MediaType> resolveMediaTypes(NativeWebRequest request) throws HttpMediaTypeNotAcceptableException {
            List<MediaType> mediaTypes = super.resolveMediaTypes(request);
            List<MediaType> publicisedInternalMediaTypes = mediaTypes.stream()
                    .filter(m -> m.getSubtype().startsWith("vnd.com.foo.private"))
                    .map(m -> new MediaType(
                            m.getType(),m.getSubtype().replace("vnd.com.foo.private","vnd.com.foo"),m.getParameters()))
                    .collect(toList());
            mediaTypes.addAll(publicisedInternalMediaTypes);
            return mediaTypes;
        }
    }));
}

到目前为止，这似乎有效，但我很想看到一种更标准或更优雅的方式来实现这一目标。

编辑：这适用于 Accept 标头，但不适用于 Content-Type。

,

在我的 DelegatingWebMvcConfiguration 中，我可以覆盖 createRequestMappingHandlerMapping() 以添加私有 consumes 和 produces 映射以添加公共映射。

@Override
protected RequestMappingInfo createRequestMappingInfo(
        RequestMapping original,RequestCondition<?> customCondition) {
    ExpandedRequestMapping expanded = expandRequestMapping(original);
    return super.createRequestMappingInfo(expanded,customCondition);
}

/**
 * If the produces or consume fields include a public media type,* add the corresponding private media type,too.
 */
private ExpandedRequestMapping expandRequestMapping(RequestMapping original) {
    return new ExpandedRequestMapping(original);
}

private static String[] expandedMediaTypes(String[] originalArray) {
    // If we ever have a controller method that produces or consumes multiple public types,// then we'll need to change this.
    return Arrays.stream(originalArray)
            .filter(mediaType -> mediaType.contains(VND_PUBLIC))
            .findAny()
            .map(publicMediaType -> publicMediaType.replace(VND_PUBLIC,VND_PRIVATE))
            .map(privateMediaType -> ArrayUtils.addAll(originalArray,privateMediaType))
            .orElse(originalArray);
}

static class ExpandedRequestMapping implements RequestMapping {

    private final RequestMapping delegate;

    private ExpandedRequestMapping(RequestMapping delegate) {
        this.delegate = delegate;
    }

    @Override
    public String[] consumes() {
        return expandedMediaTypes(delegate.consumes());
    }

    @Override
    public String[] produces() {
        return expandedMediaTypes(delegate.produces());
    }

    // delegate other methods
}