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: allow disabling of stream tracers (#10609) 

This adds the ability to disable the installation of stream tracers in
each test call in AbstractInteropTest. The tracers are stored in a list
and used to make assertions by normal tests, but a long running stress
test will accumulate too many entries in this list and the heap gets
wuickly filled up.
This commit is contained in:
Terry Wilson 2023-10-17 08:37:15 -07:00 committed by GitHub
parent 0d39bf5018
commit ccf9101828
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 17 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
interop-testing/src/main/java/io/grpc/testing/integration/AbstractInteropTest.java
View File

@ -313,6 +313,11 @@ public abstract class AbstractInteropTest {
private final LinkedBlockingQueue<TestClientStreamTracer> clientStreamTracers =
new LinkedBlockingQueue<>();
private boolean enableClientStreamTracers = true;
void setEnableClientStreamTracers(boolean enableClientStreamTracers) {
this.enableClientStreamTracers = enableClientStreamTracers;
}
private final ClientStreamTracer.Factory clientStreamTracerFactory =
new ClientStreamTracer.Factory() {
@ -343,9 +348,14 @@ public abstract class AbstractInteropTest {
startServer(serverBuilder);
channel = createChannel();
blockingStub =
TestServiceGrpc.newBlockingStub(channel).withInterceptors(tracerSetupInterceptor);
asyncStub = TestServiceGrpc.newStub(channel).withInterceptors(tracerSetupInterceptor);
if (enableClientStreamTracers) {
blockingStub =
TestServiceGrpc.newBlockingStub(channel).withInterceptors(tracerSetupInterceptor);
asyncStub = TestServiceGrpc.newStub(channel).withInterceptors(tracerSetupInterceptor);
} else {
blockingStub = TestServiceGrpc.newBlockingStub(channel);
asyncStub = TestServiceGrpc.newStub(channel);
}
ClientInterceptor[] additionalInterceptors = getAdditionalInterceptors();
if (additionalInterceptors != null) {

4
interop-testing/src/main/java/io/grpc/testing/integration/StressTestClient.java
View File

@ -466,6 +466,10 @@ public class StressTestClient {
Tester tester = new Tester();
tester.setUp();
// The client stream tracers that AbstractInteropTest installs by default would fill up the
// heap in no time in a long running stress test with many requests.
tester.setEnableClientStreamTracers(false);
WeightedTestCaseSelector testCaseSelector = new WeightedTestCaseSelector(testCaseWeightPairs);
Long endTime = durationSec == null ? null : System.nanoTime() + SECONDS.toNanos(durationSecs);
long lastMetricsCollectionTime = initLastMetricsCollectionTime();