Teach sal-remoterpc-connector to route actions

[controller.git] / opendaylight / md-sal / sal-remoterpc-connector / src / main / java / org / opendaylight / controller / remote / rpc / registry / mbeans / RemoteActionRegistryMXBeanImpl.java

diff --git a/opendaylight/md-sal/sal-remoterpc-connector/src/main/java/org/opendaylight/controller/remote/rpc/registry/mbeans/RemoteActionRegistryMXBeanImpl.java b/opendaylight/md-sal/sal-remoterpc-connector/src/main/java/org/opendaylight/controller/remote/rpc/registry/mbeans/RemoteActionRegistryMXBeanImpl.java
new file mode 100644 (file)
index 0000000..cc117bd
--- /dev/null
+++ b/opendaylight/md-sal/sal-remoterpc-connector/src/main/java/org/opendaylight/controller/remote/rpc/registry/mbeans/RemoteActionRegistryMXBeanImpl.java
@@ -0,0 +1,129 @@
+/*
+ * Copyright (c) 2019 Nordix Foundation.  All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
+package org.opendaylight.controller.remote.rpc.registry.mbeans;
+
+import akka.actor.Address;
+import akka.util.Timeout;
+import java.util.Collection;
+import java.util.HashMap;
+import java.util.HashSet;
+import java.util.Map;
+import java.util.Set;
+import org.opendaylight.controller.remote.rpc.registry.ActionRoutingTable;
+import org.opendaylight.controller.remote.rpc.registry.gossip.Bucket;
+import org.opendaylight.controller.remote.rpc.registry.gossip.BucketStoreAccess;
+import org.opendaylight.mdsal.dom.api.DOMActionInstance;
+import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
+import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
+
+public class RemoteActionRegistryMXBeanImpl extends AbstractRegistryMXBean<ActionRoutingTable, DOMActionInstance>
+        implements RemoteActionRegistryMXBean {
+
+    public RemoteActionRegistryMXBeanImpl(final BucketStoreAccess actionRegistryAccess, final Timeout timeout) {
+        super("RemoteActionRegistry", "RemoteActionBroker", actionRegistryAccess, timeout);
+    }
+
+    @Override
+    public Set<String> getLocalRegisteredAction() {
+        ActionRoutingTable table = localData();
+        Set<String> routedAction = new HashSet<>(table.getItems().size());
+        for (DOMActionInstance route : table.getItems()) {
+            if (route.getType().getLastComponent() != null) {
+                final YangInstanceIdentifier actionPath = YangInstanceIdentifier.create(
+                        new NodeIdentifier(route.getType().getLastComponent()));
+                if (!actionPath.isEmpty()) {
+                    routedAction.add(ROUTE_CONSTANT + actionPath + NAME_CONSTANT + route.getType());
+                }
+            }
+        }
+
+        log.debug("Locally registered routed RPCs {}", routedAction);
+        return routedAction;
+    }
+
+    @Override
+    public Map<String, String> findActionByName(final String name) {
+        ActionRoutingTable localTable = localData();
+        // Get all Actions from local bucket
+        Map<String, String> rpcMap = new HashMap<>(getActionMemberMapByName(localTable, name, LOCAL_CONSTANT));
+
+        // Get all Actions from remote bucket
+        Map<Address, Bucket<ActionRoutingTable>> buckets = remoteBuckets();
+        for (Map.Entry<Address, Bucket<ActionRoutingTable>> entry : buckets.entrySet()) {
+            ActionRoutingTable table = entry.getValue().getData();
+            rpcMap.putAll(getActionMemberMapByName(table, name, entry.getKey().toString()));
+        }
+
+        log.debug("list of Actions {} searched by name {}", rpcMap, name);
+        return rpcMap;
+    }
+
+    @Override
+    public Map<String, String> findActionByRoute(final String routeId) {
+        ActionRoutingTable localTable = localData();
+        Map<String, String> rpcMap = new HashMap<>(getActionMemberMapByAction(localTable, routeId, LOCAL_CONSTANT));
+
+        Map<Address, Bucket<ActionRoutingTable>> buckets = remoteBuckets();
+        for (Map.Entry<Address, Bucket<ActionRoutingTable>> entry : buckets.entrySet()) {
+            ActionRoutingTable table = entry.getValue().getData();
+            rpcMap.putAll(getActionMemberMapByAction(table, routeId, entry.getKey().toString()));
+        }
+
+        log.debug("list of Actions {} searched by route {}", rpcMap, routeId);
+        return rpcMap;
+    }
+
+    /**
+     * Search if the routing table route String contains routeName.
+     */
+    private static Map<String, String> getActionMemberMapByAction(final ActionRoutingTable table,
+                                                                  final String routeName, final String address) {
+        Collection<DOMActionInstance> routes = table.getItems();
+        Map<String, String> actionMap = new HashMap<>(routes.size());
+        for (DOMActionInstance route : routes) {
+            if (route.getType().getLastComponent() != null) {
+                final YangInstanceIdentifier actionPath = YangInstanceIdentifier.create(
+                        new NodeIdentifier(route.getType().getLastComponent()));
+                if (!actionPath.isEmpty()) {
+                    String routeString = actionPath.toString();
+                    if (routeString.contains(routeName)) {
+                        actionMap.put(ROUTE_CONSTANT + routeString + NAME_CONSTANT + route.getType(), address);
+                    }
+                }
+            }
+        }
+        return actionMap;
+    }
+
+    /**
+     * Search if the routing table route type contains name.
+     */
+    private static Map<String, String> getActionMemberMapByName(final ActionRoutingTable table, final String name,
+                                                                final String address) {
+        Collection<DOMActionInstance> routes = table.getItems();
+        Map<String, String> actionMap = new HashMap<>(routes.size());
+        for (DOMActionInstance route : routes) {
+            if (route.getType().getLastComponent() != null) {
+                final YangInstanceIdentifier actionPath = YangInstanceIdentifier.create(
+                        new NodeIdentifier(route.getType().getLastComponent()));
+                if (!actionPath.isEmpty()) {
+                    String type = route.getType().toString();
+                    if (type.contains(name)) {
+                        actionMap.put(ROUTE_CONSTANT + actionPath + NAME_CONSTANT + type, address);
+                    }
+                }
+            }
+        }
+        return actionMap;
+    }
+
+    @Override
+    public String getBucketVersions() {
+        return bucketVersions();
+    }
+}