Skip to content

Commit c3e3df5

Browse files
committed
Polish
1 parent 3009e29 commit c3e3df5

File tree

2 files changed

+23
-22
lines changed

2 files changed

+23
-22
lines changed

spring-web/src/main/java/org/springframework/http/codec/json/AbstractJackson2Decoder.java

Lines changed: 9 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -45,21 +45,24 @@
4545
import org.springframework.util.MimeType;
4646

4747
/**
48-
* Base class providing support methods for Jackson 2.9 decoding.
48+
* Abstract base class for Jackson JSON 2.9 decoding.
4949
*
5050
* @author Sebastien Deleuze
5151
* @author Rossen Stoyanchev
52+
* @author Arjen Poutsma
5253
* @since 5.0
5354
*/
5455
public abstract class AbstractJackson2Decoder extends Jackson2CodecSupport implements HttpMessageDecoder<Object> {
5556

57+
5658
/**
5759
* Constructor with a Jackson {@link ObjectMapper} to use.
5860
*/
5961
protected AbstractJackson2Decoder(ObjectMapper mapper, MimeType... mimeTypes) {
6062
super(mapper, mimeTypes);
6163
}
6264

65+
6366
@Override
6467
public boolean canDecode(ResolvableType elementType, @Nullable MimeType mimeType) {
6568
JavaType javaType = objectMapper().getTypeFactory().constructType(elementType.getType());
@@ -87,22 +90,17 @@ public Mono<Object> decodeToMono(Publisher<DataBuffer> input, ResolvableType ele
8790
private Flux<TokenBuffer> tokenize(Publisher<DataBuffer> input, boolean tokenizeArrayElements) {
8891
try {
8992
JsonFactory factory = objectMapper().getFactory();
90-
JsonParser nonBlockingParser = factory.createNonBlockingByteArrayParser();
91-
Jackson2Tokenizer tokenizer = new Jackson2Tokenizer(nonBlockingParser,
92-
tokenizeArrayElements);
93-
return Flux.from(input)
94-
.flatMap(tokenizer)
95-
.doFinally(t -> tokenizer.endOfInput());
93+
JsonParser parser = factory.createNonBlockingByteArrayParser();
94+
Jackson2Tokenizer tokenizer = new Jackson2Tokenizer(parser, tokenizeArrayElements);
95+
return Flux.from(input).flatMap(tokenizer).doFinally(t -> tokenizer.endOfInput());
9696
}
9797
catch (IOException ex) {
9898
return Flux.error(new UncheckedIOException(ex));
9999
}
100-
101100
}
102101

103-
private Flux<Object> decodeInternal(Flux<TokenBuffer> tokens,
104-
ResolvableType elementType, @Nullable MimeType mimeType,
105-
@Nullable Map<String, Object> hints) {
102+
private Flux<Object> decodeInternal(Flux<TokenBuffer> tokens, ResolvableType elementType,
103+
@Nullable MimeType mimeType, @Nullable Map<String, Object> hints) {
106104

107105
Assert.notNull(tokens, "'tokens' must not be null");
108106
Assert.notNull(elementType, "'elementType' must not be null");

spring-web/src/main/java/org/springframework/http/codec/json/Jackson2Tokenizer.java

Lines changed: 14 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -34,8 +34,9 @@
3434
import org.springframework.util.Assert;
3535

3636
/**
37-
* Function that transforms an arbitrary split byte stream representing JSON objects into a
38-
* {@code Flux<TokenBuffer>}, where each token buffer is a well-formed JSON object.
37+
* {@link Function} to transform a JSON stream of arbitrary size, byte array
38+
* chunks into a {@code Flux<TokenBuffer>} where each token buffer is a
39+
* well-formed JSON object.
3940
*
4041
* @author Arjen Poutsma
4142
* @since 5.0
@@ -53,14 +54,16 @@ class Jackson2Tokenizer implements Function<DataBuffer, Flux<TokenBuffer>> {
5354
private int arrayDepth;
5455

5556
// TODO: change to ByteBufferFeeder when supported by Jackson
56-
private ByteArrayFeeder inputFeeder;
57+
private final ByteArrayFeeder inputFeeder;
58+
5759

5860
/**
5961
* Create a new instance of the {@code Jackson2Tokenizer}.
6062
* @param parser the non-blocking parser, obtained via
6163
* {@link com.fasterxml.jackson.core.JsonFactory#createNonBlockingByteArrayParser}
62-
* @param tokenizeArrayElements if {@code true} and the "top level" JSON object is an array,
63-
* each of its elements is returned individually and immediately after it was fully received
64+
* @param tokenizeArrayElements if {@code true} and the "top level" JSON
65+
* object is an array, each element is returned individually, immediately
66+
* after it is received.
6467
*/
6568
public Jackson2Tokenizer(JsonParser parser, boolean tokenizeArrayElements) {
6669
Assert.notNull(parser, "'parser' must not be null");
@@ -71,6 +74,7 @@ public Jackson2Tokenizer(JsonParser parser, boolean tokenizeArrayElements) {
7174
this.inputFeeder = (ByteArrayFeeder) this.parser.getNonBlockingInputFeeder();
7275
}
7376

77+
7478
@Override
7579
public Flux<TokenBuffer> apply(DataBuffer dataBuffer) {
7680
byte[] bytes = new byte[dataBuffer.readableByteCount()];
@@ -86,7 +90,7 @@ public Flux<TokenBuffer> apply(DataBuffer dataBuffer) {
8690
if (token == JsonToken.NOT_AVAILABLE) {
8791
break;
8892
}
89-
calculateDepth(token);
93+
updateDepth(token);
9094

9195
if (!this.tokenizeArrayElements) {
9296
processTokenNormal(token, result);
@@ -106,11 +110,7 @@ public Flux<TokenBuffer> apply(DataBuffer dataBuffer) {
106110
}
107111
}
108112

109-
public void endOfInput() {
110-
this.inputFeeder.endOfInput();
111-
}
112-
113-
private void calculateDepth(JsonToken token) {
113+
private void updateDepth(JsonToken token) {
114114
switch (token) {
115115
case START_OBJECT:
116116
this.objectDepth++;
@@ -149,7 +149,10 @@ private void processTokenArray(JsonToken token, List<TokenBuffer> result) throws
149149
result.add(this.tokenBuffer);
150150
this.tokenBuffer = new TokenBuffer(this.parser);
151151
}
152+
}
152153

154+
public void endOfInput() {
155+
this.inputFeeder.endOfInput();
153156
}
154157

155158
}

0 commit comments

Comments
 (0)