debian-mirror-gitlab/lib/gitlab/popen.rb

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

65 lines
2 KiB
Ruby
Raw Normal View History

2018-12-13 13:39:08 +05:30
# frozen_string_literal: true
2014-09-02 18:07:02 +05:30
require 'fileutils'
require 'open3'
module Gitlab
module Popen
extend self
2018-03-17 18:26:18 +05:30
Result = Struct.new(:cmd, :stdout, :stderr, :status, :duration)
# Returns [stdout + stderr, status]
2022-04-04 11:22:00 +05:30
# status is either the exit code or the signal that killed the process
2018-03-17 18:26:18 +05:30
def popen(cmd, path = nil, vars = {}, &block)
result = popen_with_detail(cmd, path, vars, &block)
2022-04-04 11:22:00 +05:30
# Process#waitpid returns Process::Status, which holds a 16-bit value.
# The higher-order 8 bits hold the exit() code (`exitstatus`).
# The lower-order bits holds whether the process was terminated.
# If the process didn't exit normally, `exitstatus` will be `nil`,
# but we still want a non-zero code, even if the value is
# platform-dependent.
status = result.status&.exitstatus || result.status.to_i
["#{result.stdout}#{result.stderr}", status]
2018-03-17 18:26:18 +05:30
end
# Returns Result
def popen_with_detail(cmd, path = nil, vars = {})
2014-09-02 18:07:02 +05:30
unless cmd.is_a?(Array)
raise "System commands must be given as an array of strings"
end
path ||= Dir.pwd
2017-08-17 22:00:37 +05:30
vars['PWD'] = path
2014-09-02 18:07:02 +05:30
options = { chdir: path }
unless File.directory?(path)
FileUtils.mkdir_p(path)
end
2018-03-17 18:26:18 +05:30
cmd_stdout = ''
cmd_stderr = ''
cmd_status = nil
start = Time.now
2014-09-02 18:07:02 +05:30
Open3.popen3(vars, *cmd, options) do |stdin, stdout, stderr, wait_thr|
2018-11-18 11:00:15 +05:30
# stderr and stdout pipes can block if stderr/stdout aren't drained: https://bugs.ruby-lang.org/issues/9082
# Mimic what Ruby does with capture3: https://github.com/ruby/ruby/blob/1ec544695fa02d714180ef9c34e755027b6a2103/lib/open3.rb#L257-L273
out_reader = Thread.new { stdout.read }
err_reader = Thread.new { stderr.read }
2016-09-29 09:46:39 +05:30
yield(stdin) if block_given?
2015-04-26 12:48:37 +05:30
stdin.close
2016-09-29 09:46:39 +05:30
2018-11-18 11:00:15 +05:30
cmd_stdout = out_reader.value
cmd_stderr = err_reader.value
2018-03-17 18:26:18 +05:30
cmd_status = wait_thr.value
2014-09-02 18:07:02 +05:30
end
2018-03-17 18:26:18 +05:30
Result.new(cmd, cmd_stdout, cmd_stderr, cmd_status, Time.now - start)
2014-09-02 18:07:02 +05:30
end
end
end