binding/mdsal-binding-runtime-api/src/main/java/org/opendaylight/mdsal/binding/runtime/api/AbstractBindingRuntimeContext.java

   1 /*
   2  * Copyright (c) 2014 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.runtime.api;
   9
  10 import static com.google.common.base.Preconditions.checkArgument;
  11 import static java.util.Objects.requireNonNull;
  12
  13 import com.google.common.annotations.Beta;
  14 import com.google.common.base.Throwables;
  15 import com.google.common.cache.CacheBuilder;
  16 import com.google.common.cache.CacheLoader;
  17 import com.google.common.cache.LoadingCache;
  18 import java.util.concurrent.ExecutionException;
  19 import org.eclipse.jdt.annotation.NonNull;
  20 import org.opendaylight.mdsal.binding.model.api.JavaTypeName;
  21 import org.opendaylight.yangtools.yang.binding.Action;
  22 import org.opendaylight.yangtools.yang.binding.Augmentation;
  23 import org.opendaylight.yangtools.yang.binding.BaseIdentity;
  24 import org.opendaylight.yangtools.yang.binding.Notification;
  25 import org.opendaylight.yangtools.yang.binding.RpcInput;
  26 import org.opendaylight.yangtools.yang.binding.RpcOutput;
  27 import org.opendaylight.yangtools.yang.common.QName;
  28 import org.opendaylight.yangtools.yang.model.api.stmt.SchemaNodeIdentifier.Absolute;
  29
  30 /**
  31  * Runtime Context for Java YANG Binding classes. It provides information derived from the backing effective model,
  32  * which is not captured in generated classes (and hence cannot be obtained from {@code BindingReflections}.
  33  */
  34 @Beta
  35 public abstract class AbstractBindingRuntimeContext implements BindingRuntimeContext {
  36     private final LoadingCache<@NonNull QName, @NonNull Class<? extends BaseIdentity>> identityClasses =
  37         CacheBuilder.newBuilder().weakValues().build(new CacheLoader<>() {
  38             @Override
  39             public Class<? extends BaseIdentity> load(final QName key) {
  40                 final var type = getTypes().findIdentity(key).orElseThrow(
  41                     () -> new IllegalArgumentException("Supplied QName " + key + " is not a valid identity"));
  42                 try {
  43                     return loadClass(type.getIdentifier()).asSubclass(BaseIdentity.class);
  44                 } catch (ClassNotFoundException e) {
  45                     throw new IllegalArgumentException("Required class " + type + " was not found.", e);
  46                 } catch (ClassCastException e) {
  47                     throw new IllegalArgumentException(key + " resolves to a non-identity class", e);
  48                 }
  49             }
  50         });
  51
  52     @Override
  53     public final <T extends Augmentation<?>> AugmentRuntimeType getAugmentationDefinition(final Class<T> augClass) {
  54         return getTypes().findSchema(JavaTypeName.create(augClass))
  55             .filter(AugmentRuntimeType.class::isInstance)
  56             .map(AugmentRuntimeType.class::cast)
  57             .orElse(null);
  58     }
  59
  60     @Override
  61     public final CompositeRuntimeType getSchemaDefinition(final Class<?> cls) {
  62         checkArgument(!Augmentation.class.isAssignableFrom(cls), "Supplied class must not be an augmentation (%s is)",
  63             cls);
  64         checkArgument(!Action.class.isAssignableFrom(cls), "Supplied class must not be an action (%s is)", cls);
  65         checkArgument(!Notification.class.isAssignableFrom(cls), "Supplied class must not be a notification (%s is)",
  66             cls);
  67         return (CompositeRuntimeType) getTypes().findSchema(JavaTypeName.create(cls)).orElse(null);
  68     }
  69
  70     @Override
  71     public final ActionRuntimeType getActionDefinition(final Class<? extends Action<?, ?, ?>> cls) {
  72         return (ActionRuntimeType) getTypes().findSchema(JavaTypeName.create(cls)).orElse(null);
  73     }
  74
  75     @Override
  76     public final RuntimeType getTypeWithSchema(final Class<?> type) {
  77         return getTypes().findSchema(JavaTypeName.create(type))
  78             .orElseThrow(() -> new IllegalArgumentException("Failed to find schema for " + type));
  79     }
  80
  81     @Override
  82     public final Class<?> getClassForSchema(final Absolute schema) {
  83         final var child = getTypes().schemaTreeChild(schema);
  84         checkArgument(child != null, "Failed to find binding type for %s", schema);
  85         return loadClass(child);
  86     }
  87
  88     @Override
  89     public final Class<? extends BaseIdentity> getIdentityClass(final QName input) {
  90         try {
  91             return identityClasses.get(requireNonNull(input));
  92         } catch (ExecutionException e) {
  93             Throwables.throwIfUnchecked(e.getCause());
  94             throw new IllegalStateException("Unexpected error looking up " + input, e);
  95         }
  96     }
  97
  98     @Override
  99     public final Class<? extends RpcInput> getRpcInput(final QName rpcName) {
 100         return loadClass(getTypes().findRpcInput(rpcName)
 101             .orElseThrow(() -> new IllegalArgumentException("Failed to find RpcInput for " + rpcName)))
 102             .asSubclass(RpcInput.class);
 103     }
 104
 105     @Override
 106     public final Class<? extends RpcOutput> getRpcOutput(final QName rpcName) {
 107         return loadClass(getTypes().findRpcOutput(rpcName)
 108             .orElseThrow(() -> new IllegalArgumentException("Failed to find RpcOutput for " + rpcName)))
 109             .asSubclass(RpcOutput.class);
 110     }
 111
 112     private Class<?> loadClass(final RuntimeType type) {
 113         try {
 114             return loadClass(type.javaType());
 115         } catch (final ClassNotFoundException e) {
 116             throw new IllegalStateException(e);
 117         }
 118     }
 119 }