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;
import java.util.Map.Entry;
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.sal.connector.api.RpcRouter;
+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 {
protected final Logger log = LoggerFactory.getLogger(getClass());
- private static final String NULL_CONSTANT = "null";
-
private static final String LOCAL_CONSTANT = "local";
private static final String ROUTE_CONSTANT = "route:";
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<Address, Bucket<RoutingTable>> getRemoteBuckets() {
+ try {
+ return (Map<Address, Bucket<RoutingTable>>) Await.result((Future)rpcRegistryAccess.getRemoteBuckets(),
+ timeout.duration());
+ } catch (Exception e) {
+ throw new RuntimeException("getRemoteBuckets failed", e);
+ }
+ }
+
@Override
public Set<String> getGlobalRpc() {
- RoutingTable table = rpcRegistry.getLocalBucket().getData();
+ RoutingTable table = getLocalData();
Set<String> globalRpc = new HashSet<>(table.getRoutes().size());
- for (RpcRouter.RouteIdentifier<?, ?, ?> route : table.getRoutes()) {
- if (route.getRoute() == null) {
- globalRpc.add(route.getType() != null ? route.getType().toString() : NULL_CONSTANT);
+ for (DOMRpcIdentifier route : table.getRoutes()) {
+ if (route.getContextReference().isEmpty()) {
+ globalRpc.add(route.getType().toString());
}
}
@Override
public Set<String> getLocalRegisteredRoutedRpc() {
- RoutingTable table = rpcRegistry.getLocalBucket().getData();
+ RoutingTable table = getLocalData();
Set<String> routedRpc = new HashSet<>(table.getRoutes().size());
- for (RpcRouter.RouteIdentifier<?, ?, ?> route : table.getRoutes()) {
- if (route.getRoute() != null) {
+ for (DOMRpcIdentifier route : table.getRoutes()) {
+ if (!route.getContextReference().isEmpty()) {
StringBuilder builder = new StringBuilder(ROUTE_CONSTANT);
- builder.append(route.getRoute().toString()).append(NAME_CONSTANT).append(route.getType() != null
- ? route.getType().toString() : NULL_CONSTANT);
+ builder.append(route.getContextReference().toString()).append(NAME_CONSTANT).append(route.getType());
routedRpc.add(builder.toString());
}
}
@Override
public Map<String, String> findRpcByName(final String name) {
- RoutingTable localTable = rpcRegistry.getLocalBucket().getData();
+ RoutingTable localTable = getLocalData();
// Get all RPCs from local bucket
Map<String, String> rpcMap = new HashMap<>(getRpcMemberMapByName(localTable, name, LOCAL_CONSTANT));
// Get all RPCs from remote bucket
- Map<Address, Bucket<RoutingTable>> buckets = rpcRegistry.getRemoteBuckets();
+ Map<Address, Bucket<RoutingTable>> buckets = getRemoteBuckets();
for (Entry<Address, Bucket<RoutingTable>> entry : buckets.entrySet()) {
RoutingTable table = entry.getValue().getData();
rpcMap.putAll(getRpcMemberMapByName(table, name, entry.getKey().toString()));
}
@Override
- public Map<String, String> findRpcByRoute(String routeId) {
- RoutingTable localTable = rpcRegistry.getLocalBucket().getData();
+ public Map<String, String> findRpcByRoute(final String routeId) {
+ RoutingTable localTable = getLocalData();
Map<String, String> rpcMap = new HashMap<>(getRpcMemberMapByRoute(localTable, routeId, LOCAL_CONSTANT));
- Map<Address, Bucket<RoutingTable>> buckets = rpcRegistry.getRemoteBuckets();
+ Map<Address, Bucket<RoutingTable>> buckets = getRemoteBuckets();
for (Entry<Address, Bucket<RoutingTable>> entry : buckets.entrySet()) {
RoutingTable table = entry.getValue().getData();
rpcMap.putAll(getRpcMemberMapByRoute(table, routeId, entry.getKey().toString()));
/**
* Search if the routing table route String contains routeName.
*/
- private Map<String,String> getRpcMemberMapByRoute(final RoutingTable table, final String routeName,
+ private static Map<String,String> getRpcMemberMapByRoute(final RoutingTable table, final String routeName,
final String address) {
- Set<RpcRouter.RouteIdentifier<?, ?, ?>> routes = table.getRoutes();
+ Set<DOMRpcIdentifier> routes = table.getRoutes();
Map<String, String> rpcMap = new HashMap<>(routes.size());
- for (RpcRouter.RouteIdentifier<?, ?, ?> route : table.getRoutes()) {
- if (route.getRoute() != null) {
- String routeString = route.getRoute().toString();
+ for (DOMRpcIdentifier route : routes) {
+ if (!route.getContextReference().isEmpty()) {
+ String routeString = route.getContextReference().toString();
if (routeString.contains(routeName)) {
StringBuilder builder = new StringBuilder(ROUTE_CONSTANT);
- builder.append(routeString).append(NAME_CONSTANT).append(route.getType() != null
- ? route.getType().toString() : NULL_CONSTANT);
+ builder.append(routeString).append(NAME_CONSTANT).append(route.getType());
rpcMap.put(builder.toString(), address);
}
}
/**
* Search if the routing table route type contains name.
*/
- private Map<String, String> getRpcMemberMapByName(final RoutingTable table, final String name,
+ private static Map<String, String> getRpcMemberMapByName(final RoutingTable table, final String name,
final String address) {
- Set<RpcRouter.RouteIdentifier<?, ?, ?>> routes = table.getRoutes();
+ Set<DOMRpcIdentifier> routes = table.getRoutes();
Map<String, String> rpcMap = new HashMap<>(routes.size());
- for (RpcRouter.RouteIdentifier<?, ?, ?> route : routes) {
- if (route.getType() != null) {
+ for (DOMRpcIdentifier route : routes) {
+ if (!route.getContextReference().isEmpty()) {
String type = route.getType().toString();
if (type.contains(name)) {
StringBuilder builder = new StringBuilder(ROUTE_CONSTANT);
- builder.append(route.getRoute() != null ? route.getRoute().toString() : NULL_CONSTANT)
- .append(NAME_CONSTANT).append(type);
+ builder.append(route.getContextReference()).append(NAME_CONSTANT).append(type);
rpcMap.put(builder.toString(), address);
}
}
}
@Override
+ @SuppressWarnings({"unchecked", "checkstyle:IllegalCatch", "rawtypes"})
public String getBucketVersions() {
- return rpcRegistry.getVersions().toString();
+ try {
+ return Await.result((Future)rpcRegistryAccess.getBucketVersions(), timeout.duration()).toString();
+ } catch (Exception e) {
+ throw new RuntimeException("getVersions failed", e);
+ }
}
}
Code Review / controller.git / blobdiff