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

interop-testing: support rpc sleep behavior for xDS test server (#7836)

This commit is contained in:
Doug Fawley 2021-01-25 10:14:37 -08:00 committed by GitHub
parent 0ffd942fd2
commit ec270a7757
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 17 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
interop-testing/src/main/java/io/grpc/testing/integration/XdsTestServer.java
View File

@ -26,6 +26,8 @@ import io.grpc.ServerCall;
import io.grpc.ServerCallHandler; import io.grpc.ServerCallHandler;
import io.grpc.ServerInterceptor; import io.grpc.ServerInterceptor;
import io.grpc.ServerInterceptors; import io.grpc.ServerInterceptors;
import io.grpc.Status;
import io.grpc.StatusRuntimeException;
import io.grpc.health.v1.HealthCheckResponse.ServingStatus; import io.grpc.health.v1.HealthCheckResponse.ServingStatus;
import io.grpc.netty.NettyServerBuilder; import io.grpc.netty.NettyServerBuilder;
import io.grpc.protobuf.services.ProtoReflectionService; import io.grpc.protobuf.services.ProtoReflectionService;
@ -51,6 +53,7 @@ public final class XdsTestServer {
private static final Context.Key<String> CALL_BEHAVIOR_KEY = private static final Context.Key<String> CALL_BEHAVIOR_KEY =
Context.key("rpc-behavior"); Context.key("rpc-behavior");
private static final String CALL_BEHAVIOR_KEEP_OPEN_VALUE = "keep-open"; private static final String CALL_BEHAVIOR_KEEP_OPEN_VALUE = "keep-open";
private static final String CALL_BEHAVIOR_SLEEP_VALUE = "sleep-";
private static final int CHANNELZ_MAX_PAGE_SIZE = 100; private static final int CHANNELZ_MAX_PAGE_SIZE = 100;
private static Logger logger = Logger.getLogger(XdsTestServer.class.getName()); private static Logger logger = Logger.getLogger(XdsTestServer.class.getName());
@ -286,6 +289,20 @@ public final class XdsTestServer {
ServerCallHandler<ReqT, RespT> next) { ServerCallHandler<ReqT, RespT> next) {
String callBehavior = requestHeaders.get(CALL_BEHAVIOR_MD_KEY); String callBehavior = requestHeaders.get(CALL_BEHAVIOR_MD_KEY);
Context newContext = Context.current().withValue(CALL_BEHAVIOR_KEY, callBehavior); Context newContext = Context.current().withValue(CALL_BEHAVIOR_KEY, callBehavior);
if (callBehavior != null && callBehavior.startsWith(CALL_BEHAVIOR_SLEEP_VALUE)) {
try {
int timeout = Integer.parseInt(
callBehavior.substring(CALL_BEHAVIOR_SLEEP_VALUE.length()));
Thread.sleep(timeout * 1000);
} catch (NumberFormatException e) {
throw new StatusRuntimeException(
Status.INVALID_ARGUMENT.withDescription(
String.format("Invalid format for rpc-behavior header (%s)", callBehavior)));
} catch (InterruptedException e) {
throw new StatusRuntimeException(
Status.ABORTED.withDescription("execution of server interrupted"));
}
}
ServerCall<ReqT, RespT> newCall = new SimpleForwardingServerCall<ReqT, RespT>(call) { ServerCall<ReqT, RespT> newCall = new SimpleForwardingServerCall<ReqT, RespT>(call) {
@Override @Override
public void sendHeaders(Metadata responseHeaders) { public void sendHeaders(Metadata responseHeaders) {