BUG-4688: Rework SchemaContext module lookups
[yangtools.git] / yang / yang-data-util / src / main / java / org / opendaylight / yangtools / yang / data / util / ModuleStringIdentityrefCodec.java
1 /*
2  * Copyright (c) 2016 Intel Corporation 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
9 package org.opendaylight.yangtools.yang.data.util;
10
11 import static java.util.Objects.requireNonNull;
12
13 import java.net.URI;
14 import java.util.Iterator;
15 import javax.annotation.Nonnull;
16 import org.opendaylight.yangtools.yang.common.QNameModule;
17 import org.opendaylight.yangtools.yang.data.api.codec.IdentityrefCodec;
18 import org.opendaylight.yangtools.yang.model.api.Module;
19 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
20
21 public abstract class ModuleStringIdentityrefCodec
22         extends AbstractModuleStringIdentityrefCodec
23         implements IdentityrefCodec<String> {
24     protected final SchemaContext context;
25     protected final QNameModule parentModuleQname;
26
27     public ModuleStringIdentityrefCodec(@Nonnull final SchemaContext context, @Nonnull final QNameModule parentModule) {
28         this.context = requireNonNull(context);
29         this.parentModuleQname = requireNonNull(parentModule);
30     }
31
32     @Override
33     protected String prefixForNamespace(@Nonnull final URI namespace) {
34         final Iterator<Module> modules = context.findModules(namespace).iterator();
35         return modules.hasNext() ? modules.next().getName() : null;
36     }
37 }