internal static IEnumerable <IContainerInstance> GetRemoteDockerContainers(IConnection connection, string hostname) { SSHConnection sshConnection = connection as SSHConnection; List <string> outputLines = new List <string>(); StringBuilder errorSB = new StringBuilder(); if (sshConnection == null) { return(null); } List <DockerContainerInstance> containers = new List <DockerContainerInstance>(); DockerCommandSettings settings = new DockerCommandSettings(hostname, true); settings.SetCommand(dockerPSCommand, dockerPSArgs); RemoteCommandRunner commandRunner = new RemoteCommandRunner(settings, sshConnection); ManualResetEvent resetEvent = new ManualResetEvent(false); int exitCode = 0; commandRunner.ErrorOccured += ((sender, args) => { errorSB.Append(args); }); commandRunner.Closed += ((sender, args) => { exitCode = args; resetEvent.Set(); }); commandRunner.OutputReceived += ((sender, args) => { if (!string.IsNullOrWhiteSpace(args)) { // If it isn't json, assume its an error message if (args.Trim()[0] != '{') { errorSB.Append(args); } // Unix line endings are '\n' so split on that for json items. foreach (var item in args.Split('\n').ToList()) { if (!string.IsNullOrWhiteSpace(item)) { outputLines.Add(item); } } } }); commandRunner.Start(); bool cancellationRequested = false; VS.VSOperationWaiter.Wait(UIResources.QueryingForContainersMessage, false, (cancellationToken) => { while (!resetEvent.WaitOne(2000) && !cancellationToken.IsCancellationRequested) { } cancellationRequested = cancellationToken.IsCancellationRequested; }); if (!cancellationRequested) { if (exitCode != 0) { // if the exit code is not zero, then the output we received possibly is the error message string exceptionMessage = UIResources.CommandExecutionErrorWithExitCodeFormat.FormatCurrentCultureWithArgs( "{0} {1}".FormatInvariantWithArgs(settings.Command, settings.CommandArgs), exitCode, errorSB.ToString()); throw new CommandFailedException(exceptionMessage); } foreach (var item in outputLines) { containers.Add(DockerContainerInstance.Create(item)); } } return(containers); }
internal static IEnumerable <DockerContainerInstance> GetRemoteDockerContainers(IConnection connection, string hostname, out int totalContainers) { totalContainers = 0; SSHConnection sshConnection = connection as SSHConnection; List <string> outputLines = new List <string>(); StringBuilder errorSB = new StringBuilder(); if (sshConnection == null) { return(null); } List <DockerContainerInstance> containers = new List <DockerContainerInstance>(); DockerCommandSettings settings = new DockerCommandSettings(hostname, true); settings.SetCommand(dockerPSCommand, dockerPSArgs); RemoteCommandRunner commandRunner = new RemoteCommandRunner(settings, sshConnection, handleRawOutput: false); ManualResetEvent resetEvent = new ManualResetEvent(false); int exitCode = 0; commandRunner.ErrorOccured += ((sender, args) => { errorSB.Append(args); }); commandRunner.Closed += ((sender, args) => { exitCode = args; resetEvent.Set(); }); commandRunner.OutputReceived += ((sender, line) => { if (!string.IsNullOrWhiteSpace(line)) { Debug.Assert(line.IndexOf('\n') < 0, "Why does `line` have embedded newline characters?"); // If it isn't json, assume its an error message if (line.Trim()[0] != '{') { errorSB.Append(line); } outputLines.Add(line); } }); commandRunner.Start(); bool cancellationRequested = false; VS.VSOperationWaiter.Wait(UIResources.QueryingForContainersMessage, false, (cancellationToken) => { while (!resetEvent.WaitOne(2000) && !cancellationToken.IsCancellationRequested) { } cancellationRequested = cancellationToken.IsCancellationRequested; }); if (!cancellationRequested) { if (exitCode != 0) { // if the exit code is not zero, then the output we received possibly is the error message string exceptionMessage = UIResources.CommandExecutionErrorWithExitCodeFormat.FormatCurrentCultureWithArgs( "{0} {1}".FormatInvariantWithArgs(settings.Command, settings.CommandArgs), exitCode, errorSB.ToString()); throw new CommandFailedException(exceptionMessage); } foreach (var item in outputLines) { if (DockerContainerInstance.TryCreate(item, out DockerContainerInstance containerInstance)) { containers.Add(containerInstance); } totalContainers++; } } return(containers); }
internal static IEnumerable <IContainerInstance> GetRemoteDockerContainers(IConnection connection) { SSHConnection sshConnection = connection as SSHConnection; List <string> outputLines = new List <string>(); StringBuilder errorSB = new StringBuilder(); if (sshConnection == null) { return(null); } List <DockerContainerInstance> containers = new List <DockerContainerInstance>(); RemoteCommandRunner commandRunner = new RemoteCommandRunner(dockerCommand, dockerPSArgs, sshConnection); ManualResetEvent resetEvent = new ManualResetEvent(false); int exitCode = 0; commandRunner.ErrorOccured += ((sender, args) => { errorSB.Append(args); }); commandRunner.Closed += ((sender, args) => { exitCode = args; resetEvent.Set(); }); commandRunner.OutputReceived += ((sender, args) => { if (!string.IsNullOrWhiteSpace(args)) { // If it isn't json, assume its an error message if (args.Trim()[0] != '{') { errorSB.Append(args); } // Unix line endings are '\n' so split on that for json items. foreach (var item in args.Split('\n').ToList()) { if (!string.IsNullOrWhiteSpace(item)) { outputLines.Add(item); } } } }); resetEvent.WaitOne(); if (exitCode != 0) { // if the exit code is not zero, then the output we received possibly is the error message string exceptionMessage = string.Format(CultureInfo.CurrentCulture, UIResources.CommandExecutionErrorWithExitCodeFormat, dockerCommand, exitCode, errorSB.ToString()); throw new CommandFailedException(exceptionMessage); } foreach (var item in outputLines) { containers.Add(DockerContainerInstance.Create(item)); } return(containers); }