如何配置JAXB在编组时解析XML引用？求替代预处理方案

我们正在处理一个包含大量ref引用的大型XML文档，已通过JAXB完成反编组。目前通过三元运算符someElement = someElement.getRef() != null ? (someClass)someElement.getRef() : someElement;解析引用，但该语句执行次数过多，希望彻底避免。此外，使用JAXB.marshal将子节点输出为字符串时，引用未被解析导致内容缺失，期望在编组时自动展开引用内容。

输入XML示例：

<Root>
  <Category>
    <SomeElement ref="someId" />
  </Category>
  <SomeElement id="someId">
    <ValuableContent>HelloWorld</ValuableContent>
  </SomeElement>
</Root>

期望输出XML示例：

<Root>
  <Category>
    <SomeElement>
      <ValuableContent>HelloWorld</ValuableContent>
    </SomeElement>
  </Category>
</Root>

JAXB本身没有内置的引用自动解析机制，但可以通过以下几种方案解决问题：

1. 反编组后一次性预解析所有引用 #

这是最直接的优化方式，彻底避免重复执行三元判断：

• 反编组完成后，先遍历根对象，将所有带id的SomeElement存入Map<String, SomeElement>，建立ID到实例的映射
• 再遍历所有包含引用的节点（比如Category下的SomeElement），直接用ref值从映射中取出实际实例替换引用节点
• 后续业务代码和编组操作直接使用处理后的对象即可

示例代码：

// 反编组得到Root实例
Root root = jaxbContext.createUnmarshaller().unmarshal(xmlSource);

// 构建ID到SomeElement的映射
Map<String, SomeElement> elementMap = new HashMap<>();
for (SomeElement elem : root.getSomeElementList()) {
    if (elem.getId() != null) {
        elementMap.put(elem.getId(), elem);
    }
}

// 替换Category中的引用节点
for (Category category : root.getCategoryList()) {
    SomeElement refElement = category.getSomeElement();
    if (refElement != null && refElement.getRef() != null) {
        category.setSomeElement(elementMap.get(refElement.getRef()));
    }
}

2. 自定义XmlAdapter实现编组时自动解析 #

通过JAXB的XmlAdapter扩展机制，在编组阶段自动将引用替换为实际对象：

• 编写继承XmlAdapter的适配器类，持有ID到实例的映射，marshal方法中判断并替换引用
• 在SomeElement类或其ref字段上添加@XmlJavaTypeAdapter注解，指定适配器
• 编组时传入包含映射的适配器实例即可

示例适配器代码：

public class SomeElementRefAdapter extends XmlAdapter<SomeElement, SomeElement> {
    private final Map<String, SomeElement> elementMap;

    // 构造器传入映射
    public SomeElementRefAdapter(Map<String, SomeElement> elementMap) {
        this.elementMap = elementMap;
    }

    @Override
    public SomeElement marshal(SomeElement elem) throws Exception {
        // 如果是引用节点，替换为实际对象
        if (elem.getRef() != null && elementMap.containsKey(elem.getRef())) {
            return elementMap.get(elem.getRef());
        }
        return elem;
    }

    @Override
    public SomeElement unmarshal(SomeElement elem) throws Exception {
        // 反编组阶段无需处理，直接返回
        return elem;
    }
}

在SomeElement类上添加注解：

@XmlRootElement(name = "SomeElement")
@XmlJavaTypeAdapter(value = SomeElementRefAdapter.class)
public class SomeElement {
    @XmlAttribute
    private String id;
    @XmlAttribute
    private String ref;
    @XmlElement(name = "ValuableContent")
    private String valuableContent;

    // getter、setter省略
}

编组时传入适配器：

Marshaller marshaller = jaxbContext.createMarshaller();
marshaller.setProperty(Marshaller.JAXB_FORMATTED_OUTPUT, true);
// 传递映射给适配器（可自定义属性键）
marshaller.setProperty("elementMap", elementMap);
// 编组输出目标节点
marshaller.marshal(root.getCategoryList().get(0), System.out);

3. XSLT预处理XML（无需修改Java代码） #

如果不想调整JAXB相关的Java代码，可以先用XSLT将原始XML中的引用节点替换为实际内容，再交给JAXB处理：

• 编写XSLT模板，用key函数匹配ID对应的节点，替换带ref的节点
• 用XSLT转换器先处理XML，得到展开引用后的新XML，再进行反编组或直接用于生成PDF

示例XSLT：

<xsl:stylesheet version="1.0" xmlns:xsl="http://www.w3.org/1999/XSL/Transform">
    <!-- 定义ID到节点的键映射 -->
    <xsl:key name="elementById" match="SomeElement[@id]" use="@id"/>

    <!-- 复制所有节点和属性的默认模板 -->
    <xsl:template match="@*|node()">
        <xsl:copy>
            <xsl:apply-templates select="@*|node()"/>
        </xsl:copy>
    </xsl:template>

    <!-- 匹配带ref的SomeElement，替换为对应ID的节点内容 -->
    <xsl:template match="SomeElement[@ref]">
        <xsl:copy-of select="key('elementById', @ref)"/>
    </xsl:template>
</xsl:stylesheet>

内容的提问来源于stack exchange，提问作者Jimmy3Sticks