binding/mdsal-binding-dom-adapter/src/main/java/org/opendaylight/mdsal/binding/dom/adapter/BindingDOMRpcProviderServiceAdapter.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.binding.dom.adapter;
   9
  10 import com.google.common.annotations.VisibleForTesting;
  11 import com.google.common.collect.ImmutableSet;
  12 import java.lang.reflect.Method;
  13 import java.util.ArrayList;
  14 import java.util.Collection;
  15 import java.util.HashSet;
  16 import java.util.Map;
  17 import java.util.Set;
  18 import org.opendaylight.mdsal.binding.api.RpcProviderService;
  19 import org.opendaylight.mdsal.dom.api.DOMRpcIdentifier;
  20 import org.opendaylight.mdsal.dom.api.DOMRpcImplementationRegistration;
  21 import org.opendaylight.mdsal.dom.api.DOMRpcProviderService;
  22 import org.opendaylight.yangtools.concepts.ObjectRegistration;
  23 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  24 import org.opendaylight.yangtools.yang.binding.RpcService;
  25 import org.opendaylight.yangtools.yang.common.QName;
  26 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  27
  28 @VisibleForTesting
  29 public class BindingDOMRpcProviderServiceAdapter extends AbstractBindingAdapter<DOMRpcProviderService>
  30         implements RpcProviderService {
  31     private static final ImmutableSet<YangInstanceIdentifier> GLOBAL = ImmutableSet.of(YangInstanceIdentifier.empty());
  32
  33     public BindingDOMRpcProviderServiceAdapter(final AdapterContext adapterContext,
  34             final DOMRpcProviderService domRpcRegistry) {
  35         super(adapterContext, domRpcRegistry);
  36     }
  37
  38     @Override
  39     public <S extends RpcService, T extends S> ObjectRegistration<T> registerRpcImplementation(final Class<S> type,
  40             final T implementation) {
  41         return register(type, implementation, GLOBAL);
  42     }
  43
  44     @Override
  45     public <S extends RpcService, T extends S> ObjectRegistration<T> registerRpcImplementation(final Class<S> type,
  46             final T implementation, final Set<InstanceIdentifier<?>> paths) {
  47         return register(type, implementation, toYangInstanceIdentifiers(paths));
  48     }
  49
  50     private <S extends RpcService, T extends S> ObjectRegistration<T> register(final Class<S> type,
  51             final T implementation, final Collection<YangInstanceIdentifier> rpcContextPaths) {
  52         final Map<QName, Method> rpcs = currentSerializer().getRpcMethodToQName(type).inverse();
  53
  54         final BindingDOMRpcImplementationAdapter adapter = new BindingDOMRpcImplementationAdapter(adapterContext(),
  55             type, rpcs, implementation);
  56         final Set<DOMRpcIdentifier> domRpcs = createDomRpcIdentifiers(rpcs.keySet(), rpcContextPaths);
  57         final DOMRpcImplementationRegistration<?> domReg = getDelegate().registerRpcImplementation(adapter, domRpcs);
  58         return new BindingRpcAdapterRegistration<>(implementation, domReg);
  59     }
  60
  61     private static Set<DOMRpcIdentifier> createDomRpcIdentifiers(final Set<QName> rpcs,
  62             final Collection<YangInstanceIdentifier> paths) {
  63         final Set<DOMRpcIdentifier> ret = new HashSet<>();
  64         for (final YangInstanceIdentifier path : paths) {
  65             for (final QName rpc : rpcs) {
  66                 ret.add(DOMRpcIdentifier.create(rpc, path));
  67             }
  68         }
  69         return ret;
  70     }
  71
  72     private Collection<YangInstanceIdentifier> toYangInstanceIdentifiers(final Set<InstanceIdentifier<?>> identifiers) {
  73         final Collection<YangInstanceIdentifier> ret = new ArrayList<>(identifiers.size());
  74         final CurrentAdapterSerializer serializer = currentSerializer();
  75         for (final InstanceIdentifier<?> binding : identifiers) {
  76             ret.add(serializer.toCachedYangInstanceIdentifier(binding));
  77         }
  78         return ret;
  79     }
  80 }