TCA: Support for VES/A&AI enrichment
[dcaegen2/analytics/tca.git] / dcae-analytics-dmaap / src / main / java / org / openecomp / dcae / apod / analytics / dmaap / service / publisher / DMaaPMRPublisherQueueImpl.java
index 936abe3..e42b6b0 100644 (file)
-/*
- * ===============================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.openecomp.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();
-    }
-}
+/*\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