XstreamXMLResponseAggregator.java

/*
 * Copyright 2016 Red Hat, Inc. and/or its affiliates.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
*/

package org.kie.server.router.proxy.aggragate;

import static org.kie.server.router.utils.Helper.readProperties;

import java.util.Arrays;
import java.util.List;
import java.util.Properties;

import org.w3c.dom.Document;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;


public class XstreamXMLResponseAggregator extends XMLResponseAggregator {

    private static final String XSTREAM_TYPE = "xstream";

    private static final Properties sortByMapping = readProperties(XstreamXMLResponseAggregator.class.getResourceAsStream("/sort-xstream.mapping"));

    protected List<String> nodes = Arrays.asList(
            "processes",
            "adHocFragments",
            "comments",
            "definitions",
            "caseInstances",
            "milestones",
            "roleAssignments",
            "stages",
            "migrationReportInstances",
            "processNodes",
            "taskNotifications",
            "taskReassignments",
            "timerInstances",
            "queries",
            "tasks",
            "documentInstances",
            "errorInfoInstances",
            "nodeInstances",
            "processInstances",
            "requestInfoInstances",
            "solver",
            "taskAttachments",
            "taskComments",
            "taskEvents",
            "tasks",
            "tasks",
            "variableInstances",
            "workItems",
            "result",
            "list");



    public XstreamXMLResponseAggregator() {

    }

    @Override
    public boolean supports(Object... acceptTypes) {
        return supports(XSTREAM_TYPE, acceptTypes);
    }

    @Override
    protected void copyNodes(NodeList children, Document target, Node targetNode) {

        for (int j = 0; j < children.getLength(); j++) {
            Node existing = children.item(j);

            if (existing.getNodeName().equals("list") && existing.hasChildNodes()) {
                Node imported = target.importNode(existing, true);
                targetNode.appendChild(imported);
            } else if (!nodes.contains(existing.getNodeName()) && existing.hasChildNodes()) {
                Node imported = target.importNode(existing, true);
                targetNode.appendChild(imported);
            }
        }
    }

    @Override
    protected List<String> knownNames() {
        return nodes;
    }

    @Override
    protected String getElementLevel(String rootNode) {
        if ("list".equals(rootNode)) {
            return "1";
        }
        return "2";
    }

    @Override
    protected String getRootNode(String rootNode) {

        return "//*[count(ancestor::*) = 1]";
    }

    @Override
    protected String sortBy(String fieldName) {
        return sortByMapping.getProperty(fieldName, fieldName);
    }
}