Использование JAXB для перекрестной ссылки XmlID из двух файлов XML

Question

Я пытаюсь маршалировать / демаршировать из двух разных XML-файлов в POJOS.Первый XML-файл выглядит следующим образом:

--Network.xml--
<Network>
  <Nodes>
    <Node id="ROD" />
    <Node id="KFI" />
    <Node id="JND" />
  </Nodes>
  <Arcs>
    <Arc fromNode="ROD" />
    <Arc fromNode="JND" />
  </Arcs>
</Network>
---------

Используя аннотации @XmlID и @XmlIDREF, я могу успешно заполнить классы Arc, чтобы указать на правильный узел, на который он ссылается.

Однако мне также нужно проанализировать этот XML:

--NetworkInputs.xml--
<NetworkInputs>
  <Flows>
    <Flow toNode="JND" />
    <Flow toNode="ROD" />
  </Flows>
</NetworkInputs>
------

В настоящее время моя программа успешно демонтирует объект Network, но нет соединения между Network и NetworkInputs, которое позволяет JAXB "видеть"узлы, которые существуют в сети.Я хочу, чтобы мои объекты Flow указывали на правильный узел в классе Network.

Я в основном хочу сделать это: http://old.nabble.com/JAXB-Unmarshalling-and-XmlIDREF-using-different-stores-td14035248.html

Я пытался реализовать это: http://weblogs.java.net/blog/kohsuke/archive/2005/08/pluggable_ididr.html иэто просто не работает, потому что я не могу получить данные узла для моей заполненной сети из статического контекста.

Можно ли вообще сделать что-то подобное?

Answer 1

Это можно сделать с помощью XmlAdapter. Хитрость в том, что XmlAdapter необходимо инициализировать всеми узлами из Network.xml и передать Unmarshaller, используемому с NetworkInputs.xml:

import java.io.File;
import javax.xml.bind.JAXBContext;
import javax.xml.bind.Marshaller;
import javax.xml.bind.Unmarshaller;

public class Demo {

    public static void main(String[] args) throws Exception {
        JAXBContext jc = JAXBContext.newInstance(Network.class, NetworkInputs.class);

        File networkXML = new File("Network.xml");
        Unmarshaller unmarshaller = jc.createUnmarshaller();
        Network network = (Network) unmarshaller.unmarshal(networkXML);

        File networkInputsXML = new File("NetworkInputs.xml");
        Unmarshaller unmarshaller2 = jc.createUnmarshaller();
        NodeAdapter nodeAdapter = new NodeAdapter();
        for(Node node : network.getNodes()) {
            nodeAdapter.getNodes().put(node.getId(), node);
        }
        unmarshaller2.setAdapter(nodeAdapter);
        NetworkInputs networkInputs = (NetworkInputs) unmarshaller2.unmarshal(networkInputsXML);

        Marshaller marshaller = jc.createMarshaller();
        marshaller.setProperty(Marshaller.JAXB_FORMATTED_OUTPUT, true);
        marshaller.marshal(networkInputs, System.out);
    }
}

Хитрость в том, чтобы отобразить свойство toNode в Flow с помощью XmlAdapter:

import javax.xml.bind.annotation.XmlAttribute;
import javax.xml.bind.annotation.adapters.XmlJavaTypeAdapter;

public class Flow {

    private Node toNode;

    @XmlAttribute
    @XmlJavaTypeAdapter(NodeAdapter.class)
    public Node getToNode() {
        return toNode;
    }

    public void setToNode(Node toNode) {
        this.toNode = toNode;
    }

}

Адаптер будет выглядеть следующим образом. Хитрость в том, что мы передадим сконфигурированный XmlAdapter, который знает обо всех узлах, маршаллеру:

import java.util.HashMap;
import java.util.Map;

import javax.xml.bind.annotation.adapters.XmlAdapter;

public class NodeAdapter extends XmlAdapter<String, Node>{

    private Map<String, Node> nodes = new HashMap<String, Node>();

    public Map<String, Node> getNodes() {
        return nodes;
    }

    @Override
    public Node unmarshal(String v) throws Exception {
        return nodes.get(v);
    }

    @Override
    public String marshal(Node v) throws Exception {
        return v.getId();
    }

}

Answer 2

Мое решение: разрешение идентификатора обрабатывается (к сожалению) внутренним классом (com.sun.xml.internal.bind.IDResolver), который можно установить из внешнего.

final Unmarshaller unmarshaller = context.createUnmarshaller();
unmarshaller.setProperty(IDResolver.class.getName(), resolver);

Где резольвер можно использовать в течение многих инстанций у маршаллера. Но дело в том, что распознаватель не очистит себя от startDocument, как стандартная реализация com.sun.xml.internal.bind.v2.runtime.unmarshaller.DefaultIDResolver:

import java.text.MessageFormat;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.Callable;

import org.xml.sax.SAXException;

import com.sun.xml.internal.bind.IDResolver;

public final class IDResolverExtension extends IDResolver {
    public static final class CallableImplementation implements Callable<Object> {
        private final Object value;

        private CallableImplementation(final Object value) {
            this.value = value;
        }

        @Override
        public Object call() {
            return value;
        }
    }

    private final Map<KeyAndClass, Object> m = new HashMap<KeyAndClass, Object>();

    @SuppressWarnings("rawtypes")
    @Override
    public synchronized CallableImplementation resolve(final String key0, final Class clazz) throws SAXException {
        assert clazz != null;
        assert key0 != null;
        final KeyAndClass key = new KeyAndClass(clazz, key0);
        final Object value = m.get(key);
        return new CallableImplementation(value);
    }

    static class KeyAndClass {
        public final Class<?> clazz;
        public final String key;

        public KeyAndClass(final Class<?> clazz, final String key) {
            this.clazz = clazz;
            this.key = key;
        }

        @Override
        public int hashCode() {
            final int prime = 31;
            int result = 1;
            result = prime * result + clazz.hashCode();
            result = prime * result + key.hashCode();
            return result;
        }

        @Override
        public boolean equals(Object obj) {
            if (this == obj) {
                return true;
            }
            if (obj == null) {
                return false;
            }
            if (getClass() != obj.getClass()) {
                return false;
            }
            final KeyAndClass other = (KeyAndClass) obj;
            if (!clazz.equals(other.clazz)) {
                return false;
            }
            if (!key.equals(other.key)) {
                return false;
            }
            return true;
        }

    }

    @Override
    public synchronized void bind(final String key0, final Object value) throws SAXException {
        assert key0 != null;
        assert value != null;
        Class<? extends Object> clazz = value.getClass();
        assert clazz != null;
        final KeyAndClass key = new KeyAndClass(clazz, key0);
        final Object oldValue = m.put(key, value);
        if (oldValue != null) {
            final String message = MessageFormat.format("duplicated key ''{0}'' => ''{1}'' - old: ''{2}''", key, value,
                    oldValue);
            throw new AssertionError(message);
        }
    }
}