Examples with SystemCommand org.apache.jorphan.exec.SystemCommand used on opensource projects

Example 1 with SystemCommand

use of org.apache.jorphan.exec.SystemCommand in project jmeter by apache.

the class SystemSampler method sample.

/**
     * Performs a test sample.
     * 
     * @param entry
     *            the Entry for this sample
     * @return test SampleResult
     */
@Override
public SampleResult sample(Entry entry) {
    SampleResult results = new SampleResult();
    results.setDataType(SampleResult.TEXT);
    results.setSampleLabel(getName());
    String command = getCommand();
    Arguments args = getArguments();
    Arguments environment = getEnvironmentVariables();
    boolean checkReturnCode = getCheckReturnCode();
    int expectedReturnCode = getExpectedReturnCode();
    List<String> cmds = new ArrayList<>(args.getArgumentCount() + 1);
    StringBuilder cmdLine = new StringBuilder((null == command) ? "" : command);
    cmds.add(command);
    for (int i = 0; i < args.getArgumentCount(); i++) {
        Argument arg = args.getArgument(i);
        cmds.add(arg.getPropertyAsString(Argument.VALUE));
        cmdLine.append(" ");
        cmdLine.append(cmds.get(i + 1));
    }
    Map<String, String> env = new HashMap<>();
    for (int i = 0; i < environment.getArgumentCount(); i++) {
        Argument arg = environment.getArgument(i);
        env.put(arg.getName(), arg.getPropertyAsString(Argument.VALUE));
    }
    File directory;
    if (StringUtils.isEmpty(getDirectory())) {
        directory = new File(FileServer.getDefaultBase());
        if (log.isDebugEnabled()) {
            log.debug("Using default directory:" + directory.getAbsolutePath());
        }
    } else {
        directory = new File(getDirectory());
        if (log.isDebugEnabled()) {
            log.debug("Using configured directory:" + directory.getAbsolutePath());
        }
    }
    if (log.isDebugEnabled()) {
        log.debug("Will run : " + cmdLine + " using working directory:" + directory.getAbsolutePath() + " with environment: " + env);
    }
    results.setSamplerData("Working Directory: " + directory.getAbsolutePath() + "\nEnvironment: " + env + "\nExecuting: " + cmdLine.toString());
    SystemCommand nativeCommand = null;
    try {
        nativeCommand = new SystemCommand(directory, getTimeout(), POLL_INTERVAL, env, getStdin(), getStdout(), getStderr());
        results.sampleStart();
        int returnCode = nativeCommand.run(cmds);
        results.sampleEnd();
        results.setResponseCode(Integer.toString(returnCode));
        if (log.isDebugEnabled()) {
            log.debug("Ran : " + cmdLine + " using working directory: " + directory.getAbsolutePath() + " with execution environment: " + nativeCommand.getExecutionEnvironment() + " => " + returnCode);
        }
        if (checkReturnCode && (returnCode != expectedReturnCode)) {
            results.setSuccessful(false);
            results.setResponseMessage("Unexpected return code.  Expected [" + expectedReturnCode + "]. Actual [" + returnCode + "].");
        } else {
            results.setSuccessful(true);
            results.setResponseMessage("OK");
        }
    } catch (IOException ioe) {
        results.sampleEnd();
        results.setSuccessful(false);
        //$NON-NLS-1$
        results.setResponseCode("500");
        results.setResponseMessage("Exception occurred whilst executing system call: " + ioe);
    } catch (InterruptedException ie) {
        results.sampleEnd();
        results.setSuccessful(false);
        //$NON-NLS-1$
        results.setResponseCode("500");
        results.setResponseMessage("System Sampler interrupted whilst executing system call: " + ie);
        Thread.currentThread().interrupt();
    }
    if (nativeCommand != null) {
        // default charset is deliberate here
        results.setResponseData(nativeCommand.getOutResult().getBytes());
    }
    return results;
}