Search in sources :

Example 1 with MonitorManager

use of org.apache.phoenix.pherf.jmx.MonitorManager in project phoenix by apache.

the class ResultTest method testMonitorResult.

@Test
public void testMonitorResult() throws Exception {
    ExecutorService executorService = Executors.newFixedThreadPool(1);
    MonitorManager monitor = new MonitorManager(100);
    Future future = executorService.submit(monitor.execute());
    List<Result> records;
    final int TIMEOUT = 30;
    int ct = 0;
    int max = 30;
    // Wait while we write some rows.
    while (!future.isDone()) {
        Thread.sleep(100);
        if (ct == max) {
            int timer = 0;
            monitor.complete();
            while (monitor.isRunning() && (timer < TIMEOUT)) {
                System.out.println("Waiting for monitor to finish. Seconds Waited :" + timer);
                Thread.sleep(1000);
                timer++;
            }
        }
        ct++;
    }
    executorService.shutdown();
    records = monitor.readResults();
    assertNotNull("Could not retrieve records", records);
    assertTrue("Failed to get correct CSV records.", records.size() > 0);
    assertFalse("Monitor was not stopped correctly.", monitor.isRunning());
}
Also used : MonitorManager(org.apache.phoenix.pherf.jmx.MonitorManager) ExecutorService(java.util.concurrent.ExecutorService) Future(java.util.concurrent.Future) Test(org.junit.Test)

Example 2 with MonitorManager

use of org.apache.phoenix.pherf.jmx.MonitorManager in project phoenix by apache.

the class Pherf method run.

public void run() throws Exception {
    MonitorManager monitorManager = null;
    List<Workload> workloads = new ArrayList<>();
    WorkloadExecutor workloadExecutor = new WorkloadExecutor(properties, workloads, !isFunctional);
    try {
        if (listFiles) {
            ResourceList list = new ResourceList(PherfConstants.RESOURCE_DATAMODEL);
            Collection<Path> schemaFiles = list.getResourceList(PherfConstants.SCHEMA_ROOT_PATTERN + ".sql");
            System.out.println("Schema Files:");
            for (Path path : schemaFiles) {
                System.out.println(path);
            }
            list = new ResourceList(PherfConstants.RESOURCE_SCENARIO);
            Collection<Path> scenarioFiles = list.getResourceList(PherfConstants.SCENARIO_ROOT_PATTERN + ".xml");
            System.out.println("Scenario Files:");
            for (Path path : scenarioFiles) {
                System.out.println(path);
            }
            return;
        }
        // Compare results and exit  
        if (null != compareResults) {
            logger.info("\nStarting to compare results and exiting for " + compareResults);
            new GoogleChartGenerator(compareResults, compareType).readAndRender();
            return;
        }
        XMLConfigParser parser = new XMLConfigParser(scenarioFile);
        // Drop tables with PHERF schema and regex comparison
        if (null != dropPherfTablesRegEx) {
            logger.info("\nDropping existing table with PHERF namename and " + dropPherfTablesRegEx + " regex expression.");
            phoenixUtil.deleteTables(dropPherfTablesRegEx);
        }
        if (monitor) {
            monitorManager = new MonitorManager(Integer.parseInt(properties.getProperty("pherf.default.monitorFrequency")));
            workloadExecutor.add(monitorManager);
        }
        if (applySchema) {
            logger.info("\nStarting to apply schema...");
            SchemaReader reader = (schemaFile == null) ? new SchemaReader(".*.sql") : new SchemaReader(schemaFile);
            reader.applySchema();
        }
        // Schema and Data Load
        if (preLoadData) {
            logger.info("\nStarting Data Load...");
            Workload workload = new WriteWorkload(parser, generateStatistics);
            try {
                workloadExecutor.add(workload);
                // Wait for dataLoad to complete
                workloadExecutor.get(workload);
            } finally {
                if (null != workload) {
                    workload.complete();
                }
            }
        } else {
            logger.info("\nSKIPPED: Data Load and schema creation as -l argument not specified");
        }
        // Execute multi-threaded query sets
        if (executeQuerySets) {
            logger.info("\nStarting to apply Execute Queries...");
            workloadExecutor.add(new QueryExecutor(parser, phoenixUtil, workloadExecutor, parser.getDataModels(), queryHint, isFunctional, writeRuntimeResults));
        } else {
            logger.info("\nSKIPPED: Multithreaded query set execution as -q argument not specified");
        }
        // Clean up the monitor explicitly
        if (monitorManager != null) {
            logger.info("Run completed. Shutting down Monitor.");
            monitorManager.complete();
        }
        // Collect any final jobs
        workloadExecutor.get();
    } finally {
        if (workloadExecutor != null) {
            logger.info("Run completed. Shutting down thread pool.");
            workloadExecutor.shutdown();
        }
    }
}
Also used : Path(java.nio.file.Path) SchemaReader(org.apache.phoenix.pherf.schema.SchemaReader) WriteWorkload(org.apache.phoenix.pherf.workload.WriteWorkload) ArrayList(java.util.ArrayList) XMLConfigParser(org.apache.phoenix.pherf.configuration.XMLConfigParser) WorkloadExecutor(org.apache.phoenix.pherf.workload.WorkloadExecutor) Workload(org.apache.phoenix.pherf.workload.Workload) WriteWorkload(org.apache.phoenix.pherf.workload.WriteWorkload) MonitorManager(org.apache.phoenix.pherf.jmx.MonitorManager) ResourceList(org.apache.phoenix.pherf.util.ResourceList) GoogleChartGenerator(org.apache.phoenix.pherf.util.GoogleChartGenerator) QueryExecutor(org.apache.phoenix.pherf.workload.QueryExecutor)

Aggregations

MonitorManager (org.apache.phoenix.pherf.jmx.MonitorManager)2 Path (java.nio.file.Path)1 ArrayList (java.util.ArrayList)1 ExecutorService (java.util.concurrent.ExecutorService)1 Future (java.util.concurrent.Future)1 XMLConfigParser (org.apache.phoenix.pherf.configuration.XMLConfigParser)1 SchemaReader (org.apache.phoenix.pherf.schema.SchemaReader)1 GoogleChartGenerator (org.apache.phoenix.pherf.util.GoogleChartGenerator)1 ResourceList (org.apache.phoenix.pherf.util.ResourceList)1 QueryExecutor (org.apache.phoenix.pherf.workload.QueryExecutor)1 Workload (org.apache.phoenix.pherf.workload.Workload)1 WorkloadExecutor (org.apache.phoenix.pherf.workload.WorkloadExecutor)1 WriteWorkload (org.apache.phoenix.pherf.workload.WriteWorkload)1 Test (org.junit.Test)1