*/
package org.opendaylight.mdsal.binding.generator.impl;
+import static com.google.common.base.Preconditions.checkState;
import static java.util.Objects.requireNonNull;
import com.google.common.collect.BiMap;
import java.util.List;
import java.util.Map;
import java.util.Set;
-import javax.annotation.concurrent.NotThreadSafe;
+import org.eclipse.jdt.annotation.NonNull;
+import org.eclipse.jdt.annotation.Nullable;
+import org.opendaylight.mdsal.binding.model.api.GeneratedType;
import org.opendaylight.mdsal.binding.model.api.JavaTypeName;
import org.opendaylight.mdsal.binding.model.api.Type;
import org.opendaylight.mdsal.binding.model.api.type.builder.GeneratedTOBuilder;
import org.opendaylight.mdsal.binding.model.api.type.builder.GeneratedTypeBuilder;
import org.opendaylight.mdsal.binding.spec.naming.BindingMapping;
+import org.opendaylight.yangtools.concepts.Mutable;
import org.opendaylight.yangtools.yang.common.QName;
import org.opendaylight.yangtools.yang.model.api.AugmentationSchemaNode;
import org.opendaylight.yangtools.yang.model.api.CaseSchemaNode;
+import org.opendaylight.yangtools.yang.model.api.ContainerSchemaNode;
import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
import org.opendaylight.yangtools.yang.model.api.DocumentedNode.WithStatus;
import org.opendaylight.yangtools.yang.model.api.GroupingDefinition;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
-@NotThreadSafe
-public final class ModuleContext {
+/**
+ * Utility class for building up Binding mapping information. This class is NOT thread-safe.
+ */
+public final class ModuleContext implements Mutable {
private static final Logger LOG = LoggerFactory.getLogger(ModuleContext.class);
private final BiMap<Type, AugmentationSchemaNode> typeToAugmentation = HashBiMap.create();
private final List<GeneratedTypeBuilder> augmentations = new ArrayList<>();
private final Multimap<Type, Type> choiceToCases = HashMultimap.create();
private final Set<GeneratedTypeBuilder> topLevelNodes = new HashSet<>();
+ private final Map<SchemaPath, JavaTypeName> aliases = new HashMap<>();
private final Map<Type, WithStatus> typeToSchema = new HashMap<>();
private final List<GeneratedTOBuilder> genTOs = new ArrayList<>();
private final Map<SchemaPath, Type> innerTypes = new HashMap<>();
private final Map<JavaTypeName, SchemaNode> nameMapping = new HashMap<>();
private GeneratedTypeBuilder moduleNode;
+ private JavaTypeName moduleInfoType;
private String modulePackageName;
ModuleContext(final Module module) {
return module;
}
- String modulePackageName() {
+ @NonNull String modulePackageName() {
String ret = modulePackageName;
if (ret == null) {
modulePackageName = ret = BindingMapping.getRootPackageName(module.getQNameModule());
return ret;
}
+ @NonNull JavaTypeName moduleInfoType() {
+ JavaTypeName ret = moduleInfoType;
+ if (ret == null) {
+ moduleInfoType = ret = JavaTypeName.create(modulePackageName(), BindingMapping.MODULE_INFO_CLASS_NAME);
+ }
+ return ret;
+ }
+
List<Type> getGeneratedTypes() {
List<Type> result = new ArrayList<>();
return moduleNode;
}
- public GeneratedTypeBuilder getChildNode(final SchemaPath p) {
- return childNodes.get(p);
+ public GeneratedTypeBuilder getChildNode(final SchemaPath path) {
+ return childNodes.get(path);
+ }
+
+ public GeneratedTypeBuilder getGrouping(final SchemaPath path) {
+ return groupings.get(path);
}
- public GeneratedTypeBuilder getGrouping(final SchemaPath p) {
- return groupings.get(p);
+ public GeneratedTypeBuilder getCase(final SchemaPath path) {
+ return cases.get(path);
}
- public GeneratedTypeBuilder getCase(final SchemaPath p) {
- return cases.get(p);
+ public void addModuleNode(final GeneratedTypeBuilder newModuleNode) {
+ this.moduleNode = newModuleNode;
}
- public void addModuleNode(final GeneratedTypeBuilder moduleNode) {
- this.moduleNode = moduleNode;
+ public void addGeneratedTOBuilder(final GeneratedTOBuilder builder) {
+ genTOs.add(builder);
}
- public void addGeneratedTOBuilder(final GeneratedTOBuilder b) {
- genTOs.add(b);
+ @NonNull GeneratedType addAliasType(final ModuleContext sourceContext, final ContainerSchemaNode source,
+ final ContainerSchemaNode alias) {
+ final GeneratedTypeBuilder builder = sourceContext.getChildNode(source.getPath());
+ checkState(builder != null, "Could not find builder for %s", source);
+
+ final JavaTypeName id = builder.getIdentifier();
+ final SchemaPath path = alias.getPath();
+ final JavaTypeName prev = aliases.putIfAbsent(path, id);
+ checkState(prev == null, "Type aliasing conflict on %s: %s vs %s", path, prev, id);
+
+ return builder.build();
}
- public void addChildNodeType(final SchemaNode def, final GeneratedTypeBuilder b) {
- checkNamingConflict(def, b.getIdentifier());
- childNodes.put(def.getPath(), b);
- typeToSchema.put(b, def);
+ @Nullable JavaTypeName getAlias(final SchemaPath path) {
+ return aliases.get(path);
}
- public void addGroupingType(final GroupingDefinition def, final GeneratedTypeBuilder b) {
- checkNamingConflict(def, b.getIdentifier());
- groupings.put(def.getPath(), b);
+ public void addChildNodeType(final SchemaNode def, final GeneratedTypeBuilder builder) {
+ checkNamingConflict(def, builder.getIdentifier());
+ childNodes.put(def.getPath(), builder);
+ typeToSchema.put(builder, def);
}
- public void addTypedefType(final TypeDefinition<?> def, final Type t) {
- final JavaTypeName name = t.getIdentifier();
+ public void addGroupingType(final GroupingDefinition def, final GeneratedTypeBuilder builder) {
+ checkNamingConflict(def, builder.getIdentifier());
+ groupings.put(def.getPath(), builder);
+ }
+
+ public void addTypedefType(final TypeDefinition<?> def, final Type type) {
+ final JavaTypeName name = type.getIdentifier();
final SchemaNode existingDef = nameMapping.putIfAbsent(name, def);
if (existingDef != null) {
if (!(existingDef instanceof TypeDefinition)) {
LOG.debug("GeneratedType conflict between {} and {} on {}", def, existingDef, name);
}
- typedefs.put(def.getPath(), t);
+ typedefs.put(def.getPath(), type);
}
- public void addCaseType(final SchemaPath p, final GeneratedTypeBuilder b) {
- cases.put(p, b);
+ public void addCaseType(final SchemaPath path, final GeneratedTypeBuilder builder) {
+ cases.put(path, builder);
}
- public void addIdentityType(final IdentitySchemaNode def, final GeneratedTypeBuilder b) {
- checkNamingConflict(def, b.getIdentifier());
- identities.put(def.getQName(), b);
+ public void addIdentityType(final IdentitySchemaNode def, final GeneratedTypeBuilder builder) {
+ checkNamingConflict(def, builder.getIdentifier());
+ identities.put(def.getQName(), builder);
}
- public void addTopLevelNodeType(final GeneratedTypeBuilder b) {
- topLevelNodes.add(b);
+ public void addTopLevelNodeType(final GeneratedTypeBuilder builder) {
+ topLevelNodes.add(builder);
}
- public void addAugmentType(final GeneratedTypeBuilder b) {
- augmentations.add(b);
+ public void addAugmentType(final GeneratedTypeBuilder builder) {
+ augmentations.add(builder);
}
public Map<SchemaPath, Type> getTypedefs() {
}
/**
- * Returns mapping of type to its schema.
- *
- * Valid values are only instances of {@link DataSchemaNode} or {@link AugmentationSchemaNode}
+ * Returns mapping of type to its schema. Valid values are only instances of {@link DataSchemaNode}
+ * or {@link AugmentationSchemaNode}.
*
* @return Mapping from type to corresponding schema
*/
/**
* Adds mapping between schema path and an inner type.
- *
- * @param path
- * @param type
*/
void addInnerTypedefType(final SchemaPath path, final Type type) {
innerTypes.put(path, type);