yang/yang-parser-impl/src/main/java/org/opendaylight/yangtools/yang/parser/spi/source/QNameToStatementDefinitionMap.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.yangtools.yang.parser.spi.source;
   9
  10 import java.net.URI;
  11 import java.util.HashMap;
  12 import java.util.Map;
  13 import javax.annotation.Nonnull;
  14 import javax.annotation.Nullable;
  15 import org.opendaylight.yangtools.yang.common.QName;
  16 import org.opendaylight.yangtools.yang.model.api.meta.StatementDefinition;
  17
  18 public class QNameToStatementDefinitionMap implements QNameToStatementDefinition {
  19
  20     private Map<QName, StatementDefinition> qNameToStmtDefMap = new HashMap<>();
  21     private Map<QName, StatementDefinition> qNameWithoutRevisionToStmtDefMap = new HashMap<>();
  22
  23     public void put(QName qName, StatementDefinition stDef) {
  24         qNameToStmtDefMap.put(qName, stDef);
  25         qNameWithoutRevisionToStmtDefMap.put(QName.create(qName.getNamespace(), null, qName.getLocalName()), stDef);
  26     }
  27
  28     @Nullable
  29     @Override
  30     public StatementDefinition get(@Nonnull QName identifier) {
  31         return qNameToStmtDefMap.get(identifier);
  32     }
  33
  34     @Nullable
  35     @Override
  36     public StatementDefinition getByNamespaceAndLocalName(URI namespace, @Nonnull String localName) {
  37         return qNameWithoutRevisionToStmtDefMap.get(QName.create(namespace, null, localName));
  38     }
  39 }