reactor.netty.http.client.HttpClientRequest Java Examples

@Override
public void accept(HttpClientRequest req, Connection connection) {
	PendingSpan pendingSpan = req.currentContext().getOrDefault(PendingSpan.class,
			null);
	if (pendingSpan == null) {
		return; // Somehow TracingMapConnect was not invoked.. skip out
	}

	// All completion hooks clear this reference. If somehow this has a span upon
	// re-entry, the state model in reactor-netty has changed and we need to
	// update this code!
	Span span = pendingSpan.getAndSet(null);
	if (span != null) {
		assert false : "span exists when it shouldn't!";
		span.abandon(); // abandon instead of break
	}

	// Start a new client span with the appropriate parent
	TraceContext parent = req.currentContext().getOrDefault(TraceContext.class,
			null);
	HttpClientRequestWrapper request = new HttpClientRequestWrapper(req);

	span = handler().handleSendWithParent(request, parent);
	parseConnectionAddress(connection, span);
	pendingSpan.set(span);
}
 

public ReactorClientHttpRequest(HttpMethod method, URI uri, HttpClientRequest request, NettyOutbound outbound) {
	this.httpMethod = method;
	this.uri = uri;
	this.request = request;
	this.outbound = outbound;
	this.bufferFactory = new NettyDataBufferFactory(outbound.alloc());
}
 

public ReactorClientHttpRequest(HttpMethod method, URI uri, HttpClientRequest request, NettyOutbound outbound) {
	this.httpMethod = method;
	this.uri = uri;
	this.request = request;
	this.outbound = outbound;
	this.bufferFactory = new NettyDataBufferFactory(outbound.alloc());
}
 

private void doTestDropData(
		BiFunction<? super HttpServerRequest, ? super
				HttpServerResponse, ? extends Publisher<Void>> serverFn,
		BiFunction<? super HttpClientRequest, ? super NettyOutbound, ? extends Publisher<Void>> clientFn)
		throws Exception {
	disposableServer =
			HttpServer.create()
			          .port(0)
			          .handle(serverFn)
			          .wiretap(true)
			          .bindNow(Duration.ofSeconds(30));

	CountDownLatch latch = new CountDownLatch(1);
	String response =
			HttpClient.create()
			          .port(disposableServer.port())
			          .wiretap(true)
			          .doOnRequest((req, conn) -> conn.onTerminate()
			                                          .subscribe(null, null, latch::countDown))
			          .request(HttpMethod.GET)
			          .uri("/")
			          .send(clientFn)
			          .responseContent()
			          .aggregate()
			          .asString()
			          .switchIfEmpty(Mono.just("Empty"))
			          .block(Duration.ofSeconds(30));

	assertThat(latch.await(30, TimeUnit.SECONDS)).isTrue();
	assertThat(response).isEqualTo("Empty");
}
 

@Override
public void beforeMethod(EnhancedInstance objInst, Method method, Object[] allArguments, Class<?>[] argumentsTypes,
                         MethodInterceptResult result) throws Throwable {
    EnhanceObjectCache enhanceObjectCache = (EnhanceObjectCache) objInst.getSkyWalkingDynamicField();
    AbstractSpan span = ContextManager.activeSpan();
    span.prepareForAsync();

    if (!StringUtil.isEmpty(enhanceObjectCache.getUrl())) {
        URL url = new URL(enhanceObjectCache.getUrl());

        ContextCarrier contextCarrier = new ContextCarrier();
        AbstractSpan abstractSpan = ContextManager.createExitSpan(
            "SpringCloudGateway/sendRequest", contextCarrier, getPeer(url));
        Tags.URL.set(abstractSpan, enhanceObjectCache.getUrl());
        abstractSpan.prepareForAsync();
        abstractSpan.setComponent(SPRING_CLOUD_GATEWAY);

        ContextManager.stopSpan(abstractSpan);
        ContextManager.stopSpan(span);

        BiFunction<? super HttpClientRequest, ? super NettyOutbound, ? extends Publisher<Void>> finalSender = (BiFunction<? super HttpClientRequest, ? super NettyOutbound, ? extends Publisher<Void>>) allArguments[0];
        allArguments[0] = new BiFunction<HttpClientRequest, NettyOutbound, Publisher<Void>>() {
            @Override
            public Publisher<Void> apply(HttpClientRequest request, NettyOutbound outbound) {
                Publisher publisher = finalSender.apply(request, outbound);

                CarrierItem next = contextCarrier.items();
                while (next.hasNext()) {
                    next = next.next();
                    request.requestHeaders().remove(next.getHeadKey());
                    request.requestHeaders().set(next.getHeadKey(), next.getHeadValue());
                }
                return publisher;
            }
        };
        enhanceObjectCache.setCacheSpan(abstractSpan);
    }

    enhanceObjectCache.setSpan1(span);
}
 

private ReactorClientHttpRequest adaptRequest(HttpMethod method, URI uri, HttpClientRequest request,
		NettyOutbound nettyOutbound) {

	return new ReactorClientHttpRequest(method, uri, request, nettyOutbound);
}
 

private ReactorClientHttpRequest adaptRequest(HttpMethod method, URI uri, HttpClientRequest request,
		NettyOutbound nettyOutbound) {

	return new ReactorClientHttpRequest(method, uri, request, nettyOutbound);
}
 

HttpClientHandler<brave.http.HttpClientRequest, brave.http.HttpClientResponse> handler() {
	if (this.handler == null) {
		this.handler = HttpClientHandler.create(httpTracing.get());
	}
	return this.handler;
}
 

@Override
public void accept(HttpClientRequest req, Throwable error) {
	handle(req.currentContext(), null, error);
}
 

HttpClientHandler<brave.http.HttpClientRequest, brave.http.HttpClientResponse> handler() {
	if (this.handler == null) {
		this.handler = HttpClientHandler.create(httpTracing.get());
	}
	return this.handler;
}
 

HttpClientRequestWrapper(HttpClientRequest delegate) {
	this.delegate = delegate;
}