io.robe.convert.xml.XMLImporter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of robe-convert Show documentation
There is a newer version: 0.5.0.0-1039
package io.robe.convert.xml;

import com.fasterxml.jackson.core.JsonParser;
import com.fasterxml.jackson.core.JsonToken;
import com.fasterxml.jackson.dataformat.xml.XmlFactory;
import io.robe.convert.common.Importer;
import io.robe.convert.common.OnItemHandler;
import io.robe.convert.xml.parsers.Parsers;

import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.lang.reflect.Field;
import java.util.LinkedList;
import java.util.List;
import java.util.Locale;
import java.util.Map;

public class XMLImporter extends Importer {


    public XMLImporter(Class dataClass) {
        super(dataClass);
    }

    @Override
    public List importStream(InputStream inputStream) throws Exception {
        return importStream(inputStream, DEFAULT_ENCODING);
    }

    @Override
    public List importStream(InputStream inputStream, String charSetName) throws Exception {
        final List list = new LinkedList();
        OnItemHandler handler = new OnItemHandler() {
            @Override
            public void onItem(T item) {
                list.add(item);
            }
        };
        importStream(inputStream, handler, charSetName);
        return list;
    }

    @Override
    public void importStream(InputStream inputStream, OnItemHandler handler) throws Exception {
        importStream(inputStream, handler, DEFAULT_ENCODING);
    }

    @Override
    public void importStream(InputStream inputStream, OnItemHandler handler, String charSetName) throws Exception {

        XmlFactory factory = new XmlFactory();
        JsonParser parser = factory.createParser(new InputStreamReader(inputStream, charSetName));
        JsonToken current;

        Map fields = getFieldMap(getDataClass());

        current = parser.nextToken();
        while (current != JsonToken.START_OBJECT) {
            throw new RuntimeException("Error: root should be object.");

        }

        while (parser.nextToken() != JsonToken.END_OBJECT) {
            try {
                parser.getCurrentName();
            } catch (Exception e) {
                continue;
            }
            if (getDataClass().getSimpleName().equals(parser.getValueAsString())) {
                T item = (T) getDataClass().newInstance();
                while (parser.nextToken() != JsonToken.END_OBJECT) {
                    if (parser.getValueAsString() == null || parser.getCurrentToken() == JsonToken.FIELD_NAME)
                        continue;
                    Field field = fields.get(parser.getCurrentName());
                    setField(parser, item, field);
                }
                handler.onItem(item);
            }

        }

    }

    private  void setField(JsonParser parser, T item, Field field) throws IllegalAccessException, IOException {
        String fieldType = field.getType().getSimpleName().toUpperCase(Locale.ENGLISH);
        if (field.getType().getGenericSuperclass() != null) {
            if (field.getType().getGenericSuperclass().toString().startsWith("java.lang.Enum")) {
                fieldType = "ENUM";
            }
        }
        boolean acc = field.isAccessible();
        field.setAccessible(true);
        field.set(item, Parsers.valueOf(fieldType).getParser().parse(parser, field));
        field.setAccessible(acc);
    }


}