Examples with ImportantThread com.axway.ats.agent.core.threading.ImportantThread used on opensource projects

Example 1 with ImportantThread

use of com.axway.ats.agent.core.threading.ImportantThread in project ats-framework by Axway.

the class DistributedLoadExecutor method executeActions.

@Override
public void executeActions(List<ActionRequest> actionRequests) throws AgentException {
    //set the block until completion param to false
    //this way we'll be able to start all the loaders on all boxes at the same time
    //and then wait for all of them to finish
    final boolean blockUntilCompletion = threadingPattern.isBlockUntilCompletion();
    threadingPattern.setBlockUntilCompletion(false);
    int maxHostCount = atsAgents.size();
    // distribute the threading pattern for each agent host
    final List<ThreadingPattern> distributedPatterns = threadingPattern.distribute(maxHostCount);
    if (distributedPatterns.size() < maxHostCount) {
        log.warn("Threading pattern cannot be distributed accross all agents, only the first " + distributedPatterns.size() + " agents will execute");
    }
    // distribute the data configurators for each agent host
    final List<LoaderDataConfig> distributedLoaderDataConfigs = loaderDataConfig.distribute(distributedPatterns.size());
    // start queue in the database and retrieve its ID
    int queueId = retrieveQueueId(queueSequence, getHostsList());
    //call the web service now
    try {
        //first schedule the loaders on all hosts
        for (int i = 0; i < distributedPatterns.size(); i++) {
            //serialize the threading pattern - it's easier to pass it to the web service that way
            byte[] serializedThreadingPattern = serializeObject(distributedPatterns.get(i));
            byte[] serializedLoaderDataConfig = serializeObject(distributedLoaderDataConfigs.get(i));
            //wrap all the action requests
            List<ActionWrapper> actionWrappers = new ArrayList<ActionWrapper>();
            for (ActionRequest actionRequest : actionRequests) {
                actionWrappers.add(wrapActionRequest(actionRequest));
            }
            //schedule the actions, but do not execute
            //get the client
            AgentService agentServicePort = AgentServicePool.getInstance().getClient(atsAgents.get(i));
            agentServicePort.scheduleActionsInMultipleThreads(queueName, queueId, actionWrappers, serializedThreadingPattern, serializedLoaderDataConfig, distributedPatterns.get(0).isUseSynchronizedIterations());
        }
        boolean useSynchronizedIterations = distributedPatterns.get(0).isUseSynchronizedIterations();
        if (useSynchronizedIterations && !blockUntilCompletion) {
            // It is non blocking, but synchronized - we have to wait until the queue finish its execution.
            // We first start the queue. This assures the queue is created before the user have the chance
            // to say "wait for completion"
            startSynchedIterations();
            // and then wait in another thread until the queue finish
            ImportantThread helpThread = new ImportantThread(new Runnable() {

                @Override
                public void run() {
                    runSynchedIterations();
                }
            });
            helpThread.setDescription(queueName);
            helpThread.start();
        } else {
            if (useSynchronizedIterations) {
                // it is blocking - we can wait until the queue finish it execution
                startSynchedIterations();
                runSynchedIterations();
            } else {
                // if blocking - we can wait until the queue finish it execution
                // if non blocking - as it is not synchronized, we will wait only until the queue is started
                runNotSynchedIterations(blockUntilCompletion);
            }
        }
    } catch (AgentException_Exception ae) {
        throw new AgentException(ae.getMessage());
    } catch (InternalComponentException_Exception ice) {
        //we need to get internal component exception info from the soap fault
        InternalComponentException faultInfo = ice.getFaultInfo();
        //then construct and throw a real InternalComponentException (not the JAXB mapping type above)
        throw new com.axway.ats.agent.core.exceptions.InternalComponentException(faultInfo.getComponentName(), faultInfo.getActionName(), faultInfo.getExceptionMessage(), faultInfo.getHostIp());
    } catch (Exception e) {
        throw new AgentException(e.getMessage(), e);
    }
    // restore this flag
    threadingPattern.setBlockUntilCompletion(blockUntilCompletion);
}

Example 2 with ImportantThread

use of com.axway.ats.agent.core.threading.ImportantThread in project ats-framework by Axway.

the class DistributedLoadExecutor method runNotSynchedIterations.

// TODO When running non blocking, this method is started from a dedicated thread,
// so throwing an error does not affect the main test execution thread,
// so the test does not fail.
// We can keep this errors in some structure which can be checked when
// closing the test case, so we can fail it when needed
private void runNotSynchedIterations(boolean blockUntilCompletion) {
    log.info("Start executing action queue '" + queueName + "' on " + atsAgents.toString() + ". " + (blockUntilCompletion ? "And wait to finish." : "Will not wait to finish."));
    try {
        // start the actions on all loaders
        for (int i = 0; i < atsAgents.size(); i++) {
            String host = atsAgents.get(i);
            AgentService agentServicePort = AgentServicePool.getInstance().getClient(host);
            agentServicePort.startQueue(queueName);
        }
    } catch (Exception e) {
        String msg = "Error starting action queue '" + queueName + "'";
        log.error(msg, e);
        throw new RuntimeException(msg, e);
    }
    try {
        //wait until finished
        if (blockUntilCompletion) {
            waitUntilQueueFinishOnAllLoaders();
        } else {
            // and then wait in another thread until the queue finish
            ImportantThread helpThread = new ImportantThread(new Runnable() {

                @Override
                public void run() {
                    waitUntilQueueFinishOnAllLoaders();
                }
            });
            helpThread.setDescription(queueName);
            helpThread.start();
        }
    } catch (Exception e) {
        String msg = "Error waiting for completion of action queue '" + queueName + "'";
        log.error(msg, e);
        throw new RuntimeException(msg, e);
    }
}

Example 3 with ImportantThread

use of com.axway.ats.agent.core.threading.ImportantThread in project ats-framework by Axway.

the class LocalLoadExecutor method executeActions.

@Override
public void executeActions(List<ActionRequest> actionRequests) throws AgentException {
    log.info("Start executing action queue '" + queueName + "'." + (threadingPattern.isBlockUntilCompletion() ? " And wait to finish." : ""));
    // start queue in the database and retrieve its ID
    int queueId = retrieveQueueId(queueSequence, HostUtils.getLocalHostIP());
    try {
        MultiThreadedActionHandler.getInstance(ThreadsPerCaller.getCaller()).executeActions("LOCAL", queueName, queueId, actionRequests, threadingPattern, loaderDataConfig);
        if (threadingPattern.isBlockUntilCompletion()) {
            log.endLoadQueue(queueName, LoadQueueResult.PASSED);
            log.info("Finished executing action queue '" + queueName + "'");
        } else {
            // wait in another thread until the queue finish
            ImportantThread helpThread = new ImportantThread(new Runnable() {

                @Override
                public void run() {
                    try {
                        MultiThreadedActionHandler.getInstance(ThreadsPerCaller.getCaller()).waitUntilQueueFinish(queueName);
                        log.endLoadQueue(queueName, LoadQueueResult.PASSED);
                        log.info("Finished executing action queue '" + queueName + "'");
                    } catch (NoSuchLoadQueueException e) {
                        log.error("Error waiting for action queue '" + queueName + "' completion", e);
                        log.endLoadQueue(queueName, LoadQueueResult.FAILED);
                    }
                }
            });
            helpThread.setDescription(queueName);
            helpThread.start();
            log.info("Action queue '" + queueName + "' scheduled for asynchronous execution");
        }
    } catch (Exception e) {
        throw new AgentException(e.getMessage(), e);
    }
}