快捷導(dǎo)航

如何使用C# 捕獲進(jìn)程輸出

更新時間：2020年08月31日 05:17:22 作者：WeihanLi

這篇文章主要介紹了如何使用C# 捕獲進(jìn)程輸出，幫助大家更好的理解和使用c#,感興趣的朋友可以了解下

Intro

很多時候我們可能會需要執(zhí)行一段命令獲取一個輸出，遇到的比較典型的就是之前我們需要用 FFMpeg 實現(xiàn)視頻的編碼壓縮水印等一系列操作，當(dāng)時使用的是 FFMpegCore 這個類庫，這個類庫的實現(xiàn)原理是啟動另外一個進(jìn)程，啟動 ffmpeg 并傳遞相應(yīng)的處理參數(shù)，并根據(jù)進(jìn)程輸出獲取處理進(jìn)度

為了方便使用，實現(xiàn)了兩個幫助類來方便的獲取進(jìn)程的輸出，分別是 ProcessExecutor 和 CommandRunner，前者更為靈活，可以通過事件添加自己的額外事件訂閱處理，后者為簡化版，主要是只獲取輸出的場景，兩者的實現(xiàn)原理大體是一樣的，啟動一個 Process，并監(jiān)聽其輸出事件獲取輸出

ProcessExecutor

使用示例，這個示例是獲取保存 nuget 包的路徑的一個示例：

using var executor = new ProcessExecutor("dotnet", "nuget locals global-packages -l");
var folder = string.Empty;
executor.OnOutputDataReceived += (sender, str) =>
{
  if(str is null)
    return;

  Console.WriteLine(str);

  if(str.StartsWith("global-packages:"))
  {
    folder = str.Substring("global-packages:".Length).Trim();          
  }
};
executor.Execute();

Console.WriteLine(folder);

ProcessExecutor 實現(xiàn)代碼如下：

public class ProcessExecutor : IDisposable
{
  public event EventHandler<int> OnExited;

  public event EventHandler<string> OnOutputDataReceived;

  public event EventHandler<string> OnErrorDataReceived;

  protected readonly Process _process;

  protected bool _started;

  public ProcessExecutor(string exePath) : this(new ProcessStartInfo(exePath))
  {
  }

  public ProcessExecutor(string exePath, string arguments) : this(new ProcessStartInfo(exePath, arguments))
  {
  }

  public ProcessExecutor(ProcessStartInfo startInfo)
  {
    _process = new Process()
    {
      StartInfo = startInfo,
      EnableRaisingEvents = true,
    };
    _process.StartInfo.UseShellExecute = false;
    _process.StartInfo.CreateNoWindow = true;
    _process.StartInfo.RedirectStandardOutput = true;
    _process.StartInfo.RedirectStandardInput = true;
    _process.StartInfo.RedirectStandardError = true;
  }

  protected virtual void InitializeEvents()
  {
    _process.OutputDataReceived += (sender, args) =>
    {
      if (args.Data != null)
      {
        OnOutputDataReceived?.Invoke(sender, args.Data);
      }
    };
    _process.ErrorDataReceived += (sender, args) =>
    {
      if (args.Data != null)
      {
        OnErrorDataReceived?.Invoke(sender, args.Data);
      }
    };
    _process.Exited += (sender, args) =>
    {
      if (sender is Process process)
      {
        OnExited?.Invoke(sender, process.ExitCode);
      }
      else
      {
        OnExited?.Invoke(sender, _process.ExitCode);
      }
    };
  }

  protected virtual void Start()
  {
    if (_started)
    {
      return;
    }
    _started = true;

    _process.Start();
    _process.BeginOutputReadLine();
    _process.BeginErrorReadLine();
    _process.WaitForExit();
  }

  public async virtual Task SendInput(string input)
  {
    try
    {
      await _process.StandardInput.WriteAsync(input!);
    }
    catch (Exception e)
    {
      OnErrorDataReceived?.Invoke(_process, e.ToString());
    }
  }

  public virtual int Execute()
  {
    InitializeEvents();
    Start();
    return _process.ExitCode;
  }

  public virtual async Task<int> ExecuteAsync()
  {
    InitializeEvents();
    return await Task.Run(() =>
    {
      Start();
      return _process.ExitCode;
    }).ConfigureAwait(false);
  }

  public virtual void Dispose()
  {
    _process.Dispose();
    OnExited = null;
    OnOutputDataReceived = null;
    OnErrorDataReceived = null;
  }
}

CommandExecutor

上面的這種方式比較靈活但有些繁瑣，于是有了下面這個版本

使用示例：

[Fact]
public void HostNameTest()
{
  if (!RuntimeInformation.IsOSPlatform(OSPlatform.Windows))
  {
    return;
  }

  var result = CommandRunner.ExecuteAndCapture("hostname");

  var hostName = Dns.GetHostName();
  Assert.Equal(hostName, result.StandardOut.TrimEnd());
  Assert.Equal(0, result.ExitCode);
}

實現(xiàn)源碼：

public static class CommandRunner
{
  public static int Execute(string commandPath, string arguments = null, string workingDirectory = null)
  {
    using var process = new Process()
    {
      StartInfo = new ProcessStartInfo(commandPath, arguments ?? string.Empty)
      {
        UseShellExecute = false,
        CreateNoWindow = true,

        WorkingDirectory = workingDirectory ?? Environment.CurrentDirectory
      }
    };

    process.Start();
    process.WaitForExit();
    return process.ExitCode;
  }

  public static CommandResult ExecuteAndCapture(string commandPath, string arguments = null, string workingDirectory = null)
  {
    using var process = new Process()
    {
      StartInfo = new ProcessStartInfo(commandPath, arguments ?? string.Empty)
      {
        UseShellExecute = false,
        CreateNoWindow = true,

        RedirectStandardOutput = true,
        RedirectStandardError = true,

        WorkingDirectory = workingDirectory ?? Environment.CurrentDirectory
      }
    };
    process.Start();
    var standardOut = process.StandardOutput.ReadToEnd();
    var standardError = process.StandardError.ReadToEnd();
    process.WaitForExit();
    return new CommandResult(process.ExitCode, standardOut, standardError);
  }
}

public sealed class CommandResult
{
  public CommandResult(int exitCode, string standardOut, string standardError)
  {
    ExitCode = exitCode;
    StandardOut = standardOut;
    StandardError = standardError;
  }

  public string StandardOut { get; }
  public string StandardError { get; }
  public int ExitCode { get; }
}

More

如果只要執(zhí)行命令獲取是否執(zhí)行成功則使用 CommandRunner.Execute 即可，只獲取輸出和是否成功可以用 CommandRunner.ExecuteAndCapture 方法，如果想要進(jìn)一步的添加事件訂閱則使用 ProcessExecutor

Reference

https://github.com/rosenbjerg/FFMpegCore
https://github.com/WeihanLi/WeihanLi.Common/blob/dev/src/WeihanLi.Common/Helpers/ProcessExecutor.cs
https://github.com/WeihanLi/WeihanLi.Common/blob/dev/test/WeihanLi.Common.Test/HelpersTest/ProcessExecutorTest.cs
https://github.com/WeihanLi/WeihanLi.Common/blob/dev/test/WeihanLi.Common.Test/HelpersTest/CommandRunnerTest.cs

以上就是如何使用C# 捕獲進(jìn)程輸出的詳細(xì)內(nèi)容，更多關(guān)于C# 捕獲進(jìn)程輸出的資料請關(guān)注腳本之家其它相關(guān)文章！

您可能感興趣的文章: