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