X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?p=controller.git;a=blobdiff_plain;f=opendaylight%2Fmd-sal%2Fsal-remoterpc-connector%2Fsrc%2Fmain%2Fjava%2Forg%2Fopendaylight%2Fcontroller%2Fremote%2Frpc%2Fregistry%2Fmbeans%2FRemoteRpcRegistryMXBeanImpl.java;h=5442c403a7ba8a3273d69ef7898d2736f75d77e1;hp=eabb170db244ce260a65dcb8e2dde86eb304be21;hb=c9aab0231686bc6aa06dcfef51fca8c272fb9382;hpb=fafb6cc1d45f0ef060cec7fe5bad2f6acbca94c4 diff --git a/opendaylight/md-sal/sal-remoterpc-connector/src/main/java/org/opendaylight/controller/remote/rpc/registry/mbeans/RemoteRpcRegistryMXBeanImpl.java b/opendaylight/md-sal/sal-remoterpc-connector/src/main/java/org/opendaylight/controller/remote/rpc/registry/mbeans/RemoteRpcRegistryMXBeanImpl.java index eabb170db2..5442c403a7 100644 --- a/opendaylight/md-sal/sal-remoterpc-connector/src/main/java/org/opendaylight/controller/remote/rpc/registry/mbeans/RemoteRpcRegistryMXBeanImpl.java +++ b/opendaylight/md-sal/sal-remoterpc-connector/src/main/java/org/opendaylight/controller/remote/rpc/registry/mbeans/RemoteRpcRegistryMXBeanImpl.java @@ -9,6 +9,7 @@ package org.opendaylight.controller.remote.rpc.registry.mbeans; import akka.actor.Address; +import akka.util.Timeout; import java.util.HashMap; import java.util.HashSet; import java.util.Map; @@ -17,10 +18,12 @@ import java.util.Set; import org.opendaylight.controller.md.sal.common.util.jmx.AbstractMXBean; import org.opendaylight.controller.md.sal.dom.api.DOMRpcIdentifier; import org.opendaylight.controller.remote.rpc.registry.RoutingTable; -import org.opendaylight.controller.remote.rpc.registry.RpcRegistry; import org.opendaylight.controller.remote.rpc.registry.gossip.Bucket; +import org.opendaylight.controller.remote.rpc.registry.gossip.BucketStoreAccess; import org.slf4j.Logger; import org.slf4j.LoggerFactory; +import scala.concurrent.Await; +import scala.concurrent.Future; public class RemoteRpcRegistryMXBeanImpl extends AbstractMXBean implements RemoteRpcRegistryMXBean { @@ -33,17 +36,38 @@ public class RemoteRpcRegistryMXBeanImpl extends AbstractMXBean implements Remot private static final String NAME_CONSTANT = " | name:"; - private final RpcRegistry rpcRegistry; + private final BucketStoreAccess rpcRegistryAccess; + private final Timeout timeout; - public RemoteRpcRegistryMXBeanImpl(final RpcRegistry rpcRegistry) { + public RemoteRpcRegistryMXBeanImpl(final BucketStoreAccess rpcRegistryAccess, Timeout timeout) { super("RemoteRpcRegistry", "RemoteRpcBroker", null); - this.rpcRegistry = rpcRegistry; + this.rpcRegistryAccess = rpcRegistryAccess; + this.timeout = timeout; registerMBean(); } + @SuppressWarnings({"unchecked", "checkstyle:IllegalCatch", "rawtypes"}) + private RoutingTable getLocalData() { + try { + return (RoutingTable) Await.result((Future) rpcRegistryAccess.getLocalData(), timeout.duration()); + } catch (Exception e) { + throw new RuntimeException("getLocalData failed", e); + } + } + + @SuppressWarnings({"unchecked", "checkstyle:IllegalCatch", "rawtypes"}) + private Map
> getRemoteBuckets() { + try { + return (Map>) Await.result((Future)rpcRegistryAccess.getRemoteBuckets(), + timeout.duration()); + } catch (Exception e) { + throw new RuntimeException("getRemoteBuckets failed", e); + } + } + @Override public Set