com.yahoo.system.ProcessExecuter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of vespajlib Show documentation

Library for use in Java components of Vespa. Shared code which do not fit anywhere else.

There is a newer version: 8.441.21

// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.system;

import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;
import java.util.StringTokenizer;

import com.yahoo.collections.Pair;

/**
 * Executes a system command synchronously.
 *
 * @author bratseth
 */
public class ProcessExecuter {

    private final boolean override_log_control;
    public ProcessExecuter(boolean override_log_control) {
        this.override_log_control = override_log_control;
    }
    public ProcessExecuter() {
        this(false);
    }

    /**
     * Executes the given command synchronously without timeout.
     * 
     * @return retcode and stdout/stderr merged
     */
    public Pair exec(String command) throws IOException {
        StringTokenizer tok = new StringTokenizer(command);
        List tokens = new ArrayList<>();
        while (tok.hasMoreElements())
            tokens.add(tok.nextToken());
        return exec(tokens.toArray(new String[0]));
    }
    
    /**
     * Executes the given command synchronously without timeout.
     * 
     * @param command tokens
     * @return Retcode and stdout/stderr merged
     */
    public Pair exec(String[] command) throws IOException {
        ProcessBuilder pb = new ProcessBuilder(command);        
        StringBuilder ret = new StringBuilder();
        pb.environment().remove("VESPA_LOG_TARGET");
        if (override_log_control) {
            pb.environment().remove("VESPA_LOG_CONTROL_FILE");
            pb.environment().put("VESPA_SERVICE_NAME", "exec-" + command[0]);
        }
        pb.redirectErrorStream(true);
        Process p = pb.start();
        InputStream is = p.getInputStream();
        while (true) {
            int b = is.read();
            if (b == -1) break;
            ret.append((char)b);
        }
        int rc = 0;
        try {
            rc = p.waitFor();
        } catch (InterruptedException e) {
            throw new RuntimeException(e);
        }
        return new Pair<>(rc, ret.toString());
    }

}