blob: 8414a9b0ca67ccae9364f1e1da250d03160eae0b (
plain) (
blame)
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
|
#!/bin/sh -e
# $Id: trace.sh 2313967 2016-05-24 12:52:38Z serxa $
# $HeadURL: svn+ssh://arcadia.yandex.ru/arc/trunk/arcadia/library/cpp/lwtrace/mon/trace.sh $
usage() {
if [ -n "$*" ]; then
echo "ERROR: $*" >&2
else
echo "Script for LWTrace tracing management" >&2
fi
echo "USAGE: `basename $0` COMMAND TRACEID HOST:PORT PARAM1=VALUE1 ..." >&2
echo "COMMANDS:" >&2
echo " new Start new tracing described by query from STDIN," >&2
echo " uniquely named with TRACEID string," >&2
echo " using HOST:PORT monitoring service." >&2
echo " query can contain \$params that are substituted with corresponding values from cmdline" >&2
echo " (alse \$\$ is replaced by \$, each param must be defined from cmdline)" >&2
echo " delete Stop existing tracing with specified name TRACEID," >&2
echo " on HOST:PORT monitoring service." >&2
exit 1
}
COMMAND=$1
ID="$2"
IDENC="$(perl -MURI::Escape -e '$id = $ARGV[0]; $id=uri_escape($id); $id =~ s{[+%]}{-}g; print $id;' "$ID")"
ADDRESS="$3"
if [ "$COMMAND" = "--help" ] || [ -z "$*" ]; then usage; fi
if [ -z "$ID" ]; then usage "TRACEID is not specified"; fi
if [ -z "$ADDRESS" ]; then usage "HOST:PORT is not specified"; fi
shift 3
STATUS=0
ERRFILE=/var/tmp/lwtrace.sh.$$
error() {
echo "ERROR: $*" >&2
[ ! -e $ERRFILE ] || cat $ERRFILE >&2
exit 1
}
stop() { rm -f $ERRFILE; }
trap stop INT ABRT EXIT
case "$COMMAND" in
new)
QUERY="$(perl -e 'use MIME::Base64;
local $/;
$a = <STDIN>;
for $arg (@ARGV) {
($k,$v) = split "=",$arg,2;
$a =~ s{\$$k}{$v}g;
}
if ($a =~ /\$([A-Za-z_][\w_]*)/) {
print STDERR "undefined param in lwtrace query: $1\n";
exit 0
}
$a =~ s{\$\$}{\$}g;
print encode_base64($a, "");
' "$@" 2>$ERRFILE)"
if [ -z "$QUERY" ]; then error "lwtrace query errors"; fi
wget --post-data="id=$IDENC&query=$QUERY" \
-O - http://$ADDRESS/trace?mode=new </dev/null 2>$ERRFILE || STATUS=$?
if [ $STATUS -ne 0 ]; then error "wget failure"; fi
;;
delete)
wget --post-data="id=$IDENC" \
-O - http://$ADDRESS/trace?mode=delete </dev/null 2>$ERRFILE || STATUS=$?
if [ $STATUS -ne 0 ]; then error "wget failure"; fi
;;
*)
echo "usage: `basename $0` new TRACEID ADDRESS < query.txt" >&2
echo " `basename $0` delete TRACEID ADDRESS" >&2
exit 1
;;
esac
echo "Done"
|
