~drizzle-trunk/drizzle/development

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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
#!/usr/bin/env perl

use strict;
use warnings;
use English qw(-no_match_vars);

my ($file, $attrib, $cmp, $val, $val2) = @ARGV;
if ( !$file || !$attrib ) {
   warn "Usage: check-query-log-values.pl FILE ATTRIBUTE [CMP VALUE [VALUE]]\n";
   exit 1;
}

# Slurp the file.  There should be only 1 event.
open my $fh, "<", $file or die "Cannot open $file: $OS_ERROR";
my $event_text = '';
{
   local $INPUT_RECORD_SEPARATOR = "\n#\n";
   while ( defined(my $event = <$fh>) ) {
      $event_text = $event;
   }
}
close $fh;
my @event_lines = split /\n/, $event_text;

my $event = parse_event(@event_lines);

my @attribs = split /,/, $attrib;
my @vals    = split /,/, $val if $val;
my @vals2   = split /,/, $val2 if $val2;

if ( $val ) {
   die "The number of ATTRIBUTEs and VALUEs does not match"
      if $#attribs != $#vals;
}

print "Checking attributes and values of query $event->{arg}\n";

for my $i (0..$#attribs) {
   my $attrib = lc $attribs[$i];
   my $val    = $vals[$i];
   my $val2   = $vals2[$i];

   if ( $attrib eq 'all' ) {
      dump_event($event);
   }
   else {
      die "A CMP argument is required if ATTRIBUTE is not ALL"
         unless $cmp;
      die "A VALUE argument is required if ATTRIBUTE is not ALL"
         unless defined $val;

      print "ERROR: attribute $attrib does not exist.\n"
         unless exists $event->{$attrib};

      $cmp = lc $cmp;

      my $ok = 0;
      my $event_val = $event->{$attrib};
      if ( $cmp eq '=' || $cmp eq 'equals' ) {
         $ok = 1 if defined $event_val && $event_val eq $val;
      }
      elsif ( $cmp eq 'matches' ) {
         $ok = 1 if defined $event_val && $event_val =~ m/$val/;
      }
      elsif ( $cmp eq 'between' ) {
         die "I need a second VALUE argument if CMP is BETWEEN"
            unless $val2;
         $ok = 1 if defined $event_val
            && $val <= $event_val && $event_val <= $val2;
      }
      else {
         die "Unknown CMP: $cmp";
      }

      if ( $ok ) {
         # Don't print the matches pattern becaues it's probably some
         # variable value like a timestamp.
         print "$attrib value $cmp "
            . ($cmp eq 'matches' ? "" : "$val ")
            . ($cmp eq 'between' ? "and $val2 " : "")
            . "OK\n";
      }
      else {
         print "$attrib value $event_val does not '$cmp' $val"
            . ($cmp eq 'between' ? " and $val2" : "") . "\n"
            . "Event dump:\n";
         dump_event($event);
      }
   }
}

sub parse_event {
   my (@event_lines) = @_;
   die "I need a event_lines argument" unless @event_lines;

   my $rs = pop @event_lines;
   if ( $rs ne "#" ) {
      print "ERROR: Event does not end with the # record separator.\n";
   }

   my @props;
   my $arg = '';
   my $lineno = 1;
   foreach my $line ( @event_lines ) { 
      next if $line =~ m/^$/;
      if ( $lineno <= 4 ) { # ts, ints, floats and bools
         push @props, $line =~ m/([a-z_]+)=(\S+)/g;
      }
      elsif ( $lineno == 5 ) { # strings
         push @props, $line =~ m/([a-z_]+)="([^"]*)"/;
      }
      else { # query
         $arg .= $line;
      }
      $lineno++;
   }

   chomp $arg;
   push @props, 'arg', $arg;

   my $event = { @props };

   return $event;
}

sub dump_event {
   my ($event) = @_;
   foreach my $attrib ( sort keys %$event ) {
      print "$attrib=$event->{$attrib}\n";
   }
}