io.rhiot.spec.IoTSpec.java Source code

Introduction

Here is the source code for io.rhiot.spec.IoTSpec.java

Source

/*
 * Licensed to the Rhiot under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The licenses this file to You 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 io.rhiot.spec;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.dataformat.yaml.YAMLFactory;
import io.rhiot.spec.report.CSVReport;
import org.apache.commons.cli.*;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.File;
import java.util.List;
import java.util.concurrent.*;

public class IoTSpec {

    private static final Logger LOG = LoggerFactory.getLogger(IoTSpec.class);
    private static final String CONFIG = "config";
    private static final String INSTANCE = "instance";
    private static final String REPORT = "report";

    public static void main(String[] args) throws Exception {

        CommandLineParser parser = new DefaultParser();

        Options options = new Options();
        options.addOption(Option.builder("c").longOpt(CONFIG).desc(
                "Location of the test configuration file. A default value is 'src/main/resources/test.yaml' for easy IDE testing")
                .hasArg().build());

        options.addOption(Option.builder("i").longOpt(INSTANCE).desc("Instance of the test; A default value is 1")
                .hasArg().build());

        options.addOption(Option.builder("r").longOpt(REPORT)
                .desc("Location of the test report. A default value is 'target/report.csv'").hasArg().build());

        CommandLine line = parser.parse(options, args);
        ObjectMapper mapper = new ObjectMapper(new YAMLFactory());
        TestProfile test = mapper.readValue(new File(line.getOptionValue(CONFIG, "src/main/resources/test.yaml")),
                TestProfile.class);
        int instance = Integer.valueOf(line.getOptionValue(INSTANCE, "1"));
        test.setInstance(instance);
        String report = line.getOptionValue(REPORT, "target/report.csv");
        test.setReport(new CSVReport(report));

        LOG.info("Test '" + test.getName() + "' instance " + instance + " started");
        final List<Driver> drivers = test.getDrivers();
        ExecutorService executorService = Executors.newFixedThreadPool(drivers.size());
        List<Future<Void>> results = executorService.invokeAll(drivers, test.getDuration(), TimeUnit.MILLISECONDS);
        executorService.shutdownNow();
        executorService.awaitTermination(5, TimeUnit.SECONDS);

        results.forEach(result -> {
            try {
                result.get();
            } catch (ExecutionException execution) {
                LOG.warn("Exception running driver", execution);
            } catch (Exception interrupted) {
            }
        });

        drivers.forEach(driver -> {
            driver.stop();

            try {
                test.getReport().print(driver);
            } catch (Exception e) {
                LOG.warn("Failed to write reports for the driver " + driver);
            }
            LOG.debug("Driver " + driver);
            LOG.debug("\t " + driver.getResult());
        });
        test.getReport().close();

        LOG.info("Test '" + test.getName() + "' instance " + instance + " finished");

    }
}