components/kpi-computation-ms/src/main/java/org/onap/dcaegen2/kpi/utils/DmaapUtils.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  *  Copyright (C) 2021 China Mobile.
   4  *  Copyright (c) 2021-2022 Wipro Limited.
   5  * ================================================================================
   6  * Licensed under the Apache License, Version 2.0 (the "License");
   7  * you may not use this file except in compliance with the License.
   8  * You may obtain a copy of the License at
   9  *
  10  *      http://www.apache.org/licenses/LICENSE-2.0
  11  *
  12  * Unless required by applicable law or agreed to in writing, software
  13  * distributed under the License is distributed on an "AS IS" BASIS,
  14  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15  * See the License for the specific language governing permissions and
  16  * limitations under the License.
  17  *
  18  * SPDX-License-Identifier: Apache-2.0
  19  * ============LICENSE_END=========================================================
  20  */
  21
  22 package org.onap.dcaegen2.kpi.utils;
  23
  24 import org.onap.dcaegen2.kpi.models.Configuration;
  25 import org.onap.dcaegen2.services.sdk.model.streams.ImmutableAafCredentials;
  26 import org.onap.dcaegen2.services.sdk.model.streams.dmaap.ImmutableMessageRouterSink;
  27 import org.onap.dcaegen2.services.sdk.model.streams.dmaap.ImmutableMessageRouterSource;
  28 import org.onap.dcaegen2.services.sdk.model.streams.dmaap.MessageRouterSink;
  29 import org.onap.dcaegen2.services.sdk.model.streams.dmaap.MessageRouterSource;
  30 import org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.api.DmaapClientFactory;
  31 import org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.api.MessageRouterPublisher;
  32 import org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.api.MessageRouterSubscriber;
  33 import org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.model.ImmutableMessageRouterPublishRequest;
  34 import org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.model.ImmutableMessageRouterSubscribeRequest;
  35 import org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.model.MessageRouterPublishRequest;
  36 import org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.model.MessageRouterSubscribeRequest;
  37 import org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.model.config.MessageRouterPublisherConfig;
  38 import org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.model.config.MessageRouterSubscriberConfig;
  39
  40 /**
  41  * Utility class to perform actions related to Dmaap.
  42  *
  43  * @author Kai Lu
  44  *
  45  */
  46 public class DmaapUtils {
  47
  48     public MessageRouterPublisher buildPublisher() {
  49         final MessageRouterPublisher publisher = DmaapClientFactory
  50              .createMessageRouterPublisher(MessageRouterPublisherConfig.createDefault());
  51         return publisher;
  52     }
  53
  54     public MessageRouterPublishRequest buildPublisherRequest(Configuration config, String topicUrl) {
  55         MessageRouterSink sinkDefinition = ImmutableMessageRouterSink.builder().topicUrl(topicUrl)
  56              .aafCredentials(ImmutableAafCredentials.builder().username(config.getAafUsername())
  57              .password(config.getAafPassword()).build())
  58              .build();
  59         MessageRouterPublishRequest request = ImmutableMessageRouterPublishRequest.builder()
  60              .sinkDefinition(sinkDefinition).build();
  61         return request;
  62     }
  63
  64     public MessageRouterSubscriber buildSubscriber() {
  65         MessageRouterSubscriber subscriber = DmaapClientFactory
  66             .createMessageRouterSubscriber(MessageRouterSubscriberConfig.createDefault());
  67         return subscriber;
  68     }
  69
  70     public MessageRouterSubscribeRequest buildSubscriberRequest(Configuration config, String topicUrl) {
  71         MessageRouterSource sourceDefinition = ImmutableMessageRouterSource.builder().topicUrl(topicUrl)
  72              .aafCredentials(ImmutableAafCredentials.builder().username(config.getAafUsername())
  73              .password(config.getAafPassword()).build())
  74              .build();
  75         MessageRouterSubscribeRequest request = ImmutableMessageRouterSubscribeRequest.builder()
  76              .consumerGroup(config.getCg()).consumerId(config.getCid()).sourceDefinition(sourceDefinition).build();
  77         return request;
  78     }
  79 }