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 = new Factory<RpcConsumerRegistry>() {
  28
  29         @Override
  30         public BindingDOMAdapterBuilder<RpcConsumerRegistry> newBuilder() {
  31             return new Builder();
  32         }
  33
  34     };
  35
  36     private final LoadingCache<Class<? extends RpcService>, RpcServiceAdapter> proxies = CacheBuilder.newBuilder()
  37             .weakKeys()
  38             .build(new CacheLoader<Class<? extends RpcService>, RpcServiceAdapter>() {
  39
  40                 @Override
  41                 public RpcServiceAdapter load(final Class<? extends RpcService> key) throws Exception {
  42                     return createProxy(key);
  43                 }
  44
  45             });
  46
  47     private final DOMRpcService domService;
  48     private final BindingToNormalizedNodeCodec codec;
  49
  50     public BindingDOMRpcServiceAdapter(final DOMRpcService domService, final BindingToNormalizedNodeCodec codec) {
  51         super();
  52         this.domService = domService;
  53         this.codec = codec;
  54     }
  55
  56     @SuppressWarnings("unchecked")
  57     @Override
  58     public <T extends RpcService> T getRpcService(final Class<T> rpcService) {
  59         Preconditions.checkArgument(rpcService != null, "Rpc Service needs to be specied.");
  60         return (T) proxies.getUnchecked(rpcService).getProxy();
  61     }
  62
  63     private RpcServiceAdapter createProxy(final Class<? extends RpcService> key) {
  64         Preconditions.checkArgument(BindingReflections.isBindingClass(key));
  65         Preconditions.checkArgument(key.isInterface(), "Supplied RPC service type must be interface.");
  66         return new RpcServiceAdapter(key, codec, domService);
  67     }
  68
  69     private static final class Builder extends BindingDOMAdapterBuilder<RpcConsumerRegistry> {
  70
  71         @Override
  72         protected RpcConsumerRegistry createInstance(final BindingToNormalizedNodeCodec codec,
  73                 final ClassToInstanceMap<DOMService> delegates) {
  74             final DOMRpcService domRpc  = delegates.getInstance(DOMRpcService.class);
  75             return new BindingDOMRpcServiceAdapter(domRpc  , codec);
  76         }
  77
  78         @Override
  79         public Set<? extends Class<? extends DOMService>> getRequiredDelegates() {
  80             return ImmutableSet.of(DOMRpcService.class);
  81         }
  82
  83     }
  84
  85 }