flasn4nt
/
grpc-java
mirror of https://github.com/grpc/grpc-java.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Avoid flushing headers when the server returns a single response (#9314) 

* core,netty: avoid flushing the headers on the server side when the server has a single response.
This commit is contained in:
amirhadadi 2023-10-09 19:43:10 +03:00 committed by GitHub
parent aec9a56eaf
commit 8c0545564f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
16 changed files with 43 additions and 43 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
binder/src/main/java/io/grpc/binder/internal/MultiMessageServerStream.java
View File

@ -77,7 +77,7 @@ final class MultiMessageServerStream implements ServerStream {
} }
@Override @Override
public void writeHeaders(Metadata headers) { public void writeHeaders(Metadata headers, boolean flush) {
try { try {
synchronized (outbound) { synchronized (outbound) {
outbound.sendHeaders(headers); outbound.sendHeaders(headers);

2
binder/src/main/java/io/grpc/binder/internal/SingleMessageServerStream.java
View File

@ -80,7 +80,7 @@ final class SingleMessageServerStream implements ServerStream {
} }
@Override @Override
public void writeHeaders(Metadata headers) { public void writeHeaders(Metadata headers, boolean flush) {
pendingHeaders = headers; pendingHeaders = headers;
} }

6
core/src/main/java/io/grpc/internal/AbstractServerStream.java
View File

@ -41,7 +41,7 @@ public abstract class AbstractServerStream extends AbstractStream
* *
* @param headers the headers to be sent to client. * @param headers the headers to be sent to client.
*/ */
void writeHeaders(Metadata headers); void writeHeaders(Metadata headers, boolean flush);
/** /**
* Sends an outbound frame to the remote end point. * Sends an outbound frame to the remote end point.
@ -96,11 +96,11 @@ public abstract class AbstractServerStream extends AbstractStream
} }
@Override @Override
public final void writeHeaders(Metadata headers) { public final void writeHeaders(Metadata headers, boolean flush) {
Preconditions.checkNotNull(headers, "headers"); Preconditions.checkNotNull(headers, "headers");
headersSent = true; headersSent = true;
abstractServerStreamSink().writeHeaders(headers); abstractServerStreamSink().writeHeaders(headers, flush);
} }
@Override @Override

2
core/src/main/java/io/grpc/internal/ServerCallImpl.java
View File

@ -142,7 +142,7 @@ final class ServerCallImpl<ReqT, RespT> extends ServerCall<ReqT, RespT> {
// Don't check if sendMessage has been called, since it requires that sendHeaders was already // Don't check if sendMessage has been called, since it requires that sendHeaders was already
// called. // called.
sendHeadersCalled = true; sendHeadersCalled = true;
stream.writeHeaders(headers); stream.writeHeaders(headers, !getMethodDescriptor().getType().serverSendsOneMessage());
} }
@Override @Override

2
core/src/main/java/io/grpc/internal/ServerStream.java
View File

@ -36,7 +36,7 @@ public interface ServerStream extends Stream {
* *
* @param headers to send to client. * @param headers to send to client.
*/ */
void writeHeaders(Metadata headers); void writeHeaders(Metadata headers, boolean flush);
/** /**
* Closes the stream for both reading and writing. A status code of * Closes the stream for both reading and writing. A status code of

16
core/src/test/java/io/grpc/internal/AbstractServerStreamTest.java
View File

@ -285,28 +285,28 @@ public class AbstractServerStreamTest {
public void writeHeaders_failsOnNullHeaders() { public void writeHeaders_failsOnNullHeaders() {
thrown.expect(NullPointerException.class); thrown.expect(NullPointerException.class);
stream.writeHeaders(null); stream.writeHeaders(null, true);
} }
@Test @Test
public void writeHeaders() { public void writeHeaders() {
Metadata headers = new Metadata(); Metadata headers = new Metadata();
stream.writeHeaders(headers); stream.writeHeaders(headers, true);
verify(sink).writeHeaders(same(headers)); verify(sink).writeHeaders(same(headers), eq(true));
} }
@Test @Test
public void writeMessage_dontWriteDuplicateHeaders() { public void writeMessage_dontWriteDuplicateHeaders() {
stream.writeHeaders(new Metadata()); stream.writeHeaders(new Metadata(), true);
stream.writeMessage(new ByteArrayInputStream(new byte[]{})); stream.writeMessage(new ByteArrayInputStream(new byte[]{}));
// Make sure it wasn't called twice // Make sure it wasn't called twice
verify(sink).writeHeaders(any(Metadata.class)); verify(sink).writeHeaders(any(Metadata.class), eq(true));
} }
@Test @Test
public void writeMessage_ignoreIfFramerClosed() { public void writeMessage_ignoreIfFramerClosed() {
stream.writeHeaders(new Metadata()); stream.writeHeaders(new Metadata(), true);
stream.endOfMessages(); stream.endOfMessages();
reset(sink); reset(sink);
@ -317,7 +317,7 @@ public class AbstractServerStreamTest {
@Test @Test
public void writeMessage() { public void writeMessage() {
stream.writeHeaders(new Metadata()); stream.writeHeaders(new Metadata(), true);
stream.writeMessage(new ByteArrayInputStream(new byte[]{})); stream.writeMessage(new ByteArrayInputStream(new byte[]{}));
stream.flush(); stream.flush();
@ -327,7 +327,7 @@ public class AbstractServerStreamTest {
@Test @Test
public void writeMessage_closesStream() throws Exception { public void writeMessage_closesStream() throws Exception {
stream.writeHeaders(new Metadata()); stream.writeHeaders(new Metadata(), true);
InputStream input = mock(InputStream.class, delegatesTo(new ByteArrayInputStream(new byte[1]))); InputStream input = mock(InputStream.class, delegatesTo(new ByteArrayInputStream(new byte[1])));
stream.writeMessage(input); stream.writeMessage(input);
verify(input).close(); verify(input).close();

4
core/src/test/java/io/grpc/internal/ServerCallImplTest.java
View File

@ -153,7 +153,7 @@ public class ServerCallImplTest {
call.sendHeaders(headers); call.sendHeaders(headers);
verify(stream).writeHeaders(headers); verify(stream).writeHeaders(headers, false);
} }
@Test @Test
@ -162,7 +162,7 @@ public class ServerCallImplTest {
headers.put(CONTENT_LENGTH_KEY, "123"); headers.put(CONTENT_LENGTH_KEY, "123");
call.sendHeaders(headers); call.sendHeaders(headers);
verify(stream).writeHeaders(headers); verify(stream).writeHeaders(headers, false);
assertNull(headers.get(CONTENT_LENGTH_KEY)); assertNull(headers.get(CONTENT_LENGTH_KEY));
} }

2
core/src/test/java/io/grpc/internal/ServerImplTest.java
View File

@ -681,7 +681,7 @@ public class ServerImplTest {
Metadata responseHeaders = new Metadata(); Metadata responseHeaders = new Metadata();
responseHeaders.put(metadataKey, "response value"); responseHeaders.put(metadataKey, "response value");
call.sendHeaders(responseHeaders); call.sendHeaders(responseHeaders);
verify(stream).writeHeaders(responseHeaders); verify(stream).writeHeaders(responseHeaders, true);
verify(stream).setCompressor(isA(Compressor.class)); verify(stream).setCompressor(isA(Compressor.class));
call.sendMessage(firstResponse); call.sendMessage(firstResponse);

24
core/src/testFixtures/java/io/grpc/internal/AbstractTransportTest.java
View File

@ -290,7 +290,7 @@ public abstract class AbstractTransportTest {
stream.flush(); stream.flush();
stream.cancel(Status.CANCELLED); stream.cancel(Status.CANCELLED);
stream.flush(); stream.flush();
serverStreamCreation.stream.writeHeaders(new Metadata()); serverStreamCreation.stream.writeHeaders(new Metadata(), true);
serverStreamCreation.stream.flush(); serverStreamCreation.stream.flush();
serverStreamCreation.stream.writeMessage(methodDescriptor.streamResponse("bar")); serverStreamCreation.stream.writeMessage(methodDescriptor.streamResponse("bar"));
serverStreamCreation.stream.flush(); serverStreamCreation.stream.flush();
@ -308,7 +308,7 @@ public abstract class AbstractTransportTest {
stream.start(mockClientStreamListener2); stream.start(mockClientStreamListener2);
serverStreamCreation serverStreamCreation
= serverTransportListener.takeStreamOrFail(TIMEOUT_MS, TimeUnit.MILLISECONDS); = serverTransportListener.takeStreamOrFail(TIMEOUT_MS, TimeUnit.MILLISECONDS);
serverStreamCreation.stream.writeHeaders(new Metadata()); serverStreamCreation.stream.writeHeaders(new Metadata(), true);
serverStreamCreation.stream.flush(); serverStreamCreation.stream.flush();
verify(mockClientStreamListener2, timeout(TIMEOUT_MS)).headersRead(any(Metadata.class)); verify(mockClientStreamListener2, timeout(TIMEOUT_MS)).headersRead(any(Metadata.class));
@ -468,7 +468,7 @@ public abstract class AbstractTransportTest {
// Try to "flush" out any listener notifications on client and server. This also ensures that // Try to "flush" out any listener notifications on client and server. This also ensures that
// the stream still functions. // the stream still functions.
serverStream.writeHeaders(new Metadata()); serverStream.writeHeaders(new Metadata(), true);
clientStream.halfClose(); clientStream.halfClose();
assertNotNull(clientStreamListener.headers.get(TIMEOUT_MS, TimeUnit.MILLISECONDS)); assertNotNull(clientStreamListener.headers.get(TIMEOUT_MS, TimeUnit.MILLISECONDS));
assertTrue(serverStreamListener.awaitHalfClosed(TIMEOUT_MS, TimeUnit.MILLISECONDS)); assertTrue(serverStreamListener.awaitHalfClosed(TIMEOUT_MS, TimeUnit.MILLISECONDS));
@ -886,7 +886,7 @@ public abstract class AbstractTransportTest {
serverHeaders.put(binaryKey, "dup,value"); serverHeaders.put(binaryKey, "dup,value");
Metadata serverHeadersCopy = new Metadata(); Metadata serverHeadersCopy = new Metadata();
serverHeadersCopy.merge(serverHeaders); serverHeadersCopy.merge(serverHeaders);
serverStream.writeHeaders(serverHeaders); serverStream.writeHeaders(serverHeaders, true);
Metadata headers = clientStreamListener.headers.get(TIMEOUT_MS, TimeUnit.MILLISECONDS); Metadata headers = clientStreamListener.headers.get(TIMEOUT_MS, TimeUnit.MILLISECONDS);
assertNotNull(headers); assertNotNull(headers);
assertAsciiMetadataValuesEqual(serverHeadersCopy.getAll(asciiKey), headers.getAll(asciiKey)); assertAsciiMetadataValuesEqual(serverHeadersCopy.getAll(asciiKey), headers.getAll(asciiKey));
@ -1010,7 +1010,7 @@ public abstract class AbstractTransportTest {
clientStream.halfClose(); clientStream.halfClose();
assertTrue(serverStreamListener.awaitHalfClosed(TIMEOUT_MS, TimeUnit.MILLISECONDS)); assertTrue(serverStreamListener.awaitHalfClosed(TIMEOUT_MS, TimeUnit.MILLISECONDS));
serverStream.writeHeaders(new Metadata()); serverStream.writeHeaders(new Metadata(), true);
assertNotNull(clientStreamListener.headers.get(TIMEOUT_MS, TimeUnit.MILLISECONDS)); assertNotNull(clientStreamListener.headers.get(TIMEOUT_MS, TimeUnit.MILLISECONDS));
Status status = Status.OK.withDescription("Nice talking to you"); Status status = Status.OK.withDescription("Nice talking to you");
@ -1047,7 +1047,7 @@ public abstract class AbstractTransportTest {
ServerStream serverStream = serverStreamCreation.stream; ServerStream serverStream = serverStreamCreation.stream;
ServerStreamListenerBase serverStreamListener = serverStreamCreation.listener; ServerStreamListenerBase serverStreamListener = serverStreamCreation.listener;
serverStream.writeHeaders(new Metadata()); serverStream.writeHeaders(new Metadata(), true);
assertNotNull(clientStreamListener.headers.get(TIMEOUT_MS, TimeUnit.MILLISECONDS)); assertNotNull(clientStreamListener.headers.get(TIMEOUT_MS, TimeUnit.MILLISECONDS));
Status strippedStatus = Status.OK.withDescription("Hello. Goodbye."); Status strippedStatus = Status.OK.withDescription("Hello. Goodbye.");
@ -1273,7 +1273,7 @@ public abstract class AbstractTransportTest {
assertTrue(serverStreamListener.awaitOnReadyAndDrain(TIMEOUT_MS, TimeUnit.MILLISECONDS)); assertTrue(serverStreamListener.awaitOnReadyAndDrain(TIMEOUT_MS, TimeUnit.MILLISECONDS));
assertTrue(serverStream.isReady()); assertTrue(serverStream.isReady());
serverStream.writeHeaders(new Metadata()); serverStream.writeHeaders(new Metadata(), true);
serverStream.writeMessage(methodDescriptor.streamRequest("foo")); serverStream.writeMessage(methodDescriptor.streamRequest("foo"));
serverStream.flush(); serverStream.flush();
@ -1362,7 +1362,7 @@ public abstract class AbstractTransportTest {
ServerStream serverStream = serverStreamCreation.stream; ServerStream serverStream = serverStreamCreation.stream;
ServerStreamListenerBase serverStreamListener = serverStreamCreation.listener; ServerStreamListenerBase serverStreamListener = serverStreamCreation.listener;
serverStream.writeHeaders(new Metadata()); serverStream.writeHeaders(new Metadata(), true);
String largeMessage; String largeMessage;
{ {
@ -1559,7 +1559,7 @@ public abstract class AbstractTransportTest {
assertNotNull(clientStreamListener.trailers.get(TIMEOUT_MS, TimeUnit.MILLISECONDS)); assertNotNull(clientStreamListener.trailers.get(TIMEOUT_MS, TimeUnit.MILLISECONDS));
// Ensure that for a closed ServerStream, interactions are noops // Ensure that for a closed ServerStream, interactions are noops
server.stream.writeHeaders(new Metadata()); server.stream.writeHeaders(new Metadata(), true);
server.stream.writeMessage(methodDescriptor.streamResponse("response")); server.stream.writeMessage(methodDescriptor.streamResponse("response"));
server.stream.close(Status.INTERNAL, new Metadata()); server.stream.close(Status.INTERNAL, new Metadata());
@ -1868,7 +1868,7 @@ public abstract class AbstractTransportTest {
assertEquals(0, clientBefore.lastMessageReceivedTimeNanos); assertEquals(0, clientBefore.lastMessageReceivedTimeNanos);
clientStream.request(1); clientStream.request(1);
serverStream.writeHeaders(new Metadata()); serverStream.writeHeaders(new Metadata(), true);
serverStream.writeMessage(methodDescriptor.streamResponse("response")); serverStream.writeMessage(methodDescriptor.streamResponse("response"));
serverStream.flush(); serverStream.flush();
verifyMessageCountAndClose(clientStreamListener.messageQueue, 1); verifyMessageCountAndClose(clientStreamListener.messageQueue, 1);
@ -1984,7 +1984,7 @@ public abstract class AbstractTransportTest {
= serverTransportListener.takeStreamOrFail(TIMEOUT_MS, TimeUnit.MILLISECONDS); = serverTransportListener.takeStreamOrFail(TIMEOUT_MS, TimeUnit.MILLISECONDS);
serverStreamCreation.stream.request(1); serverStreamCreation.stream.request(1);
serverStreamCreation.stream.writeHeaders(tooLargeMetadata); serverStreamCreation.stream.writeHeaders(tooLargeMetadata, true);
serverStreamCreation.stream.writeMessage(methodDescriptor.streamResponse("response")); serverStreamCreation.stream.writeMessage(methodDescriptor.streamResponse("response"));
serverStreamCreation.stream.close(Status.OK, new Metadata()); serverStreamCreation.stream.close(Status.OK, new Metadata());
@ -2029,7 +2029,7 @@ public abstract class AbstractTransportTest {
= serverTransportListener.takeStreamOrFail(TIMEOUT_MS, TimeUnit.MILLISECONDS); = serverTransportListener.takeStreamOrFail(TIMEOUT_MS, TimeUnit.MILLISECONDS);
serverStreamCreation.stream.request(1); serverStreamCreation.stream.request(1);
serverStreamCreation.stream.writeHeaders(new Metadata()); serverStreamCreation.stream.writeHeaders(new Metadata(), true);
serverStreamCreation.stream.writeMessage(methodDescriptor.streamResponse("response")); serverStreamCreation.stream.writeMessage(methodDescriptor.streamResponse("response"));
serverStreamCreation.stream.close(Status.OK, tooLargeMetadata); serverStreamCreation.stream.close(Status.OK, tooLargeMetadata);

2
inprocess/src/main/java/io/grpc/inprocess/InProcessTransport.java
View File

@ -546,7 +546,7 @@ final class InProcessTransport implements ServerTransport, ConnectionClientTrans
} }
@Override @Override
public void writeHeaders(Metadata headers) { public void writeHeaders(Metadata headers, boolean flush) {
if (clientMaxInboundMetadataSize != Integer.MAX_VALUE) { if (clientMaxInboundMetadataSize != Integer.MAX_VALUE) {
int metadataSize = metadataSize(headers); int metadataSize = metadataSize(headers);
if (metadataSize > clientMaxInboundMetadataSize) { if (metadataSize > clientMaxInboundMetadataSize) {

4
netty/src/main/java/io/grpc/netty/NettyServerStream.java
View File

@ -94,13 +94,13 @@ class NettyServerStream extends AbstractServerStream {
private class Sink implements AbstractServerStream.Sink { private class Sink implements AbstractServerStream.Sink {
@Override @Override
public void writeHeaders(Metadata headers) { public void writeHeaders(Metadata headers, boolean flush) {
try (TaskCloseable ignore = PerfMark.traceTask("NettyServerStream$Sink.writeHeaders")) { try (TaskCloseable ignore = PerfMark.traceTask("NettyServerStream$Sink.writeHeaders")) {
writeQueue.enqueue( writeQueue.enqueue(
SendResponseHeadersCommand.createHeaders( SendResponseHeadersCommand.createHeaders(
transportState(), transportState(),
Utils.convertServerHeaders(headers)), Utils.convertServerHeaders(headers)),
true); flush);
} }
} }

2
netty/src/test/java/io/grpc/netty/NettyClientTransportTest.java
View File

@ -923,7 +923,7 @@ public class NettyClientTransportTest {
public void streamCreated(ServerStream stream, String method, Metadata headers) { public void streamCreated(ServerStream stream, String method, Metadata headers) {
EchoServerStreamListener listener = new EchoServerStreamListener(stream, headers); EchoServerStreamListener listener = new EchoServerStreamListener(stream, headers);
stream.setListener(listener); stream.setListener(listener);
stream.writeHeaders(new Metadata()); stream.writeHeaders(new Metadata(), true);
stream.request(1); stream.request(1);
streamListeners.add(listener); streamListeners.add(listener);
} }

6
netty/src/test/java/io/grpc/netty/NettyServerStreamTest.java
View File

@ -104,7 +104,7 @@ public class NettyServerStreamTest extends NettyStreamTestBase<NettyServerStream
.status(Utils.STATUS_OK) .status(Utils.STATUS_OK)
.set(Utils.CONTENT_TYPE_HEADER, Utils.CONTENT_TYPE_GRPC)); .set(Utils.CONTENT_TYPE_HEADER, Utils.CONTENT_TYPE_GRPC));
stream.writeHeaders(new Metadata()); stream.writeHeaders(new Metadata(), true);
ArgumentCaptor<SendResponseHeadersCommand> sendHeadersCap = ArgumentCaptor<SendResponseHeadersCommand> sendHeadersCap =
ArgumentCaptor.forClass(SendResponseHeadersCommand.class); ArgumentCaptor.forClass(SendResponseHeadersCommand.class);
@ -130,7 +130,7 @@ public class NettyServerStreamTest extends NettyStreamTestBase<NettyServerStream
ListMultimap<CharSequence, CharSequence> expectedHeaders = ListMultimap<CharSequence, CharSequence> expectedHeaders =
ImmutableListMultimap.copyOf(Utils.convertServerHeaders(headers)); ImmutableListMultimap.copyOf(Utils.convertServerHeaders(headers));
stream().writeHeaders(headers); stream().writeHeaders(headers, true);
ArgumentCaptor<SendResponseHeadersCommand> sendHeadersCap = ArgumentCaptor<SendResponseHeadersCommand> sendHeadersCap =
ArgumentCaptor.forClass(SendResponseHeadersCommand.class); ArgumentCaptor.forClass(SendResponseHeadersCommand.class);
@ -300,7 +300,7 @@ public class NettyServerStreamTest extends NettyStreamTestBase<NettyServerStream
@Override @Override
protected void sendHeadersIfServer() { protected void sendHeadersIfServer() {
stream.writeHeaders(new Metadata()); stream.writeHeaders(new Metadata(), true);
} }
@Override @Override

2
okhttp/src/main/java/io/grpc/okhttp/OkHttpServerStream.java
View File

@ -83,7 +83,7 @@ class OkHttpServerStream extends AbstractServerStream {
class Sink implements AbstractServerStream.Sink { class Sink implements AbstractServerStream.Sink {
@Override @Override
public void writeHeaders(Metadata metadata) { public void writeHeaders(Metadata metadata, boolean flush) {
try (TaskCloseable ignore = try (TaskCloseable ignore =
PerfMark.traceTask("OkHttpServerStream$Sink.writeHeaders")) { PerfMark.traceTask("OkHttpServerStream$Sink.writeHeaders")) {
List<Header> responseHeaders = Headers.createResponseHeaders(metadata); List<Header> responseHeaders = Headers.createResponseHeaders(metadata);

8
okhttp/src/test/java/io/grpc/okhttp/OkHttpServerTransportTest.java
View File

@ -371,7 +371,7 @@ public class OkHttpServerTransportTest {
assertThat(streamListener.messages.pop()).isEqualTo("Hello server"); assertThat(streamListener.messages.pop()).isEqualTo("Hello server");
assertThat(streamListener.halfClosedCalled).isTrue(); assertThat(streamListener.halfClosedCalled).isTrue();
streamListener.stream.writeHeaders(metadata("User-Data", "best data")); streamListener.stream.writeHeaders(metadata("User-Data", "best data"), false);
streamListener.stream.writeMessage(new ByteArrayInputStream("Howdy client".getBytes(UTF_8))); streamListener.stream.writeMessage(new ByteArrayInputStream("Howdy client".getBytes(UTF_8)));
streamListener.stream.close(Status.OK, metadata("End-Metadata", "bye")); streamListener.stream.close(Status.OK, metadata("End-Metadata", "bye"));
@ -434,7 +434,7 @@ public class OkHttpServerTransportTest {
assertThat(streamListener.messages.pop()).isEqualTo("Hello server"); assertThat(streamListener.messages.pop()).isEqualTo("Hello server");
assertThat(streamListener.halfClosedCalled).isTrue(); assertThat(streamListener.halfClosedCalled).isTrue();
streamListener.stream.writeHeaders(new Metadata()); streamListener.stream.writeHeaders(new Metadata(), false);
streamListener.stream.writeMessage(new ByteArrayInputStream("Howdy client".getBytes(UTF_8))); streamListener.stream.writeMessage(new ByteArrayInputStream("Howdy client".getBytes(UTF_8)));
streamListener.stream.flush(); streamListener.stream.flush();
@ -1034,7 +1034,7 @@ public class OkHttpServerTransportTest {
streamListener.stream.request(1); streamListener.stream.request(1);
pingPong(); pingPong();
assertThat(streamListener.messages.pop()).isEqualTo("Hello Server Pad Me!"); assertThat(streamListener.messages.pop()).isEqualTo("Hello Server Pad Me!");
streamListener.stream.writeHeaders(metadata("User-Data", "best data")); streamListener.stream.writeHeaders(metadata("User-Data", "best data"), false);
streamListener.stream.writeMessage(new ByteArrayInputStream("Howdy client".getBytes(UTF_8))); streamListener.stream.writeMessage(new ByteArrayInputStream("Howdy client".getBytes(UTF_8)));
List<Header> responseHeaders = Arrays.asList( List<Header> responseHeaders = Arrays.asList(
new Header(":status", "200"), new Header(":status", "200"),
@ -1211,7 +1211,7 @@ public class OkHttpServerTransportTest {
pingPong(); pingPong();
assertThat(streamListener.messages.pop()).isEqualTo("Hello server"); assertThat(streamListener.messages.pop()).isEqualTo("Hello server");
streamListener.stream.writeHeaders(metadata("User-Data", "best data")); streamListener.stream.writeHeaders(metadata("User-Data", "best data"), false);
streamListener.stream.writeMessage(new ByteArrayInputStream("Howdy client".getBytes(UTF_8))); streamListener.stream.writeMessage(new ByteArrayInputStream("Howdy client".getBytes(UTF_8)));
streamListener.stream.flush(); streamListener.stream.flush();
assertThat(clientFrameReader.nextFrame(clientFramesRead)).isTrue(); assertThat(clientFrameReader.nextFrame(clientFramesRead)).isTrue();

2
servlet/src/main/java/io/grpc/servlet/ServletServerStream.java
View File

@ -225,7 +225,7 @@ final class ServletServerStream extends AbstractServerStream {
final TrailerSupplier trailerSupplier = new TrailerSupplier(); final TrailerSupplier trailerSupplier = new TrailerSupplier();
@Override @Override
public void writeHeaders(Metadata headers) { public void writeHeaders(Metadata headers, boolean flush) {
writeHeadersToServletResponse(headers); writeHeadersToServletResponse(headers);
resp.setTrailerFields(trailerSupplier); resp.setTrailerFields(trailerSupplier);
try { try {