Skip to content

Commit f589a4a

Browse files
simonmaejona86
authored andcommitted
Remove traces of gRPC v1 protocol vs gRPC v2 protocol.
------------- Created by MOE: http://code.google.com/p/moe-java MOE_MIGRATED_REVID=84087425
1 parent 3df1c33 commit f589a4a

File tree

8 files changed

+42
-52
lines changed

8 files changed

+42
-52
lines changed

core/src/main/java/com/google/net/stubby/transport/AbstractClientStream.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -208,7 +208,7 @@ public void transportReportStatus(final Status newStatus, boolean stopDelivery,
208208
closeListenerTask = null;
209209

210210
// Determine if the deframer is stalled (i.e. currently has no complete messages to deliver).
211-
boolean deliveryStalled = !deframer2.isDeliveryOutstanding();
211+
boolean deliveryStalled = !deframer.isDeliveryOutstanding();
212212

213213
if (stopDelivery || deliveryStalled) {
214214
// Close the listener immediately.

core/src/main/java/com/google/net/stubby/transport/AbstractStream.java

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@ protected enum Phase {
5858
}
5959

6060
private volatile IdT id;
61-
private final MessageFramer2 framer;
61+
private final MessageFramer framer;
6262
private final FutureCallback<Object> deframerErrorCallback = new FutureCallback<Object>() {
6363
@Override
6464
public void onSuccess(Object result) {}
@@ -69,7 +69,7 @@ public void onFailure(Throwable t) {
6969
}
7070
};
7171

72-
final MessageDeframer2 deframer2;
72+
final MessageDeframer deframer;
7373

7474
/**
7575
* Inbound phase is exclusively written to by the transport thread.
@@ -82,7 +82,7 @@ public void onFailure(Throwable t) {
8282
private Phase outboundPhase = Phase.HEADERS;
8383

8484
AbstractStream(Executor deframerExecutor) {
85-
MessageDeframer2.Listener inboundMessageHandler = new MessageDeframer2.Listener() {
85+
MessageDeframer.Listener inboundMessageHandler = new MessageDeframer.Listener() {
8686
@Override
8787
public void bytesRead(int numBytes) {
8888
returnProcessedBytes(numBytes);
@@ -109,15 +109,15 @@ public void endOfStream() {
109109
remoteEndClosed();
110110
}
111111
};
112-
MessageFramer2.Sink<ByteBuffer> outboundFrameHandler = new MessageFramer2.Sink<ByteBuffer>() {
112+
MessageFramer.Sink<ByteBuffer> outboundFrameHandler = new MessageFramer.Sink<ByteBuffer>() {
113113
@Override
114114
public void deliverFrame(ByteBuffer frame, boolean endOfStream) {
115115
internalSendFrame(frame, endOfStream);
116116
}
117117
};
118118

119-
framer = new MessageFramer2(outboundFrameHandler, 4096);
120-
this.deframer2 = new MessageDeframer2(inboundMessageHandler, deframerExecutor);
119+
framer = new MessageFramer(outboundFrameHandler, 4096);
120+
this.deframer = new MessageDeframer(inboundMessageHandler, deframerExecutor);
121121
}
122122

123123
/**
@@ -219,7 +219,7 @@ public void dispose() {
219219
*/
220220
protected final void deframe(Buffer frame, boolean endOfStream) {
221221
ListenableFuture<?> future;
222-
future = deframer2.deframe(frame, endOfStream);
222+
future = deframer.deframe(frame, endOfStream);
223223
if (future != null) {
224224
Futures.addCallback(future, deframerErrorCallback);
225225
}
@@ -229,7 +229,7 @@ protected final void deframe(Buffer frame, boolean endOfStream) {
229229
* Delays delivery from the deframer until the given future completes.
230230
*/
231231
protected final void delayDeframer(ListenableFuture<?> future) {
232-
ListenableFuture<?> deliveryFuture = deframer2.delayProcessing(future);
232+
ListenableFuture<?> deliveryFuture = deframer.delayProcessing(future);
233233
if (deliveryFuture != null) {
234234
Futures.addCallback(deliveryFuture, deframerErrorCallback);
235235
}

core/src/main/java/com/google/net/stubby/transport/Http2ClientStream.java

Lines changed: 4 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -123,20 +123,10 @@ protected void transportDataReceived(Buffer frame, boolean endOfStream) {
123123
} else {
124124
inboundDataReceived(frame);
125125
if (endOfStream) {
126-
if (false) {
127-
// This is a protocol violation as we expect to receive trailers.
128-
transportError = Status.INTERNAL.withDescription("Recevied EOS on DATA frame");
129-
frame.close();
130-
inboundTransportError(transportError);
131-
} else {
132-
// TODO(user): Delete this hack when trailers are supported by GFE with v2. Currently
133-
// GFE doesn't support trailers, so when using gRPC v2 protocol GFE will not send any
134-
// status. We paper over this for now by just assuming OK. For all properly functioning
135-
// servers (both v1 and v2), stashedStatus should not be null here.
136-
Metadata.Trailers trailers = new Metadata.Trailers();
137-
trailers.put(Status.CODE_KEY, Status.OK);
138-
inboundTrailersReceived(trailers, Status.OK);
139-
}
126+
// This is a protocol violation as we expect to receive trailers.
127+
transportError = Status.INTERNAL.withDescription("Recevied EOS on DATA frame");
128+
frame.close();
129+
inboundTransportError(transportError);
140130
}
141131
}
142132
}

core/src/main/java/com/google/net/stubby/transport/MessageDeframer2.java renamed to core/src/main/java/com/google/net/stubby/transport/MessageDeframer.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@
5656
* executor provided during creation. That executor must not allow concurrent execution of tasks.
5757
*/
5858
@NotThreadSafe
59-
public class MessageDeframer2 implements Closeable {
59+
public class MessageDeframer implements Closeable {
6060
private static final int HEADER_LENGTH = 5;
6161
private static final int COMPRESSED_FLAG_MASK = 1;
6262
private static final int RESERVED_MASK = 0xFE;
@@ -123,7 +123,7 @@ private enum State {
123123
* @param listener listener for deframer events.
124124
* @param executor used for internal event processing
125125
*/
126-
public MessageDeframer2(Listener listener, Executor executor) {
126+
public MessageDeframer(Listener listener, Executor executor) {
127127
this(listener, executor, Compression.NONE);
128128
}
129129

@@ -136,7 +136,7 @@ public MessageDeframer2(Listener listener, Executor executor) {
136136
* @param compression the compression used if a compressed frame is encountered, with NONE meaning
137137
* unsupported
138138
*/
139-
public MessageDeframer2(Listener listener, Executor executor, Compression compression) {
139+
public MessageDeframer(Listener listener, Executor executor, Compression compression) {
140140
this.listener = Preconditions.checkNotNull(listener, "sink");
141141
this.executor = Preconditions.checkNotNull(executor, "executor");
142142
this.compression = Preconditions.checkNotNull(compression, "compression");

core/src/main/java/com/google/net/stubby/transport/MessageFramer2.java renamed to core/src/main/java/com/google/net/stubby/transport/MessageFramer.java

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -45,9 +45,9 @@
4545

4646
/**
4747
* Encodes gRPC messages to be delivered via the transport layer which implements {@link
48-
* MessageFramer2.Sink}.
48+
* MessageFramer.Sink}.
4949
*/
50-
public class MessageFramer2 {
50+
public class MessageFramer {
5151
/**
5252
* Sink implemented by the transport layer to receive frames and forward them to their destination
5353
*/
@@ -75,11 +75,11 @@ public enum Compression {
7575
private final OutputStreamAdapter outputStreamAdapter = new OutputStreamAdapter();
7676
private final byte[] headerScratch = new byte[HEADER_LENGTH];
7777

78-
public MessageFramer2(Sink<ByteBuffer> sink, int maxFrameSize) {
78+
public MessageFramer(Sink<ByteBuffer> sink, int maxFrameSize) {
7979
this(sink, maxFrameSize, Compression.NONE);
8080
}
8181

82-
public MessageFramer2(Sink<ByteBuffer> sink, int maxFrameSize, Compression compression) {
82+
public MessageFramer(Sink<ByteBuffer> sink, int maxFrameSize, Compression compression) {
8383
this.sink = Preconditions.checkNotNull(sink, "sink");
8484
this.bytebuf = ByteBuffer.allocate(maxFrameSize);
8585
this.compression = Preconditions.checkNotNull(compression, "compression");

core/src/test/java/com/google/net/stubby/transport/MessageDeframer2Test.java renamed to core/src/test/java/com/google/net/stubby/transport/MessageDeframerTest.java

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@
5151
import com.google.common.util.concurrent.ListenableFuture;
5252
import com.google.common.util.concurrent.MoreExecutors;
5353
import com.google.common.util.concurrent.SettableFuture;
54-
import com.google.net.stubby.transport.MessageDeframer2.Listener;
54+
import com.google.net.stubby.transport.MessageDeframer.Listener;
5555

5656
import org.junit.Test;
5757
import org.junit.runner.RunWith;
@@ -66,13 +66,13 @@
6666
import java.util.zip.GZIPOutputStream;
6767

6868
/**
69-
* Tests for {@link MessageDeframer2}.
69+
* Tests for {@link MessageDeframer}.
7070
*/
7171
@RunWith(JUnit4.class)
72-
public class MessageDeframer2Test {
72+
public class MessageDeframerTest {
7373
private Listener listener = mock(Listener.class);
74-
private MessageDeframer2 deframer =
75-
new MessageDeframer2(listener, MoreExecutors.directExecutor());
74+
private MessageDeframer deframer =
75+
new MessageDeframer(listener, MoreExecutors.directExecutor());
7676
private ArgumentCaptor<InputStream> messages = ArgumentCaptor.forClass(InputStream.class);
7777

7878
@Test
@@ -259,8 +259,8 @@ public void futureFailureShouldStopAndPropagateFailure() throws InterruptedExcep
259259

260260
@Test
261261
public void compressed() {
262-
deframer = new MessageDeframer2(
263-
listener, MoreExecutors.directExecutor(), MessageDeframer2.Compression.GZIP);
262+
deframer = new MessageDeframer(
263+
listener, MoreExecutors.directExecutor(), MessageDeframer.Compression.GZIP);
264264
byte[] payload = compress(new byte[1000]);
265265
assertTrue(payload.length < 100);
266266
byte[] header = new byte[] {1, 0, 0, 0, (byte) payload.length};

core/src/test/java/com/google/net/stubby/transport/MessageFramer2Test.java renamed to core/src/test/java/com/google/net/stubby/transport/MessageFramerTest.java

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -53,16 +53,16 @@
5353
import java.util.List;
5454

5555
/**
56-
* Tests for {@link MessageFramer2}
56+
* Tests for {@link MessageFramer}
5757
*/
5858
@RunWith(JUnit4.class)
59-
public class MessageFramer2Test {
59+
public class MessageFramerTest {
6060
private static final int TRANSPORT_FRAME_SIZE = 12;
6161

6262
@Mock
63-
private MessageFramer2.Sink<List<Byte>> sink;
64-
private MessageFramer2.Sink<ByteBuffer> copyingSink;
65-
private MessageFramer2 framer;
63+
private MessageFramer.Sink<List<Byte>> sink;
64+
private MessageFramer.Sink<ByteBuffer> copyingSink;
65+
private MessageFramer framer;
6666

6767
@Captor
6868
private ArgumentCaptor<List<Byte>> frameCaptor;
@@ -72,7 +72,7 @@ public void setup() {
7272
MockitoAnnotations.initMocks(this);
7373

7474
copyingSink = new ByteArrayConverterSink(sink);
75-
framer = new MessageFramer2(copyingSink, TRANSPORT_FRAME_SIZE);
75+
framer = new MessageFramer(copyingSink, TRANSPORT_FRAME_SIZE);
7676
}
7777

7878
@Test
@@ -156,7 +156,7 @@ public void flushIsIdempotent() {
156156
@Test
157157
public void largerFrameSize() throws Exception {
158158
final int transportFrameSize = 10000;
159-
MessageFramer2 framer = new MessageFramer2(copyingSink, transportFrameSize);
159+
MessageFramer framer = new MessageFramer(copyingSink, transportFrameSize);
160160
writePayload(framer, new byte[1000]);
161161
framer.flush();
162162
verify(sink).deliverFrame(frameCaptor.capture(), eq(false));
@@ -169,8 +169,8 @@ public void largerFrameSize() throws Exception {
169169
@Test
170170
public void compressed() throws Exception {
171171
final int transportFrameSize = 100;
172-
MessageFramer2 framer = new MessageFramer2(copyingSink, transportFrameSize,
173-
MessageFramer2.Compression.GZIP);
172+
MessageFramer framer = new MessageFramer(copyingSink, transportFrameSize,
173+
MessageFramer.Compression.GZIP);
174174
writePayload(framer, new byte[1000]);
175175
framer.flush();
176176
verify(sink).deliverFrame(frameCaptor.capture(), eq(false));
@@ -183,18 +183,18 @@ public void compressed() throws Exception {
183183
verifyNoMoreInteractions(sink);
184184
}
185185

186-
private static void writePayload(MessageFramer2 framer, byte[] bytes) {
186+
private static void writePayload(MessageFramer framer, byte[] bytes) {
187187
framer.writePayload(new ByteArrayInputStream(bytes), bytes.length);
188188
}
189189

190190
/**
191191
* Since ByteBuffers are reused, this sink copies their value at the time of the call. Converting
192192
* to List<Byte> is convenience.
193193
*/
194-
private static class ByteArrayConverterSink implements MessageFramer2.Sink<ByteBuffer> {
195-
private final MessageFramer2.Sink<List<Byte>> delegate;
194+
private static class ByteArrayConverterSink implements MessageFramer.Sink<ByteBuffer> {
195+
private final MessageFramer.Sink<List<Byte>> delegate;
196196

197-
public ByteArrayConverterSink(MessageFramer2.Sink<List<Byte>> delegate) {
197+
public ByteArrayConverterSink(MessageFramer.Sink<List<Byte>> delegate) {
198198
this.delegate = delegate;
199199
}
200200

netty/src/test/java/com/google/net/stubby/transport/netty/NettyServerHandlerTest.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@
5353
import com.google.net.stubby.Metadata;
5454
import com.google.net.stubby.Status;
5555
import com.google.net.stubby.Status.Code;
56-
import com.google.net.stubby.transport.MessageFramer2;
56+
import com.google.net.stubby.transport.MessageFramer;
5757
import com.google.net.stubby.transport.ServerStream;
5858
import com.google.net.stubby.transport.ServerStreamListener;
5959
import com.google.net.stubby.transport.ServerTransportListener;
@@ -271,7 +271,7 @@ private void createStream() throws Exception {
271271

272272
private ByteBuf dataFrame(int streamId, boolean endStream) {
273273
final ByteBuf compressionFrame = Unpooled.buffer(CONTENT.length);
274-
MessageFramer2 framer = new MessageFramer2(new MessageFramer2.Sink<ByteBuffer>() {
274+
MessageFramer framer = new MessageFramer(new MessageFramer.Sink<ByteBuffer>() {
275275
@Override
276276
public void deliverFrame(ByteBuffer frame, boolean endOfStream) {
277277
compressionFrame.writeBytes(frame);

0 commit comments

Comments
 (0)