DepthBasedRecordSplitter.java 2 KB
/*
 * Decompiled with CFR 0_118.
 * 
 * Could not load the following classes:
 *  com.adobe.aemds.guide.batch.api.RecordSplitter
 *  com.adobe.aemfd.docmanager.Document
 *  org.apache.felix.scr.annotations.Component
 *  org.apache.felix.scr.annotations.Service
 *  org.slf4j.Logger
 *  org.slf4j.LoggerFactory
 */
package com.adobe.aemds.guide.batch.impl;

import com.adobe.aemds.guide.batch.api.RecordSplitter;
import com.adobe.aemds.guide.batch.impl.AbstractRecordSplitter;
import com.adobe.aemds.guide.batch.impl.XMLSplitter;
import com.adobe.aemfd.docmanager.Document;
import java.io.InputStream;
import java.util.Map;
import org.apache.felix.scr.annotations.Component;
import org.apache.felix.scr.annotations.Service;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.xml.sax.ext.DefaultHandler2;

@Component(immediate=1, metatype=1, label="Depth-based Record Splitter", description="Default depth-based record splitter service for batch generator.")
@Service(value={RecordSplitter.class})
public class DepthBasedRecordSplitter
extends AbstractRecordSplitter {
    private Logger logger = LoggerFactory.getLogger(DepthBasedRecordSplitter.class);
    private Integer recordLevel;
    private String recordIdentifier;

    public void configure(Object splitter, String identifier) {
        this.recordLevel = splitter != null && splitter instanceof Integer ? (Integer)splitter : XMLSplitter.DEFAULT_RECORD_LEVEL;
        this.recordIdentifier = identifier != null && identifier.trim().length() > 0 ? identifier : "id";
    }

    public Map<Document, String> split(Document dataXML) {
        XMLSplitter xmlSplitter = new XMLSplitter();
        xmlSplitter.setRecordLevel(this.recordLevel);
        xmlSplitter.setRecordIdentifier(this.recordIdentifier);
        try {
            this.parse(dataXML.getInputStream(), xmlSplitter);
        }
        catch (Exception ex) {
            this.logger.error("Exception while splitting XML document", (Throwable)ex);
        }
        return xmlSplitter.getRecordDocuments();
    }
}