File: QueryIterator.t

package info (click to toggle)
percona-toolkit 3.2.1-1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm, bullseye, forky, sid, trixie
  • size: 68,916 kB
  • sloc: perl: 241,287; sql: 22,868; sh: 19,746; javascript: 6,799; makefile: 353; awk: 38; python: 30; sed: 1
file content (87 lines) | stat: -rw-r--r-- 2,090 bytes parent folder | download | duplicates (2)
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
#!/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;

use QueryRewriter;
use FileIterator;
use QueryIterator;
use RawLogParser;
use PerconaTest;

use constant PTDEBUG => $ENV{PTDEBUG} || 0;

use Data::Dumper;
$Data::Dumper::Indent    = 1;
$Data::Dumper::Sortkeys  = 1;
$Data::Dumper::Quotekeys = 0;

my $sample = "$trunk/t/lib/samples/";

my $file_iter = FileIterator->new();
my $parser    = RawLogParser->new();
my $qr        = QueryRewriter->new();
my $query_iter;

sub test_query_iter {
   my (%args) = @_;
   my $file = $args{file};
   my $name = $args{name};

   my @events;
   while ( my $event = $query_iter->next() ) {
      push @events, $event;
   }

   is_deeply(
      \@events,
      $args{expect},
      $args{name}
   ) or diag(Dumper(\@events));
}

my $files = $file_iter->get_file_itr(
   "$sample/rawlogs/rawlog002.txt",
);

$query_iter = QueryIterator->new(
   file_iter   => $files,
   parser      => sub { return $parser->parse_event(@_) },
   fingerprint => sub { return $qr->fingerprint(@_) },
   oktorun     => sub { return 1 },
   read_only   => 1,
);

test_query_iter(
   name   => "rawlog002.txt read-only",
   expect => [
      {  pos_in_log  => 0,
         arg         => 'SELECT c FROM t WHERE id=1',
         bytes       => 26,
         cmd         => 'Query',
         Query_time  => 0,
         fingerprint => 'select c from t where id=?',
      },
      {  pos_in_log  => 27,
         arg         => '/* Hello, world! */ SELECT * FROM t2 LIMIT 1',
         bytes       => 44,
         cmd         => 'Query',
         Query_time  => 0,
         fingerprint => 'select * from t? limit ?',
      }

   ],
);

# #############################################################################
# Done.
# #############################################################################
done_testing;