dom/mdsal-dom-broker/src/main/java/org/opendaylight/mdsal/dom/broker/DOMRpcRoutingTable.java

   1 /*
   2  * Copyright (c) 2015 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.mdsal.dom.broker;
   9
  10 import com.google.common.collect.ImmutableMap;
  11 import com.google.common.collect.LinkedListMultimap;
  12 import com.google.common.collect.ListMultimap;
  13 import java.util.List;
  14 import java.util.Map;
  15 import java.util.Set;
  16 import org.eclipse.jdt.annotation.Nullable;
  17 import org.opendaylight.mdsal.dom.api.DOMRpcAvailabilityListener;
  18 import org.opendaylight.mdsal.dom.api.DOMRpcIdentifier;
  19 import org.opendaylight.mdsal.dom.api.DOMRpcImplementation;
  20 import org.opendaylight.mdsal.dom.spi.ContentRoutedRpcContext;
  21 import org.opendaylight.yangtools.yang.common.QName;
  22 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  23 import org.opendaylight.yangtools.yang.model.api.EffectiveModelContext;
  24 import org.opendaylight.yangtools.yang.model.api.stmt.RpcEffectiveStatement;
  25
  26 final class DOMRpcRoutingTable extends AbstractDOMRoutingTable<DOMRpcIdentifier, YangInstanceIdentifier,
  27         DOMRpcImplementation, DOMRpcAvailabilityListener, QName, AbstractDOMRpcRoutingTableEntry> {
  28     static final DOMRpcRoutingTable EMPTY = new DOMRpcRoutingTable(ImmutableMap.of(), null);
  29
  30     private DOMRpcRoutingTable(final Map<QName, AbstractDOMRpcRoutingTableEntry> rpcs,
  31             final EffectiveModelContext schemaContext) {
  32         super(rpcs, schemaContext);
  33     }
  34
  35     boolean contains(final DOMRpcIdentifier input) {
  36         final AbstractDOMRpcRoutingTableEntry contexts = (AbstractDOMRpcRoutingTableEntry) getEntry(input.getType());
  37         return contexts != null && contexts.containsContext(input.getContextReference());
  38     }
  39
  40     @Override
  41     protected DOMRpcRoutingTable newInstance(final Map<QName, AbstractDOMRpcRoutingTableEntry> operations,
  42             final EffectiveModelContext schemaContext) {
  43         return new DOMRpcRoutingTable(operations, schemaContext);
  44     }
  45
  46     @Override
  47     protected ListMultimap<QName, YangInstanceIdentifier> decomposeIdentifiers(final Set<DOMRpcIdentifier> rpcs) {
  48         final ListMultimap<QName, YangInstanceIdentifier> ret = LinkedListMultimap.create();
  49         for (DOMRpcIdentifier i : rpcs) {
  50             ret.put(i.getType(), i.getContextReference());
  51         }
  52         return ret;
  53     }
  54
  55     @Override
  56     AbstractDOMRpcRoutingTableEntry createOperationEntry(final EffectiveModelContext context, final QName key,
  57             final Map<YangInstanceIdentifier, List<DOMRpcImplementation>> implementations) {
  58         final var rpcDef = findRpcDefinition(context, key);
  59         if (rpcDef == null) {
  60             return new UnknownDOMRpcRoutingTableEntry(key, implementations);
  61         }
  62
  63         final var contentContext = ContentRoutedRpcContext.forRpc(rpcDef);
  64         return contentContext == null ? new GlobalDOMRpcRoutingTableEntry(rpcDef.argument(), implementations)
  65             : new RoutedDOMRpcRoutingTableEntry(rpcDef.argument(), YangInstanceIdentifier.of(contentContext.leaf()),
  66                 implementations);
  67     }
  68
  69     private static @Nullable RpcEffectiveStatement findRpcDefinition(final EffectiveModelContext context,
  70             final QName qname) {
  71         return context == null ? null : context.findModuleStatement(qname.getModule())
  72             .flatMap(module -> module.findSchemaTreeNode(qname))
  73             .filter(RpcEffectiveStatement.class::isInstance)
  74             .map(RpcEffectiveStatement.class::cast)
  75             .orElse(null);
  76     }
  77 }