Migrate Queue stats collection

[yangtools.git] / common / util / src / main / java / org / opendaylight / yangtools / util / concurrent / QueuedNotificationManagerMXBeanImpl.java

diff --git a/common/util/src/main/java/org/opendaylight/yangtools/util/concurrent/QueuedNotificationManagerMXBeanImpl.java b/common/util/src/main/java/org/opendaylight/yangtools/util/concurrent/QueuedNotificationManagerMXBeanImpl.java
index 309a84e60961a18228965e564f649f9b05c4a96a..0ed76fdc69b4f500a698891208430209e80f8fae 100644 (file)
--- a/common/util/src/main/java/org/opendaylight/yangtools/util/concurrent/QueuedNotificationManagerMXBeanImpl.java
+++ b/common/util/src/main/java/org/opendaylight/yangtools/util/concurrent/QueuedNotificationManagerMXBeanImpl.java
@@ -10,6 +10,7 @@ package org.opendaylight.yangtools.util.concurrent;
 import static java.util.Objects.requireNonNull;
 
 import java.util.List;
+import java.util.stream.Collectors;
 
 final class QueuedNotificationManagerMXBeanImpl implements QueuedNotificationManagerMXBean {
     private final AbstractQueuedNotificationManager<?, ?, ?> manager;
@@ -23,7 +24,8 @@ final class QueuedNotificationManagerMXBeanImpl implements QueuedNotificationMan
      */
     @Override
     public List<ListenerNotificationQueueStats> getCurrentListenerQueueStats() {
-        return manager.getListenerNotificationQueueStats();
+        return manager.streamTasks().map(t -> new ListenerNotificationQueueStats(t.key().toString(), t.size()))
+                .collect(Collectors.toList());
     }
 
     /**