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

Add a timeout to AltsHandshakerStub

This commit is contained in:
susinmotion 2020-11-03 18:44:03 -05:00 committed by Eric Anderson
parent beb3232c0a
commit d154aa3328
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
alts/src/main/java/io/grpc/alts/internal/AltsHandshakerStub.java
View File

@ -16,6 +16,8 @@
package io.grpc.alts.internal; package io.grpc.alts.internal;
import static java.util.concurrent.TimeUnit.SECONDS;
import com.google.common.annotations.VisibleForTesting; import com.google.common.annotations.VisibleForTesting;
import com.google.common.base.Optional; import com.google.common.base.Optional;
import io.grpc.alts.internal.HandshakerServiceGrpc.HandshakerServiceStub; import io.grpc.alts.internal.HandshakerServiceGrpc.HandshakerServiceStub;
@ -32,8 +34,13 @@ class AltsHandshakerStub {
new ArrayBlockingQueue<>(1); new ArrayBlockingQueue<>(1);
private final AtomicReference<String> exceptionMessage = new AtomicReference<>(); private final AtomicReference<String> exceptionMessage = new AtomicReference<>();
private static final long HANDSHAKE_RPC_DEADLINE_SECS = 20;
AltsHandshakerStub(HandshakerServiceStub serviceStub) { AltsHandshakerStub(HandshakerServiceStub serviceStub) {
this.writer = serviceStub.doHandshake(this.reader); this.writer =
serviceStub
.withDeadlineAfter(HANDSHAKE_RPC_DEADLINE_SECS, SECONDS)
.doHandshake(this.reader);
} }
@VisibleForTesting @VisibleForTesting