System Class Loader to Thread ClassLoader

Author: Muhammad Hammad

README.md

@@ -1,19 +1,46 @@
 # Dynamic Loader [![Build Status](https://travis-ci.org/dvare/DynamicLoader.svg?branch=master)](https://travis-ci.org/dvare/DynamicLoader)
-Library for Loading java code and dependencies at runtime.
+Dynamic Loader is In Memory Java Compiler and classpath loader which enables to compile both java file and code string in memory and also provide support to load supportive jars in classpath which need to compile the dynamic code.
+
+## Example
+
+##### In Memory Compilation..
+
+```java
+public class SourceTest {
+
+    @Test
+    public void sourceCodeTest() throws Exception {
+        
+        StringBuilder sourceCode = new StringBuilder();
+        sourceCode.append("package org.dvare.dynamic;\n");
+        sourceCode.append("public class SourceTestClass {\n");
+        sourceCode.append("   public String testMethod() { return \"inside test method\"; }");
+        sourceCode.append("}");
+
+        DynamicCompiler dynamicCompiler = new DynamicCompiler();
+        dynamicCompiler.addSource("org.dvare.dynamic.SourceTestClass", sourceCode.toString());
+        Map<String, Class<?>> compiled = dynamicCompiler.build();
+        Class<?> aClass = compiled.get("org.dvare.dynamic.SourceTestClass");
+        Assert.assertNotNull(aClass);
+        Assert.assertEquals(1, aClass.getDeclaredMethods().length);
+        }
+    }
+```
+
 
 ## Current version
 
-* The current  version is `1.0` 
+* The current stable version is `1.1` : [![Maven Central](https://maven-badges.herokuapp.com/maven-central/org.dvare/dynamic-loader/badge.svg?style=flat)](http://search.maven.org/#artifactdetails|org.dvare|dynamic-loader|1.1|)
 
- you need to add the following maven repository in your `pom.xml`:
+ Maven dependency:
 
 ```xml
 
 <dependencies>
         <dependency>
             <groupId>org.dvare</groupId>
             <artifactId>dynamic-loader</artifactId>
-            <version>1.0</version>
+            <version>1.1</version>
         </dependency>            
 <dependencies>
 ```

pom.xml

@@ -4,9 +4,12 @@
 
     <groupId>org.dvare</groupId>
     <artifactId>dynamic-loader</artifactId>
-    <version>1.0</version>
+    <version>1.1</version>
 
-    <description>Library for Loading java code and dependencies at runtime.</description>
+    <description>Dynamic Loader is in memory java compiler and classpath loader which enables to compile both java file
+        and code string in memory and also provide support to load supportive jars in classpath which need
+        to compile the dynamic code.
+    </description>
     <packaging>jar</packaging>
 
     <name>Dynamic Loader</name>
@@ -16,7 +19,7 @@
         <developer>
             <id>hammadirshad</id>
             <name>Muhammad Hammad</name>
-            <url></url>
+            <url>https://github.com/hammadirshad</url>
             <email>hammadirshad046@gmail.com</email>
             <roles>
                 <role>Developer</role>
@@ -92,6 +95,7 @@
             <groupId>junit</groupId>
             <artifactId>junit</artifactId>
             <version>4.11</version>
+            <scope>test</scope>
         </dependency>
 
         <dependency>
@@ -105,11 +109,7 @@
             <artifactId>slf4j-log4j12</artifactId>
             <version>1.7.7</version>
         </dependency>
-        <dependency>
-            <groupId>org.apache.commons</groupId>
-            <artifactId>commons-lang3</artifactId>
-            <version>3.4</version>
-        </dependency>
+
 
     </dependencies>
 

src/main/java/org/dvare/dynamic/compiler/DynamicCompiler.java

@@ -29,6 +29,8 @@ of this software and associated documentation files (the "Software"), to deal
 import org.dvare.dynamic.resources.DynamicClassLoader;
 import org.dvare.dynamic.resources.DynamicJavaFileManager;
 import org.dvare.dynamic.resources.SourceCode;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 
 import javax.tools.*;
 import java.io.File;
@@ -39,19 +41,22 @@ of this software and associated documentation files (the "Software"), to deal
 
 public class DynamicCompiler {
     private static final List<String> options = new ArrayList<>(Arrays.asList("-Xlint:unchecked"));
+    private static Logger logger = LoggerFactory.getLogger(DynamicCompiler.class);
     private static javax.tools.JavaCompiler javac = ToolProvider.getSystemJavaCompiler();
     private StandardJavaFileManager standardFileManager =
             javac.getStandardFileManager(null, null, null);
     private DynamicClassLoader dynamicClassLoader;
     private Map<String, JavaFileObject> sourceCodes = new HashMap<>();
     private List<URL> jars = new ArrayList<>();
+    private String classpath = "";
     private boolean separateContext = false;
     private boolean updateContextClassLoader = false;
 
     public void addSource(String className, String testSourceCode) {
         sourceCodes.put(className, new SourceCode(className, testSourceCode));
     }
 
+
     public void addSource(String className, File sourceFile) {
         Iterable<? extends JavaFileObject> compilationUnits = standardFileManager.getJavaFileObjectsFromFiles(Arrays.asList(sourceFile));
         for (JavaFileObject javaFileObject : compilationUnits) {
@@ -67,45 +72,63 @@ public void addJar(URL url) {
 
     public Map<String, Class<?>> build() throws DynamicCompilerException {
 
-        //ClassLoader classLoader = ClassLoader.getSystemClassLoader();
         ClassLoader classLoader = Thread.currentThread()
                 .getContextClassLoader();
         if (separateContext) {
             classLoader = new CustomClassLoader().getCustomURLClassLoader();
         }
 
-        URLClassLoader urlClassLoader = (URLClassLoader) classLoader;
 
-        try {
-            if (!jars.isEmpty()) {
+        if (!(classLoader instanceof URLClassLoader)) {
+            classLoader = this.getClass().getClassLoader();
+        }
+
+
+        if (classLoader instanceof URLClassLoader) {
+
+            URLClassLoader urlClassLoader = (URLClassLoader) classLoader;
 
-                for (URL url : jars) {
-                    Method method = URLClassLoader.class.getDeclaredMethod("addURL", URL.class);
-                    method.setAccessible(true);
-                    method.invoke(urlClassLoader, new Object[]{url});
+            try {
+                if (!jars.isEmpty()) {
+
+                    for (URL url : jars) {
+                        Method method = URLClassLoader.class.getDeclaredMethod("addURL", URL.class);
+                        method.setAccessible(true);
+                        method.invoke(urlClassLoader, new Object[]{url});
+                    }
                 }
-            }
 
-            Method method2 = URLClassLoader.class.getDeclaredMethod("getURLs");
-            method2.setAccessible(true);
-            URL[] urls = (URL[]) method2.invoke((URLClassLoader) classLoader);
-            String classpath = "";
-            for (URL url : urls) {
-                classpath += url + File.pathSeparator;
-            }
+                Method method2 = URLClassLoader.class.getDeclaredMethod("getURLs");
+                method2.setAccessible(true);
+                URL[] urls = (URL[]) method2.invoke((URLClassLoader) classLoader);
 
-            options.addAll(Arrays.asList("-classpath", classpath));
+                if (urls.length == 0) {
+
+                    urls = (URL[]) method2.invoke((URLClassLoader) getClass().getClassLoader());
+                }
+                for (URL url : urls) {
+                    File file = new File(url.getFile());
+
+                    classpath += file.getAbsolutePath() + File.pathSeparator;
+                    logger.debug(file.getAbsolutePath() + File.pathSeparator);
+                }
+
+                if (!classpath.trim().isEmpty()) {
+                    options.addAll(Arrays.asList("-classpath", classpath));
+                }
+
+
+            } catch (Exception e) {
+                e.printStackTrace();
+            }
 
 
-        } catch (Exception e) {
-            e.printStackTrace();
         }
 
+        dynamicClassLoader = new DynamicClassLoader(classLoader);
         Collection<JavaFileObject> compilationUnits = sourceCodes.values();
         List<CompiledCode> compiledCodes = new ArrayList<>();
 
-
-        dynamicClassLoader = new DynamicClassLoader(urlClassLoader);
         DynamicJavaFileManager fileManager = new DynamicJavaFileManager(standardFileManager, compiledCodes, dynamicClassLoader);
 
 
@@ -159,5 +182,11 @@ public void setUpdateContextClassLoader(boolean updateContextClassLoader) {
         this.updateContextClassLoader = updateContextClassLoader;
     }
 
+    public String getClasspath() {
+        return classpath;
+    }
 
+    public void setClasspath(String classpath) {
+        this.classpath = classpath;
+    }
 }

src/main/resources/log4j.properties

@@ -0,0 +1,5 @@
+log4j.rootLogger=DEBUG, stdout
+log4j.appender.stdout=org.apache.log4j.ConsoleAppender
+log4j.appender.stdout.Target=System.out
+log4j.appender.stdout.layout=org.apache.log4j.PatternLayout
+log4j.appender.stdout.layout.ConversionPattern=%d{yyyy-MM-dd'T'HH:mm:ss.SSS} %-5p [%c] - %m%n

src/test/org/dvare/dynamic/SourceTest.java

@@ -35,7 +35,6 @@ public class SourceTest {
     @Test
     public void compile_whenTypical() throws Exception {
         StringBuilder sourceCode = new StringBuilder();
-
         sourceCode.append("package org.dvare.dynamic;\n");
         sourceCode.append("public class SourceClass {\n");
         sourceCode.append("   public String test() { return \"inside test method\"; }");