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
90
91
92
93
94
95
96
97
98
99
100
|
#!/usr/bin/perl
BEGIN {
die "The PERCONA_TOOLKIT_BRANCH environment variable is not set.\n"
unless $ENV{PERCONA_TOOLKIT_BRANCH} && -d $ENV{PERCONA_TOOLKIT_BRANCH};
unshift @INC, "$ENV{PERCONA_TOOLKIT_BRANCH}/lib";
};
use strict;
use warnings FATAL => 'all';
use English qw(-no_match_vars);
use Test::More tests => 4;
use SlowLogParser;
use SlowLogWriter;
use PerconaTest;
my $p = new SlowLogParser;
my $w = new SlowLogWriter;
sub __no_diff {
my ( $filename, $expected ) = @_;
# Parse and rewrite the original file.
my $tmp_file = '/tmp/SlowLogWriter-test.txt';
open my $rewritten_fh, '>', $tmp_file
or die "Cannot write to $tmp_file: $OS_ERROR";
open my $fh, "<", "$trunk/$filename"
or die "Cannot open $trunk/$filename: $OS_ERROR";
my %args = (
next_event => sub { return <$fh>; },
tell => sub { return tell $fh; },
);
while ( my $e = $p->parse_event(%args) ) {
$w->write($rewritten_fh, $e);
}
close $fh;
close $rewritten_fh;
# Compare the contents of the two files.
my $retval = system("diff $tmp_file $trunk/$expected");
`rm -rf $tmp_file`;
$retval = $retval >> 8;
return !$retval;
}
sub write_event {
my ( $event, $expected_output ) = @_;
my $tmp_file = '/tmp/SlowLogWriter-output.txt';
open my $fh, '>', $tmp_file or die "Cannot open $tmp_file: $OS_ERROR";
$w->write($fh, $event);
close $fh;
my $retval = system("diff $tmp_file $trunk/$expected_output");
`rm -rf $tmp_file`;
$retval = $retval >> 8;
return !$retval;
}
# Check that I can write a slow log in the default slow log format.
ok(
__no_diff('t/lib/samples/slowlogs/slow001.txt', 't/lib/samples/slowlogs/slow001-rewritten.txt'),
'slow001.txt rewritten'
);
# Test writing a Percona-patched slow log with Thread_id and hi-res Query_time.
ok(
__no_diff('t/lib/samples/slowlogs/slow032.txt', 't/lib/samples/slowlogs/slow032-rewritten.txt'),
'slow032.txt rewritten'
);
ok(
write_event(
{
Query_time => '1',
arg => 'select * from foo',
ip => '127.0.0.1',
port => '12345',
},
't/lib/samples/slowlogs/slowlogwriter001.txt',
),
'Writes Client attrib from tcpdump',
);
ok(
write_event(
{
Query_time => '1.123456',
Lock_time => '0.000001',
arg => 'select * from foo',
},
't/lib/samples/slowlogs/slowlogwriter002.txt',
),
'Writes microsecond times'
);
# #############################################################################
# Done.
# #############################################################################
diag(`rm -rf SlowLogWriter-test.txt >/dev/null 2>&1`);
exit;
|
