5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
|
# File 'lib/hq/tools/cron.rb', line 5
def self.wrap log = nil
is_tty = $stderr.tty?
if log
log = String.new log
log.gsub! /@ts(@|\b)/, Time.now.utc.strftime("%Y-%m-%dT%H:%M:%SZ")
log.gsub! /@pid(@|\b)/, $$.to_s
else
require "tempfile"
temp = Tempfile.new "hq-cron-"
log = temp.path
end
begin
caught_exception = nil
unless is_tty
stdout_saved = $stdout.dup unless is_tty
stderr_saved = $stderr.dup unless is_tty
$stdout.reopen log, "a" unless is_tty
$stderr.reopen log, "a" unless is_tty
end
begin
yield log
rescue => e
caught_exception = e
end
$stdout.flush unless is_tty
$stderr.flush unless is_tty
$stdout.reopen stdout_saved unless is_tty
$stderr.reopen stderr_saved unless is_tty
stdout_saved.close unless is_tty
stderr_saved.close unless is_tty
if caught_exception
system "cat #{log} >&2" unless is_tty
$stderr.puts "Caught #{caught_exception.class}: #{caught_exception.message}"
$stderr.puts caught_exception.backtrace.join "\n"
end
ensure
temp.unlink if temp
end
end
|