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

Use async requests for high concurrency test. (#3534)

This commit is contained in:
Anuraag Agrawal 2021-07-10 13:20:53 +09:00 committed by GitHub
parent 7969bdf18e
commit 9be9e40556
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
testing-common/src/main/groovy/io/opentelemetry/instrumentation/test/base/HttpServerTest.groovy
View File

@ -35,7 +35,6 @@ import io.opentelemetry.testing.internal.armeria.common.HttpRequest
import io.opentelemetry.testing.internal.armeria.common.HttpRequestBuilder import io.opentelemetry.testing.internal.armeria.common.HttpRequestBuilder
import java.util.concurrent.Callable import java.util.concurrent.Callable
import java.util.concurrent.CountDownLatch import java.util.concurrent.CountDownLatch
import java.util.concurrent.Executors
import spock.lang.Unroll import spock.lang.Unroll
@Unroll @Unroll
@ -390,9 +389,8 @@ abstract class HttpServerTest<SERVER> extends InstrumentationSpecification imple
int count = 100 int count = 100
def endpoint = INDEXED_CHILD def endpoint = INDEXED_CHILD
def latch = new CountDownLatch(1) def latch = new CountDownLatch(count)
def pool = Executors.newFixedThreadPool(4)
def propagator = GlobalOpenTelemetry.getPropagators().getTextMapPropagator() def propagator = GlobalOpenTelemetry.getPropagators().getTextMapPropagator()
def setter = { HttpRequestBuilder carrier, String name, String value -> def setter = { HttpRequestBuilder carrier, String name, String value ->
carrier.header(name, value) carrier.header(name, value)
@ -400,8 +398,6 @@ abstract class HttpServerTest<SERVER> extends InstrumentationSpecification imple
when: when:
count.times { index -> count.times { index ->
def job = {
latch.await()
HttpRequestBuilder request = HttpRequest.builder() HttpRequestBuilder request = HttpRequest.builder()
// Force HTTP/1 via h1c so upgrade requests don't show up as traces // Force HTTP/1 via h1c so upgrade requests don't show up as traces
.get(endpoint.resolvePath(address).toString().replace("http://", "h1c://")) .get(endpoint.resolvePath(address).toString().replace("http://", "h1c://"))
@ -409,13 +405,12 @@ abstract class HttpServerTest<SERVER> extends InstrumentationSpecification imple
runUnderTrace("client " + index) { runUnderTrace("client " + index) {
Span.current().setAttribute(ServerEndpoint.ID_ATTRIBUTE_NAME, index) Span.current().setAttribute(ServerEndpoint.ID_ATTRIBUTE_NAME, index)
propagator.inject(Context.current(), request, setter) propagator.inject(Context.current(), request, setter)
client.execute(request.build()).aggregate().join() client.execute(request.build()).aggregate().thenRun {
}
}
pool.submit(job)
}
latch.countDown() latch.countDown()
}
}
}
latch.await()
then: then:
assertTraces(count) { assertTraces(count) {
@ -442,9 +437,6 @@ abstract class HttpServerTest<SERVER> extends InstrumentationSpecification imple
} }
} }
} }
cleanup:
pool.shutdownNow()
} }
//FIXME: add tests for POST with large/chunked data //FIXME: add tests for POST with large/chunked data