-/*\r
- * ===============================LICENSE_START======================================\r
- * dcae-analytics\r
- * ================================================================================\r
- * Copyright © 2017 AT&T Intellectual Property. All rights reserved.\r
- * ================================================================================\r
- * Licensed under the Apache License, Version 2.0 (the "License");\r
- * you may not use this file except in compliance with the License.\r
- * You may obtain a copy of the License at\r
- *\r
- * http://www.apache.org/licenses/LICENSE-2.0\r
- *\r
- * Unless required by applicable law or agreed to in writing, software\r
- * distributed under the License is distributed on an "AS IS" BASIS,\r
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
- * See the License for the specific language governing permissions and\r
- * limitations under the License.\r
- * ============================LICENSE_END===========================================\r
- */\r
-\r
-package org.openecomp.dcae.apod.analytics.dmaap.service.publisher;\r
-\r
-import com.google.inject.Inject;\r
-import com.google.inject.assistedinject.Assisted;\r
-import org.slf4j.Logger;\r
-import org.slf4j.LoggerFactory;\r
-\r
-import java.util.LinkedList;\r
-import java.util.List;\r
-import java.util.concurrent.LinkedBlockingDeque;\r
-\r
-import static com.google.common.collect.Iterables.concat;\r
-import static com.google.common.collect.Lists.newLinkedList;\r
-import static java.util.Collections.unmodifiableList;\r
-\r
-/**\r
- * <p>\r
- * An implementation of {@link DMaaPMRPublisherQueue} which uses {@link java.util.concurrent.BlockingDeque}\r
- * for batch and recovery queues\r
- * </p>\r
- *\r
- *\r
- * @author Rajiv Singla . Creation Date: 11/1/2016.\r
- */\r
-public class DMaaPMRPublisherQueueImpl implements DMaaPMRPublisherQueue {\r
-\r
- private static final Logger LOG = LoggerFactory.getLogger(DMaaPMRPublisherQueueImpl.class);\r
-\r
- private final LinkedBlockingDeque<String> batchQueue;\r
- private final LinkedBlockingDeque<String> recoveryQueue;\r
-\r
- @Inject\r
- public DMaaPMRPublisherQueueImpl(@Assisted("batchQueueSize") int batchQueueSize,\r
- @Assisted("recoveryQueueSize") int recoveryQueueSize) {\r
- batchQueue = new LinkedBlockingDeque<>(batchQueueSize);\r
- recoveryQueue = new LinkedBlockingDeque<>(recoveryQueueSize);\r
- LOG.debug("Creating Instance of DMaaP Publisher Queue. BatchQueueSize: {}, RecoveryQueueSize: {}",\r
- batchQueueSize, recoveryQueueSize);\r
- }\r
-\r
- @Override\r
- public synchronized int addBatchMessages(List<String> batchMessages) {\r
-\r
- // checks if batchMessages size does not exceed batch queue capacity\r
- if (batchMessages.size() > batchQueue.remainingCapacity()) {\r
- throw new IllegalStateException("Not enough capacity to add batchMessages in batch queue");\r
- }\r
-\r
- // Add batchMessages to batch queue\r
- for (String message : batchMessages) {\r
- batchQueue.add(message);\r
- }\r
-\r
- // returns current elements size in batch queue\r
- return batchQueue.size();\r
- }\r
-\r
- @Override\r
- public synchronized int addRecoverableMessages(List<String> recoverableMessages) {\r
-\r
- // checks if messages size does not exceed recovery queue size\r
- if (recoverableMessages.size() > recoveryQueue.remainingCapacity()) {\r
- throw new IllegalStateException("Not enough capacity to add messages in recovery queue");\r
- }\r
-\r
- // add messages to recovery queue\r
- for (String recoverableMessage : recoverableMessages) {\r
- recoveryQueue.add(recoverableMessage);\r
- }\r
-\r
- // returns current size of recovery queue\r
- return recoveryQueue.size();\r
- }\r
-\r
- @Override\r
- public synchronized List<String> getMessageForPublishing() {\r
-\r
- final List<String> recoveryMessageList = new LinkedList<>();\r
- final List<String> batchMessagesList = new LinkedList<>();\r
-\r
- // get messages from recovery queue if present\r
- if (!recoveryQueue.isEmpty()) {\r
- final int recoveryQueueSize = recoveryQueue.drainTo(recoveryMessageList);\r
- LOG.debug("Drained Recovery Queue elements for flushing: {}", recoveryQueueSize);\r
- }\r
-\r
- // get messages from batch queue if present\r
- if (!batchQueue.isEmpty()) {\r
- final int batchQueueSize = batchQueue.drainTo(batchMessagesList);\r
- LOG.debug("Drained Batch Queue elements for flushing: {}", batchQueueSize);\r
- }\r
-\r
- // concat recovery and batch queue elements\r
- return unmodifiableList(newLinkedList(concat(recoveryMessageList, batchMessagesList)));\r
- }\r
-\r
- @Override\r
- public synchronized int getBatchQueueRemainingSize() {\r
- return batchQueue.remainingCapacity();\r
- }\r
-\r
- @Override\r
- public synchronized int getRecoveryQueueRemainingSize() {\r
- return recoveryQueue.remainingCapacity();\r
- }\r
-}\r
+/*
+ * ===============================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.dcae.apod.analytics.dmaap.service.publisher;
+
+import com.google.inject.Inject;
+import com.google.inject.assistedinject.Assisted;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.util.LinkedList;
+import java.util.List;
+import java.util.concurrent.LinkedBlockingDeque;
+
+import static com.google.common.collect.Iterables.concat;
+import static com.google.common.collect.Lists.newLinkedList;
+import static java.util.Collections.unmodifiableList;
+
+/**
+ * <p>
+ * An implementation of {@link DMaaPMRPublisherQueue} which uses {@link java.util.concurrent.BlockingDeque}
+ * for batch and recovery queues
+ * </p>
+ *
+ *
+ * @author Rajiv Singla . Creation Date: 11/1/2016.
+ */
+public class DMaaPMRPublisherQueueImpl implements DMaaPMRPublisherQueue {
+
+ private static final Logger LOG = LoggerFactory.getLogger(DMaaPMRPublisherQueueImpl.class);
+
+ private final LinkedBlockingDeque<String> batchQueue;
+ private final LinkedBlockingDeque<String> recoveryQueue;
+
+ @Inject
+ public DMaaPMRPublisherQueueImpl(@Assisted("batchQueueSize") int batchQueueSize,
+ @Assisted("recoveryQueueSize") int recoveryQueueSize) {
+ batchQueue = new LinkedBlockingDeque<>(batchQueueSize);
+ recoveryQueue = new LinkedBlockingDeque<>(recoveryQueueSize);
+ LOG.debug("Creating Instance of DMaaP Publisher Queue. BatchQueueSize: {}, RecoveryQueueSize: {}",
+ batchQueueSize, recoveryQueueSize);
+ }
+
+ @Override
+ public synchronized int addBatchMessages(List<String> batchMessages) {
+
+ // checks if batchMessages size does not exceed batch queue capacity
+ if (batchMessages.size() > batchQueue.remainingCapacity()) {
+ throw new IllegalStateException("Not enough capacity to add batchMessages in batch queue");
+ }
+
+ // Add batchMessages to batch queue
+ for (String message : batchMessages) {
+ batchQueue.add(message);
+ }
+
+ // returns current elements size in batch queue
+ return batchQueue.size();
+ }
+
+ @Override
+ public synchronized int addRecoverableMessages(List<String> recoverableMessages) {
+
+ // checks if messages size does not exceed recovery queue size
+ if (recoverableMessages.size() > recoveryQueue.remainingCapacity()) {
+ throw new IllegalStateException("Not enough capacity to add messages in recovery queue");
+ }
+
+ // add messages to recovery queue
+ for (String recoverableMessage : recoverableMessages) {
+ recoveryQueue.add(recoverableMessage);
+ }
+
+ // returns current size of recovery queue
+ return recoveryQueue.size();
+ }
+
+ @Override
+ public synchronized List<String> getMessageForPublishing() {
+
+ final List<String> recoveryMessageList = new LinkedList<>();
+ final List<String> batchMessagesList = new LinkedList<>();
+
+ // get messages from recovery queue if present
+ if (!recoveryQueue.isEmpty()) {
+ final int recoveryQueueSize = recoveryQueue.drainTo(recoveryMessageList);
+ LOG.debug("Drained Recovery Queue elements for flushing: {}", recoveryQueueSize);
+ }
+
+ // get messages from batch queue if present
+ if (!batchQueue.isEmpty()) {
+ final int batchQueueSize = batchQueue.drainTo(batchMessagesList);
+ LOG.debug("Drained Batch Queue elements for flushing: {}", batchQueueSize);
+ }
+
+ // concat recovery and batch queue elements
+ return unmodifiableList(newLinkedList(concat(recoveryMessageList, batchMessagesList)));
+ }
+
+ @Override
+ public synchronized int getBatchQueueRemainingSize() {
+ return batchQueue.remainingCapacity();
+ }
+
+ @Override
+ public synchronized int getRecoveryQueueRemainingSize() {
+ return recoveryQueue.remainingCapacity();
+ }
+}
Code Review / dcaegen2 / analytics / tca.git / blobdiff