/*
* Copyright 2000-2016 JetBrains s.r.o.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.intellij.execution.process;
import com.intellij.execution.CommandLineUtil;
import com.intellij.execution.TaskExecutor;
import com.intellij.openapi.diagnostic.Logger;
import com.intellij.openapi.util.Key;
import com.intellij.openapi.util.registry.Registry;
import com.intellij.openapi.util.text.StringUtil;
import com.intellij.util.Consumer;
import com.intellij.util.concurrency.AppExecutorUtil;
import com.intellij.util.io.BaseDataReader;
import com.intellij.util.io.BaseInputStreamReader;
import com.intellij.util.io.BaseOutputReader;
import com.intellij.util.io.BaseOutputReader.Options;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.Reader;
import java.nio.charset.Charset;
import java.util.concurrent.Future;
import java.util.concurrent.TimeUnit;
public class BaseOSProcessHandler extends ProcessHandler implements TaskExecutor {
private static final Logger LOG = Logger.getInstance(BaseOSProcessHandler.class);
private static final Options ADAPTIVE_NON_BLOCKING = new Options() {
@Override
public BaseDataReader.SleepingPolicy policy() {
return new BaseDataReader.AdaptiveSleepingPolicy();
}
};
protected final Process myProcess;
protected final String myCommandLine;
protected final Charset myCharset;
protected final String myPresentableName;
protected final ProcessWaitFor myWaitFor;
/**
* {@code commandLine} must not be not empty (for correct thread attribution in the stacktrace)
*/
public BaseOSProcessHandler(@NotNull Process process, /*@NotNull*/ String commandLine, @Nullable Charset charset) {
myProcess = process;
myCommandLine = commandLine;
myCharset = charset;
if (StringUtil.isEmpty(commandLine)) {
LOG.warn(new IllegalArgumentException("Must specify non-empty 'commandLine' parameter"));
}
myPresentableName = CommandLineUtil.extractPresentableName(StringUtil.notNullize(commandLine));
myWaitFor = new ProcessWaitFor(process, this, myPresentableName);
}
/**
* Override this method in order to execute the task with a custom pool
*
* @param task a task to run
*/
@NotNull
protected Future<?> executeOnPooledThread(@NotNull final Runnable task) {
return ProcessIOExecutorService.INSTANCE.submit(task);
}
@Override
@NotNull
public Future<?> executeTask(@NotNull Runnable task) {
return executeOnPooledThread(task);
}
@NotNull
public Process getProcess() {
return myProcess;
}
/** @deprecated use {@link #readerOptions()} (to be removed in IDEA 18) */
protected boolean useAdaptiveSleepingPolicyWhenReadingOutput() {
return false;
}
/** @deprecated use {@link #readerOptions()} (to be removed in IDEA 18) */
protected boolean useNonBlockingRead() {
return !Registry.is("output.reader.blocking.mode", false);
}
/**
* Override this method to fine-tune {@link BaseOutputReader} behavior.
*/
@NotNull
@SuppressWarnings("deprecation")
protected Options readerOptions() {
if (!useNonBlockingRead()) {
return Options.BLOCKING;
}
else if (useAdaptiveSleepingPolicyWhenReadingOutput()) {
return ADAPTIVE_NON_BLOCKING;
}
else {
return Options.NON_BLOCKING;
}
}
protected boolean processHasSeparateErrorStream() {
return true;
}
@Override
public void startNotify() {
if (myCommandLine != null) {
notifyTextAvailable(myCommandLine + '\n', ProcessOutputTypes.SYSTEM);
}
addProcessListener(new ProcessAdapter() {
@Override
public void startNotified(final ProcessEvent event) {
try {
Options options = readerOptions();
@SuppressWarnings("deprecation") final BaseDataReader stdOutReader = createOutputDataReader(options.policy());
@SuppressWarnings("deprecation") final BaseDataReader stdErrReader = processHasSeparateErrorStream() ? createErrorDataReader(options.policy()) : null;
myWaitFor.setTerminationCallback(new Consumer<Integer>() {
@Override
public void consume(Integer exitCode) {
try {
// tell readers that no more attempts to read process' output should be made
if (stdErrReader != null) stdErrReader.stop();
stdOutReader.stop();
try {
if (stdErrReader != null) stdErrReader.waitFor();
stdOutReader.waitFor();
}
catch (InterruptedException ignore) { }
}
finally {
onOSProcessTerminated(exitCode);
}
}
});
}
finally {
removeProcessListener(this);
}
}
});
super.startNotify();
}
/** @deprecated override {@link #createOutputDataReader()} (to be removed in IDEA 18) */
protected BaseDataReader createErrorDataReader(@SuppressWarnings("UnusedParameters") BaseDataReader.SleepingPolicy policy) {
return createErrorDataReader();
}
/** @deprecated override {@link #createOutputDataReader()} (to be removed in IDEA 18) */
protected BaseDataReader createOutputDataReader(@SuppressWarnings("UnusedParameters") BaseDataReader.SleepingPolicy policy) {
return createOutputDataReader();
}
@NotNull
protected BaseDataReader createErrorDataReader() {
return new SimpleOutputReader(createProcessErrReader(), ProcessOutputTypes.STDERR, readerOptions(), "error stream of " + myPresentableName);
}
@NotNull
protected BaseDataReader createOutputDataReader() {
return new SimpleOutputReader(createProcessOutReader(), ProcessOutputTypes.STDOUT, readerOptions(), "output stream of " + myPresentableName);
}
protected void onOSProcessTerminated(final int exitCode) {
notifyProcessTerminated(exitCode);
}
@NotNull
protected Reader createProcessOutReader() {
return createInputStreamReader(myProcess.getInputStream());
}
@NotNull
protected Reader createProcessErrReader() {
return createInputStreamReader(myProcess.getErrorStream());
}
@NotNull
private Reader createInputStreamReader(@NotNull InputStream streamToRead) {
Charset charset = getCharset();
if (charset == null) charset = Charset.defaultCharset();
return new BaseInputStreamReader(streamToRead, charset);
}
@Override
protected void destroyProcessImpl() {
try {
closeStreams();
}
finally {
doDestroyProcess();
}
}
protected void doDestroyProcess() {
getProcess().destroy();
}
@Override
protected void detachProcessImpl() {
final Runnable runnable = new Runnable() {
@Override
public void run() {
closeStreams();
myWaitFor.detach();
notifyProcessDetached();
}
};
executeOnPooledThread(runnable);
}
protected void closeStreams() {
try {
myProcess.getOutputStream().close();
}
catch (IOException e) {
LOG.warn(e);
}
}
@Override
public boolean detachIsDefault() {
return false;
}
@Override
public OutputStream getProcessInput() {
return myProcess.getOutputStream();
}
/*@NotNull*/
public String getCommandLine() {
return myCommandLine;
}
@Nullable
public Charset getCharset() {
return myCharset;
}
@SuppressWarnings("unused")
public static class ExecutorServiceHolder {
/** @deprecated use {@link BaseOSProcessHandler#executeTask(Runnable)} instead (to be removed in IDEA 17) */
public static Future<?> submit(@NotNull Runnable task) {
LOG.warn("Deprecated method. Please use com.intellij.execution.process.BaseOSProcessHandler.executeTask() instead", new Throwable());
return AppExecutorUtil.getAppExecutorService().submit(task);
}
}
private class SimpleOutputReader extends BaseOutputReader {
private final Key myProcessOutputType;
private SimpleOutputReader(Reader reader, Key outputType, Options options, @NotNull String presentableName) {
super(reader, options);
myProcessOutputType = outputType;
start(presentableName);
}
@NotNull
@Override
protected Future<?> executeOnPooledThread(@NotNull Runnable runnable) {
return BaseOSProcessHandler.this.executeOnPooledThread(runnable);
}
@Override
protected void onTextAvailable(@NotNull String text) {
notifyTextAvailable(text, myProcessOutputType);
}
}
@Override
public String toString() {
return myCommandLine;
}
@Override
public boolean waitFor() {
boolean result = super.waitFor();
try {
myWaitFor.waitFor();
}
catch (InterruptedException e) {
throw new RuntimeException(e);
}
return result;
}
@Override
public boolean waitFor(long timeoutInMilliseconds) {
boolean result = super.waitFor(timeoutInMilliseconds);
try {
result &= myWaitFor.waitFor(timeoutInMilliseconds, TimeUnit.MILLISECONDS);
}
catch (InterruptedException e) {
throw new RuntimeException(e);
}
return result;
}
}