com.enioka.jqm.tools.XmlQueueParser Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jqm-engine Show documentation
A library containing the JQM engine. Cannot be used alone.
There is a newer version: 2.2.9
/**
 * Copyright © 2013 enioka. All rights reserved
 * Authors: Marc-Antoine GOUILLART ([email protected])
 *          Pierre COPPEE ([email protected])
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * 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 com.enioka.jqm.tools;

import java.io.File;

import javax.persistence.EntityManager;
import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;

import org.apache.log4j.Logger;
import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.NodeList;

import com.enioka.jqm.jpamodel.JobDef;
import com.enioka.jqm.jpamodel.Queue;

class XmlQueueParser
{
    private static Logger jqmlogger = Logger.getLogger(XmlQueueParser.class);

    private XmlQueueParser()
    {
        // Utility class.
    }

    static void parse(String path, EntityManager em) throws JqmEngineException
    {
        // Argument checks
        jqmlogger.trace(path);
        if (path == null || path.isEmpty())
        {
            throw new IllegalArgumentException("XML file path cannot be empty");
        }
        if (em == null)
        {
            throw new IllegalArgumentException("EntityManager cannot be null");
        }
        File f = new File(path);
        if (f == null || !f.isFile() || !f.canRead())
        {
            throw new IllegalArgumentException("The XML file " + f + " was not found or cannot be read.");
        }

        // Create parsers
        DocumentBuilderFactory dbFactory = DocumentBuilderFactory.newInstance();
        DocumentBuilder dBuilder;

        try
        {
            Queue q;

            dBuilder = dbFactory.newDocumentBuilder();
            Document doc = dBuilder.parse(f);
            doc.getDocumentElement().normalize();

            NodeList qList = doc.getElementsByTagName("queue");

            for (int qIndex = 0; qIndex < qList.getLength(); qIndex++)
            {
                Element qElement = (Element) qList.item(qIndex);
                em.getTransaction().begin();

                // Insert or update?
                String qName = qElement.getElementsByTagName("name").item(0).getTextContent();
                q = Helpers.findQueue(qName, em);
                if (q == null)
                {
                    q = new Queue();
                }

                // Simple fields
                q.setName(qElement.getElementsByTagName("name").item(0).getTextContent());
                q.setDescription(qElement.getElementsByTagName("description").item(0).getTextContent());
                if (qElement.getElementsByTagName("timeToLive").getLength() == 1)
                {
                    q.setTimeToLive(Integer.parseInt(qElement.getElementsByTagName("timeToLive").item(0).getTextContent()));
                }

                // We now merge & commit - we will need to reference the queue in the next paragraph.
                q = em.merge(q);
                em.getTransaction().commit();

                // Applications that should use this queue
                em.getTransaction().begin();
                NodeList appList = qElement.getElementsByTagName("applicationName");
                for (int appIndex = 0; appIndex < appList.getLength(); appIndex++)
                {
                    Element appElement = (Element) appList.item(appIndex);

                    jqmlogger.debug("Default queue of the job " + appElement.getTextContent() + " must be changed");
                    JobDef jd = Helpers.findJobDef(appElement.getTextContent(), em);
                    if (jd != null)
                    {
                        jd.setQueue(q);
                    }
                }
                em.getTransaction().commit();
            }
        }
        catch (Exception e)
        {
            throw new JqmEngineException("Could not parse the Queue XML", e);
        }
    }
}