2 * ============LICENSE_START====================================
3 * DCAEGEN2-SERVICES-SDK
4 * =========================================================
5 * Copyright (C) 2019-2020 Nokia. All rights reserved.
6 * =========================================================
7 * Licensed under the Apache License, Version 2.0 (the "License");
8 * you may not use this file except in compliance with the License.
9 * You may obtain a copy of the License at
11 * http://www.apache.org/licenses/LICENSE-2.0
13 * Unless required by applicable law or agreed to in writing, software
14 * distributed under the License is distributed on an "AS IS" BASIS,
15 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16 * See the License for the specific language governing permissions and
17 * limitations under the License.
18 * ============LICENSE_END=====================================
20 package org.onap.dcaegen2.services.sdk.rest.services.adapters.http;
22 import io.vavr.collection.Stream;
23 import io.vavr.control.Option;
24 import org.onap.dcaegen2.services.sdk.rest.services.model.logging.RequestDiagnosticContext;
25 import org.slf4j.Logger;
26 import org.slf4j.LoggerFactory;
27 import reactor.core.publisher.Mono;
28 import reactor.netty.http.client.HttpClient;
29 import reactor.netty.http.client.HttpClient.ResponseReceiver;
30 import reactor.netty.http.client.HttpClientRequest;
31 import reactor.netty.http.client.HttpClientResponse;
33 import java.util.stream.Collectors;
38 public class RxHttpClient {
40 private static final Logger LOGGER = LoggerFactory.getLogger(RxHttpClient.class);
41 private final HttpClient httpClient;
43 RxHttpClient(HttpClient httpClient) {
44 this.httpClient = httpClient;
47 public Mono<HttpResponse> call(HttpRequest request) {
48 return prepareRequest(request)
49 .responseSingle((resp, content) ->
51 .defaultIfEmpty(new byte[0])
52 .map(bytes -> new NettyHttpResponse(request.url(), resp.status(), bytes)));
55 ResponseReceiver<?> prepareRequest(HttpRequest request) {
56 final HttpClient simpleClient = httpClient
57 .doOnRequest((req, conn) -> logRequest(request.diagnosticContext(), req))
58 .doOnResponse((rsp, conn) -> logResponse(request.diagnosticContext(), rsp))
59 .headers(hdrs -> request.headers().forEach(hdr -> hdrs.set(hdr._1, hdr._2)));
61 final HttpClient theClient = Option.of(request.timeout())
62 .map(simpleClient::responseTimeout)
63 .getOrElse(simpleClient);
65 return prepareBody(request, theClient);
68 private ResponseReceiver<?> prepareBody(HttpRequest request, HttpClient theClient) {
69 if (request.body() == null) {
70 return prepareBodyWithoutContents(request, theClient);
72 return request.body().length() == null
73 ? prepareBodyChunked(request, theClient)
74 : prepareBodyUnchunked(request, theClient);
78 private ResponseReceiver<?> prepareBodyChunked(HttpRequest request, HttpClient theClient) {
80 .headers(hdrs -> hdrs.set(HttpHeaders.TRANSFER_ENCODING_TYPE, HttpHeaders.CHUNKED))
81 .request(request.method().asNetty())
82 .send(request.body().contents())
86 private ResponseReceiver<?> prepareBodyUnchunked(HttpRequest request, HttpClient theClient) {
88 .headers(hdrs -> hdrs.set(HttpHeaders.CONTENT_LENGTH, request.body().length().toString()))
89 .request(request.method().asNetty())
90 .send(request.body().contents())
94 private ResponseReceiver<?> prepareBodyWithoutContents(HttpRequest request, HttpClient theClient) {
96 .request(request.method().asNetty())
100 private void logRequest(RequestDiagnosticContext context, HttpClientRequest httpClientRequest) {
101 context.withSlf4jMdc(LOGGER.isDebugEnabled(), () -> {
102 LOGGER.debug("Request: {} {} {}", httpClientRequest.method(), httpClientRequest.uri(),
103 httpClientRequest.requestHeaders());
104 if (LOGGER.isTraceEnabled()) {
105 final String headers = Stream.ofAll(httpClientRequest.requestHeaders())
106 .map(entry -> entry.getKey() + "=" + entry.getValue())
107 .collect(Collectors.joining("\n"));
108 LOGGER.trace(headers);
113 private void logResponse(RequestDiagnosticContext context, HttpClientResponse httpClientResponse) {
114 context.withSlf4jMdc(LOGGER.isDebugEnabled(),
115 () -> LOGGER.debug("Response status: {}", httpClientResponse.status()));