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
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=====================================
21 package org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.impl;
23 import static org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.impl.Commons.extractFailReason;
25 import com.google.gson.JsonArray;
26 import com.google.gson.JsonElement;
27 import io.vavr.collection.HashMap;
28 import io.vavr.collection.List;
29 import java.time.Duration;
30 import java.util.stream.Collectors;
31 import org.jetbrains.annotations.NotNull;
32 import org.onap.dcaegen2.services.sdk.rest.services.adapters.http.HttpHeaders;
33 import org.onap.dcaegen2.services.sdk.rest.services.adapters.http.HttpMethod;
34 import org.onap.dcaegen2.services.sdk.rest.services.adapters.http.HttpRequest;
35 import org.onap.dcaegen2.services.sdk.rest.services.adapters.http.HttpResponse;
36 import org.onap.dcaegen2.services.sdk.rest.services.adapters.http.ImmutableHttpRequest;
37 import org.onap.dcaegen2.services.sdk.rest.services.adapters.http.RequestBody;
38 import org.onap.dcaegen2.services.sdk.rest.services.adapters.http.RxHttpClient;
39 import org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.ContentType;
40 import org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.api.MessageRouterPublisher;
41 import org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.model.ImmutableMessageRouterPublishResponse;
42 import org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.model.MessageRouterPublishRequest;
43 import org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.model.MessageRouterPublishResponse;
44 import org.reactivestreams.Publisher;
45 import org.slf4j.Logger;
46 import org.slf4j.LoggerFactory;
47 import reactor.core.publisher.Flux;
48 import reactor.core.publisher.Mono;
51 * @author <a href="mailto:piotr.jaszczyk@nokia.com">Piotr Jaszczyk</a>
54 public class MessageRouterPublisherImpl implements MessageRouterPublisher {
55 private final RxHttpClient httpClient;
56 private final int maxBatchSize;
57 private final Duration maxBatchDuration;
58 private static final Logger LOGGER = LoggerFactory.getLogger(MessageRouterPublisherImpl.class);
60 public MessageRouterPublisherImpl(RxHttpClient httpClient, int maxBatchSize, Duration maxBatchDuration) {
61 this.httpClient = httpClient;
62 this.maxBatchSize = maxBatchSize;
63 this.maxBatchDuration = maxBatchDuration;
67 public Flux<MessageRouterPublishResponse> put(
68 MessageRouterPublishRequest request,
69 Flux<? extends JsonElement> items) {
70 return items.bufferTimeout(maxBatchSize, maxBatchDuration)
71 .flatMap(subItems -> subItems.isEmpty() ? Mono.empty() : pushBatchToMr(request, List.ofAll(subItems)));
74 private Publisher<? extends MessageRouterPublishResponse> pushBatchToMr(
75 MessageRouterPublishRequest request,
76 List<JsonElement> batch) {
77 LOGGER.debug("Sending a batch of {} items to DMaaP MR", batch.size());
78 LOGGER.trace("The items to be sent: {}", batch);
79 return httpClient.call(buildHttpRequest(request, createBody(batch, request.contentType())))
80 .map(httpResponse -> buildResponse(httpResponse, batch));
83 private @NotNull RequestBody createBody(List<? extends JsonElement> subItems, ContentType contentType) {
84 if(contentType == ContentType.APPLICATION_JSON) {
85 final JsonArray elements = new JsonArray(subItems.size());
86 subItems.forEach(elements::add);
87 return RequestBody.fromJson(elements);
88 }else if(contentType == ContentType.TEXT_PLAIN){
89 String messages = subItems.map(JsonElement::toString)
90 .collect(Collectors.joining("\n"));
91 return RequestBody.fromString(messages);
92 }else throw new IllegalArgumentException("Unsupported content type: " + contentType);
95 private @NotNull HttpRequest buildHttpRequest(MessageRouterPublishRequest request, RequestBody body) {
96 return ImmutableHttpRequest.builder()
97 .method(HttpMethod.POST)
98 .url(request.sinkDefinition().topicUrl())
99 .diagnosticContext(request.diagnosticContext().withNewInvocationId())
100 .customHeaders(HashMap.of(HttpHeaders.CONTENT_TYPE, request.contentType().toString()))
105 private MessageRouterPublishResponse buildResponse(
106 HttpResponse httpResponse, List<JsonElement> batch) {
107 final ImmutableMessageRouterPublishResponse.Builder builder =
108 ImmutableMessageRouterPublishResponse.builder();
110 return httpResponse.successful()
111 ? builder.items(batch).build()
112 : builder.failReason(extractFailReason(httpResponse)).build();