TCA: Support for VES/A&AI enrichment
[dcaegen2/analytics/tca.git] / dcae-analytics-cdap-tca / src / main / java / org / openecomp / dcae / apod / analytics / cdap / tca / utils / AppPreferencesToSubscriberConfigMapper.java
1 /*\r
2  * ===============================LICENSE_START======================================\r
3  *  dcae-analytics\r
4  * ================================================================================\r
5  *    Copyright © 2017 AT&T Intellectual Property. All rights reserved.\r
6  * ================================================================================\r
7  *  Licensed under the Apache License, Version 2.0 (the "License");\r
8  *  you may not use this file except in compliance with the License.\r
9  *   You may obtain a copy of the License at\r
10  *\r
11  *          http://www.apache.org/licenses/LICENSE-2.0\r
12  *\r
13  *  Unless required by applicable law or agreed to in writing, software\r
14  *  distributed under the License is distributed on an "AS IS" BASIS,\r
15  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
16  *  See the License for the specific language governing permissions and\r
17  *  limitations under the License.\r
18  *  ============================LICENSE_END===========================================\r
19  */\r
20 \r
21 package org.openecomp.dcae.apod.analytics.cdap.tca.utils;\r
22 \r
23 import com.google.common.base.Function;\r
24 import org.openecomp.dcae.apod.analytics.cdap.tca.settings.TCAAppPreferences;\r
25 import org.openecomp.dcae.apod.analytics.dmaap.domain.config.DMaaPMRSubscriberConfig;\r
26 \r
27 import javax.annotation.Nonnull;\r
28 \r
29 import static org.openecomp.dcae.apod.analytics.cdap.common.utils.ValidationUtils.isPresent;\r
30 \r
31 \r
32 /**\r
33  * Function which translates {@link TCAAppPreferences} to {@link DMaaPMRSubscriberConfig}\r
34  *\r
35  * @author Rajiv Singla . Creation Date: 11/17/2016.\r
36  */\r
37 public class AppPreferencesToSubscriberConfigMapper implements Function<TCAAppPreferences, DMaaPMRSubscriberConfig> {\r
38 \r
39     /**\r
40      * Factory Method to converts {@link TCAAppPreferences} to {@link DMaaPMRSubscriberConfig} object\r
41      *\r
42      * @param tcaAppPreferences tca app preferences\r
43      * @return DMaaP Subscriber Config\r
44      */\r
45     public static DMaaPMRSubscriberConfig map(final TCAAppPreferences tcaAppPreferences) {\r
46         return new AppPreferencesToSubscriberConfigMapper().apply(tcaAppPreferences);\r
47     }\r
48 \r
49     /**\r
50      * Implementation to convert {@link TCAAppPreferences} to {@link DMaaPMRSubscriberConfig} object\r
51      *\r
52      * @param tcaAppPreferences tca app preferences\r
53      *\r
54      * @return DMaaP Subscriber Config\r
55      */\r
56     @Nonnull\r
57     @Override\r
58     public DMaaPMRSubscriberConfig apply(@Nonnull TCAAppPreferences tcaAppPreferences) {\r
59 \r
60         // Create a new subscriber settings builder\r
61         final DMaaPMRSubscriberConfig.Builder subscriberConfigBuilder = new DMaaPMRSubscriberConfig.Builder(\r
62                 tcaAppPreferences.getSubscriberHostName(), tcaAppPreferences.getSubscriberTopicName());\r
63 \r
64         // Setup up any optional subscriber parameters if they are present\r
65         final Integer subscriberHostPortNumber = tcaAppPreferences.getSubscriberHostPort();\r
66         if (subscriberHostPortNumber != null) {\r
67             subscriberConfigBuilder.setPortNumber(subscriberHostPortNumber);\r
68         }\r
69 \r
70         final String subscriberProtocol = tcaAppPreferences.getSubscriberProtocol();\r
71         if (isPresent(subscriberProtocol)) {\r
72             subscriberConfigBuilder.setProtocol(subscriberProtocol);\r
73         }\r
74 \r
75         final String subscriberUserName = tcaAppPreferences.getSubscriberUserName();\r
76         if (isPresent(subscriberUserName)) {\r
77             subscriberConfigBuilder.setUserName(subscriberUserName);\r
78         }\r
79 \r
80         final String subscriberUserPassword = tcaAppPreferences.getSubscriberUserPassword();\r
81         if (isPresent(subscriberUserPassword)) {\r
82             subscriberConfigBuilder.setUserPassword(subscriberUserPassword);\r
83         }\r
84 \r
85         final String subscriberContentType = tcaAppPreferences.getSubscriberContentType();\r
86         if (isPresent(subscriberContentType)) {\r
87             subscriberConfigBuilder.setContentType(subscriberContentType);\r
88         }\r
89 \r
90         final String subscriberConsumerId = tcaAppPreferences.getSubscriberConsumerId();\r
91         if (isPresent(subscriberConsumerId)) {\r
92             subscriberConfigBuilder.setConsumerId(subscriberConsumerId);\r
93         }\r
94 \r
95         final String subscriberConsumerGroup = tcaAppPreferences.getSubscriberConsumerGroup();\r
96         if (isPresent(subscriberConsumerGroup)) {\r
97             subscriberConfigBuilder.setConsumerGroup(subscriberConsumerGroup);\r
98         }\r
99 \r
100         final Integer subscriberTimeoutMS = tcaAppPreferences.getSubscriberTimeoutMS();\r
101         if (subscriberTimeoutMS != null) {\r
102             subscriberConfigBuilder.setTimeoutMS(subscriberTimeoutMS);\r
103         }\r
104         final Integer subscriberMessageLimit = tcaAppPreferences.getSubscriberMessageLimit();\r
105         if (subscriberMessageLimit != null) {\r
106             subscriberConfigBuilder.setMessageLimit(subscriberMessageLimit);\r
107         }\r
108 \r
109         // return Subscriber settings\r
110         return subscriberConfigBuilder.build();\r
111 \r
112     }\r
113 }\r