opendaylight/md-sal/sal-binding-broker/src/main/java/org/opendaylight/controller/md/sal/binding/impl/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.controller.md.sal.binding.impl;
   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.controller.md.sal.binding.impl.BindingDOMAdapterBuilder.Factory;
  18 import org.opendaylight.controller.md.sal.dom.api.DOMRpcService;
  19 import org.opendaylight.controller.md.sal.dom.api.DOMService;
  20 import org.opendaylight.controller.sal.binding.api.RpcConsumerRegistry;
  21 import org.opendaylight.yangtools.yang.binding.RpcService;
  22 import org.opendaylight.yangtools.yang.binding.util.BindingReflections;
  23
  24 public class BindingDOMRpcServiceAdapter implements RpcConsumerRegistry {
  25
  26     protected static final Factory<RpcConsumerRegistry> BUILDER_FACTORY = new Factory<RpcConsumerRegistry>() {
  27
  28         @Override
  29         public BindingDOMAdapterBuilder<RpcConsumerRegistry> newBuilder() {
  30             return new Builder();
  31         }
  32
  33     };
  34
  35     private final LoadingCache<Class<? extends RpcService>, RpcServiceAdapter> proxies = CacheBuilder.newBuilder()
  36             .weakKeys()
  37             .build(new CacheLoader<Class<? extends RpcService>, RpcServiceAdapter>() {
  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     @Override
  56     public <T extends RpcService> T getRpcService(final Class<T> rpcService) {
  57         Preconditions.checkArgument(rpcService != null, "Rpc Service needs to be specied.");
  58         @SuppressWarnings("unchecked")
  59         final
  60         T proxy = (T) proxies.getUnchecked(rpcService).getProxy();
  61         return proxy;
  62     }
  63
  64     private RpcServiceAdapter createProxy(final Class<? extends RpcService> key) {
  65         Preconditions.checkArgument(BindingReflections.isBindingClass(key));
  66         Preconditions.checkArgument(key.isInterface(), "Supplied RPC service type must be interface.");
  67         return new RpcServiceAdapter(key, codec, domService);
  68     }
  69
  70     private static final class Builder extends BindingDOMAdapterBuilder<RpcConsumerRegistry> {
  71
  72         @Override
  73         protected RpcConsumerRegistry createInstance(final BindingToNormalizedNodeCodec codec,
  74                 final ClassToInstanceMap<DOMService> delegates) {
  75             final DOMRpcService domRpc  = delegates.getInstance(DOMRpcService.class);
  76             return new BindingDOMRpcServiceAdapter(domRpc  , codec);
  77         }
  78
  79         @Override
  80         public Set<? extends Class<? extends DOMService>> getRequiredDelegates() {
  81             return ImmutableSet.of(DOMRpcService.class);
  82         }
  83
  84     }
  85
  86 }