com.dattack.dbtools.drules.DrulesClient.java Source code

Introduction

Here is the source code for com.dattack.dbtools.drules.DrulesClient.java

Source

/*
 * Copyright (c) 2016, The Dattack team (http://www.dattack.com)
 *
 * 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.dattack.dbtools.drules;

import java.util.List;

import org.apache.commons.cli.CommandLine;
import org.apache.commons.cli.CommandLineParser;
import org.apache.commons.cli.DefaultParser;
import org.apache.commons.cli.HelpFormatter;
import org.apache.commons.cli.Option;
import org.apache.commons.cli.Options;
import org.apache.commons.cli.ParseException;
import org.apache.commons.configuration.CompositeConfiguration;
import org.apache.commons.configuration.ConfigurationException;
import org.apache.commons.configuration.PropertiesConfiguration;
import org.apache.commons.lang.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.dattack.dbtools.GlobalConfiguration;
import com.dattack.dbtools.drules.beans.Identifier;
import com.dattack.dbtools.drules.beans.Identifier.IdentifierBuilder;
import com.dattack.dbtools.drules.engine.DrulesEngine;
import com.dattack.dbtools.drules.exceptions.DrulesNestableException;
import com.dattack.jtoolbox.commons.configuration.ConfigurationUtil;

/**
 * @author cvarela
 * @since 0.1
 */
public final class DrulesClient {

    private static final Logger LOGGER = LoggerFactory.getLogger(DrulesClient.class);

    private static final String DRULES_OPTION = "d";
    private static final String TASK_OPTION = "t";
    private static final String PROPERTIES_OPTION = "p";

    private static Options createOptions() {

        final Options options = new Options();

        options.addOption(Option.builder(DRULES_OPTION) //
                .required(true) //
                .longOpt("drules") //
                .hasArg(true) //
                .argName("DRULES_FILE") //
                .desc("the path of the file containing the DRules configuration") //
                .build());

        options.addOption(Option.builder(TASK_OPTION) //
                .required(false) //
                .longOpt("task") //
                .hasArg(true) //
                .argName("TASK_NAME") //
                .desc("the name of the task being performed") //
                .build());

        options.addOption(Option.builder(PROPERTIES_OPTION) //
                .required(false) //
                .longOpt("properties") //
                .hasArg(true) //
                .argName("PROPERTIES") //
                .desc("the path of the file containing custom runtime properties") //
                .build());

        return options;
    }

    private static void execute(final String[] args) throws ConfigurationException, DrulesNestableException {

        final Options options = createOptions();

        try {
            final CommandLineParser parser = new DefaultParser();
            final CommandLine cmd = parser.parse(options, args);
            final String filename = cmd.getOptionValue(DRULES_OPTION);

            CompositeConfiguration configuration = null;
            if (cmd.hasOption(PROPERTIES_OPTION)) {
                configuration = loadProperties(cmd.getOptionValues(PROPERTIES_OPTION));
            }

            final DrulesEngine engine = new DrulesEngine(filename, configuration);

            if (cmd.hasOption(TASK_OPTION)) {
                for (final String taskName : cmd.getOptionValues(TASK_OPTION)) {
                    final Identifier taskId = new IdentifierBuilder().withValue(taskName).build();
                    engine.execute(taskId);
                }
            } else {
                // list available tasks
                showTaskList(engine.listTasks());

            }
        } catch (@SuppressWarnings("unused") final ParseException e) {
            showUsage(options);
        }
    }

    private static CompositeConfiguration loadProperties(final String[] propertyFiles)
            throws ConfigurationException {

        final CompositeConfiguration configuration = new CompositeConfiguration();
        configuration.addConfiguration(GlobalConfiguration.getConfiguration());

        if (propertyFiles != null) {
            for (final String filename : propertyFiles) {
                if (StringUtils.isNotBlank(filename)) {
                    configuration.addConfiguration(
                            new PropertiesConfiguration(ConfigurationUtil.interpolate(filename, configuration)));
                }
            }
        }
        return configuration;
    }

    /**
     * Main() method.
     *
     * @param args
     *            the arguments
     */
    public static void main(final String[] args) {

        try {
            execute(args);
            System.exit(0);
        } catch (final Throwable e) { // NOPMD by cvarela on 20/02/16 13:19
            LOGGER.error(e.getMessage(), e);
            System.exit(-1);
        }
    }

    private static void showTaskList(final List<Identifier> taskList) {

        if (taskList == null || taskList.isEmpty()) {
            System.out.println("There are no tasks available to you yet");
        } else {
            System.out.println("Available tasks:");
            for (int index = 0; index < taskList.size(); index++) {
                System.out.format("  % 3d) %s%n", index + 1, taskList.get(index).getValue());
            }
        }
    }

    private static void showUsage(final Options options) {
        final HelpFormatter formatter = new HelpFormatter();
        final int descPadding = 5;
        final int leftPadding = 4;
        formatter.setDescPadding(descPadding);
        formatter.setLeftPadding(leftPadding);
        final String header = "\n";
        final String footer = "\nPlease report issues at https://github.com/dattack/dbtools/issues";
        formatter.printHelp("drules ", header, options, footer, true);
    }

    private DrulesClient() {
        // static class
    }
}