X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?a=blobdiff_plain;f=yang%2Fyang-data-codec-xml%2Fsrc%2Fmain%2Fjava%2Forg%2Fopendaylight%2Fyangtools%2Fyang%2Fdata%2Fcodec%2Fxml%2FRandomPrefix.java;h=bc3213b9e9a5aa69ed8793b61a9f809bd2204daa;hb=c4517068c0183a892703b26baf098c97bfb2a854;hp=674efb19085c2e9c4743d2a9ec0b9f2c259a7dd7;hpb=6f4042c68d54ac6043104623fcb880fced5b237b;p=yangtools.git

diff --git a/yang/yang-data-codec-xml/src/main/java/org/opendaylight/yangtools/yang/data/codec/xml/RandomPrefix.java b/yang/yang-data-codec-xml/src/main/java/org/opendaylight/yangtools/yang/data/codec/xml/RandomPrefix.java
index 674efb1908..bc3213b9e9 100644
--- a/yang/yang-data-codec-xml/src/main/java/org/opendaylight/yangtools/yang/data/codec/xml/RandomPrefix.java
+++ b/yang/yang-data-codec-xml/src/main/java/org/opendaylight/yangtools/yang/data/codec/xml/RandomPrefix.java
@@ -12,10 +12,10 @@ import static com.google.common.base.Preconditions.checkArgument;
 import com.google.common.annotations.VisibleForTesting;
 import com.google.common.collect.BiMap;
 import com.google.common.collect.HashBiMap;
-import java.net.URI;
 import java.util.Map.Entry;
 import javax.xml.XMLConstants;
 import javax.xml.namespace.NamespaceContext;
+import org.opendaylight.yangtools.yang.common.XMLNamespace;
 
 class RandomPrefix {
     // 32 characters, carefully chosen
@@ -26,18 +26,18 @@ class RandomPrefix {
     private int counter = 0;
 
     // BiMap to make values lookup faster
-    private final BiMap<URI, String> prefixes = HashBiMap.create();
+    private final BiMap<XMLNamespace, String> prefixes = HashBiMap.create();
     private final NamespaceContext context;
 
     RandomPrefix(final NamespaceContext context) {
         this.context = context;
     }
 
-    Iterable<Entry<URI, String>> getPrefixes() {
+    Iterable<Entry<XMLNamespace, String>> getPrefixes() {
         return prefixes.entrySet();
     }
 
-    String encodePrefix(final URI namespace) {
+    String encodePrefix(final XMLNamespace namespace) {
         String prefix = prefixes.get(namespace);
         if (prefix != null) {
             return prefix;