[#517] customize with mapinit:class local class for map type

highsource · Apr 10, 2024 · 6b0c11e · 6b0c11e
1 parent d491509
commit 6b0c11e
Show file tree

Hide file tree

Showing 9 changed files with 173 additions and 6 deletions.
diff --git a/basics/basic/src/main/java/org/jvnet/jaxb2_commons/plugin/map_init/Customizations.java b/basics/basic/src/main/java/org/jvnet/jaxb2_commons/plugin/map_init/Customizations.java
@@ -1,5 +1,13 @@
 package org.jvnet.jaxb2_commons.plugin.map_init;
 
+import com.sun.tools.xjc.model.CClassInfo;
+import com.sun.tools.xjc.model.CPluginCustomization;
+import javax.xml.bind.JAXBContext;
+import javax.xml.bind.JAXBException;
+
+import com.sun.xml.bind.v2.ContextFactory;
+import org.jvnet.jaxb2_commons.util.CustomizationUtils;
+
 import javax.xml.namespace.QName;
 
 public class Customizations {
@@ -8,4 +16,33 @@ public class Customizations {
 
     public static QName IGNORED_ELEMENT_NAME = new QName(NAMESPACE_URI, "ignored");
 
+    public static QName OBJECT_FACTORY_ELEMENT_NAME = new QName(NAMESPACE_URI, "objectFactory");
+
+    public static QName INIT_CLASS_ELEMENT_NAME = new QName(NAMESPACE_URI, "class");
+
+
+    private static final JAXBContext context;
+    static {
+        try {
+            context = ContextFactory.createContext(
+                org.jvnet.jaxb2_commons.plugin.map_init.ObjectFactory.class.getPackage().getName(),
+                ObjectFactory.class.getClassLoader(),
+                null);
+        } catch (JAXBException e) {
+            throw new ExceptionInInitializerError(e);
+        }
+    }
+
+    public static JAXBContext getContext() {
+        return context;
+    }
+
+    public static void _initClass(CClassInfo classInfo, String className) {
+        final InitClass initClass = new InitClass();
+        initClass.setClassName(className);
+        final CPluginCustomization customization = CustomizationUtils.marshal(getContext(), Customizations.INIT_CLASS_ELEMENT_NAME, initClass);
+        classInfo.getCustomizations().add(customization);
+        customization.markAsAcknowledged();
+
+    }
 }
diff --git a/basics/basic/src/main/java/org/jvnet/jaxb2_commons/plugin/map_init/InitClass.java b/basics/basic/src/main/java/org/jvnet/jaxb2_commons/plugin/map_init/InitClass.java
@@ -0,0 +1,26 @@
+package org.jvnet.jaxb2_commons.plugin.map_init;
+
+import javax.xml.bind.annotation.XmlAccessType;
+import javax.xml.bind.annotation.XmlAccessorType;
+import javax.xml.bind.annotation.XmlRootElement;
+import javax.xml.bind.annotation.XmlValue;
+import javax.xml.bind.annotation.adapters.CollapsedStringAdapter;
+import javax.xml.bind.annotation.adapters.XmlJavaTypeAdapter;
+
+@XmlRootElement(namespace = "urn:jaxb.jvnet.org:plugin:mapinit", name = "class")
+@XmlAccessorType(XmlAccessType.PROPERTY)
+public class InitClass {
+
+	private String className;
+
+	@XmlValue
+	@XmlJavaTypeAdapter(value = CollapsedStringAdapter.class)
+	String getClassName() {
+		return className;
+	}
+
+	public void setClassName(String className) {
+		this.className = className;
+	}
+
+}
diff --git a/basics/basic/src/main/java/org/jvnet/jaxb2_commons/plugin/map_init/MapInitPlugin.java b/basics/basic/src/main/java/org/jvnet/jaxb2_commons/plugin/map_init/MapInitPlugin.java
@@ -9,13 +9,15 @@
 import com.sun.codemodel.JMethod;
 import com.sun.codemodel.JType;
 import com.sun.tools.xjc.Options;
+import com.sun.tools.xjc.model.CPluginCustomization;
 import com.sun.tools.xjc.outline.ClassOutline;
 import com.sun.tools.xjc.outline.FieldOutline;
 import com.sun.tools.xjc.outline.Outline;
 import org.jvnet.jaxb2_commons.plugin.AbstractParameterizablePlugin;
 import org.jvnet.jaxb2_commons.plugin.CustomizedIgnoring;
 import org.jvnet.jaxb2_commons.plugin.Ignoring;
 import org.jvnet.jaxb2_commons.plugin.util.FieldOutlineUtils;
+import org.jvnet.jaxb2_commons.util.CustomizationUtils;
 import org.xml.sax.ErrorHandler;
 
 import javax.xml.namespace.QName;
@@ -78,21 +80,32 @@ private void generateMapInit(ClassOutline classOutline,
 
             if (getter != null && getter.type().erasure().boxify().isAssignableFrom(codeModel.ref(Map.class))) {
                 final JFieldVar field = theClass.fields().get(fieldOutline.getPropertyInfo().getName(false));
+
+                final CPluginCustomization initClassCustomization = CustomizationUtils
+                    .findCustomization(fieldOutline, Customizations.INIT_CLASS_ELEMENT_NAME);
+                String mapClassName = getMapClass();
+                if (initClassCustomization != null) {
+                    final InitClass initClass = (InitClass) CustomizationUtils
+                        .unmarshall(Customizations.getContext(), initClassCustomization);
+                    if (initClass != null && initClass.getClassName() != null && !"".equals(initClass.getClassName())) {
+                        mapClassName = initClass.getClassName();
+                    }
+                }
                 if (field != null) {
                     getter.body().pos(0);
                     getter.body()._if(field.eq(JExpr._null()))._then()
-                        .assign(field, newCoreMap(codeModel, getter.type().boxify().getTypeParameters()));
+                        .assign(field, newCoreMap(codeModel, mapClassName, getter.type().boxify().getTypeParameters()));
                 }
             }
         }
     }
 
-    private JExpression newCoreMap(JCodeModel codeModel, List<JClass> typeParameters) {
-        return JExpr._new(codeModel.ref(getMapClass()).narrow(typeParameters));
+    private JExpression newCoreMap(JCodeModel codeModel, String mapClassName, List<JClass> typeParameters) {
+        return JExpr._new(codeModel.ref(mapClassName).narrow(typeParameters));
     }
 
     private Ignoring ignoring = new CustomizedIgnoring(
-            org.jvnet.jaxb2_commons.plugin.map_init.Customizations.IGNORED_ELEMENT_NAME);
+            Customizations.IGNORED_ELEMENT_NAME);
 
     public Ignoring getIgnoring() {
         return ignoring;
@@ -105,6 +118,7 @@ public void setIgnoring(Ignoring ignoring) {
     @Override
     public Collection<QName> getCustomizationElementNames() {
         return Arrays.asList(
-            org.jvnet.jaxb2_commons.plugin.map_init.Customizations.IGNORED_ELEMENT_NAME);
+            Customizations.IGNORED_ELEMENT_NAME,
+            Customizations.INIT_CLASS_ELEMENT_NAME);
     }
 }
diff --git a/basics/basic/src/main/java/org/jvnet/jaxb2_commons/plugin/map_init/ObjectFactory.java b/basics/basic/src/main/java/org/jvnet/jaxb2_commons/plugin/map_init/ObjectFactory.java
@@ -0,0 +1,16 @@
+package org.jvnet.jaxb2_commons.plugin.map_init;
+
+import javax.xml.bind.annotation.XmlRegistry;
+
+@XmlRegistry
+public class ObjectFactory {
+
+    public InitClass createInitClass() {
+        return new InitClass();
+    }
+
+    public ObjectFactoryCustomization createObjectFactoryCustomization() {
+        return new ObjectFactoryCustomization();
+    }
+
+}
diff --git a/...sic/src/main/java/org/jvnet/jaxb2_commons/plugin/map_init/ObjectFactoryCustomization.java b/...sic/src/main/java/org/jvnet/jaxb2_commons/plugin/map_init/ObjectFactoryCustomization.java
@@ -0,0 +1,25 @@
+package org.jvnet.jaxb2_commons.plugin.map_init;
+
+import javax.xml.bind.annotation.XmlAccessType;
+import javax.xml.bind.annotation.XmlAccessorType;
+import javax.xml.bind.annotation.XmlElement;
+import javax.xml.bind.annotation.XmlRootElement;
+import javax.xml.bind.annotation.XmlType;
+
+@XmlRootElement(namespace = "urn:jaxb.jvnet.org:plugin:mapinit", name = "objectFactory")
+@XmlType(propOrder = { "initClass" })
+@XmlAccessorType(XmlAccessType.PUBLIC_MEMBER)
+public class ObjectFactoryCustomization {
+
+    private InitClass initClass;
+
+    @XmlElement(namespace = "urn:jaxb.jvnet.org:plugin:mapinit", name = "class")
+    public InitClass getInitClass() {
+        return initClass;
+    }
+
+    public void setInitClass(InitClass initClass) {
+        this.initClass = initClass;
+    }
+
+}
diff --git a/basics/tests/map_init/src/main/resources/example-definedMapClass.xsd b/basics/tests/map_init/src/main/resources/example-definedMapClass.xsd
@@ -31,6 +31,28 @@
           </xsd:sequence>
         </xsd:complexType>
       </xsd:element>
+      <xsd:element name="weakDataMap">
+        <xsd:annotation>
+          <xsd:appinfo>
+            <mapinit:class>java.util.WeakHashMap</mapinit:class>
+            <jaxb:property>
+              <jaxb:baseType name="java.util.Map&lt;String,Object&gt;" />
+            </jaxb:property>
+          </xsd:appinfo>
+        </xsd:annotation>
+        <xsd:complexType>
+          <xsd:sequence>
+            <xsd:element name="entry" minOccurs="0" maxOccurs="unbounded">
+              <xsd:complexType>
+                <xsd:sequence>
+                  <xsd:element name="key" minOccurs="0" type="xsd:string"/>
+                  <xsd:element name="value" minOccurs="0" type="xsd:anyType"/>
+                </xsd:sequence>
+              </xsd:complexType>
+            </xsd:element>
+          </xsd:sequence>
+        </xsd:complexType>
+      </xsd:element>
       <xsd:element name="ignoredDataMap">
         <xsd:annotation>
           <xsd:appinfo>

diff --git a/basics/tests/map_init/src/main/resources/example-undefinedMapClass.xsd b/basics/tests/map_init/src/main/resources/example-undefinedMapClass.xsd
@@ -31,6 +31,28 @@
           </xsd:sequence>
         </xsd:complexType>
       </xsd:element>
+      <xsd:element name="weakDataMap">
+        <xsd:annotation>
+          <xsd:appinfo>
+            <mapinit:class>java.util.WeakHashMap</mapinit:class>
+            <jaxb:property>
+              <jaxb:baseType name="java.util.Map&lt;String,Object&gt;" />
+            </jaxb:property>
+          </xsd:appinfo>
+        </xsd:annotation>
+        <xsd:complexType>
+          <xsd:sequence>
+            <xsd:element name="entry" minOccurs="0" maxOccurs="unbounded">
+              <xsd:complexType>
+                <xsd:sequence>
+                  <xsd:element name="key" minOccurs="0" type="xsd:string"/>
+                  <xsd:element name="value" minOccurs="0" type="xsd:anyType"/>
+                </xsd:sequence>
+              </xsd:complexType>
+            </xsd:element>
+          </xsd:sequence>
+        </xsd:complexType>
+      </xsd:element>
       <xsd:element name="ignoredDataMap">
         <xsd:annotation>
           <xsd:appinfo>

diff --git a/basics/tests/map_init/src/test/java/org/jvnet/jaxb/tests/mapinit/EmployeeDefinedTest.java b/basics/tests/map_init/src/test/java/org/jvnet/jaxb/tests/mapinit/EmployeeDefinedTest.java
@@ -5,8 +5,8 @@
 import org.jvnet.jaxb.example.defined.Employee;
 
 import java.util.ArrayList;
-import java.util.HashMap;
 import java.util.TreeMap;
+import java.util.WeakHashMap;
 
 public class EmployeeDefinedTest {
 
@@ -15,6 +15,8 @@ public void testEmployee() {
         Employee a = new Employee();
         Assert.assertNotNull(a.getDataMap());
         Assert.assertTrue(a.getDataMap() instanceof TreeMap);
+        Assert.assertNotNull(a.getWeakDataMap());
+        Assert.assertTrue(a.getWeakDataMap() instanceof WeakHashMap);
         Assert.assertNull(a.getIgnoredDataMap());
         Assert.assertNull(a.getValue());
         Assert.assertNotNull(a.getListValue());

diff --git a/basics/tests/map_init/src/test/java/org/jvnet/jaxb/tests/mapinit/EmployeeUndefinedTest.java b/basics/tests/map_init/src/test/java/org/jvnet/jaxb/tests/mapinit/EmployeeUndefinedTest.java
@@ -6,6 +6,7 @@
 
 import java.util.ArrayList;
 import java.util.HashMap;
+import java.util.WeakHashMap;
 
 public class EmployeeUndefinedTest {
 
@@ -14,6 +15,8 @@ public void testEmployee() {
         Employee a = new Employee();
         Assert.assertNotNull(a.getDataMap());
         Assert.assertTrue(a.getDataMap() instanceof HashMap);
+        Assert.assertNotNull(a.getWeakDataMap());
+        Assert.assertTrue(a.getWeakDataMap() instanceof WeakHashMap);
         Assert.assertNull(a.getIgnoredDataMap());
         Assert.assertNull(a.getValue());
         Assert.assertNotNull(a.getListValue());