Support Protobuf serialization in WebFlux
This commit introduces Protobuf support in WebFlux via dedicated codecs. Flux<Message> are serialized/deserialized using delimited Protobuf messages with the size of each message specified before the message itself. In that case, a "delimited=true" parameter is added to the content type. Mono<Message> are expected to use regular Protobuf message format (without the size prepended before the message). Related HttpMessageReader/Writer are automatically registered when the "com.google.protobuf:protobuf-java" library is detected in the classpath, and can be customized easily if needed via CodecConfigurer, for example to specify protocol extensions via the ExtensionRegistry based constructors. Both "application/x-protobuf" and "application/octet-stream" mime types are supported. Issue: SPR-15776
This commit is contained in:
committed by
Sebastien Deleuze
parent
4475c67ba8
commit
36a07aa897
@@ -110,6 +110,22 @@ public interface CodecConfigurer {
|
||||
*/
|
||||
void jackson2JsonEncoder(Encoder<?> encoder);
|
||||
|
||||
/**
|
||||
* Override the default Protobuf {@code Decoder}.
|
||||
* @param decoder the decoder instance to use
|
||||
* @since 5.1
|
||||
* @see org.springframework.http.codec.protobuf.ProtobufDecoder
|
||||
*/
|
||||
void protobufDecoder(Decoder<?> decoder);
|
||||
|
||||
/**
|
||||
* Override the default Protobuf {@code HttpMessageReader}.
|
||||
* @param decoder the decoder instance to use
|
||||
* @since 5.1
|
||||
* @see org.springframework.http.codec.protobuf.ProtobufHttpMessageWriter
|
||||
*/
|
||||
void protobufWriter(HttpMessageWriter<?> decoder);
|
||||
|
||||
/**
|
||||
* Whether to log form data at DEBUG level, and headers at TRACE level.
|
||||
* Both may contain sensitive information.
|
||||
|
||||
@@ -160,7 +160,8 @@ public class EncoderHttpMessageWriter<T> implements HttpMessageWriter<T> {
|
||||
private boolean isStreamingMediaType(@Nullable MediaType contentType) {
|
||||
return (contentType != null && this.encoder instanceof HttpMessageEncoder &&
|
||||
((HttpMessageEncoder<?>) this.encoder).getStreamingMediaTypes().stream()
|
||||
.anyMatch(contentType::isCompatibleWith));
|
||||
.anyMatch(streamingMediaType -> contentType.isCompatibleWith(streamingMediaType) &&
|
||||
contentType.getParameters().entrySet().containsAll(streamingMediaType.getParameters().keySet())));
|
||||
}
|
||||
|
||||
|
||||
|
||||
@@ -0,0 +1,51 @@
|
||||
/*
|
||||
* Copyright 2002-2018 the original author or authors.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package org.springframework.http.codec.protobuf;
|
||||
|
||||
import java.util.Arrays;
|
||||
import java.util.Collections;
|
||||
import java.util.List;
|
||||
|
||||
import org.springframework.lang.Nullable;
|
||||
import org.springframework.util.MimeType;
|
||||
|
||||
/**
|
||||
* Base class providing support methods for Protobuf encoding and decoding.
|
||||
*
|
||||
* @author Sebastien Deleuze
|
||||
* @since 5.1
|
||||
*/
|
||||
public abstract class ProtobufCodecSupport {
|
||||
|
||||
static final List<MimeType> MIME_TYPES = Collections.unmodifiableList(
|
||||
Arrays.asList(
|
||||
new MimeType("application", "x-protobuf"),
|
||||
new MimeType("application", "octet-stream")));
|
||||
|
||||
static final String DELIMITED_KEY = "delimited";
|
||||
|
||||
static final String DELIMITED_VALUE = "true";
|
||||
|
||||
|
||||
protected boolean supportsMimeType(@Nullable MimeType mimeType) {
|
||||
return (mimeType == null || MIME_TYPES.stream().anyMatch(m -> m.isCompatibleWith(mimeType)));
|
||||
}
|
||||
|
||||
protected List<MimeType> getMimeTypes() {
|
||||
return MIME_TYPES;
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,212 @@
|
||||
/*
|
||||
* Copyright 2002-2018 the original author or authors.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package org.springframework.http.codec.protobuf;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.lang.reflect.Method;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.concurrent.ConcurrentHashMap;
|
||||
import java.util.function.Function;
|
||||
|
||||
import com.google.protobuf.CodedInputStream;
|
||||
import com.google.protobuf.ExtensionRegistry;
|
||||
import com.google.protobuf.Message;
|
||||
import org.reactivestreams.Publisher;
|
||||
import reactor.core.publisher.Flux;
|
||||
import reactor.core.publisher.Mono;
|
||||
|
||||
import org.springframework.core.ResolvableType;
|
||||
import org.springframework.core.codec.Decoder;
|
||||
import org.springframework.core.codec.DecodingException;
|
||||
import org.springframework.core.io.buffer.DataBuffer;
|
||||
import org.springframework.core.io.buffer.DataBufferUtils;
|
||||
import org.springframework.util.Assert;
|
||||
import org.springframework.util.MimeType;
|
||||
|
||||
/**
|
||||
* A {@code Decoder} that reads {@link com.google.protobuf.Message}s
|
||||
* using <a href="https://developers.google.com/protocol-buffers/">Google Protocol Buffers</a>.
|
||||
*
|
||||
* Flux deserialized via
|
||||
* {@link #decode(Publisher, ResolvableType, MimeType, Map)} are expected to use
|
||||
* <a href="https://developers.google.com/protocol-buffers/docs/techniques?hl=en#streaming">delimited Protobuf messages</a>
|
||||
* with the size of each message specified before the message itself. Single values deserialized
|
||||
* via {@link #decodeToMono(Publisher, ResolvableType, MimeType, Map)} are expected to use
|
||||
* regular Protobuf message format (without the size prepended before the message).
|
||||
*
|
||||
* <p>To generate {@code Message} Java classes, you need to install the {@code protoc} binary.
|
||||
*
|
||||
* <p>This decoder requires Protobuf 3 or higher, and supports
|
||||
* {@code "application/x-protobuf"} and {@code "application/octet-stream"} with the official
|
||||
* {@code "com.google.protobuf:protobuf-java"} library.
|
||||
*
|
||||
* @author Sébastien Deleuze
|
||||
* @since 5.1
|
||||
* @see ProtobufEncoder
|
||||
*/
|
||||
public class ProtobufDecoder extends ProtobufCodecSupport implements Decoder<Message> {
|
||||
|
||||
/**
|
||||
* The default max size for aggregating messages.
|
||||
*/
|
||||
protected static final int DEFAULT_MESSAGE_MAX_SIZE = 64 * 1024;
|
||||
|
||||
private static final ConcurrentHashMap<Class<?>, Method> methodCache = new ConcurrentHashMap<>();
|
||||
|
||||
private final ExtensionRegistry extensionRegistry;
|
||||
|
||||
private int maxMessageSize = DEFAULT_MESSAGE_MAX_SIZE;
|
||||
|
||||
|
||||
/**
|
||||
* Construct a new {@code ProtobufDecoder}.
|
||||
*/
|
||||
public ProtobufDecoder() {
|
||||
this(ExtensionRegistry.newInstance());
|
||||
}
|
||||
|
||||
/**
|
||||
* Construct a new {@code ProtobufDecoder} with an initializer that allows the
|
||||
* registration of message extensions.
|
||||
* @param extensionRegistry a message extension registry
|
||||
*/
|
||||
public ProtobufDecoder(ExtensionRegistry extensionRegistry) {
|
||||
Assert.notNull(extensionRegistry, "ExtensionRegistry must not be null");
|
||||
this.extensionRegistry = extensionRegistry;
|
||||
}
|
||||
|
||||
public void setMaxMessageSize(int maxMessageSize) {
|
||||
this.maxMessageSize = maxMessageSize;
|
||||
}
|
||||
|
||||
@Override
|
||||
public boolean canDecode(ResolvableType elementType, MimeType mimeType) {
|
||||
return Message.class.isAssignableFrom(elementType.getRawClass()) && supportsMimeType(mimeType);
|
||||
}
|
||||
|
||||
@Override
|
||||
public Flux<Message> decode(Publisher<DataBuffer> inputStream, ResolvableType elementType,
|
||||
MimeType mimeType, Map<String, Object> hints) {
|
||||
|
||||
return Flux.from(inputStream)
|
||||
.concatMap(new MessageDecoderFunction(elementType, this.maxMessageSize));
|
||||
}
|
||||
|
||||
@Override
|
||||
public Mono<Message> decodeToMono(Publisher<DataBuffer> inputStream, ResolvableType elementType,
|
||||
MimeType mimeType, Map<String, Object> hints) {
|
||||
return DataBufferUtils.join(inputStream).map(dataBuffer -> {
|
||||
try {
|
||||
Message.Builder builder = getMessageBuilder(elementType.getRawClass());
|
||||
builder.mergeFrom(CodedInputStream.newInstance(dataBuffer.asByteBuffer()), this.extensionRegistry);
|
||||
Message message = builder.build();
|
||||
DataBufferUtils.release(dataBuffer);
|
||||
return message;
|
||||
}
|
||||
catch (IOException ex) {
|
||||
throw new DecodingException("I/O error while parsing input stream", ex);
|
||||
}
|
||||
catch (Exception ex) {
|
||||
throw new DecodingException("Could not read Protobuf message: " + ex.getMessage(), ex);
|
||||
}
|
||||
}
|
||||
);
|
||||
}
|
||||
|
||||
/**
|
||||
* Create a new {@code Message.Builder} instance for the given class.
|
||||
* <p>This method uses a ConcurrentHashMap for caching method lookups.
|
||||
*/
|
||||
private static Message.Builder getMessageBuilder(Class<?> clazz) throws Exception {
|
||||
Method method = methodCache.get(clazz);
|
||||
if (method == null) {
|
||||
method = clazz.getMethod("newBuilder");
|
||||
methodCache.put(clazz, method);
|
||||
}
|
||||
return (Message.Builder) method.invoke(clazz);
|
||||
}
|
||||
|
||||
@Override
|
||||
public List<MimeType> getDecodableMimeTypes() {
|
||||
return getMimeTypes();
|
||||
}
|
||||
|
||||
|
||||
private class MessageDecoderFunction implements Function<DataBuffer, Publisher<? extends Message>> {
|
||||
|
||||
private final ResolvableType elementType;
|
||||
|
||||
private final int maxMessageSize;
|
||||
|
||||
private DataBuffer output;
|
||||
|
||||
private int messageBytesToRead;
|
||||
|
||||
public MessageDecoderFunction(ResolvableType elementType, int maxMessageSize) {
|
||||
this.elementType = elementType;
|
||||
this.maxMessageSize = maxMessageSize;
|
||||
}
|
||||
|
||||
// TODO Instead of the recursive call, loop over the current DataBuffer, produce a list of as many messages as are contained, and save any remaining bytes with flatMapIterable
|
||||
@Override
|
||||
public Publisher<? extends Message> apply(DataBuffer input) {
|
||||
|
||||
try {
|
||||
if (this.output == null) {
|
||||
int firstByte = input.read();
|
||||
if (firstByte == -1) {
|
||||
return Flux.error(new DecodingException("Can't parse message size"));
|
||||
}
|
||||
this.messageBytesToRead = CodedInputStream.readRawVarint32(firstByte, input.asInputStream());
|
||||
if (this.messageBytesToRead > this.maxMessageSize) {
|
||||
return Flux.error(new DecodingException(
|
||||
"The number of bytes to read parsed in the incoming stream (" +
|
||||
this.messageBytesToRead + ") exceeds the configured limit (" + this.maxMessageSize + ")"));
|
||||
}
|
||||
this.output = input.factory().allocateBuffer(this.messageBytesToRead);
|
||||
}
|
||||
int chunkBytesToRead = this.messageBytesToRead >= input.readableByteCount() ?
|
||||
input.readableByteCount() : this.messageBytesToRead;
|
||||
int remainingBytesToRead = input.readableByteCount() - chunkBytesToRead;
|
||||
this.output.write(input.slice(input.readPosition(), chunkBytesToRead));
|
||||
this.messageBytesToRead -= chunkBytesToRead;
|
||||
Message message = null;
|
||||
if (this.messageBytesToRead == 0) {
|
||||
Message.Builder builder = getMessageBuilder(this.elementType.getRawClass());
|
||||
builder.mergeFrom(CodedInputStream.newInstance(this.output.asByteBuffer()), extensionRegistry);
|
||||
message = builder.build();
|
||||
DataBufferUtils.release(this.output);
|
||||
this.output = null;
|
||||
}
|
||||
if (remainingBytesToRead > 0) {
|
||||
return Mono.justOrEmpty(message).concatWith(
|
||||
apply(input.slice(input.readPosition() + chunkBytesToRead, remainingBytesToRead)));
|
||||
}
|
||||
else {
|
||||
return Mono.justOrEmpty(message);
|
||||
}
|
||||
}
|
||||
catch (IOException ex) {
|
||||
return Flux.error(new DecodingException("I/O error while parsing input stream", ex));
|
||||
}
|
||||
catch (Exception ex) {
|
||||
return Flux.error(new DecodingException("Could not read Protobuf message: " + ex.getMessage(), ex));
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,103 @@
|
||||
/*
|
||||
* Copyright 2002-2018 the original author or authors.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package org.springframework.http.codec.protobuf;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.io.OutputStream;
|
||||
import java.util.Collections;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.stream.Collectors;
|
||||
|
||||
import com.google.protobuf.Message;
|
||||
import org.reactivestreams.Publisher;
|
||||
import reactor.core.publisher.Flux;
|
||||
import reactor.core.publisher.Mono;
|
||||
|
||||
import org.springframework.core.ResolvableType;
|
||||
import org.springframework.core.io.buffer.DataBuffer;
|
||||
import org.springframework.core.io.buffer.DataBufferFactory;
|
||||
import org.springframework.http.MediaType;
|
||||
import org.springframework.http.codec.HttpMessageEncoder;
|
||||
import org.springframework.util.MimeType;
|
||||
|
||||
/**
|
||||
* An {@code Encoder} that writes {@link com.google.protobuf.Message}s
|
||||
* using <a href="https://developers.google.com/protocol-buffers/">Google Protocol Buffers</a>.
|
||||
*
|
||||
* Flux are serialized using
|
||||
* <a href="https://developers.google.com/protocol-buffers/docs/techniques?hl=en#streaming">delimited Protobuf messages</a>
|
||||
* with the size of each message specified before the message itself. Single values are
|
||||
* serialized using regular Protobuf message format (without the size prepended before the message).
|
||||
*
|
||||
* <p>To generate {@code Message} Java classes, you need to install the {@code protoc} binary.
|
||||
*
|
||||
* <p>This encoder requires Protobuf 3 or higher, and supports
|
||||
* {@code "application/x-protobuf"} and {@code "application/octet-stream"} with the official
|
||||
* {@code "com.google.protobuf:protobuf-java"} library.
|
||||
*
|
||||
* @author Sébastien Deleuze
|
||||
* @since 5.1
|
||||
* @see ProtobufDecoder
|
||||
*/
|
||||
public class ProtobufEncoder extends ProtobufCodecSupport implements HttpMessageEncoder<Message> {
|
||||
|
||||
private static final List<MediaType> streamingMediaTypes = MIME_TYPES
|
||||
.stream()
|
||||
.map(mimeType -> new MediaType(mimeType.getType(), mimeType.getSubtype(), Collections.singletonMap(DELIMITED_KEY, DELIMITED_VALUE)))
|
||||
.collect(Collectors.toList());
|
||||
|
||||
@Override
|
||||
public boolean canEncode(ResolvableType elementType, MimeType mimeType) {
|
||||
return Message.class.isAssignableFrom(elementType.getRawClass()) && supportsMimeType(mimeType);
|
||||
}
|
||||
|
||||
@Override
|
||||
public Flux<DataBuffer> encode(Publisher<? extends Message> inputStream,
|
||||
DataBufferFactory bufferFactory, ResolvableType elementType, MimeType mimeType, Map<String, Object> hints) {
|
||||
return Flux
|
||||
.from(inputStream)
|
||||
.map(message -> encodeMessage(message, bufferFactory, !(inputStream instanceof Mono)));
|
||||
}
|
||||
|
||||
private DataBuffer encodeMessage(Message message, DataBufferFactory bufferFactory, boolean streaming) {
|
||||
DataBuffer buffer = bufferFactory.allocateBuffer();
|
||||
OutputStream outputStream = buffer.asOutputStream();
|
||||
try {
|
||||
if (streaming) {
|
||||
message.writeDelimitedTo(outputStream);
|
||||
}
|
||||
else {
|
||||
message.writeTo(outputStream);
|
||||
}
|
||||
return buffer;
|
||||
}
|
||||
catch (IOException ex) {
|
||||
throw new IllegalStateException("Unexpected I/O error while writing to data buffer", ex);
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public List<MediaType> getStreamingMediaTypes() {
|
||||
return streamingMediaTypes;
|
||||
}
|
||||
|
||||
@Override
|
||||
public List<MimeType> getEncodableMimeTypes() {
|
||||
return getMimeTypes();
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,108 @@
|
||||
/*
|
||||
* Copyright 2002-2018 the original author or authors.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package org.springframework.http.codec.protobuf;
|
||||
|
||||
import java.lang.reflect.Method;
|
||||
import java.util.HashMap;
|
||||
import java.util.Map;
|
||||
import java.util.concurrent.ConcurrentHashMap;
|
||||
|
||||
import com.google.protobuf.Descriptors;
|
||||
import com.google.protobuf.Message;
|
||||
import org.reactivestreams.Publisher;
|
||||
import reactor.core.publisher.Flux;
|
||||
import reactor.core.publisher.Mono;
|
||||
|
||||
import org.springframework.core.ResolvableType;
|
||||
import org.springframework.core.codec.DecodingException;
|
||||
import org.springframework.http.MediaType;
|
||||
import org.springframework.http.ReactiveHttpOutputMessage;
|
||||
import org.springframework.http.codec.EncoderHttpMessageWriter;
|
||||
import org.springframework.http.codec.HttpMessageEncoder;
|
||||
import org.springframework.lang.Nullable;
|
||||
|
||||
/**
|
||||
* {@code HttpMessageWriter} that can write a protobuf {@link Message} and adds
|
||||
* {@code X-Protobuf-Schema}, {@code X-Protobuf-Message} headers and a
|
||||
* {@code delimited=true} parameter is added to the content type if a flux is serialized.
|
||||
*
|
||||
* <p>For {@code HttpMessageReader}, just use
|
||||
* {@code new DecoderHttpMessageReader(new ProtobufDecoder())}.
|
||||
*
|
||||
* @author Sébastien Deleuze
|
||||
* @since 5.1
|
||||
* @see ProtobufEncoder
|
||||
*/
|
||||
public class ProtobufHttpMessageWriter extends EncoderHttpMessageWriter<Message> {
|
||||
|
||||
private static final ConcurrentHashMap<Class<?>, Method> methodCache = new ConcurrentHashMap<>();
|
||||
|
||||
private static final String X_PROTOBUF_SCHEMA_HEADER = "X-Protobuf-Schema";
|
||||
|
||||
private static final String X_PROTOBUF_MESSAGE_HEADER = "X-Protobuf-Message";
|
||||
|
||||
|
||||
public ProtobufHttpMessageWriter() {
|
||||
super(new ProtobufEncoder());
|
||||
}
|
||||
|
||||
public ProtobufHttpMessageWriter(ProtobufEncoder encoder) {
|
||||
super(encoder);
|
||||
}
|
||||
|
||||
|
||||
@SuppressWarnings("unchecked")
|
||||
@Override
|
||||
public Mono<Void> write(Publisher<? extends Message> inputStream, ResolvableType elementType,
|
||||
@Nullable MediaType mediaType, ReactiveHttpOutputMessage message, Map<String, Object> hints) {
|
||||
|
||||
try {
|
||||
Message.Builder builder = getMessageBuilder(elementType.getRawClass());
|
||||
Descriptors.Descriptor descriptor = builder.getDescriptorForType();
|
||||
message.getHeaders().add(X_PROTOBUF_SCHEMA_HEADER, descriptor.getFile().getName());
|
||||
message.getHeaders().add(X_PROTOBUF_MESSAGE_HEADER, descriptor.getFullName());
|
||||
if (inputStream instanceof Flux) {
|
||||
if (mediaType == null) {
|
||||
message.getHeaders().setContentType(((HttpMessageEncoder<?>)getEncoder()).getStreamingMediaTypes().get(0));
|
||||
}
|
||||
else if (!ProtobufEncoder.DELIMITED_VALUE.equals(mediaType.getParameters().get(ProtobufEncoder.DELIMITED_KEY))) {
|
||||
Map<String, String> parameters = new HashMap<>(mediaType.getParameters());
|
||||
parameters.put(ProtobufEncoder.DELIMITED_KEY, ProtobufEncoder.DELIMITED_VALUE);
|
||||
message.getHeaders().setContentType(new MediaType(mediaType.getType(), mediaType.getSubtype(), parameters));
|
||||
}
|
||||
}
|
||||
return super.write(inputStream, elementType, mediaType, message, hints);
|
||||
}
|
||||
catch (Exception ex) {
|
||||
return Mono.error(new DecodingException("Could not read Protobuf message: " + ex.getMessage(), ex));
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Create a new {@code Message.Builder} instance for the given class.
|
||||
* <p>This method uses a ConcurrentHashMap for caching method lookups.
|
||||
*/
|
||||
private static Message.Builder getMessageBuilder(Class<?> clazz) throws Exception {
|
||||
Method method = methodCache.get(clazz);
|
||||
if (method == null) {
|
||||
method = clazz.getMethod("newBuilder");
|
||||
methodCache.put(clazz, method);
|
||||
}
|
||||
return (Message.Builder) method.invoke(clazz);
|
||||
}
|
||||
|
||||
}
|
||||
@@ -41,6 +41,8 @@ import org.springframework.http.codec.json.Jackson2JsonDecoder;
|
||||
import org.springframework.http.codec.json.Jackson2JsonEncoder;
|
||||
import org.springframework.http.codec.json.Jackson2SmileDecoder;
|
||||
import org.springframework.http.codec.json.Jackson2SmileEncoder;
|
||||
import org.springframework.http.codec.protobuf.ProtobufDecoder;
|
||||
import org.springframework.http.codec.protobuf.ProtobufHttpMessageWriter;
|
||||
import org.springframework.http.codec.xml.Jaxb2XmlDecoder;
|
||||
import org.springframework.http.codec.xml.Jaxb2XmlEncoder;
|
||||
import org.springframework.lang.Nullable;
|
||||
@@ -54,26 +56,37 @@ import org.springframework.util.ClassUtils;
|
||||
*/
|
||||
class BaseDefaultCodecs implements CodecConfigurer.DefaultCodecs {
|
||||
|
||||
private static final ClassLoader classLoader = BaseCodecConfigurer.class.getClassLoader();
|
||||
|
||||
static final boolean jackson2Present =
|
||||
ClassUtils.isPresent("com.fasterxml.jackson.databind.ObjectMapper",
|
||||
BaseCodecConfigurer.class.getClassLoader()) &&
|
||||
ClassUtils.isPresent("com.fasterxml.jackson.core.JsonGenerator",
|
||||
BaseCodecConfigurer.class.getClassLoader());
|
||||
classLoader);
|
||||
|
||||
private static final boolean jackson2SmilePresent =
|
||||
ClassUtils.isPresent("com.fasterxml.jackson.dataformat.smile.SmileFactory",
|
||||
BaseCodecConfigurer.class.getClassLoader());
|
||||
classLoader);
|
||||
|
||||
private static final boolean jaxb2Present =
|
||||
ClassUtils.isPresent("javax.xml.bind.Binder", BaseCodecConfigurer.class.getClassLoader());
|
||||
ClassUtils.isPresent("javax.xml.bind.Binder", classLoader);
|
||||
|
||||
private static final boolean protobufPresent =
|
||||
ClassUtils.isPresent("com.google.protobuf.Message", classLoader);
|
||||
|
||||
|
||||
@Nullable
|
||||
private Decoder<?> jackson2JsonDecoder;
|
||||
|
||||
@Nullable
|
||||
private Decoder<?> protobufDecoder;
|
||||
|
||||
@Nullable
|
||||
private Encoder<?> jackson2JsonEncoder;
|
||||
|
||||
@Nullable
|
||||
private HttpMessageWriter<?> protobufWriter;
|
||||
|
||||
private boolean enableLoggingRequestDetails = false;
|
||||
|
||||
private boolean registerDefaults = true;
|
||||
@@ -89,6 +102,16 @@ class BaseDefaultCodecs implements CodecConfigurer.DefaultCodecs {
|
||||
this.jackson2JsonEncoder = encoder;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void protobufDecoder(Decoder<?> decoder) {
|
||||
this.protobufDecoder = decoder;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void protobufWriter(HttpMessageWriter<?> writer) {
|
||||
this.protobufWriter = writer;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void enableLoggingRequestDetails(boolean enable) {
|
||||
this.enableLoggingRequestDetails = enable;
|
||||
@@ -119,6 +142,9 @@ class BaseDefaultCodecs implements CodecConfigurer.DefaultCodecs {
|
||||
readers.add(new DecoderHttpMessageReader<>(new DataBufferDecoder()));
|
||||
readers.add(new DecoderHttpMessageReader<>(new ResourceDecoder()));
|
||||
readers.add(new DecoderHttpMessageReader<>(StringDecoder.textPlainOnly()));
|
||||
if (protobufPresent) {
|
||||
readers.add(new DecoderHttpMessageReader<>(getProtobufDecoder()));
|
||||
}
|
||||
|
||||
FormHttpMessageReader formReader = new FormHttpMessageReader();
|
||||
formReader.setEnableLoggingRequestDetails(this.enableLoggingRequestDetails);
|
||||
@@ -194,6 +220,9 @@ class BaseDefaultCodecs implements CodecConfigurer.DefaultCodecs {
|
||||
if (!forMultipart) {
|
||||
extendTypedWriters(writers);
|
||||
}
|
||||
if (protobufPresent) {
|
||||
writers.add(getProtobufWriter());
|
||||
}
|
||||
return writers;
|
||||
}
|
||||
|
||||
@@ -255,9 +284,17 @@ class BaseDefaultCodecs implements CodecConfigurer.DefaultCodecs {
|
||||
return (this.jackson2JsonDecoder != null ? this.jackson2JsonDecoder : new Jackson2JsonDecoder());
|
||||
}
|
||||
|
||||
protected Decoder<?> getProtobufDecoder() {
|
||||
return (this.protobufDecoder != null ? this.protobufDecoder : new ProtobufDecoder());
|
||||
}
|
||||
|
||||
protected Encoder<?> getJackson2JsonEncoder() {
|
||||
return (this.jackson2JsonEncoder != null ? this.jackson2JsonEncoder : new Jackson2JsonEncoder());
|
||||
}
|
||||
|
||||
protected HttpMessageWriter<?> getProtobufWriter() {
|
||||
return (this.protobufWriter != null ? this.protobufWriter : new ProtobufHttpMessageWriter());
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user