edu.usc.pgroup.floe.client.commands.Signal.java Source code

Introduction

Here is the source code for edu.usc.pgroup.floe.client.commands.Signal.java

Source

/*
 * Copyright 2014 University of Southern California
 *
 * 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 edu.usc.pgroup.floe.client.commands;

import edu.usc.pgroup.floe.client.FloeClient;
import edu.usc.pgroup.floe.thriftgen.TSignal;
import edu.usc.pgroup.floe.utils.Utils;
import org.apache.commons.cli.BasicParser;
import org.apache.commons.cli.CommandLine;
import org.apache.commons.cli.CommandLineParser;
import org.apache.commons.cli.HelpFormatter;
import org.apache.commons.cli.Option;
import org.apache.commons.cli.OptionBuilder;
import org.apache.commons.cli.Options;
import org.apache.commons.cli.ParseException;
import org.apache.thrift.TException;
import org.slf4j.LoggerFactory;

/**
 * @author kumbhare
 */
public final class Signal {

    /**
     * Hiding the default constructor.
     */
    private Signal() {

    }

    /**
     * the global logger instance.
     */
    private static final org.slf4j.Logger LOGGER = LoggerFactory.getLogger(Scale.class);

    /**
     * Entry point for Scale command.
     * @param args command line arguments sent by the floe.py script.
     */
    public static void main(final String[] args) {

        Options options = new Options();

        Option appOption = OptionBuilder.withArgName("name").hasArg().isRequired()
                .withDescription("Application Name").create("app");

        Option pelletNameOption = OptionBuilder.withArgName("name").hasArg().isRequired()
                .withDescription("Pellet Name").create("pellet");

        Option signalOption = OptionBuilder.withArgName("data").hasArg().withType(new String())
                .withDescription("signal data to send to the pellet").create("data");

        options.addOption(appOption);
        options.addOption(pelletNameOption);
        options.addOption(signalOption);

        CommandLineParser parser = new BasicParser();
        CommandLine line;

        try {
            line = parser.parse(options, args);

        } catch (ParseException e) {
            LOGGER.error("Invalid command: " + e.getMessage());
            HelpFormatter formatter = new HelpFormatter();
            formatter.printHelp("scale options", options);
            return;
        }

        String app = line.getOptionValue("app");
        String pellet = line.getOptionValue("pellet");
        String data = line.getOptionValue("data");

        LOGGER.info("Application: {}", app);
        LOGGER.info("Pellet: {}", pellet);
        LOGGER.info("data: {}", data);

        byte[] datab = Utils.serialize(data);
        try {
            TSignal signal = new TSignal();
            signal.set_destApp(app);
            signal.set_destPellet(pellet);
            signal.set_data(datab);
            FloeClient.getInstance().getClient().signal(signal);
        } catch (TException e) {
            LOGGER.error("Error while connecting to the coordinator: {}", e);
        }
    }
}