ScanManager.java

package ciscoroutertool.scanner;

import ciscoroutertool.rules.Rule;
import ciscoroutertool.rules.RuleParser;
import ciscoroutertool.utils.Host;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.Future;
import java.util.logging.Level;
import java.util.logging.Logger;

/**
 * Runs the scan on each host and prepares data for the report
 * @version 0.01ALPHA
 * @author Andrew Johnston
 */
public class ScanManager {

    /**
     * The list of rules to run against each host
     */
    private ArrayList<Rule> rules;
    
    /**
     * The hosts we are going to scan
     */
    private ArrayList<Host> hosts;
    
    /**
     * The maximum number of scanners to run at once
     */
    private static final int MAX_THREADS = 5;
    
    /**
     * Sets up the ScanManager so it can run scans against the hosts
     * @param _hosts The hosts you want to scan
     */
    public ScanManager(ArrayList<Host> _hosts) {
        hosts = _hosts;
        rules = RuleParser.getRules();
        //Give the rules to the scanner class
        Scanner.rules = rules;
    }


    /**
     * Runs the scan on all hosts and returns the full report
     * @return The report covering all hosts
     */
    public FullReport run() {
        FullReport fullReport = null;
        try {
            ExecutorService executor = Executors.newFixedThreadPool(MAX_THREADS);
            List<Future<HostReport>> futureReports;
            ArrayList<HostReport> reports = new ArrayList<>();
            ArrayList<Scanner> scanners = new ArrayList<>();
            for (Host h : hosts) {
                Scanner s = new Scanner(h);
                scanners.add(s);
            }
            futureReports = executor.invokeAll(scanners);
            for (Future<HostReport> fReport : futureReports) {
                reports.add(fReport.get());
            }
            fullReport = new FullReport(reports);
        } catch (InterruptedException | ExecutionException ex) {
            Logger.getLogger(ScanManager.class.getName()).log(Level.SEVERE, null, ex);
        }
        return fullReport;
    }

}