Commit 4a13f9b2 authored by Christos Tranoris's avatar Christos Tranoris
Browse files

fix for #15

parent a1676c1a
Loading
Loading
Loading
Loading
Loading
+17 −11
Original line number Diff line number Diff line
@@ -16,7 +16,7 @@ import org.springframework.stereotype.Component;
import org.springframework.web.reactive.function.client.WebClient;
import org.springframework.web.util.UriComponentsBuilder;
import reactor.core.publisher.Mono;

import java.net.URI;
import java.time.Duration;
import java.time.OffsetDateTime;
import java.util.concurrent.TimeUnit;
@@ -37,25 +37,31 @@ public class PrometheusQueries {

    public String sendQueryToPrometheus(String prometheusUrl, String query, MeasurementCollectionJob mcj, Job job) {
        UriComponentsBuilder builder = UriComponentsBuilder.fromHttpUrl(prometheusUrl).path("/api/v1/query").query(query);
        String url = builder.toUriString();
        URI uri = builder.build().toUri();
        
        logger.atInfo().log("Sent query at prometheus with URL: " + prometheusUrl + " with query: " + query);
        logger.atInfo().log("Sent query  builder.toUriString(): " + builder.toUriString());
        logger.atInfo().log("Sent query  builder.toUriString(): " + url);
        logger.atInfo().log("Sent query  builder.build().toUri(): " + uri.toString() );

        WebClient webclient = WebClient.create();
        String response;

        if (webclient != null) {
            try {
                String url = builder.toUriString();
                response = webclient.get().uri(url)
              
                response = webclient.get().uri( uri ) 
                        //.header("Authorization", "Basic " + encodedClientData)
                        //.attributes( ServletOAuth2AuthorizedClientExchangeFilterFunction.clientRegistrationId("authOpensliceProvider"))
                        .retrieve().onStatus(HttpStatusCode::is4xxClientError, r -> {
                            logger.error("4xx eror");
                            return Mono.error(new RuntimeException("4xx"));
                        }).onStatus(HttpStatusCode::is5xxServerError, r -> {
                            logger.error("5xx eror");
                            return Mono.error(new RuntimeException("5xx"));
                        }).bodyToMono(new ParameterizedTypeReference<String>() {
                        .retrieve()
//                        .onStatus(HttpStatusCode::is4xxClientError, r -> {
//                            logger.error("4xx eror");
//                            return Mono.error(new RuntimeException("4xx"));
//                        }).onStatus(HttpStatusCode::is5xxServerError, r -> {
//                            logger.error("5xx eror");
//                            return Mono.error(new RuntimeException("5xx"));
//                        })
                        .bodyToMono(new ParameterizedTypeReference<String>() {
                        }).block();
                logger.atDebug().log("Received " + response);
                String[] promResponse = response.split("\n");