utils/timestamped_logger.sh at master · ttsiodras/utils · GitHub

1
2
3
4
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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
#!/usr/bin/env bash
set -u
set -o pipefail

usage() {
    cat >&2 <<'EOF'
Usage:
  timestamped_logger.sh LOGFILE -- PROGRAM [ARG...]

Example:
  timestamped_logger.sh log.txt -- /path/to/program arg1 arg2

Creates:
  LOGFILE          merged timestamped stdout/stderr log
  LOGFILE.stdout   timestamped stdout log
  LOGFILE.stderr   timestamped stderr log

Requires:
  moreutils  for ts
  expect     for unbuffer
EOF
}

die() {
    printf 'timestamped_logger.sh: %s\n' "$*" >&2
    exit 2
}

need_cmd() {
    command -v "$1" >/dev/null 2>&1 || die "missing required command: $1"
}

if [ "$#" -lt 3 ]; then
    usage
    exit 2
fi

logfile=$1
shift

if [ "${1-}" != "--" ]; then
    usage
    die "expected -- after logfile"
fi
shift

if [ "$#" -lt 1 ]; then
    usage
    die "missing program"
fi

need_cmd ts
need_cmd unbuffer
need_cmd tee
need_cmd sort
need_cmd mktemp

stdout_log=$logfile.stdout
stderr_log=$logfile.stderr

tmpdir=$(mktemp -d "${TMPDIR:-/tmp}/timestamped_logger.XXXXXXXXXX") ||
    die "could not create temporary directory"

tmp_stdout=$tmpdir/stdout
tmp_stderr=$tmpdir/stderr

cleanup() {
    rm -rf "$tmpdir"
}
trap cleanup EXIT

: > "$stdout_log" || die "cannot write $stdout_log"
: > "$stderr_log" || die "cannot write $stderr_log"
: > "$logfile"    || die "cannot write $logfile"
: > "$tmp_stdout" || die "cannot write temporary stdout log"
: > "$tmp_stderr" || die "cannot write temporary stderr log"

unbuffer "$@" \
    > >(ts '%Y-%m-%d %H:%M:%S.%.S stdout' | tee "$stdout_log" "$tmp_stdout") \
    2> >(ts '%Y-%m-%d %H:%M:%S.%.S stderr' | tee "$stderr_log" "$tmp_stderr" >&2)

status=$?

wait 2>/dev/null || true

sort -s "$tmp_stdout" "$tmp_stderr" > "$logfile" ||
    die "could not write merged logfile: $logfile"

exit "$status"