Fix Protobuf support - HTTP headers already written

Prior to this commit, the `ProtobufHttpMessageConverter` would call
`outputMessage.getBody()` at the beginning of the `writeInternal`
method, thus writing HTTP headers. Since this method is trying to write
"x-protobuf" headers after that, protobuf support wasn't working
properly for the default "x-protobuf" media type.

Thanks Toshiaki Maki for the repro project!

Also fixed:
* improve `MockHttpOutputMessage` behavior to reproduce the read-only
state of HTTP headers once they've been written.

Issue: SPR-12229
master
Brian Clozel 10 years ago
parent 2989fe4203
commit b9348bb89c
  1. 11
      spring-web/src/main/java/org/springframework/http/converter/protobuf/ProtobufHttpMessageConverter.java
  2. 8
      spring-web/src/test/java/org/springframework/http/MockHttpOutputMessage.java

@ -164,19 +164,18 @@ public class ProtobufHttpMessageConverter extends AbstractHttpMessageConverter<M
MediaType contentType = outputMessage.getHeaders().getContentType(); MediaType contentType = outputMessage.getHeaders().getContentType();
Charset charset = getCharset(contentType); Charset charset = getCharset(contentType);
OutputStreamWriter writer = new OutputStreamWriter(outputMessage.getBody(), charset);
if (MediaType.TEXT_HTML.isCompatibleWith(contentType)) { if (MediaType.TEXT_HTML.isCompatibleWith(contentType)) {
HtmlFormat.print(message, writer); HtmlFormat.print(message, new OutputStreamWriter(outputMessage.getBody(), charset));
} }
else if (MediaType.APPLICATION_JSON.isCompatibleWith(contentType)) { else if (MediaType.APPLICATION_JSON.isCompatibleWith(contentType)) {
JsonFormat.print(message, writer); JsonFormat.print(message, new OutputStreamWriter(outputMessage.getBody(), charset));
} }
else if (MediaType.TEXT_PLAIN.isCompatibleWith(contentType)) { else if (MediaType.TEXT_PLAIN.isCompatibleWith(contentType)) {
TextFormat.print(message, writer); TextFormat.print(message, new OutputStreamWriter(outputMessage.getBody(), charset));
} }
else if (MediaType.APPLICATION_XML.isCompatibleWith(contentType)) { else if (MediaType.APPLICATION_XML.isCompatibleWith(contentType)) {
XmlFormat.print(message, writer); XmlFormat.print(message, new OutputStreamWriter(outputMessage.getBody(), charset));
} }
else if (PROTOBUF.isCompatibleWith(contentType)) { else if (PROTOBUF.isCompatibleWith(contentType)) {
setProtoHeader(outputMessage, message); setProtoHeader(outputMessage, message);
@ -191,6 +190,8 @@ public class ProtobufHttpMessageConverter extends AbstractHttpMessageConverter<M
/** /**
* Set the "X-Protobuf-*" HTTP headers when responding with a message of * Set the "X-Protobuf-*" HTTP headers when responding with a message of
* content type "application/x-protobuf" * content type "application/x-protobuf"
* <p><b>Note:</b> <code>outputMessage.getBody()</code> should not have been called
* before because it writes HTTP headers (making them read only).</p>
*/ */
private void setProtoHeader(HttpOutputMessage response, Message message) { private void setProtoHeader(HttpOutputMessage response, Message message) {
response.getHeaders().set(X_PROTOBUF_SCHEMA_HEADER, message.getDescriptorForType().getFile().getName()); response.getHeaders().set(X_PROTOBUF_SCHEMA_HEADER, message.getDescriptorForType().getFile().getName());

@ -1,5 +1,5 @@
/* /*
* Copyright 2002-2013 the original author or authors. * Copyright 2002-2014 the original author or authors.
* *
* Licensed under the Apache License, Version 2.0 (the "License"); * Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License. * you may not use this file except in compliance with the License.
@ -32,17 +32,21 @@ public class MockHttpOutputMessage implements HttpOutputMessage {
private final ByteArrayOutputStream body = spy(new ByteArrayOutputStream()); private final ByteArrayOutputStream body = spy(new ByteArrayOutputStream());
private boolean headersWritten = false;
@Override @Override
public HttpHeaders getHeaders() { public HttpHeaders getHeaders() {
return headers; return (this.headersWritten ? HttpHeaders.readOnlyHttpHeaders(this.headers) : this.headers);
} }
@Override @Override
public OutputStream getBody() throws IOException { public OutputStream getBody() throws IOException {
this.headersWritten = true;
return body; return body;
} }
public byte[] getBodyAsBytes() { public byte[] getBodyAsBytes() {
this.headersWritten = true;
return body.toByteArray(); return body.toByteArray();
} }

Loading…
Cancel
Save