com.enioka.jqm.cli.CommandInstallNodeTemplate Maven / Gradle / Ivy

Go to download
package com.enioka.jqm.cli;

import java.util.ArrayList;
import java.util.List;

import org.apache.commons.lang.StringUtils;

import com.beust.jcommander.Parameter;
import com.beust.jcommander.Parameters;
import com.enioka.admin.MetaService;
import com.enioka.api.admin.NodeDto;
import com.enioka.api.admin.QueueMappingDto;
import com.enioka.jqm.cli.api.CommandBase;
import com.enioka.jqm.jdbc.DbConn;
import com.enioka.jqm.jdbc.DbManager;

@Parameters(commandNames = "Install-NodeTemplate", commandDescription = "Apply a template, i.e. the configuration of a node, to another node.")
class CommandInstallNodeTemplate extends CommandBase
{
    @Parameter(names = { "-t", "--template" }, description = "Name of the template to apply (i.e. node to copy from).", required = true)
    private String templateName;

    @Parameter(names = { "-n", "--node" }, description = "Name of the target node (i.e. node to copy to).", required = true)
    private String nodeName;

    @Parameter(names = { "-i", "--interface" }, description = "Network interface to bind to", required = false)
    private String networkInterface;

    @Override
    public int doWork()
    {
        try (DbConn cnx = DbManager.getDb().getConn())
        {
            // Throws exception if nodes not found.
            NodeDto template = MetaService.getNode(cnx, templateName);
            NodeDto target = MetaService.getNode(cnx, nodeName);

            // Apply deployments parameters
            ArrayList mappings = new ArrayList<>(MetaService.getQueueMappings(cnx));
            List toRemove = new ArrayList<>(10);
            List toAdd = new ArrayList<>(10);
            for (QueueMappingDto mapping : mappings)
            {
                if (mapping.getNodeId().equals(template.getId()))
                {
                    QueueMappingDto r = new QueueMappingDto();
                    r.setEnabled(mapping.getEnabled());
                    r.setNbThread(mapping.getNbThread());
                    r.setNodeId(target.getId());
                    r.setNodeName(target.getName());
                    r.setPollingInterval(mapping.getPollingInterval());
                    r.setQueueId(mapping.getQueueId());
                    r.setQueueName(mapping.getQueueName());
                    toAdd.add(r);
                }
                if (mapping.getNodeId().equals(target.getId()))
                {
                    toRemove.add(mapping);
                }
            }

            mappings.addAll(toAdd);
            mappings.removeAll(toRemove);
            MetaService.syncQueueMappings(cnx, mappings);

            // Basic properties
            target.setEnabled(template.getEnabled());
            target.setJmxRegistryPort(template.getJmxRegistryPort());
            target.setJmxServerPort(template.getJmxServerPort());
            target.setJobRepoDirectory(template.getJobRepoDirectory());
            target.setLoadApiAdmin(template.getLoadApiAdmin());
            target.setLoadApiClient(template.getLoadApiClient());
            target.setLoapApiSimple(template.getLoapApiSimple());
            target.setOutputDirectory(template.getOutputDirectory());
            target.setDns(StringUtils.isNotEmpty(networkInterface) ? networkInterface : template.getDns());
            target.setPort(template.getPort());
            target.setRootLogLevel(template.getRootLogLevel());
            target.setTmpDirectory(template.getTmpDirectory());
            MetaService.upsertNode(cnx, target);

            // Done - meta service does not commit
            cnx.commit();
            return 0;
        }
    }
}