gerrit.onap Code Review - ccsdk/oran.git/blob

   1 /*-
   2  * ========================LICENSE_START=================================
   3  * ONAP : ccsdk oran
   4  * ======================================================================
   5  * Copyright (C) 2024 OpenInfra Foundation Europe. 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
  21 package org.onap.ccsdk.oran.a1policymanagementservice.util.v3;
  22
  23 import org.reactivestreams.Publisher;
  24 import org.slf4j.Logger;
  25 import org.slf4j.LoggerFactory;
  26 import org.springframework.core.io.buffer.DataBuffer;
  27 import org.springframework.http.server.reactive.ServerHttpRequest;
  28 import org.springframework.http.server.reactive.ServerHttpRequestDecorator;
  29 import org.springframework.http.server.reactive.ServerHttpResponse;
  30 import org.springframework.http.server.reactive.ServerHttpResponseDecorator;
  31 import org.springframework.stereotype.Component;
  32 import org.springframework.util.MultiValueMap;
  33 import org.springframework.web.server.ServerWebExchange;
  34 import org.springframework.web.server.WebFilter;
  35 import org.springframework.web.server.WebFilterChain;
  36 import reactor.core.publisher.Flux;
  37 import reactor.core.publisher.Mono;
  38
  39 import java.io.ByteArrayOutputStream;
  40 import java.io.IOException;
  41 import java.lang.invoke.MethodHandles;
  42 import java.nio.channels.Channels;
  43 import java.nio.charset.StandardCharsets;
  44
  45 public class ReactiveEntryExitFilter implements WebFilter {
  46
  47     private static final Logger logger = LoggerFactory.getLogger(MethodHandles.lookup().lookupClass());
  48     @Override
  49     public Mono<Void> filter(ServerWebExchange exchange, WebFilterChain chain) {
  50         ServerHttpRequest httpRequest = exchange.getRequest();
  51         ServerHttpResponse httpResponse = exchange.getResponse();
  52         MultiValueMap<String, String> queryParams = httpRequest.getQueryParams();
  53         logger.info("Request received with path: {} and the Request Id: {}", httpRequest.getPath(), exchange.getRequest().getId());
  54         if (!queryParams.isEmpty())
  55             logger.trace("For the request Id: {}, the Query parameters are: {}", exchange.getRequest().getId(), queryParams);
  56
  57         ServerHttpRequestDecorator loggingServerHttpRequestDecorator = new ServerHttpRequestDecorator(exchange.getRequest()) {
  58             String requestBody = "";
  59             @Override
  60             public Flux<DataBuffer> getBody() {
  61                 return super.getBody().doOnNext(dataBuffer -> {
  62                     try (ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream()) {
  63                         Channels.newChannel(byteArrayOutputStream).write(dataBuffer.asByteBuffer().asReadOnlyBuffer());
  64                         requestBody = byteArrayOutputStream.toString(StandardCharsets.UTF_8);
  65                         if (requestBody != null)
  66                             logger.trace("For the request ID: {} the received request body: {}", exchange.getRequest().getId(), requestBody);
  67                     } catch (IOException e) {
  68                         logger.info("For the request ID: {} we got the error during processing: {}", exchange.getRequest().getId(), e.getMessage());
  69                         logger.trace("For the request ID: {} the received request body: {}", exchange.getRequest().getId(), requestBody);
  70                     }
  71                 });
  72             }
  73         };
  74
  75         ServerHttpResponseDecorator loggingServerHttpResponseDecorator = new ServerHttpResponseDecorator(exchange.getResponse()) {
  76             String responseBody = "";
  77             @Override
  78             public Mono<Void> writeWith(Publisher<? extends DataBuffer> body) {
  79                 Flux<DataBuffer> buffer = Flux.from(body);
  80                 return super.writeWith(buffer.doOnNext(dataBuffer -> {
  81                     try (ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream()) {
  82                         Channels.newChannel(byteArrayOutputStream).write(dataBuffer.toByteBuffer().asReadOnlyBuffer());
  83                         responseBody = byteArrayOutputStream.toString(StandardCharsets.UTF_8);
  84                         logger.info("For the request ID: {} the Status code of the response: {}", exchange.getRequest().getId(), httpResponse.getStatusCode());
  85                         logger.trace("For the request ID: {} the response is: {} ", exchange.getRequest().getId(), responseBody);
  86                     } catch (Exception e) {
  87                         logger.info("For the request ID: {} we got an Error during processing: {}", exchange.getRequest().getId(), e.getMessage());
  88                         logger.trace("For the request ID: {} the response body :: {}", exchange.getRequest().getId(), responseBody);
  89                     }
  90                 }));
  91             }
  92         };
  93         return chain.filter(exchange.mutate().request(loggingServerHttpRequestDecorator).response(loggingServerHttpResponseDecorator).build());
  94     }
  95
  96
  97 }