binding/mdsal-binding-dom-adapter/src/main/java/org/opendaylight/mdsal/binding/dom/adapter/BindingDOMRpcServiceAdapter.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.base.Preconditions;
  11 import com.google.common.cache.CacheBuilder;
  12 import com.google.common.cache.CacheLoader;
  13 import com.google.common.cache.LoadingCache;
  14 import com.google.common.collect.ClassToInstanceMap;
  15 import com.google.common.collect.ImmutableSet;
  16 import java.util.Set;
  17 import org.opendaylight.mdsal.binding.api.RpcConsumerRegistry;
  18 import org.opendaylight.mdsal.binding.dom.adapter.BindingDOMAdapterBuilder.Factory;
  19 import org.opendaylight.mdsal.dom.api.DOMRpcService;
  20 import org.opendaylight.mdsal.dom.api.DOMService;
  21 import org.opendaylight.yangtools.yang.binding.RpcService;
  22 import org.opendaylight.yangtools.yang.binding.util.BindingReflections;
  23
  24
  25 public class BindingDOMRpcServiceAdapter implements RpcConsumerRegistry {
  26
  27     protected static final Factory<RpcConsumerRegistry> BUILDER_FACTORY = Builder::new;
  28
  29     private final LoadingCache<Class<? extends RpcService>, RpcServiceAdapter> proxies = CacheBuilder.newBuilder()
  30             .weakKeys()
  31             .build(new CacheLoader<Class<? extends RpcService>, RpcServiceAdapter>() {
  32
  33                 private RpcServiceAdapter createProxy(final Class<? extends RpcService> key) {
  34                     Preconditions.checkArgument(BindingReflections.isBindingClass(key));
  35                     Preconditions.checkArgument(key.isInterface(), "Supplied RPC service type must be interface.");
  36                     return new RpcServiceAdapter(key, codec, domService);
  37                 }
  38
  39                 @Override
  40                 public RpcServiceAdapter load(final Class<? extends RpcService> key) throws Exception {
  41                     return createProxy(key);
  42                 }
  43
  44             });
  45
  46     private final DOMRpcService domService;
  47     private final BindingToNormalizedNodeCodec codec;
  48
  49     public BindingDOMRpcServiceAdapter(final DOMRpcService domService, final BindingToNormalizedNodeCodec codec) {
  50         super();
  51         this.domService = domService;
  52         this.codec = codec;
  53     }
  54
  55     @SuppressWarnings("unchecked")
  56     @Override
  57     public <T extends RpcService> T getRpcService(final Class<T> rpcService) {
  58         Preconditions.checkArgument(rpcService != null, "Rpc Service needs to be specied.");
  59         return (T) proxies.getUnchecked(rpcService).getProxy();
  60     }
  61
  62     private static final class Builder extends BindingDOMAdapterBuilder<RpcConsumerRegistry> {
  63
  64         @Override
  65         protected RpcConsumerRegistry createInstance(final BindingToNormalizedNodeCodec codec,
  66                 final ClassToInstanceMap<DOMService> delegates) {
  67             final DOMRpcService domRpc  = delegates.getInstance(DOMRpcService.class);
  68             return new BindingDOMRpcServiceAdapter(domRpc  , codec);
  69         }
  70
  71         @Override
  72         public Set<? extends Class<? extends DOMService>> getRequiredDelegates() {
  73             return ImmutableSet.of(DOMRpcService.class);
  74         }
  75
  76     }
  77
  78 }