Merge pull request #9 from BSI-Business-Systems-Integration-AG/patch_reference_caching_for_direct_classes

Enable reference caching for classes which are not visible to the classloader

Author: UnquietCode

codemodel/src/main/java/com/sun/codemodel/JCodeModel.java

@@ -101,7 +101,9 @@ public final class JCodeModel {
     /** All JReferencedClasses are pooled here. */
     private final HashMap<Class<?>,JReferencedClass> refClasses = new HashMap<Class<?>,JReferencedClass>();
 
-    
+    /** All JDirectClass are pooled here. */
+    private final Map<String,JDirectClass> refDirectClasses = new HashMap<String,JDirectClass>();
+
     /** Obtains a reference to the special "null" type. */
     public final JNullType NULL = new JNullType(this);
     // primitive types 
@@ -382,7 +384,12 @@ public JClass ref(String fullyQualifiedClassName) {
         }
 
         // assume it's not visible to us.
-        return new JDirectClass(this,fullyQualifiedClassName);
+        JDirectClass jdrc = refDirectClasses.get(fullyQualifiedClassName);
+        if (jdrc == null) {
+        	jdrc = new JDirectClass(this,fullyQualifiedClassName);
+        	refDirectClasses.put(fullyQualifiedClassName, jdrc);
+        }
+        return jdrc;
     }
 
     /**

codemodel/src/test/java/com/sun/codemodel/tests/JDirectClassRefTest.java

@@ -0,0 +1,18 @@
+package com.sun.codemodel.tests;
+
+import static org.junit.Assert.assertSame;
+
+import org.junit.Test;
+
+import com.sun.codemodel.JClassAlreadyExistsException;
+import com.sun.codemodel.JCodeModel;
+
+public class JDirectClassRefTest {
+
+	@Test
+	public void testDirectClassRef() throws JClassAlreadyExistsException {
+		final JCodeModel codeModel = new JCodeModel();
+
+		assertSame("Should be same JClass", codeModel.ref("org.test.ClassNotOnClasspath"), codeModel.ref("org.test.ClassNotOnClasspath"));
+	}
+}