gerrit.onap Code Review - dcaegen2/services/sdk.git/blob

   1 /*
   2  * ============LICENSE_START====================================
   3  * DCAEGEN2-SERVICES-SDK
   4  * =========================================================
   5  * Copyright (C) 2019 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
  10  *
  11  *       http://www.apache.org/licenses/LICENSE-2.0
  12  *
  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=====================================
  19  */
  20 package org.onap.dcaegen2.services.sdk.rest.services.adapters.http;
  21
  22 import io.netty.handler.ssl.SslContext;
  23 import io.vavr.collection.Stream;
  24 import java.util.stream.Collectors;
  25 import org.jetbrains.annotations.NotNull;
  26 import org.onap.dcaegen2.services.sdk.rest.services.model.logging.RequestDiagnosticContext;
  27 import org.slf4j.Logger;
  28 import org.slf4j.LoggerFactory;
  29 import reactor.core.publisher.Mono;
  30 import reactor.netty.http.client.HttpClient;
  31 import reactor.netty.http.client.HttpClient.ResponseReceiver;
  32 import reactor.netty.http.client.HttpClientRequest;
  33 import reactor.netty.http.client.HttpClientResponse;
  34
  35 /**
  36  * @since 1.1.4
  37  */
  38 public class RxHttpClient {
  39
  40     private static final Logger LOGGER = LoggerFactory.getLogger(RxHttpClient.class);
  41     private final HttpClient httpClient;
  42
  43     public static RxHttpClient create() {
  44         return new RxHttpClient(HttpClient.create());
  45     }
  46
  47     // TODO: hide netty from public api (io.netty.handler.ssl.SslContext)
  48     public static RxHttpClient create(@NotNull
  49             SslContext sslContext) {
  50         return new RxHttpClient(HttpClient.create().secure(sslContextSpec -> sslContextSpec.sslContext(sslContext)));
  51     }
  52
  53     RxHttpClient(HttpClient httpClient) {
  54         this.httpClient = httpClient;
  55     }
  56
  57     public Mono<HttpResponse> call(HttpRequest request) {
  58         return prepareRequest(request)
  59                 .responseSingle((resp, content) ->
  60                         content.asByteArray()
  61                                 .defaultIfEmpty(new byte[0])
  62                                 .map(bytes -> new NettyHttpResponse(request.url(), resp.status(), bytes)));
  63     }
  64
  65     ResponseReceiver<?> prepareRequest(HttpRequest request) {
  66         final HttpClient theClient = httpClient
  67                 .doOnRequest((req, conn) -> logRequest(request.diagnosticContext(), req))
  68                 .doOnResponse((rsp, conn) -> logResponse(request.diagnosticContext(), rsp))
  69                 .headers(hdrs -> request.headers().forEach(hdr -> hdrs.set(hdr._1, hdr._2)));
  70
  71         return prepareBody(request, theClient);
  72     }
  73
  74     private ResponseReceiver<?> prepareBody(HttpRequest request, HttpClient theClient) {
  75         if (request.body() == null) {
  76             return prepareBodyWithoutContents(request, theClient);
  77         } else {
  78             return request.body().length() == null
  79                     ? prepareBodyChunked(request, theClient)
  80                     : prepareBodyUnchunked(request, theClient);
  81         }
  82     }
  83
  84     private ResponseReceiver<?> prepareBodyChunked(HttpRequest request, HttpClient theClient) {
  85         return theClient
  86                 .chunkedTransfer(true)
  87                 .request(request.method().asNetty())
  88                 .send(request.body().contents())
  89                 .uri(request.url());
  90     }
  91
  92     private ResponseReceiver<?> prepareBodyUnchunked(HttpRequest request, HttpClient theClient) {
  93         return theClient
  94                 .headers(hdrs -> hdrs.set(HttpHeaders.CONTENT_LENGTH, request.body().length().toString()))
  95                 .request(request.method().asNetty())
  96                 .send(request.body().contents())
  97                 .uri(request.url());
  98     }
  99
 100     private ResponseReceiver<?> prepareBodyWithoutContents(HttpRequest request, HttpClient theClient) {
 101         return theClient
 102                 .request(request.method().asNetty())
 103                 .uri(request.url());
 104     }
 105
 106     private void logRequest(RequestDiagnosticContext context, HttpClientRequest httpClientRequest) {
 107         context.withSlf4jMdc(LOGGER.isDebugEnabled(), () -> {
 108             LOGGER.debug("Request: {} {} {}", httpClientRequest.method(), httpClientRequest.uri(),
 109                     httpClientRequest.requestHeaders());
 110             if (LOGGER.isTraceEnabled()) {
 111                 final String headers = Stream.ofAll(httpClientRequest.requestHeaders())
 112                         .map(entry -> entry.getKey() + "=" + entry.getValue())
 113                         .collect(Collectors.joining("\n"));
 114                 LOGGER.trace(headers);
 115             }
 116         });
 117     }
 118
 119     private void logResponse(RequestDiagnosticContext context, HttpClientResponse httpClientResponse) {
 120         context.withSlf4jMdc(LOGGER.isDebugEnabled(),
 121                 () -> LOGGER.debug("Response status: {}", httpClientResponse.status()));
 122     }
 123 }