1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
|
/*
* ===============================LICENSE_START======================================
* dcae-analytics
* ================================================================================
* Copyright © 2017 AT&T Intellectual Property. All rights reserved.
* ================================================================================
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
* ============================LICENSE_END===========================================
*/
package org.onap.universalvesadapter.dmaap.MRPublisher;
import static java.lang.String.format;
import java.io.IOException;
import java.net.URI;
import java.util.Date;
import java.util.List;
import org.apache.commons.lang3.tuple.Pair;
import org.apache.http.HttpHeaders;
import org.apache.http.client.HttpClient;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.entity.ContentType;
import org.apache.http.entity.StringEntity;
import org.apache.http.impl.client.CloseableHttpClient;
import org.onap.universalvesadapter.configs.DMaaPMRPublisherConfig;
import org.onap.universalvesadapter.dmaap.BaseDMaaPMRComponent;
import org.onap.universalvesadapter.exception.DMaapException;
import org.onap.universalvesadapter.utils.HTTPUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Value;
import com.att.aft.dme2.internal.springframework.context.annotation.ComponentScan;
import com.google.common.base.Optional;
import com.google.common.collect.Iterables;
import com.google.common.collect.Lists;
import com.google.inject.Inject;
import com.google.inject.assistedinject.Assisted;
/**
* Concrete Implementation of {@link DMaaPMRPublisher} which uses {@link HttpClient}
*
* @author Rajiv Singla . Creation Date: 10/13/2016.
*/
@ComponentScan
public class DMaaPMRPublisherImpl extends BaseDMaaPMRComponent implements DMaaPMRPublisher {
@Value("${mr.publisher.PUBLISHER_MAX_FLUSH_RETRIES_ON_CLOSE}")
private int publisherMaxFlushRetries;
private static final Logger LOG = LoggerFactory.getLogger(DMaaPMRPublisherImpl.class);
public int a =2;
private final DMaaPMRPublisherConfig publisherConfig;
private final CloseableHttpClient closeableHttpClient;
private final DMaaPMRPublisherQueue publisherQueue;
private final Date publisherCreationTime;
private URI publisherUri;
@Inject
public DMaaPMRPublisherImpl(@Assisted DMaaPMRPublisherConfig publisherConfig,
DMaaPMRPublisherQueueFactory dMaaPMRPublisherQueueFactory,
CloseableHttpClient closeableHttpClient){
this.publisherConfig = publisherConfig;
final int maxBatchSize = publisherConfig.getMaxBatchSize() > 0 ? publisherConfig.getMaxBatchSize() : 1;
this.publisherQueue = dMaaPMRPublisherQueueFactory.create(
maxBatchSize, publisherConfig.getMaxRecoveryQueueSize());
this.closeableHttpClient = closeableHttpClient;
this.publisherUri = createPublisherURI(publisherConfig);
this.publisherCreationTime = new Date();
}
@Override
public DMaaPMRPublisherResponse publish(List<String> messages) {
final int batchQueueRemainingSize = publisherQueue.getBatchQueueRemainingSize();
// if messages size is less than batch queue size - just queue them for batch publishing
if (batchQueueRemainingSize > messages.size()) {
LOG.debug("Adding messages to batch Queue. No flushing required. Messages Size:{}. Batch Queue Size:{}",
messages.size(), batchQueueRemainingSize);
final int batchQueueSize = publisherQueue.addBatchMessages(messages);
return createPublisherAcceptedResponse(batchQueueSize);
} else {
// grab all already queued messages, append current messages and force publish them to DMaaP MR topic
final List<String> queueMessages = publisherQueue.getMessageForPublishing();
LOG.debug("Batch Queue capacity exceeds messages size. Flushing of all pending messages to DMaaP MR " +
"Publisher Topic.");
return forcePublish(Lists.newLinkedList(Iterables.concat(queueMessages, messages)));
}
}
@Override
public DMaaPMRPublisherResponse forcePublish(List<String> messages) {
LOG.debug("Force publishing messages to DMaaP MR Topic. Messages Size: {}", messages.size());
final String contentType = publisherConfig.getContentType();
final String userName =(publisherConfig.getUserName().equals("null")) ? null : publisherConfig.getUserName();
final String userPassword = (publisherConfig.getUserPassword().equals("null")) ? null : publisherConfig.getUserPassword();
final HttpPost postRequest = new HttpPost(publisherUri);
// add Authorization Header if username and password are present
final Optional<String> authHeader = getAuthHeader(userName, userPassword);
if (authHeader.isPresent()) {
postRequest.addHeader(HttpHeaders.AUTHORIZATION, authHeader.get());
} else {
LOG.debug("DMaaP MR Publisher Authentication is disabled as username or password is not present.");
}
// Create post string entity
final String messagesJson = convertToJsonString(messages);
final StringEntity requestEntity =
new StringEntity(messagesJson, ContentType.create(contentType, "UTF-8"));
postRequest.setEntity(requestEntity);
try {
final Pair<Integer, String> responsePair = closeableHttpClient.execute(postRequest, responseHandler());
final Integer responseCode = responsePair.getLeft();
final String responseBody = responsePair.getRight();
// if messages were published successfully, return successful response
if (HTTPUtils.isSuccessfulResponseCode(responseCode)) {
LOG.debug("DMaaP MR Messages published successfully. DMaaP Response Code: {}. DMaaP Response " +
"Body: {}, Number of Messages published: {}",
responseCode, responseBody, messages.size());
} else {
LOG.warn("Unable to publish messages to DMaaP MR Topic. DMaaP Response Code: {}, DMaaP Response " +
"Body: {}. Messages will be queued in recovery queue", responseCode, responseBody);
addMessagesToRecoveryQueue(publisherQueue, messages);
}
return createPublisherResponse(responseCode, responseBody,
getPendingMessages(publisherQueue, publisherConfig));
} catch (IOException e) {
// If IO Error then we need to also put messages in recovery queue
addMessagesToRecoveryQueue(publisherQueue, messages);
final String errorMessage = format("IO Exception while publishing messages to DMaaP Topic. " +
"Messages will be queued in recovery queue. Messages Size: %d", messages.size());
throw new DMaapException(errorMessage, LOG, e);
}
}
@Override
public DMaaPMRPublisherResponse flush() {
final List<String> queueMessages = publisherQueue.getMessageForPublishing();
// If there are no message return 204 (No Content) response code
if (queueMessages.isEmpty()) {
LOG.debug("No messages to publish to batch queue. Returning 204 status code");
return createPublisherNoContentResponse();
} else {
// force publish messages in queue
return forcePublish(queueMessages);
}
}
@Override
public Date getPublisherCreationTime() {
return new Date(publisherCreationTime.getTime());
}
@Override
public void close() throws Exception {
// flush current message in the queue
int retrialNumber = 0;
int flushResponseCode;
// automatic retries if messages cannot be flushed
do {
retrialNumber++;
DMaaPMRPublisherResponse flushResponse = flush();
flushResponseCode = flushResponse.getResponseCode();
if (!HTTPUtils.isSuccessfulResponseCode(flushResponseCode)) {
LOG.warn("Unable to flush batch messages to publisher due to DMaaP MR invalid Response: {}. " +
"Retrial No: {} of Max {} Retries", flushResponseCode, retrialNumber,
publisherMaxFlushRetries);
Thread.sleep(publisherMaxFlushRetries);
}
} while (retrialNumber <= publisherMaxFlushRetries &&
!HTTPUtils.isSuccessfulResponseCode(flushResponseCode));
if (!HTTPUtils.isSuccessfulResponseCode(flushResponseCode)) {
LOG.error("Unable to flush batch messages to publisher. Messages loss cannot be prevented");
} else {
LOG.info("Successfully published all batched messages to publisher.");
}
// close http client
closeableHttpClient.close();
}
}
|