cs-codex-dist-tests/KubernetesWorkflow/CommandRunner.cs

53 lines
1.6 KiB
C#
Raw Normal View History

2023-04-14 07:54:07 +00:00
using k8s;
using Utils;
namespace KubernetesWorkflow
{
public class CommandRunner
{
2023-05-04 06:25:48 +00:00
private readonly K8sClient client;
2023-04-14 07:54:07 +00:00
private readonly string k8sNamespace;
private readonly RunningPod pod;
private readonly string containerName;
private readonly string command;
private readonly string[] arguments;
private readonly List<string> lines = new List<string>();
2023-05-04 06:25:48 +00:00
public CommandRunner(K8sClient client, string k8sNamespace, RunningPod pod, string containerName, string command, string[] arguments)
2023-04-14 07:54:07 +00:00
{
this.client = client;
this.k8sNamespace = k8sNamespace;
this.pod = pod;
this.containerName = containerName;
this.command = command;
this.arguments = arguments;
}
public void Run()
{
var input = new[] { command }.Concat(arguments).ToArray();
2023-05-04 06:25:48 +00:00
Time.Wait(client.Run(c => c.NamespacedPodExecAsync(
pod.Name, k8sNamespace, containerName, input, false, Callback, new CancellationToken())));
2023-04-14 07:54:07 +00:00
}
public string GetStdOut()
{
return string.Join(Environment.NewLine, lines);
}
private Task Callback(Stream stdIn, Stream stdOut, Stream stdErr)
{
using var streamReader = new StreamReader(stdOut);
var line = streamReader.ReadLine();
while (line != null)
{
lines.Add(line);
line = streamReader.ReadLine();
}
return Task.CompletedTask;
}
}
}