Allow loading included schemas on the class-path

xlate · Dec 21, 2021 · eedfb16 · eedfb16
1 parent ef86d59
commit eedfb16
Show file tree

Hide file tree

Showing 4 changed files with 42 additions and 3 deletions.
diff --git a/src/main/java/io/xlate/edi/internal/schema/ClasspathURLStreamHandler.java b/src/main/java/io/xlate/edi/internal/schema/ClasspathURLStreamHandler.java
@@ -0,0 +1,28 @@
+package io.xlate.edi.internal.schema;
+
+import java.io.FileNotFoundException;
+import java.io.IOException;
+import java.net.URL;
+import java.net.URLConnection;
+import java.net.URLStreamHandler;
+
+class ClasspathURLStreamHandler extends URLStreamHandler {
+    /** The classloader to find resources from. */
+    private final ClassLoader classLoader;
+
+    public ClasspathURLStreamHandler(ClassLoader classLoader) {
+        this.classLoader = classLoader;
+    }
+
+    @Override
+    protected URLConnection openConnection(URL u) throws IOException {
+        final String resourcePath = u.getPath();
+        final URL resourceUrl = classLoader.getResource(resourcePath);
+
+        if (resourceUrl == null) {
+            throw new FileNotFoundException("Class-path resource not found: " + resourcePath);
+        }
+
+        return resourceUrl.openConnection();
+    }
+}
diff --git a/src/main/java/io/xlate/edi/internal/schema/SchemaReaderV4.java b/src/main/java/io/xlate/edi/internal/schema/SchemaReaderV4.java
@@ -48,7 +48,18 @@ protected void readInclude(XMLStreamReader reader, Map<String, EDIType> types) t
                 }
             }
 
-            URL schemaLocation = context != null ? new URL(context, location) : new URL(location);
+            URL schemaLocation;
+
+            if (location.startsWith("classpath:")) {
+                ClassLoader loader = Thread.currentThread().getContextClassLoader();
+                if (location.startsWith("classpath:/")) {
+                    location = location.substring(0, "classpath:".length()) + location.substring("classpath:/".length());
+                }
+                schemaLocation = new URL(null, location, new ClasspathURLStreamHandler(loader));
+            } else {
+                schemaLocation = new URL(context, location);
+            }
+
             types.putAll(StaEDISchemaFactory.readSchemaTypes(schemaLocation, super.properties));
             reader.nextTag(); // End of include
         } catch (Exception e) {

diff --git a/src/test/resources/x12/005010X222/837_REF_impls.xml b/src/test/resources/x12/005010X222/837_REF_impls.xml
@@ -1,7 +1,7 @@
 <?xml version="1.0" encoding="UTF-8" standalone="yes"?>
 <schema xmlns="http://xlate.io/EDISchema/v4">
 
-  <include schemaLocation="file:./target/test-classes/x12/005010/837.xml"/>
+  <include schemaLocation="classpath:/x12/005010/837.xml"/>
 
   <implementation>
     <sequence>

diff --git a/src/test/resources/x12/005010X222/837_loop1000_only.xml b/src/test/resources/x12/005010X222/837_loop1000_only.xml
@@ -1,7 +1,7 @@
 <?xml version="1.0" encoding="UTF-8" standalone="yes"?>
 <schema xmlns="http://xlate.io/EDISchema/v4">
 
-  <include schemaLocation="file:./target/test-classes/x12/005010/837.xml"/>
+  <include schemaLocation="classpath:x12/005010/837.xml"/>
 
   <implementation>
     <sequence>