Merge lp:~percona-toolkit-dev/percona-toolkit/fix-table-status-bug-960513 into lp:percona-toolkit/2.0
- fix-table-status-bug-960513
- Merge into 2.0
Proposed by
Daniel Nichter
Status: | Rejected | ||||
---|---|---|---|---|---|
Rejected by: | Daniel Nichter | ||||
Proposed branch: | lp:~percona-toolkit-dev/percona-toolkit/fix-table-status-bug-960513 | ||||
Merge into: | lp:percona-toolkit/2.0 | ||||
Diff against target: |
15458 lines (+13654/-776) 41 files modified
bin/pt-duplicate-key-checker (+171/-92) bin/pt-fingerprint (+2143/-0) bin/pt-index-usage (+170/-92) bin/pt-table-checksum (+42/-67) bin/pt-table-sync (+553/-404) bin/pt-table-usage (+7320/-0) lib/NibbleIterator.pm (+8/-15) lib/QueryRewriter.pm (+24/-4) lib/SQLParser.pm (+98/-8) lib/SchemaIterator.pm (+27/-45) lib/TableUsage.pm (+1060/-0) t/lib/QueryRewriter.t (+59/-1) t/lib/SchemaIterator.t (+2/-47) t/lib/TableUsage.t (+817/-0) t/lib/samples/SchemaIterator/all-dbs-tbls.txt (+486/-0) t/lib/samples/SchemaIterator/resume-from-ignored-sakila-payment.txt (+49/-0) t/lib/samples/SchemaIterator/resume-from-sakila-payment.txt (+66/-0) t/pt-fingerprint/basics.t (+101/-0) t/pt-fingerprint/samples/query001 (+2/-0) t/pt-fingerprint/samples/query001.fingerprint (+1/-0) t/pt-fingerprint/samples/query002 (+2/-0) t/pt-fingerprint/samples/query002.fingerprint (+1/-0) t/pt-table-sync/issue_408.t (+1/-1) t/pt-table-usage/basics.t (+138/-0) t/pt-table-usage/create_table_definitions.t (+41/-0) t/pt-table-usage/explain_extended.t (+79/-0) t/pt-table-usage/samples/ee.out (+6/-0) t/pt-table-usage/samples/ee.sql (+26/-0) t/pt-table-usage/samples/in/slow001.txt (+24/-0) t/pt-table-usage/samples/in/slow002.txt (+20/-0) t/pt-table-usage/samples/in/slow003.txt (+3/-0) t/pt-table-usage/samples/out/create-table-defs-001.txt (+4/-0) t/pt-table-usage/samples/out/create001.txt (+5/-0) t/pt-table-usage/samples/out/drop-table-if-exists.txt (+3/-0) t/pt-table-usage/samples/out/query001.txt (+6/-0) t/pt-table-usage/samples/out/query002.txt (+5/-0) t/pt-table-usage/samples/out/slow001.txt (+31/-0) t/pt-table-usage/samples/out/slow002.txt (+40/-0) t/pt-table-usage/samples/out/slow003-001.txt (+6/-0) t/pt-table-usage/samples/out/slow003-002.txt (+8/-0) t/pt-table-usage/samples/out/slow003-003.txt (+6/-0) |
||||
To merge this branch: | bzr merge lp:~percona-toolkit-dev/percona-toolkit/fix-table-status-bug-960513 | ||||
Related bugs: |
|
Reviewer | Review Type | Date Requested | Status |
---|---|---|---|
Daniel Nichter | Approve | ||
Review via email: mp+100296@code.launchpad.net |
Commit message
Description of the change
To post a comment you must log in.
Revision history for this message
Daniel Nichter (daniel-nichter) : | # |
review:
Approve
Unmerged revisions
- 228. By Daniel Nichter
-
Update SchemaIterator, TableParser, and NibbleIterator in tools that use them. All tools' tests still pass.
- 227. By Daniel Nichter
-
Don't use TABLE STATUS, use CREATE TABLE instead for the table's engine.
- 226. By Daniel Nichter
-
Update SchemaIterator in pt-table-sync. Make the module handle NAME_lc or not.
- 225. By Daniel Nichter
-
Add pt-table-usage and update SQLParser.pm from Maatkit.
- 224. By Daniel Nichter
-
Add pt-fingerprint.
- 223. By Daniel Nichter
-
Merge fix-no-stalk-bug-955860.
- 222. By Daniel Nichter
-
Merge fix-pt-
upgrade- docs-bug-953461. - 221. By Daniel Nichter
-
Merge fix-quoting-bug-967451.
- 220. By Daniel Nichter
-
Merge fix-risks-docs-bug-949653.
- 219. By Daniel Nichter
-
Merge lp:~percona-toolkit-dev/percona-toolkit/update-relay-log-space-limit-docs-fix-bug-949154.
Preview Diff
[H/L] Next/Prev Comment, [J/K] Next/Prev File, [N/P] Next/Prev Hunk
1 | === modified file 'bin/pt-duplicate-key-checker' | |||
2 | --- bin/pt-duplicate-key-checker 2012-03-07 23:41:54 +0000 | |||
3 | +++ bin/pt-duplicate-key-checker 2012-03-31 16:07:24 +0000 | |||
4 | @@ -199,19 +199,58 @@ | |||
5 | 199 | return bless $self, $class; | 199 | return bless $self, $class; |
6 | 200 | } | 200 | } |
7 | 201 | 201 | ||
8 | 202 | sub get_create_table { | ||
9 | 203 | my ( $self, $dbh, $db, $tbl ) = @_; | ||
10 | 204 | die "I need a dbh parameter" unless $dbh; | ||
11 | 205 | die "I need a db parameter" unless $db; | ||
12 | 206 | die "I need a tbl parameter" unless $tbl; | ||
13 | 207 | my $q = $self->{Quoter}; | ||
14 | 208 | |||
15 | 209 | my $new_sql_mode | ||
16 | 210 | = '/*!40101 SET @OLD_SQL_MODE := @@SQL_MODE, ' | ||
17 | 211 | . q{@@SQL_MODE := REPLACE(REPLACE(@@SQL_MODE, 'ANSI_QUOTES', ''), ',,', ','), } | ||
18 | 212 | . '@OLD_QUOTE := @@SQL_QUOTE_SHOW_CREATE, ' | ||
19 | 213 | . '@@SQL_QUOTE_SHOW_CREATE := 1 */'; | ||
20 | 214 | |||
21 | 215 | my $old_sql_mode = '/*!40101 SET @@SQL_MODE := @OLD_SQL_MODE, ' | ||
22 | 216 | . '@@SQL_QUOTE_SHOW_CREATE := @OLD_QUOTE */'; | ||
23 | 217 | |||
24 | 218 | PTDEBUG && _d($new_sql_mode); | ||
25 | 219 | eval { $dbh->do($new_sql_mode); }; | ||
26 | 220 | PTDEBUG && $EVAL_ERROR && _d($EVAL_ERROR); | ||
27 | 221 | |||
28 | 222 | my $use_sql = 'USE ' . $q->quote($db); | ||
29 | 223 | PTDEBUG && _d($dbh, $use_sql); | ||
30 | 224 | $dbh->do($use_sql); | ||
31 | 225 | |||
32 | 226 | my $show_sql = "SHOW CREATE TABLE " . $q->quote($db, $tbl); | ||
33 | 227 | PTDEBUG && _d($show_sql); | ||
34 | 228 | my $href; | ||
35 | 229 | eval { $href = $dbh->selectrow_hashref($show_sql); }; | ||
36 | 230 | if ( $EVAL_ERROR ) { | ||
37 | 231 | PTDEBUG && _d($EVAL_ERROR); | ||
38 | 232 | |||
39 | 233 | PTDEBUG && _d($old_sql_mode); | ||
40 | 234 | $dbh->do($old_sql_mode); | ||
41 | 235 | |||
42 | 236 | return; | ||
43 | 237 | } | ||
44 | 238 | |||
45 | 239 | PTDEBUG && _d($old_sql_mode); | ||
46 | 240 | $dbh->do($old_sql_mode); | ||
47 | 241 | |||
48 | 242 | my ($key) = grep { m/create (?:table|view)/i } keys %$href; | ||
49 | 243 | if ( !$key ) { | ||
50 | 244 | die "Error: no 'Create Table' or 'Create View' in result set from " | ||
51 | 245 | . "$show_sql: " . Dumper($href); | ||
52 | 246 | } | ||
53 | 247 | |||
54 | 248 | return $href->{$key}; | ||
55 | 249 | } | ||
56 | 250 | |||
57 | 202 | sub parse { | 251 | sub parse { |
58 | 203 | my ( $self, $ddl, $opts ) = @_; | 252 | my ( $self, $ddl, $opts ) = @_; |
59 | 204 | return unless $ddl; | 253 | return unless $ddl; |
60 | 205 | if ( ref $ddl eq 'ARRAY' ) { | ||
61 | 206 | if ( lc $ddl->[0] eq 'table' ) { | ||
62 | 207 | $ddl = $ddl->[1]; | ||
63 | 208 | } | ||
64 | 209 | else { | ||
65 | 210 | return { | ||
66 | 211 | engine => 'VIEW', | ||
67 | 212 | }; | ||
68 | 213 | } | ||
69 | 214 | } | ||
70 | 215 | 254 | ||
71 | 216 | if ( $ddl !~ m/CREATE (?:TEMPORARY )?TABLE `/ ) { | 255 | if ( $ddl !~ m/CREATE (?:TEMPORARY )?TABLE `/ ) { |
72 | 217 | die "Cannot parse table definition; is ANSI quoting " | 256 | die "Cannot parse table definition; is ANSI quoting " |
73 | @@ -518,41 +557,31 @@ | |||
74 | 518 | return $ddl; | 557 | return $ddl; |
75 | 519 | } | 558 | } |
76 | 520 | 559 | ||
112 | 521 | sub remove_secondary_indexes { | 560 | sub get_table_status { |
113 | 522 | my ( $self, $ddl ) = @_; | 561 | my ( $self, $dbh, $db, $like ) = @_; |
114 | 523 | my $sec_indexes_ddl; | 562 | my $q = $self->{Quoter}; |
115 | 524 | my $tbl_struct = $self->parse($ddl); | 563 | my $sql = "SHOW TABLE STATUS FROM " . $q->quote($db); |
116 | 525 | 564 | my @params; | |
117 | 526 | if ( ($tbl_struct->{engine} || '') =~ m/InnoDB/i ) { | 565 | if ( $like ) { |
118 | 527 | my $clustered_key = $tbl_struct->{clustered_key}; | 566 | $sql .= ' LIKE ?'; |
119 | 528 | $clustered_key ||= ''; | 567 | push @params, $like; |
120 | 529 | 568 | } | |
121 | 530 | my @sec_indexes = map { | 569 | PTDEBUG && _d($sql, @params); |
122 | 531 | my $key_def = $_->{ddl}; | 570 | my $sth = $dbh->prepare($sql); |
123 | 532 | $key_def =~ s/([\(\)])/\\$1/g; | 571 | eval { $sth->execute(@params); }; |
124 | 533 | $ddl =~ s/\s+$key_def//i; | 572 | if ($EVAL_ERROR) { |
125 | 534 | 573 | PTDEBUG && _d($EVAL_ERROR); | |
126 | 535 | my $key_ddl = "ADD $_->{ddl}"; | 574 | return; |
127 | 536 | $key_ddl .= ',' unless $key_ddl =~ m/,$/; | 575 | } |
128 | 537 | $key_ddl; | 576 | my @tables = @{$sth->fetchall_arrayref({})}; |
129 | 538 | } | 577 | @tables = map { |
130 | 539 | grep { $_->{name} ne $clustered_key } | 578 | my %tbl; # Make a copy with lowercased keys |
131 | 540 | values %{$tbl_struct->{keys}}; | 579 | @tbl{ map { lc $_ } keys %$_ } = values %$_; |
132 | 541 | PTDEBUG && _d('Secondary indexes:', Dumper(\@sec_indexes)); | 580 | $tbl{engine} ||= $tbl{type} || $tbl{comment}; |
133 | 542 | 581 | delete $tbl{type}; | |
134 | 543 | if ( @sec_indexes ) { | 582 | \%tbl; |
135 | 544 | $sec_indexes_ddl = join(' ', @sec_indexes); | 583 | } @tables; |
136 | 545 | $sec_indexes_ddl =~ s/,$//; | 584 | return @tables; |
102 | 546 | } | ||
103 | 547 | |||
104 | 548 | $ddl =~ s/,(\n\) )/$1/s; | ||
105 | 549 | } | ||
106 | 550 | else { | ||
107 | 551 | PTDEBUG && _d('Not removing secondary indexes from', | ||
108 | 552 | $tbl_struct->{engine}, 'table'); | ||
109 | 553 | } | ||
110 | 554 | |||
111 | 555 | return $ddl, $sec_indexes_ddl, $tbl_struct; | ||
137 | 556 | } | 585 | } |
138 | 557 | 586 | ||
139 | 558 | sub _d { | 587 | sub _d { |
140 | @@ -3195,7 +3224,7 @@ | |||
141 | 3195 | 3224 | ||
142 | 3196 | sub new { | 3225 | sub new { |
143 | 3197 | my ( $class, %args ) = @_; | 3226 | my ( $class, %args ) = @_; |
145 | 3198 | my @required_args = qw(OptionParser Quoter); | 3227 | my @required_args = qw(OptionParser TableParser Quoter); |
146 | 3199 | foreach my $arg ( @required_args ) { | 3228 | foreach my $arg ( @required_args ) { |
147 | 3200 | die "I need a $arg argument" unless $args{$arg}; | 3229 | die "I need a $arg argument" unless $args{$arg}; |
148 | 3201 | } | 3230 | } |
149 | @@ -3204,8 +3233,19 @@ | |||
150 | 3204 | die "I need either a dbh or file_itr argument" | 3233 | die "I need either a dbh or file_itr argument" |
151 | 3205 | if (!$dbh && !$file_itr) || ($dbh && $file_itr); | 3234 | if (!$dbh && !$file_itr) || ($dbh && $file_itr); |
152 | 3206 | 3235 | ||
153 | 3236 | my %resume; | ||
154 | 3237 | if ( my $table = $args{resume} ) { | ||
155 | 3238 | PTDEBUG && _d('Will resume from or after', $table); | ||
156 | 3239 | my ($db, $tbl) = $args{Quoter}->split_unquote($table); | ||
157 | 3240 | die "Resume table must be database-qualified: $table" | ||
158 | 3241 | unless $db && $tbl; | ||
159 | 3242 | $resume{db} = $db; | ||
160 | 3243 | $resume{tbl} = $tbl; | ||
161 | 3244 | } | ||
162 | 3245 | |||
163 | 3207 | my $self = { | 3246 | my $self = { |
164 | 3208 | %args, | 3247 | %args, |
165 | 3248 | resume => \%resume, | ||
166 | 3209 | filters => _make_filters(%args), | 3249 | filters => _make_filters(%args), |
167 | 3210 | }; | 3250 | }; |
168 | 3211 | 3251 | ||
169 | @@ -3266,9 +3306,19 @@ | |||
170 | 3266 | return \%filters; | 3306 | return \%filters; |
171 | 3267 | } | 3307 | } |
172 | 3268 | 3308 | ||
174 | 3269 | sub next_schema_object { | 3309 | sub next { |
175 | 3270 | my ( $self ) = @_; | 3310 | my ( $self ) = @_; |
176 | 3271 | 3311 | ||
177 | 3312 | if ( !$self->{initialized} ) { | ||
178 | 3313 | $self->{initialized} = 1; | ||
179 | 3314 | if ( $self->{resume}->{tbl} | ||
180 | 3315 | && !$self->table_is_allowed(@{$self->{resume}}{qw(db tbl)}) ) { | ||
181 | 3316 | PTDEBUG && _d('Will resume after', | ||
182 | 3317 | join('.', @{$self->{resume}}{qw(db tbl)})); | ||
183 | 3318 | $self->{resume}->{after} = 1; | ||
184 | 3319 | } | ||
185 | 3320 | } | ||
186 | 3321 | |||
187 | 3272 | my $schema_obj; | 3322 | my $schema_obj; |
188 | 3273 | if ( $self->{file_itr} ) { | 3323 | if ( $self->{file_itr} ) { |
189 | 3274 | $schema_obj= $self->_iterate_files(); | 3324 | $schema_obj= $self->_iterate_files(); |
190 | @@ -3278,24 +3328,18 @@ | |||
191 | 3278 | } | 3328 | } |
192 | 3279 | 3329 | ||
193 | 3280 | if ( $schema_obj ) { | 3330 | if ( $schema_obj ) { |
194 | 3281 | if ( $schema_obj->{ddl} && $self->{TableParser} ) { | ||
195 | 3282 | $schema_obj->{tbl_struct} | ||
196 | 3283 | = $self->{TableParser}->parse($schema_obj->{ddl}); | ||
197 | 3284 | } | ||
198 | 3285 | |||
199 | 3286 | delete $schema_obj->{ddl} unless $self->{keep_ddl}; | ||
200 | 3287 | |||
201 | 3288 | if ( my $schema = $self->{Schema} ) { | 3331 | if ( my $schema = $self->{Schema} ) { |
202 | 3289 | $schema->add_schema_object($schema_obj); | 3332 | $schema->add_schema_object($schema_obj); |
203 | 3290 | } | 3333 | } |
204 | 3334 | PTDEBUG && _d('Next schema object:', | ||
205 | 3335 | $schema_obj->{db}, $schema_obj->{tbl}); | ||
206 | 3291 | } | 3336 | } |
207 | 3292 | 3337 | ||
208 | 3293 | PTDEBUG && _d('Next schema object:', $schema_obj->{db}, $schema_obj->{tbl}); | ||
209 | 3294 | return $schema_obj; | 3338 | return $schema_obj; |
210 | 3295 | } | 3339 | } |
211 | 3296 | 3340 | ||
212 | 3297 | sub _iterate_files { | 3341 | sub _iterate_files { |
214 | 3298 | my ( $self ) = @_; | 3342 | my ( $self ) = @_; |
215 | 3299 | 3343 | ||
216 | 3300 | if ( !$self->{fh} ) { | 3344 | if ( !$self->{fh} ) { |
217 | 3301 | my ($fh, $file) = $self->{file_itr}->(); | 3345 | my ($fh, $file) = $self->{file_itr}->(); |
218 | @@ -3316,7 +3360,8 @@ | |||
219 | 3316 | my $db = $1; # XXX | 3360 | my $db = $1; # XXX |
220 | 3317 | $db =~ s/^`//; # strip leading ` | 3361 | $db =~ s/^`//; # strip leading ` |
221 | 3318 | $db =~ s/`$//; # and trailing ` | 3362 | $db =~ s/`$//; # and trailing ` |
223 | 3319 | if ( $self->database_is_allowed($db) ) { | 3363 | if ( $self->database_is_allowed($db) |
224 | 3364 | && $self->_resume_from_database($db) ) { | ||
225 | 3320 | $self->{db} = $db; | 3365 | $self->{db} = $db; |
226 | 3321 | } | 3366 | } |
227 | 3322 | } | 3367 | } |
228 | @@ -3329,21 +3374,22 @@ | |||
229 | 3329 | my ($tbl) = $chunk =~ m/$tbl_name/; | 3374 | my ($tbl) = $chunk =~ m/$tbl_name/; |
230 | 3330 | $tbl =~ s/^\s*`//; | 3375 | $tbl =~ s/^\s*`//; |
231 | 3331 | $tbl =~ s/`\s*$//; | 3376 | $tbl =~ s/`\s*$//; |
233 | 3332 | if ( $self->table_is_allowed($self->{db}, $tbl) ) { | 3377 | if ( $self->_resume_from_table($tbl) |
234 | 3378 | && $self->table_is_allowed($self->{db}, $tbl) ) { | ||
235 | 3333 | my ($ddl) = $chunk =~ m/^(?:$open_comment)?(CREATE TABLE.+?;)$/ms; | 3379 | my ($ddl) = $chunk =~ m/^(?:$open_comment)?(CREATE TABLE.+?;)$/ms; |
236 | 3334 | if ( !$ddl ) { | 3380 | if ( !$ddl ) { |
237 | 3335 | warn "Failed to parse CREATE TABLE from\n" . $chunk; | 3381 | warn "Failed to parse CREATE TABLE from\n" . $chunk; |
238 | 3336 | next CHUNK; | 3382 | next CHUNK; |
239 | 3337 | } | 3383 | } |
240 | 3338 | $ddl =~ s/ \*\/;\Z/;/; # remove end of version comment | 3384 | $ddl =~ s/ \*\/;\Z/;/; # remove end of version comment |
245 | 3339 | 3385 | my $tbl_struct = $self->{TableParser}->parse($ddl); | |
246 | 3340 | my ($engine) = $ddl =~ m/\).*?(?:ENGINE|TYPE)=(\w+)/; | 3386 | if ( $self->engine_is_allowed($tbl_struct->{engine}) ) { |
243 | 3341 | |||
244 | 3342 | if ( !$engine || $self->engine_is_allowed($engine) ) { | ||
247 | 3343 | return { | 3387 | return { |
251 | 3344 | db => $self->{db}, | 3388 | db => $self->{db}, |
252 | 3345 | tbl => $tbl, | 3389 | tbl => $tbl, |
253 | 3346 | ddl => $ddl, | 3390 | name => $self->{Quoter}->quote($self->{db}, $tbl), |
254 | 3391 | ddl => $ddl, | ||
255 | 3392 | tbl_struct => $tbl_struct, | ||
256 | 3347 | }; | 3393 | }; |
257 | 3348 | } | 3394 | } |
258 | 3349 | } | 3395 | } |
259 | @@ -3360,6 +3406,7 @@ | |||
260 | 3360 | sub _iterate_dbh { | 3406 | sub _iterate_dbh { |
261 | 3361 | my ( $self ) = @_; | 3407 | my ( $self ) = @_; |
262 | 3362 | my $q = $self->{Quoter}; | 3408 | my $q = $self->{Quoter}; |
263 | 3409 | my $tp = $self->{TableParser}; | ||
264 | 3363 | my $dbh = $self->{dbh}; | 3410 | my $dbh = $self->{dbh}; |
265 | 3364 | PTDEBUG && _d('Getting next schema object from dbh', $dbh); | 3411 | PTDEBUG && _d('Getting next schema object from dbh', $dbh); |
266 | 3365 | 3412 | ||
267 | @@ -3373,7 +3420,9 @@ | |||
268 | 3373 | } | 3420 | } |
269 | 3374 | 3421 | ||
270 | 3375 | if ( !$self->{db} ) { | 3422 | if ( !$self->{db} ) { |
272 | 3376 | $self->{db} = shift @{$self->{dbs}}; | 3423 | do { |
273 | 3424 | $self->{db} = shift @{$self->{dbs}}; | ||
274 | 3425 | } until $self->_resume_from_database($self->{db}); | ||
275 | 3377 | PTDEBUG && _d('Next database:', $self->{db}); | 3426 | PTDEBUG && _d('Next database:', $self->{db}); |
276 | 3378 | return unless $self->{db}; | 3427 | return unless $self->{db}; |
277 | 3379 | } | 3428 | } |
278 | @@ -3386,8 +3435,9 @@ | |||
279 | 3386 | } | 3435 | } |
280 | 3387 | grep { | 3436 | grep { |
281 | 3388 | my ($tbl, $type) = @$_; | 3437 | my ($tbl, $type) = @$_; |
284 | 3389 | $self->table_is_allowed($self->{db}, $tbl) | 3438 | (!$type || ($type ne 'VIEW')) |
285 | 3390 | && (!$type || ($type ne 'VIEW')); | 3439 | && $self->_resume_from_table($tbl) |
286 | 3440 | && $self->table_is_allowed($self->{db}, $tbl); | ||
287 | 3391 | } | 3441 | } |
288 | 3392 | @{$dbh->selectall_arrayref($sql)}; | 3442 | @{$dbh->selectall_arrayref($sql)}; |
289 | 3393 | PTDEBUG && _d('Found', scalar @tbls, 'tables in database', $self->{db}); | 3443 | PTDEBUG && _d('Found', scalar @tbls, 'tables in database', $self->{db}); |
290 | @@ -3395,27 +3445,15 @@ | |||
291 | 3395 | } | 3445 | } |
292 | 3396 | 3446 | ||
293 | 3397 | while ( my $tbl = shift @{$self->{tbls}} ) { | 3447 | while ( my $tbl = shift @{$self->{tbls}} ) { |
311 | 3398 | my $engine; | 3448 | my $ddl = $tp->get_create_table($dbh, $self->{db}, $tbl); |
312 | 3399 | if ( $self->{filters}->{'engines'} | 3449 | my $tbl_struct = $tp->parse($ddl); |
313 | 3400 | || $self->{filters}->{'ignore-engines'} ) { | 3450 | if ( $self->engine_is_allowed($tbl_struct->{engine}) ) { |
297 | 3401 | my $sql = "SHOW TABLE STATUS FROM " . $q->quote($self->{db}) | ||
298 | 3402 | . " LIKE \'$tbl\'"; | ||
299 | 3403 | PTDEBUG && _d($sql); | ||
300 | 3404 | $engine = $dbh->selectrow_hashref($sql)->{engine}; | ||
301 | 3405 | PTDEBUG && _d($tbl, 'uses', $engine, 'engine'); | ||
302 | 3406 | } | ||
303 | 3407 | |||
304 | 3408 | |||
305 | 3409 | if ( !$engine || $self->engine_is_allowed($engine) ) { | ||
306 | 3410 | my $ddl; | ||
307 | 3411 | if ( my $du = $self->{MySQLDump} ) { | ||
308 | 3412 | $ddl = $du->get_create_table($dbh, $q, $self->{db}, $tbl)->[1]; | ||
309 | 3413 | } | ||
310 | 3414 | |||
314 | 3415 | return { | 3451 | return { |
318 | 3416 | db => $self->{db}, | 3452 | db => $self->{db}, |
319 | 3417 | tbl => $tbl, | 3453 | tbl => $tbl, |
320 | 3418 | ddl => $ddl, | 3454 | name => $q->quote($self->{db}, $tbl), |
321 | 3455 | ddl => $ddl, | ||
322 | 3456 | tbl_struct => $tbl_struct, | ||
323 | 3419 | }; | 3457 | }; |
324 | 3420 | } | 3458 | } |
325 | 3421 | } | 3459 | } |
326 | @@ -3476,6 +3514,10 @@ | |||
327 | 3476 | 3514 | ||
328 | 3477 | my $filter = $self->{filters}; | 3515 | my $filter = $self->{filters}; |
329 | 3478 | 3516 | ||
330 | 3517 | if ( $db eq 'mysql' && ($tbl eq 'general_log' || $tbl eq 'slow_log') ) { | ||
331 | 3518 | return 0; | ||
332 | 3519 | } | ||
333 | 3520 | |||
334 | 3479 | if ( $filter->{'ignore-tables'}->{$tbl} | 3521 | if ( $filter->{'ignore-tables'}->{$tbl} |
335 | 3480 | && ($filter->{'ignore-tables'}->{$tbl} eq '*' | 3522 | && ($filter->{'ignore-tables'}->{$tbl} eq '*' |
336 | 3481 | || $filter->{'ignore-tables'}->{$tbl} eq $db) ) { | 3523 | || $filter->{'ignore-tables'}->{$tbl} eq $db) ) { |
337 | @@ -3515,7 +3557,11 @@ | |||
338 | 3515 | 3557 | ||
339 | 3516 | sub engine_is_allowed { | 3558 | sub engine_is_allowed { |
340 | 3517 | my ( $self, $engine ) = @_; | 3559 | my ( $self, $engine ) = @_; |
342 | 3518 | die "I need an engine argument" unless $engine; | 3560 | |
343 | 3561 | if ( !$engine ) { | ||
344 | 3562 | PTDEBUG && _d('No engine specified; allowing the table'); | ||
345 | 3563 | return 1; | ||
346 | 3564 | } | ||
347 | 3519 | 3565 | ||
348 | 3520 | $engine = lc $engine; | 3566 | $engine = lc $engine; |
349 | 3521 | 3567 | ||
350 | @@ -3535,6 +3581,40 @@ | |||
351 | 3535 | return 1; | 3581 | return 1; |
352 | 3536 | } | 3582 | } |
353 | 3537 | 3583 | ||
354 | 3584 | sub _resume_from_database { | ||
355 | 3585 | my ($self, $db) = @_; | ||
356 | 3586 | |||
357 | 3587 | return 1 unless $self->{resume}->{db}; | ||
358 | 3588 | |||
359 | 3589 | if ( $db eq $self->{resume}->{db} ) { | ||
360 | 3590 | PTDEBUG && _d('At resume db', $db); | ||
361 | 3591 | delete $self->{resume}->{db}; | ||
362 | 3592 | return 1; | ||
363 | 3593 | } | ||
364 | 3594 | |||
365 | 3595 | return 0; | ||
366 | 3596 | } | ||
367 | 3597 | |||
368 | 3598 | sub _resume_from_table { | ||
369 | 3599 | my ($self, $tbl) = @_; | ||
370 | 3600 | |||
371 | 3601 | return 1 unless $self->{resume}->{tbl}; | ||
372 | 3602 | |||
373 | 3603 | if ( $tbl eq $self->{resume}->{tbl} ) { | ||
374 | 3604 | if ( !$self->{resume}->{after} ) { | ||
375 | 3605 | PTDEBUG && _d('Resuming from table', $tbl); | ||
376 | 3606 | delete $self->{resume}->{tbl}; | ||
377 | 3607 | return 1; | ||
378 | 3608 | } | ||
379 | 3609 | else { | ||
380 | 3610 | PTDEBUG && _d('Resuming after table', $tbl); | ||
381 | 3611 | delete $self->{resume}->{tbl}; | ||
382 | 3612 | } | ||
383 | 3613 | } | ||
384 | 3614 | |||
385 | 3615 | return 0; | ||
386 | 3616 | } | ||
387 | 3617 | |||
388 | 3538 | sub _d { | 3618 | sub _d { |
389 | 3539 | my ($package, undef, $line) = caller 0; | 3619 | my ($package, undef, $line) = caller 0; |
390 | 3540 | @_ = map { (my $temp = $_) =~ s/\n/\n# /g; $temp; } | 3620 | @_ = map { (my $temp = $_) =~ s/\n/\n# /g; $temp; } |
391 | @@ -3644,11 +3724,10 @@ | |||
392 | 3644 | MySQLDump => $du, | 3724 | MySQLDump => $du, |
393 | 3645 | TableParser => $tp, | 3725 | TableParser => $tp, |
394 | 3646 | Schema => $schema, | 3726 | Schema => $schema, |
395 | 3647 | keep_ddl => 1, | ||
396 | 3648 | ); | 3727 | ); |
397 | 3649 | TABLE: | 3728 | TABLE: |
400 | 3650 | while ( my $tbl = $schema_itr->next_schema_object() ) { | 3729 | while ( my $tbl = $schema_itr->next() ) { |
401 | 3651 | $tbl->{engine} = $tp->get_engine($tbl->{ddl}); | 3730 | $tbl->{engine} = $tbl->{tbl_struct}->{engine}; |
402 | 3652 | 3731 | ||
403 | 3653 | my ($keys, $clustered_key, $fks); | 3732 | my ($keys, $clustered_key, $fks); |
404 | 3654 | if ( $get_keys ) { | 3733 | if ( $get_keys ) { |
405 | 3655 | 3734 | ||
406 | === added file 'bin/pt-fingerprint' | |||
407 | --- bin/pt-fingerprint 1970-01-01 00:00:00 +0000 | |||
408 | +++ bin/pt-fingerprint 2012-03-31 16:07:24 +0000 | |||
409 | @@ -0,0 +1,2143 @@ | |||
410 | 1 | #!/usr/bin/env perl | ||
411 | 2 | |||
412 | 3 | # This program is part of Percona Toolkit: http://www.percona.com/software/ | ||
413 | 4 | # See "COPYRIGHT, LICENSE, AND WARRANTY" at the end of this file for legal | ||
414 | 5 | # notices and disclaimers. | ||
415 | 6 | |||
416 | 7 | use strict; | ||
417 | 8 | use warnings FATAL => 'all'; | ||
418 | 9 | use constant MKDEBUG => $ENV{MKDEBUG} || 0; | ||
419 | 10 | |||
420 | 11 | # ########################################################################### | ||
421 | 12 | # OptionParser package | ||
422 | 13 | # This package is a copy without comments from the original. The original | ||
423 | 14 | # with comments and its test file can be found in the Bazaar repository at, | ||
424 | 15 | # lib/OptionParser.pm | ||
425 | 16 | # t/lib/OptionParser.t | ||
426 | 17 | # See https://launchpad.net/percona-toolkit for more information. | ||
427 | 18 | # ########################################################################### | ||
428 | 19 | { | ||
429 | 20 | package OptionParser; | ||
430 | 21 | |||
431 | 22 | use strict; | ||
432 | 23 | use warnings FATAL => 'all'; | ||
433 | 24 | use English qw(-no_match_vars); | ||
434 | 25 | use constant PTDEBUG => $ENV{PTDEBUG} || 0; | ||
435 | 26 | |||
436 | 27 | use List::Util qw(max); | ||
437 | 28 | use Getopt::Long; | ||
438 | 29 | |||
439 | 30 | my $POD_link_re = '[LC]<"?([^">]+)"?>'; | ||
440 | 31 | |||
441 | 32 | sub new { | ||
442 | 33 | my ( $class, %args ) = @_; | ||
443 | 34 | my @required_args = qw(); | ||
444 | 35 | foreach my $arg ( @required_args ) { | ||
445 | 36 | die "I need a $arg argument" unless $args{$arg}; | ||
446 | 37 | } | ||
447 | 38 | |||
448 | 39 | my ($program_name) = $PROGRAM_NAME =~ m/([.A-Za-z-]+)$/; | ||
449 | 40 | $program_name ||= $PROGRAM_NAME; | ||
450 | 41 | my $home = $ENV{HOME} || $ENV{HOMEPATH} || $ENV{USERPROFILE} || '.'; | ||
451 | 42 | |||
452 | 43 | my %attributes = ( | ||
453 | 44 | 'type' => 1, | ||
454 | 45 | 'short form' => 1, | ||
455 | 46 | 'group' => 1, | ||
456 | 47 | 'default' => 1, | ||
457 | 48 | 'cumulative' => 1, | ||
458 | 49 | 'negatable' => 1, | ||
459 | 50 | ); | ||
460 | 51 | |||
461 | 52 | my $self = { | ||
462 | 53 | head1 => 'OPTIONS', # These args are used internally | ||
463 | 54 | skip_rules => 0, # to instantiate another Option- | ||
464 | 55 | item => '--(.*)', # Parser obj that parses the | ||
465 | 56 | attributes => \%attributes, # DSN OPTIONS section. Tools | ||
466 | 57 | parse_attributes => \&_parse_attribs, # don't tinker with these args. | ||
467 | 58 | |||
468 | 59 | %args, | ||
469 | 60 | |||
470 | 61 | strict => 1, # disabled by a special rule | ||
471 | 62 | program_name => $program_name, | ||
472 | 63 | opts => {}, | ||
473 | 64 | got_opts => 0, | ||
474 | 65 | short_opts => {}, | ||
475 | 66 | defaults => {}, | ||
476 | 67 | groups => {}, | ||
477 | 68 | allowed_groups => {}, | ||
478 | 69 | errors => [], | ||
479 | 70 | rules => [], # desc of rules for --help | ||
480 | 71 | mutex => [], # rule: opts are mutually exclusive | ||
481 | 72 | atleast1 => [], # rule: at least one opt is required | ||
482 | 73 | disables => {}, # rule: opt disables other opts | ||
483 | 74 | defaults_to => {}, # rule: opt defaults to value of other opt | ||
484 | 75 | DSNParser => undef, | ||
485 | 76 | default_files => [ | ||
486 | 77 | "/etc/percona-toolkit/percona-toolkit.conf", | ||
487 | 78 | "/etc/percona-toolkit/$program_name.conf", | ||
488 | 79 | "$home/.percona-toolkit.conf", | ||
489 | 80 | "$home/.$program_name.conf", | ||
490 | 81 | ], | ||
491 | 82 | types => { | ||
492 | 83 | string => 's', # standard Getopt type | ||
493 | 84 | int => 'i', # standard Getopt type | ||
494 | 85 | float => 'f', # standard Getopt type | ||
495 | 86 | Hash => 'H', # hash, formed from a comma-separated list | ||
496 | 87 | hash => 'h', # hash as above, but only if a value is given | ||
497 | 88 | Array => 'A', # array, similar to Hash | ||
498 | 89 | array => 'a', # array, similar to hash | ||
499 | 90 | DSN => 'd', # DSN | ||
500 | 91 | size => 'z', # size with kMG suffix (powers of 2^10) | ||
501 | 92 | time => 'm', # time, with an optional suffix of s/h/m/d | ||
502 | 93 | }, | ||
503 | 94 | }; | ||
504 | 95 | |||
505 | 96 | return bless $self, $class; | ||
506 | 97 | } | ||
507 | 98 | |||
508 | 99 | sub get_specs { | ||
509 | 100 | my ( $self, $file ) = @_; | ||
510 | 101 | $file ||= $self->{file} || __FILE__; | ||
511 | 102 | my @specs = $self->_pod_to_specs($file); | ||
512 | 103 | $self->_parse_specs(@specs); | ||
513 | 104 | |||
514 | 105 | open my $fh, "<", $file or die "Cannot open $file: $OS_ERROR"; | ||
515 | 106 | my $contents = do { local $/ = undef; <$fh> }; | ||
516 | 107 | close $fh; | ||
517 | 108 | if ( $contents =~ m/^=head1 DSN OPTIONS/m ) { | ||
518 | 109 | PTDEBUG && _d('Parsing DSN OPTIONS'); | ||
519 | 110 | my $dsn_attribs = { | ||
520 | 111 | dsn => 1, | ||
521 | 112 | copy => 1, | ||
522 | 113 | }; | ||
523 | 114 | my $parse_dsn_attribs = sub { | ||
524 | 115 | my ( $self, $option, $attribs ) = @_; | ||
525 | 116 | map { | ||
526 | 117 | my $val = $attribs->{$_}; | ||
527 | 118 | if ( $val ) { | ||
528 | 119 | $val = $val eq 'yes' ? 1 | ||
529 | 120 | : $val eq 'no' ? 0 | ||
530 | 121 | : $val; | ||
531 | 122 | $attribs->{$_} = $val; | ||
532 | 123 | } | ||
533 | 124 | } keys %$attribs; | ||
534 | 125 | return { | ||
535 | 126 | key => $option, | ||
536 | 127 | %$attribs, | ||
537 | 128 | }; | ||
538 | 129 | }; | ||
539 | 130 | my $dsn_o = new OptionParser( | ||
540 | 131 | description => 'DSN OPTIONS', | ||
541 | 132 | head1 => 'DSN OPTIONS', | ||
542 | 133 | dsn => 0, # XXX don't infinitely recurse! | ||
543 | 134 | item => '\* (.)', # key opts are a single character | ||
544 | 135 | skip_rules => 1, # no rules before opts | ||
545 | 136 | attributes => $dsn_attribs, | ||
546 | 137 | parse_attributes => $parse_dsn_attribs, | ||
547 | 138 | ); | ||
548 | 139 | my @dsn_opts = map { | ||
549 | 140 | my $opts = { | ||
550 | 141 | key => $_->{spec}->{key}, | ||
551 | 142 | dsn => $_->{spec}->{dsn}, | ||
552 | 143 | copy => $_->{spec}->{copy}, | ||
553 | 144 | desc => $_->{desc}, | ||
554 | 145 | }; | ||
555 | 146 | $opts; | ||
556 | 147 | } $dsn_o->_pod_to_specs($file); | ||
557 | 148 | $self->{DSNParser} = DSNParser->new(opts => \@dsn_opts); | ||
558 | 149 | } | ||
559 | 150 | |||
560 | 151 | if ( $contents =~ m/^=head1 VERSION\n\n^(.+)$/m ) { | ||
561 | 152 | $self->{version} = $1; | ||
562 | 153 | PTDEBUG && _d($self->{version}); | ||
563 | 154 | } | ||
564 | 155 | |||
565 | 156 | return; | ||
566 | 157 | } | ||
567 | 158 | |||
568 | 159 | sub DSNParser { | ||
569 | 160 | my ( $self ) = @_; | ||
570 | 161 | return $self->{DSNParser}; | ||
571 | 162 | }; | ||
572 | 163 | |||
573 | 164 | sub get_defaults_files { | ||
574 | 165 | my ( $self ) = @_; | ||
575 | 166 | return @{$self->{default_files}}; | ||
576 | 167 | } | ||
577 | 168 | |||
578 | 169 | sub _pod_to_specs { | ||
579 | 170 | my ( $self, $file ) = @_; | ||
580 | 171 | $file ||= $self->{file} || __FILE__; | ||
581 | 172 | open my $fh, '<', $file or die "Cannot open $file: $OS_ERROR"; | ||
582 | 173 | |||
583 | 174 | my @specs = (); | ||
584 | 175 | my @rules = (); | ||
585 | 176 | my $para; | ||
586 | 177 | |||
587 | 178 | local $INPUT_RECORD_SEPARATOR = ''; | ||
588 | 179 | while ( $para = <$fh> ) { | ||
589 | 180 | next unless $para =~ m/^=head1 $self->{head1}/; | ||
590 | 181 | last; | ||
591 | 182 | } | ||
592 | 183 | |||
593 | 184 | while ( $para = <$fh> ) { | ||
594 | 185 | last if $para =~ m/^=over/; | ||
595 | 186 | next if $self->{skip_rules}; | ||
596 | 187 | chomp $para; | ||
597 | 188 | $para =~ s/\s+/ /g; | ||
598 | 189 | $para =~ s/$POD_link_re/$1/go; | ||
599 | 190 | PTDEBUG && _d('Option rule:', $para); | ||
600 | 191 | push @rules, $para; | ||
601 | 192 | } | ||
602 | 193 | |||
603 | 194 | die "POD has no $self->{head1} section" unless $para; | ||
604 | 195 | |||
605 | 196 | do { | ||
606 | 197 | if ( my ($option) = $para =~ m/^=item $self->{item}/ ) { | ||
607 | 198 | chomp $para; | ||
608 | 199 | PTDEBUG && _d($para); | ||
609 | 200 | my %attribs; | ||
610 | 201 | |||
611 | 202 | $para = <$fh>; # read next paragraph, possibly attributes | ||
612 | 203 | |||
613 | 204 | if ( $para =~ m/: / ) { # attributes | ||
614 | 205 | $para =~ s/\s+\Z//g; | ||
615 | 206 | %attribs = map { | ||
616 | 207 | my ( $attrib, $val) = split(/: /, $_); | ||
617 | 208 | die "Unrecognized attribute for --$option: $attrib" | ||
618 | 209 | unless $self->{attributes}->{$attrib}; | ||
619 | 210 | ($attrib, $val); | ||
620 | 211 | } split(/; /, $para); | ||
621 | 212 | if ( $attribs{'short form'} ) { | ||
622 | 213 | $attribs{'short form'} =~ s/-//; | ||
623 | 214 | } | ||
624 | 215 | $para = <$fh>; # read next paragraph, probably short help desc | ||
625 | 216 | } | ||
626 | 217 | else { | ||
627 | 218 | PTDEBUG && _d('Option has no attributes'); | ||
628 | 219 | } | ||
629 | 220 | |||
630 | 221 | $para =~ s/\s+\Z//g; | ||
631 | 222 | $para =~ s/\s+/ /g; | ||
632 | 223 | $para =~ s/$POD_link_re/$1/go; | ||
633 | 224 | |||
634 | 225 | $para =~ s/\.(?:\n.*| [A-Z].*|\Z)//s; | ||
635 | 226 | PTDEBUG && _d('Short help:', $para); | ||
636 | 227 | |||
637 | 228 | die "No description after option spec $option" if $para =~ m/^=item/; | ||
638 | 229 | |||
639 | 230 | if ( my ($base_option) = $option =~ m/^\[no\](.*)/ ) { | ||
640 | 231 | $option = $base_option; | ||
641 | 232 | $attribs{'negatable'} = 1; | ||
642 | 233 | } | ||
643 | 234 | |||
644 | 235 | push @specs, { | ||
645 | 236 | spec => $self->{parse_attributes}->($self, $option, \%attribs), | ||
646 | 237 | desc => $para | ||
647 | 238 | . (defined $attribs{default} ? " (default $attribs{default})" : ''), | ||
648 | 239 | group => ($attribs{'group'} ? $attribs{'group'} : 'default'), | ||
649 | 240 | }; | ||
650 | 241 | } | ||
651 | 242 | while ( $para = <$fh> ) { | ||
652 | 243 | last unless $para; | ||
653 | 244 | if ( $para =~ m/^=head1/ ) { | ||
654 | 245 | $para = undef; # Can't 'last' out of a do {} block. | ||
655 | 246 | last; | ||
656 | 247 | } | ||
657 | 248 | last if $para =~ m/^=item /; | ||
658 | 249 | } | ||
659 | 250 | } while ( $para ); | ||
660 | 251 | |||
661 | 252 | die "No valid specs in $self->{head1}" unless @specs; | ||
662 | 253 | |||
663 | 254 | close $fh; | ||
664 | 255 | return @specs, @rules; | ||
665 | 256 | } | ||
666 | 257 | |||
667 | 258 | sub _parse_specs { | ||
668 | 259 | my ( $self, @specs ) = @_; | ||
669 | 260 | my %disables; # special rule that requires deferred checking | ||
670 | 261 | |||
671 | 262 | foreach my $opt ( @specs ) { | ||
672 | 263 | if ( ref $opt ) { # It's an option spec, not a rule. | ||
673 | 264 | PTDEBUG && _d('Parsing opt spec:', | ||
674 | 265 | map { ($_, '=>', $opt->{$_}) } keys %$opt); | ||
675 | 266 | |||
676 | 267 | my ( $long, $short ) = $opt->{spec} =~ m/^([\w-]+)(?:\|([^!+=]*))?/; | ||
677 | 268 | if ( !$long ) { | ||
678 | 269 | die "Cannot parse long option from spec $opt->{spec}"; | ||
679 | 270 | } | ||
680 | 271 | $opt->{long} = $long; | ||
681 | 272 | |||
682 | 273 | die "Duplicate long option --$long" if exists $self->{opts}->{$long}; | ||
683 | 274 | $self->{opts}->{$long} = $opt; | ||
684 | 275 | |||
685 | 276 | if ( length $long == 1 ) { | ||
686 | 277 | PTDEBUG && _d('Long opt', $long, 'looks like short opt'); | ||
687 | 278 | $self->{short_opts}->{$long} = $long; | ||
688 | 279 | } | ||
689 | 280 | |||
690 | 281 | if ( $short ) { | ||
691 | 282 | die "Duplicate short option -$short" | ||
692 | 283 | if exists $self->{short_opts}->{$short}; | ||
693 | 284 | $self->{short_opts}->{$short} = $long; | ||
694 | 285 | $opt->{short} = $short; | ||
695 | 286 | } | ||
696 | 287 | else { | ||
697 | 288 | $opt->{short} = undef; | ||
698 | 289 | } | ||
699 | 290 | |||
700 | 291 | $opt->{is_negatable} = $opt->{spec} =~ m/!/ ? 1 : 0; | ||
701 | 292 | $opt->{is_cumulative} = $opt->{spec} =~ m/\+/ ? 1 : 0; | ||
702 | 293 | $opt->{is_required} = $opt->{desc} =~ m/required/ ? 1 : 0; | ||
703 | 294 | |||
704 | 295 | $opt->{group} ||= 'default'; | ||
705 | 296 | $self->{groups}->{ $opt->{group} }->{$long} = 1; | ||
706 | 297 | |||
707 | 298 | $opt->{value} = undef; | ||
708 | 299 | $opt->{got} = 0; | ||
709 | 300 | |||
710 | 301 | my ( $type ) = $opt->{spec} =~ m/=(.)/; | ||
711 | 302 | $opt->{type} = $type; | ||
712 | 303 | PTDEBUG && _d($long, 'type:', $type); | ||
713 | 304 | |||
714 | 305 | |||
715 | 306 | $opt->{spec} =~ s/=./=s/ if ( $type && $type =~ m/[HhAadzm]/ ); | ||
716 | 307 | |||
717 | 308 | if ( (my ($def) = $opt->{desc} =~ m/default\b(?: ([^)]+))?/) ) { | ||
718 | 309 | $self->{defaults}->{$long} = defined $def ? $def : 1; | ||
719 | 310 | PTDEBUG && _d($long, 'default:', $def); | ||
720 | 311 | } | ||
721 | 312 | |||
722 | 313 | if ( $long eq 'config' ) { | ||
723 | 314 | $self->{defaults}->{$long} = join(',', $self->get_defaults_files()); | ||
724 | 315 | } | ||
725 | 316 | |||
726 | 317 | if ( (my ($dis) = $opt->{desc} =~ m/(disables .*)/) ) { | ||
727 | 318 | $disables{$long} = $dis; | ||
728 | 319 | PTDEBUG && _d('Deferring check of disables rule for', $opt, $dis); | ||
729 | 320 | } | ||
730 | 321 | |||
731 | 322 | $self->{opts}->{$long} = $opt; | ||
732 | 323 | } | ||
733 | 324 | else { # It's an option rule, not a spec. | ||
734 | 325 | PTDEBUG && _d('Parsing rule:', $opt); | ||
735 | 326 | push @{$self->{rules}}, $opt; | ||
736 | 327 | my @participants = $self->_get_participants($opt); | ||
737 | 328 | my $rule_ok = 0; | ||
738 | 329 | |||
739 | 330 | if ( $opt =~ m/mutually exclusive|one and only one/ ) { | ||
740 | 331 | $rule_ok = 1; | ||
741 | 332 | push @{$self->{mutex}}, \@participants; | ||
742 | 333 | PTDEBUG && _d(@participants, 'are mutually exclusive'); | ||
743 | 334 | } | ||
744 | 335 | if ( $opt =~ m/at least one|one and only one/ ) { | ||
745 | 336 | $rule_ok = 1; | ||
746 | 337 | push @{$self->{atleast1}}, \@participants; | ||
747 | 338 | PTDEBUG && _d(@participants, 'require at least one'); | ||
748 | 339 | } | ||
749 | 340 | if ( $opt =~ m/default to/ ) { | ||
750 | 341 | $rule_ok = 1; | ||
751 | 342 | $self->{defaults_to}->{$participants[0]} = $participants[1]; | ||
752 | 343 | PTDEBUG && _d($participants[0], 'defaults to', $participants[1]); | ||
753 | 344 | } | ||
754 | 345 | if ( $opt =~ m/restricted to option groups/ ) { | ||
755 | 346 | $rule_ok = 1; | ||
756 | 347 | my ($groups) = $opt =~ m/groups ([\w\s\,]+)/; | ||
757 | 348 | my @groups = split(',', $groups); | ||
758 | 349 | %{$self->{allowed_groups}->{$participants[0]}} = map { | ||
759 | 350 | s/\s+//; | ||
760 | 351 | $_ => 1; | ||
761 | 352 | } @groups; | ||
762 | 353 | } | ||
763 | 354 | if( $opt =~ m/accepts additional command-line arguments/ ) { | ||
764 | 355 | $rule_ok = 1; | ||
765 | 356 | $self->{strict} = 0; | ||
766 | 357 | PTDEBUG && _d("Strict mode disabled by rule"); | ||
767 | 358 | } | ||
768 | 359 | |||
769 | 360 | die "Unrecognized option rule: $opt" unless $rule_ok; | ||
770 | 361 | } | ||
771 | 362 | } | ||
772 | 363 | |||
773 | 364 | foreach my $long ( keys %disables ) { | ||
774 | 365 | my @participants = $self->_get_participants($disables{$long}); | ||
775 | 366 | $self->{disables}->{$long} = \@participants; | ||
776 | 367 | PTDEBUG && _d('Option', $long, 'disables', @participants); | ||
777 | 368 | } | ||
778 | 369 | |||
779 | 370 | return; | ||
780 | 371 | } | ||
781 | 372 | |||
782 | 373 | sub _get_participants { | ||
783 | 374 | my ( $self, $str ) = @_; | ||
784 | 375 | my @participants; | ||
785 | 376 | foreach my $long ( $str =~ m/--(?:\[no\])?([\w-]+)/g ) { | ||
786 | 377 | die "Option --$long does not exist while processing rule $str" | ||
787 | 378 | unless exists $self->{opts}->{$long}; | ||
788 | 379 | push @participants, $long; | ||
789 | 380 | } | ||
790 | 381 | PTDEBUG && _d('Participants for', $str, ':', @participants); | ||
791 | 382 | return @participants; | ||
792 | 383 | } | ||
793 | 384 | |||
794 | 385 | sub opts { | ||
795 | 386 | my ( $self ) = @_; | ||
796 | 387 | my %opts = %{$self->{opts}}; | ||
797 | 388 | return %opts; | ||
798 | 389 | } | ||
799 | 390 | |||
800 | 391 | sub short_opts { | ||
801 | 392 | my ( $self ) = @_; | ||
802 | 393 | my %short_opts = %{$self->{short_opts}}; | ||
803 | 394 | return %short_opts; | ||
804 | 395 | } | ||
805 | 396 | |||
806 | 397 | sub set_defaults { | ||
807 | 398 | my ( $self, %defaults ) = @_; | ||
808 | 399 | $self->{defaults} = {}; | ||
809 | 400 | foreach my $long ( keys %defaults ) { | ||
810 | 401 | die "Cannot set default for nonexistent option $long" | ||
811 | 402 | unless exists $self->{opts}->{$long}; | ||
812 | 403 | $self->{defaults}->{$long} = $defaults{$long}; | ||
813 | 404 | PTDEBUG && _d('Default val for', $long, ':', $defaults{$long}); | ||
814 | 405 | } | ||
815 | 406 | return; | ||
816 | 407 | } | ||
817 | 408 | |||
818 | 409 | sub get_defaults { | ||
819 | 410 | my ( $self ) = @_; | ||
820 | 411 | return $self->{defaults}; | ||
821 | 412 | } | ||
822 | 413 | |||
823 | 414 | sub get_groups { | ||
824 | 415 | my ( $self ) = @_; | ||
825 | 416 | return $self->{groups}; | ||
826 | 417 | } | ||
827 | 418 | |||
828 | 419 | sub _set_option { | ||
829 | 420 | my ( $self, $opt, $val ) = @_; | ||
830 | 421 | my $long = exists $self->{opts}->{$opt} ? $opt | ||
831 | 422 | : exists $self->{short_opts}->{$opt} ? $self->{short_opts}->{$opt} | ||
832 | 423 | : die "Getopt::Long gave a nonexistent option: $opt"; | ||
833 | 424 | |||
834 | 425 | $opt = $self->{opts}->{$long}; | ||
835 | 426 | if ( $opt->{is_cumulative} ) { | ||
836 | 427 | $opt->{value}++; | ||
837 | 428 | } | ||
838 | 429 | else { | ||
839 | 430 | $opt->{value} = $val; | ||
840 | 431 | } | ||
841 | 432 | $opt->{got} = 1; | ||
842 | 433 | PTDEBUG && _d('Got option', $long, '=', $val); | ||
843 | 434 | } | ||
844 | 435 | |||
845 | 436 | sub get_opts { | ||
846 | 437 | my ( $self ) = @_; | ||
847 | 438 | |||
848 | 439 | foreach my $long ( keys %{$self->{opts}} ) { | ||
849 | 440 | $self->{opts}->{$long}->{got} = 0; | ||
850 | 441 | $self->{opts}->{$long}->{value} | ||
851 | 442 | = exists $self->{defaults}->{$long} ? $self->{defaults}->{$long} | ||
852 | 443 | : $self->{opts}->{$long}->{is_cumulative} ? 0 | ||
853 | 444 | : undef; | ||
854 | 445 | } | ||
855 | 446 | $self->{got_opts} = 0; | ||
856 | 447 | |||
857 | 448 | $self->{errors} = []; | ||
858 | 449 | |||
859 | 450 | if ( @ARGV && $ARGV[0] eq "--config" ) { | ||
860 | 451 | shift @ARGV; | ||
861 | 452 | $self->_set_option('config', shift @ARGV); | ||
862 | 453 | } | ||
863 | 454 | if ( $self->has('config') ) { | ||
864 | 455 | my @extra_args; | ||
865 | 456 | foreach my $filename ( split(',', $self->get('config')) ) { | ||
866 | 457 | eval { | ||
867 | 458 | push @extra_args, $self->_read_config_file($filename); | ||
868 | 459 | }; | ||
869 | 460 | if ( $EVAL_ERROR ) { | ||
870 | 461 | if ( $self->got('config') ) { | ||
871 | 462 | die $EVAL_ERROR; | ||
872 | 463 | } | ||
873 | 464 | elsif ( PTDEBUG ) { | ||
874 | 465 | _d($EVAL_ERROR); | ||
875 | 466 | } | ||
876 | 467 | } | ||
877 | 468 | } | ||
878 | 469 | unshift @ARGV, @extra_args; | ||
879 | 470 | } | ||
880 | 471 | |||
881 | 472 | Getopt::Long::Configure('no_ignore_case', 'bundling'); | ||
882 | 473 | GetOptions( | ||
883 | 474 | map { $_->{spec} => sub { $self->_set_option(@_); } } | ||
884 | 475 | grep { $_->{long} ne 'config' } # --config is handled specially above. | ||
885 | 476 | values %{$self->{opts}} | ||
886 | 477 | ) or $self->save_error('Error parsing options'); | ||
887 | 478 | |||
888 | 479 | if ( exists $self->{opts}->{version} && $self->{opts}->{version}->{got} ) { | ||
889 | 480 | if ( $self->{version} ) { | ||
890 | 481 | print $self->{version}, "\n"; | ||
891 | 482 | } | ||
892 | 483 | else { | ||
893 | 484 | print "Error parsing version. See the VERSION section of the tool's documentation.\n"; | ||
894 | 485 | } | ||
895 | 486 | exit 0; | ||
896 | 487 | } | ||
897 | 488 | |||
898 | 489 | if ( @ARGV && $self->{strict} ) { | ||
899 | 490 | $self->save_error("Unrecognized command-line options @ARGV"); | ||
900 | 491 | } | ||
901 | 492 | |||
902 | 493 | foreach my $mutex ( @{$self->{mutex}} ) { | ||
903 | 494 | my @set = grep { $self->{opts}->{$_}->{got} } @$mutex; | ||
904 | 495 | if ( @set > 1 ) { | ||
905 | 496 | my $err = join(', ', map { "--$self->{opts}->{$_}->{long}" } | ||
906 | 497 | @{$mutex}[ 0 .. scalar(@$mutex) - 2] ) | ||
907 | 498 | . ' and --'.$self->{opts}->{$mutex->[-1]}->{long} | ||
908 | 499 | . ' are mutually exclusive.'; | ||
909 | 500 | $self->save_error($err); | ||
910 | 501 | } | ||
911 | 502 | } | ||
912 | 503 | |||
913 | 504 | foreach my $required ( @{$self->{atleast1}} ) { | ||
914 | 505 | my @set = grep { $self->{opts}->{$_}->{got} } @$required; | ||
915 | 506 | if ( @set == 0 ) { | ||
916 | 507 | my $err = join(', ', map { "--$self->{opts}->{$_}->{long}" } | ||
917 | 508 | @{$required}[ 0 .. scalar(@$required) - 2] ) | ||
918 | 509 | .' or --'.$self->{opts}->{$required->[-1]}->{long}; | ||
919 | 510 | $self->save_error("Specify at least one of $err"); | ||
920 | 511 | } | ||
921 | 512 | } | ||
922 | 513 | |||
923 | 514 | $self->_check_opts( keys %{$self->{opts}} ); | ||
924 | 515 | $self->{got_opts} = 1; | ||
925 | 516 | return; | ||
926 | 517 | } | ||
927 | 518 | |||
928 | 519 | sub _check_opts { | ||
929 | 520 | my ( $self, @long ) = @_; | ||
930 | 521 | my $long_last = scalar @long; | ||
931 | 522 | while ( @long ) { | ||
932 | 523 | foreach my $i ( 0..$#long ) { | ||
933 | 524 | my $long = $long[$i]; | ||
934 | 525 | next unless $long; | ||
935 | 526 | my $opt = $self->{opts}->{$long}; | ||
936 | 527 | if ( $opt->{got} ) { | ||
937 | 528 | if ( exists $self->{disables}->{$long} ) { | ||
938 | 529 | my @disable_opts = @{$self->{disables}->{$long}}; | ||
939 | 530 | map { $self->{opts}->{$_}->{value} = undef; } @disable_opts; | ||
940 | 531 | PTDEBUG && _d('Unset options', @disable_opts, | ||
941 | 532 | 'because', $long,'disables them'); | ||
942 | 533 | } | ||
943 | 534 | |||
944 | 535 | if ( exists $self->{allowed_groups}->{$long} ) { | ||
945 | 536 | |||
946 | 537 | my @restricted_groups = grep { | ||
947 | 538 | !exists $self->{allowed_groups}->{$long}->{$_} | ||
948 | 539 | } keys %{$self->{groups}}; | ||
949 | 540 | |||
950 | 541 | my @restricted_opts; | ||
951 | 542 | foreach my $restricted_group ( @restricted_groups ) { | ||
952 | 543 | RESTRICTED_OPT: | ||
953 | 544 | foreach my $restricted_opt ( | ||
954 | 545 | keys %{$self->{groups}->{$restricted_group}} ) | ||
955 | 546 | { | ||
956 | 547 | next RESTRICTED_OPT if $restricted_opt eq $long; | ||
957 | 548 | push @restricted_opts, $restricted_opt | ||
958 | 549 | if $self->{opts}->{$restricted_opt}->{got}; | ||
959 | 550 | } | ||
960 | 551 | } | ||
961 | 552 | |||
962 | 553 | if ( @restricted_opts ) { | ||
963 | 554 | my $err; | ||
964 | 555 | if ( @restricted_opts == 1 ) { | ||
965 | 556 | $err = "--$restricted_opts[0]"; | ||
966 | 557 | } | ||
967 | 558 | else { | ||
968 | 559 | $err = join(', ', | ||
969 | 560 | map { "--$self->{opts}->{$_}->{long}" } | ||
970 | 561 | grep { $_ } | ||
971 | 562 | @restricted_opts[0..scalar(@restricted_opts) - 2] | ||
972 | 563 | ) | ||
973 | 564 | . ' or --'.$self->{opts}->{$restricted_opts[-1]}->{long}; | ||
974 | 565 | } | ||
975 | 566 | $self->save_error("--$long is not allowed with $err"); | ||
976 | 567 | } | ||
977 | 568 | } | ||
978 | 569 | |||
979 | 570 | } | ||
980 | 571 | elsif ( $opt->{is_required} ) { | ||
981 | 572 | $self->save_error("Required option --$long must be specified"); | ||
982 | 573 | } | ||
983 | 574 | |||
984 | 575 | $self->_validate_type($opt); | ||
985 | 576 | if ( $opt->{parsed} ) { | ||
986 | 577 | delete $long[$i]; | ||
987 | 578 | } | ||
988 | 579 | else { | ||
989 | 580 | PTDEBUG && _d('Temporarily failed to parse', $long); | ||
990 | 581 | } | ||
991 | 582 | } | ||
992 | 583 | |||
993 | 584 | die "Failed to parse options, possibly due to circular dependencies" | ||
994 | 585 | if @long == $long_last; | ||
995 | 586 | $long_last = @long; | ||
996 | 587 | } | ||
997 | 588 | |||
998 | 589 | return; | ||
999 | 590 | } | ||
1000 | 591 | |||
1001 | 592 | sub _validate_type { | ||
1002 | 593 | my ( $self, $opt ) = @_; | ||
1003 | 594 | return unless $opt; | ||
1004 | 595 | |||
1005 | 596 | if ( !$opt->{type} ) { | ||
1006 | 597 | $opt->{parsed} = 1; | ||
1007 | 598 | return; | ||
1008 | 599 | } | ||
1009 | 600 | |||
1010 | 601 | my $val = $opt->{value}; | ||
1011 | 602 | |||
1012 | 603 | if ( $val && $opt->{type} eq 'm' ) { # type time | ||
1013 | 604 | PTDEBUG && _d('Parsing option', $opt->{long}, 'as a time value'); | ||
1014 | 605 | my ( $prefix, $num, $suffix ) = $val =~ m/([+-]?)(\d+)([a-z])?$/; | ||
1015 | 606 | if ( !$suffix ) { | ||
1016 | 607 | my ( $s ) = $opt->{desc} =~ m/\(suffix (.)\)/; | ||
1017 | 608 | $suffix = $s || 's'; | ||
1018 | 609 | PTDEBUG && _d('No suffix given; using', $suffix, 'for', | ||
1019 | 610 | $opt->{long}, '(value:', $val, ')'); | ||
1020 | 611 | } | ||
1021 | 612 | if ( $suffix =~ m/[smhd]/ ) { | ||
1022 | 613 | $val = $suffix eq 's' ? $num # Seconds | ||
1023 | 614 | : $suffix eq 'm' ? $num * 60 # Minutes | ||
1024 | 615 | : $suffix eq 'h' ? $num * 3600 # Hours | ||
1025 | 616 | : $num * 86400; # Days | ||
1026 | 617 | $opt->{value} = ($prefix || '') . $val; | ||
1027 | 618 | PTDEBUG && _d('Setting option', $opt->{long}, 'to', $val); | ||
1028 | 619 | } | ||
1029 | 620 | else { | ||
1030 | 621 | $self->save_error("Invalid time suffix for --$opt->{long}"); | ||
1031 | 622 | } | ||
1032 | 623 | } | ||
1033 | 624 | elsif ( $val && $opt->{type} eq 'd' ) { # type DSN | ||
1034 | 625 | PTDEBUG && _d('Parsing option', $opt->{long}, 'as a DSN'); | ||
1035 | 626 | my $prev = {}; | ||
1036 | 627 | my $from_key = $self->{defaults_to}->{ $opt->{long} }; | ||
1037 | 628 | if ( $from_key ) { | ||
1038 | 629 | PTDEBUG && _d($opt->{long}, 'DSN copies from', $from_key, 'DSN'); | ||
1039 | 630 | if ( $self->{opts}->{$from_key}->{parsed} ) { | ||
1040 | 631 | $prev = $self->{opts}->{$from_key}->{value}; | ||
1041 | 632 | } | ||
1042 | 633 | else { | ||
1043 | 634 | PTDEBUG && _d('Cannot parse', $opt->{long}, 'until', | ||
1044 | 635 | $from_key, 'parsed'); | ||
1045 | 636 | return; | ||
1046 | 637 | } | ||
1047 | 638 | } | ||
1048 | 639 | my $defaults = $self->{DSNParser}->parse_options($self); | ||
1049 | 640 | $opt->{value} = $self->{DSNParser}->parse($val, $prev, $defaults); | ||
1050 | 641 | } | ||
1051 | 642 | elsif ( $val && $opt->{type} eq 'z' ) { # type size | ||
1052 | 643 | PTDEBUG && _d('Parsing option', $opt->{long}, 'as a size value'); | ||
1053 | 644 | $self->_parse_size($opt, $val); | ||
1054 | 645 | } | ||
1055 | 646 | elsif ( $opt->{type} eq 'H' || (defined $val && $opt->{type} eq 'h') ) { | ||
1056 | 647 | $opt->{value} = { map { $_ => 1 } split(/(?<!\\),\s*/, ($val || '')) }; | ||
1057 | 648 | } | ||
1058 | 649 | elsif ( $opt->{type} eq 'A' || (defined $val && $opt->{type} eq 'a') ) { | ||
1059 | 650 | $opt->{value} = [ split(/(?<!\\),\s*/, ($val || '')) ]; | ||
1060 | 651 | } | ||
1061 | 652 | else { | ||
1062 | 653 | PTDEBUG && _d('Nothing to validate for option', | ||
1063 | 654 | $opt->{long}, 'type', $opt->{type}, 'value', $val); | ||
1064 | 655 | } | ||
1065 | 656 | |||
1066 | 657 | $opt->{parsed} = 1; | ||
1067 | 658 | return; | ||
1068 | 659 | } | ||
1069 | 660 | |||
1070 | 661 | sub get { | ||
1071 | 662 | my ( $self, $opt ) = @_; | ||
1072 | 663 | my $long = (length $opt == 1 ? $self->{short_opts}->{$opt} : $opt); | ||
1073 | 664 | die "Option $opt does not exist" | ||
1074 | 665 | unless $long && exists $self->{opts}->{$long}; | ||
1075 | 666 | return $self->{opts}->{$long}->{value}; | ||
1076 | 667 | } | ||
1077 | 668 | |||
1078 | 669 | sub got { | ||
1079 | 670 | my ( $self, $opt ) = @_; | ||
1080 | 671 | my $long = (length $opt == 1 ? $self->{short_opts}->{$opt} : $opt); | ||
1081 | 672 | die "Option $opt does not exist" | ||
1082 | 673 | unless $long && exists $self->{opts}->{$long}; | ||
1083 | 674 | return $self->{opts}->{$long}->{got}; | ||
1084 | 675 | } | ||
1085 | 676 | |||
1086 | 677 | sub has { | ||
1087 | 678 | my ( $self, $opt ) = @_; | ||
1088 | 679 | my $long = (length $opt == 1 ? $self->{short_opts}->{$opt} : $opt); | ||
1089 | 680 | return defined $long ? exists $self->{opts}->{$long} : 0; | ||
1090 | 681 | } | ||
1091 | 682 | |||
1092 | 683 | sub set { | ||
1093 | 684 | my ( $self, $opt, $val ) = @_; | ||
1094 | 685 | my $long = (length $opt == 1 ? $self->{short_opts}->{$opt} : $opt); | ||
1095 | 686 | die "Option $opt does not exist" | ||
1096 | 687 | unless $long && exists $self->{opts}->{$long}; | ||
1097 | 688 | $self->{opts}->{$long}->{value} = $val; | ||
1098 | 689 | return; | ||
1099 | 690 | } | ||
1100 | 691 | |||
1101 | 692 | sub save_error { | ||
1102 | 693 | my ( $self, $error ) = @_; | ||
1103 | 694 | push @{$self->{errors}}, $error; | ||
1104 | 695 | return; | ||
1105 | 696 | } | ||
1106 | 697 | |||
1107 | 698 | sub errors { | ||
1108 | 699 | my ( $self ) = @_; | ||
1109 | 700 | return $self->{errors}; | ||
1110 | 701 | } | ||
1111 | 702 | |||
1112 | 703 | sub usage { | ||
1113 | 704 | my ( $self ) = @_; | ||
1114 | 705 | warn "No usage string is set" unless $self->{usage}; # XXX | ||
1115 | 706 | return "Usage: " . ($self->{usage} || '') . "\n"; | ||
1116 | 707 | } | ||
1117 | 708 | |||
1118 | 709 | sub descr { | ||
1119 | 710 | my ( $self ) = @_; | ||
1120 | 711 | warn "No description string is set" unless $self->{description}; # XXX | ||
1121 | 712 | my $descr = ($self->{description} || $self->{program_name} || '') | ||
1122 | 713 | . " For more details, please use the --help option, " | ||
1123 | 714 | . "or try 'perldoc $PROGRAM_NAME' " | ||
1124 | 715 | . "for complete documentation."; | ||
1125 | 716 | $descr = join("\n", $descr =~ m/(.{0,80})(?:\s+|$)/g) | ||
1126 | 717 | unless $ENV{DONT_BREAK_LINES}; | ||
1127 | 718 | $descr =~ s/ +$//mg; | ||
1128 | 719 | return $descr; | ||
1129 | 720 | } | ||
1130 | 721 | |||
1131 | 722 | sub usage_or_errors { | ||
1132 | 723 | my ( $self, $file, $return ) = @_; | ||
1133 | 724 | $file ||= $self->{file} || __FILE__; | ||
1134 | 725 | |||
1135 | 726 | if ( !$self->{description} || !$self->{usage} ) { | ||
1136 | 727 | PTDEBUG && _d("Getting description and usage from SYNOPSIS in", $file); | ||
1137 | 728 | my %synop = $self->_parse_synopsis($file); | ||
1138 | 729 | $self->{description} ||= $synop{description}; | ||
1139 | 730 | $self->{usage} ||= $synop{usage}; | ||
1140 | 731 | PTDEBUG && _d("Description:", $self->{description}, | ||
1141 | 732 | "\nUsage:", $self->{usage}); | ||
1142 | 733 | } | ||
1143 | 734 | |||
1144 | 735 | if ( $self->{opts}->{help}->{got} ) { | ||
1145 | 736 | print $self->print_usage() or die "Cannot print usage: $OS_ERROR"; | ||
1146 | 737 | exit 0 unless $return; | ||
1147 | 738 | } | ||
1148 | 739 | elsif ( scalar @{$self->{errors}} ) { | ||
1149 | 740 | print $self->print_errors() or die "Cannot print errors: $OS_ERROR"; | ||
1150 | 741 | exit 0 unless $return; | ||
1151 | 742 | } | ||
1152 | 743 | |||
1153 | 744 | return; | ||
1154 | 745 | } | ||
1155 | 746 | |||
1156 | 747 | sub print_errors { | ||
1157 | 748 | my ( $self ) = @_; | ||
1158 | 749 | my $usage = $self->usage() . "\n"; | ||
1159 | 750 | if ( (my @errors = @{$self->{errors}}) ) { | ||
1160 | 751 | $usage .= join("\n * ", 'Errors in command-line arguments:', @errors) | ||
1161 | 752 | . "\n"; | ||
1162 | 753 | } | ||
1163 | 754 | return $usage . "\n" . $self->descr(); | ||
1164 | 755 | } | ||
1165 | 756 | |||
1166 | 757 | sub print_usage { | ||
1167 | 758 | my ( $self ) = @_; | ||
1168 | 759 | die "Run get_opts() before print_usage()" unless $self->{got_opts}; | ||
1169 | 760 | my @opts = values %{$self->{opts}}; | ||
1170 | 761 | |||
1171 | 762 | my $maxl = max( | ||
1172 | 763 | map { | ||
1173 | 764 | length($_->{long}) # option long name | ||
1174 | 765 | + ($_->{is_negatable} ? 4 : 0) # "[no]" if opt is negatable | ||
1175 | 766 | + ($_->{type} ? 2 : 0) # "=x" where x is the opt type | ||
1176 | 767 | } | ||
1177 | 768 | @opts); | ||
1178 | 769 | |||
1179 | 770 | my $maxs = max(0, | ||
1180 | 771 | map { | ||
1181 | 772 | length($_) | ||
1182 | 773 | + ($self->{opts}->{$_}->{is_negatable} ? 4 : 0) | ||
1183 | 774 | + ($self->{opts}->{$_}->{type} ? 2 : 0) | ||
1184 | 775 | } | ||
1185 | 776 | values %{$self->{short_opts}}); | ||
1186 | 777 | |||
1187 | 778 | my $lcol = max($maxl, ($maxs + 3)); | ||
1188 | 779 | my $rcol = 80 - $lcol - 6; | ||
1189 | 780 | my $rpad = ' ' x ( 80 - $rcol ); | ||
1190 | 781 | |||
1191 | 782 | $maxs = max($lcol - 3, $maxs); | ||
1192 | 783 | |||
1193 | 784 | my $usage = $self->descr() . "\n" . $self->usage(); | ||
1194 | 785 | |||
1195 | 786 | my @groups = reverse sort grep { $_ ne 'default'; } keys %{$self->{groups}}; | ||
1196 | 787 | push @groups, 'default'; | ||
1197 | 788 | |||
1198 | 789 | foreach my $group ( reverse @groups ) { | ||
1199 | 790 | $usage .= "\n".($group eq 'default' ? 'Options' : $group).":\n\n"; | ||
1200 | 791 | foreach my $opt ( | ||
1201 | 792 | sort { $a->{long} cmp $b->{long} } | ||
1202 | 793 | grep { $_->{group} eq $group } | ||
1203 | 794 | @opts ) | ||
1204 | 795 | { | ||
1205 | 796 | my $long = $opt->{is_negatable} ? "[no]$opt->{long}" : $opt->{long}; | ||
1206 | 797 | my $short = $opt->{short}; | ||
1207 | 798 | my $desc = $opt->{desc}; | ||
1208 | 799 | |||
1209 | 800 | $long .= $opt->{type} ? "=$opt->{type}" : ""; | ||
1210 | 801 | |||
1211 | 802 | if ( $opt->{type} && $opt->{type} eq 'm' ) { | ||
1212 | 803 | my ($s) = $desc =~ m/\(suffix (.)\)/; | ||
1213 | 804 | $s ||= 's'; | ||
1214 | 805 | $desc =~ s/\s+\(suffix .\)//; | ||
1215 | 806 | $desc .= ". Optional suffix s=seconds, m=minutes, h=hours, " | ||
1216 | 807 | . "d=days; if no suffix, $s is used."; | ||
1217 | 808 | } | ||
1218 | 809 | $desc = join("\n$rpad", grep { $_ } $desc =~ m/(.{0,$rcol})(?:\s+|$)/g); | ||
1219 | 810 | $desc =~ s/ +$//mg; | ||
1220 | 811 | if ( $short ) { | ||
1221 | 812 | $usage .= sprintf(" --%-${maxs}s -%s %s\n", $long, $short, $desc); | ||
1222 | 813 | } | ||
1223 | 814 | else { | ||
1224 | 815 | $usage .= sprintf(" --%-${lcol}s %s\n", $long, $desc); | ||
1225 | 816 | } | ||
1226 | 817 | } | ||
1227 | 818 | } | ||
1228 | 819 | |||
1229 | 820 | $usage .= "\nOption types: s=string, i=integer, f=float, h/H/a/A=comma-separated list, d=DSN, z=size, m=time\n"; | ||
1230 | 821 | |||
1231 | 822 | if ( (my @rules = @{$self->{rules}}) ) { | ||
1232 | 823 | $usage .= "\nRules:\n\n"; | ||
1233 | 824 | $usage .= join("\n", map { " $_" } @rules) . "\n"; | ||
1234 | 825 | } | ||
1235 | 826 | if ( $self->{DSNParser} ) { | ||
1236 | 827 | $usage .= "\n" . $self->{DSNParser}->usage(); | ||
1237 | 828 | } | ||
1238 | 829 | $usage .= "\nOptions and values after processing arguments:\n\n"; | ||
1239 | 830 | foreach my $opt ( sort { $a->{long} cmp $b->{long} } @opts ) { | ||
1240 | 831 | my $val = $opt->{value}; | ||
1241 | 832 | my $type = $opt->{type} || ''; | ||
1242 | 833 | my $bool = $opt->{spec} =~ m/^[\w-]+(?:\|[\w-])?!?$/; | ||
1243 | 834 | $val = $bool ? ( $val ? 'TRUE' : 'FALSE' ) | ||
1244 | 835 | : !defined $val ? '(No value)' | ||
1245 | 836 | : $type eq 'd' ? $self->{DSNParser}->as_string($val) | ||
1246 | 837 | : $type =~ m/H|h/ ? join(',', sort keys %$val) | ||
1247 | 838 | : $type =~ m/A|a/ ? join(',', @$val) | ||
1248 | 839 | : $val; | ||
1249 | 840 | $usage .= sprintf(" --%-${lcol}s %s\n", $opt->{long}, $val); | ||
1250 | 841 | } | ||
1251 | 842 | return $usage; | ||
1252 | 843 | } | ||
1253 | 844 | |||
1254 | 845 | sub prompt_noecho { | ||
1255 | 846 | shift @_ if ref $_[0] eq __PACKAGE__; | ||
1256 | 847 | my ( $prompt ) = @_; | ||
1257 | 848 | local $OUTPUT_AUTOFLUSH = 1; | ||
1258 | 849 | print $prompt | ||
1259 | 850 | or die "Cannot print: $OS_ERROR"; | ||
1260 | 851 | my $response; | ||
1261 | 852 | eval { | ||
1262 | 853 | require Term::ReadKey; | ||
1263 | 854 | Term::ReadKey::ReadMode('noecho'); | ||
1264 | 855 | chomp($response = <STDIN>); | ||
1265 | 856 | Term::ReadKey::ReadMode('normal'); | ||
1266 | 857 | print "\n" | ||
1267 | 858 | or die "Cannot print: $OS_ERROR"; | ||
1268 | 859 | }; | ||
1269 | 860 | if ( $EVAL_ERROR ) { | ||
1270 | 861 | die "Cannot read response; is Term::ReadKey installed? $EVAL_ERROR"; | ||
1271 | 862 | } | ||
1272 | 863 | return $response; | ||
1273 | 864 | } | ||
1274 | 865 | |||
1275 | 866 | sub _read_config_file { | ||
1276 | 867 | my ( $self, $filename ) = @_; | ||
1277 | 868 | open my $fh, "<", $filename or die "Cannot open $filename: $OS_ERROR\n"; | ||
1278 | 869 | my @args; | ||
1279 | 870 | my $prefix = '--'; | ||
1280 | 871 | my $parse = 1; | ||
1281 | 872 | |||
1282 | 873 | LINE: | ||
1283 | 874 | while ( my $line = <$fh> ) { | ||
1284 | 875 | chomp $line; | ||
1285 | 876 | next LINE if $line =~ m/^\s*(?:\#|\;|$)/; | ||
1286 | 877 | $line =~ s/\s+#.*$//g; | ||
1287 | 878 | $line =~ s/^\s+|\s+$//g; | ||
1288 | 879 | if ( $line eq '--' ) { | ||
1289 | 880 | $prefix = ''; | ||
1290 | 881 | $parse = 0; | ||
1291 | 882 | next LINE; | ||
1292 | 883 | } | ||
1293 | 884 | if ( $parse | ||
1294 | 885 | && (my($opt, $arg) = $line =~ m/^\s*([^=\s]+?)(?:\s*=\s*(.*?)\s*)?$/) | ||
1295 | 886 | ) { | ||
1296 | 887 | push @args, grep { defined $_ } ("$prefix$opt", $arg); | ||
1297 | 888 | } | ||
1298 | 889 | elsif ( $line =~ m/./ ) { | ||
1299 | 890 | push @args, $line; | ||
1300 | 891 | } | ||
1301 | 892 | else { | ||
1302 | 893 | die "Syntax error in file $filename at line $INPUT_LINE_NUMBER"; | ||
1303 | 894 | } | ||
1304 | 895 | } | ||
1305 | 896 | close $fh; | ||
1306 | 897 | return @args; | ||
1307 | 898 | } | ||
1308 | 899 | |||
1309 | 900 | sub read_para_after { | ||
1310 | 901 | my ( $self, $file, $regex ) = @_; | ||
1311 | 902 | open my $fh, "<", $file or die "Can't open $file: $OS_ERROR"; | ||
1312 | 903 | local $INPUT_RECORD_SEPARATOR = ''; | ||
1313 | 904 | my $para; | ||
1314 | 905 | while ( $para = <$fh> ) { | ||
1315 | 906 | next unless $para =~ m/^=pod$/m; | ||
1316 | 907 | last; | ||
1317 | 908 | } | ||
1318 | 909 | while ( $para = <$fh> ) { | ||
1319 | 910 | next unless $para =~ m/$regex/; | ||
1320 | 911 | last; | ||
1321 | 912 | } | ||
1322 | 913 | $para = <$fh>; | ||
1323 | 914 | chomp($para); | ||
1324 | 915 | close $fh or die "Can't close $file: $OS_ERROR"; | ||
1325 | 916 | return $para; | ||
1326 | 917 | } | ||
1327 | 918 | |||
1328 | 919 | sub clone { | ||
1329 | 920 | my ( $self ) = @_; | ||
1330 | 921 | |||
1331 | 922 | my %clone = map { | ||
1332 | 923 | my $hashref = $self->{$_}; | ||
1333 | 924 | my $val_copy = {}; | ||
1334 | 925 | foreach my $key ( keys %$hashref ) { | ||
1335 | 926 | my $ref = ref $hashref->{$key}; | ||
1336 | 927 | $val_copy->{$key} = !$ref ? $hashref->{$key} | ||
1337 | 928 | : $ref eq 'HASH' ? { %{$hashref->{$key}} } | ||
1338 | 929 | : $ref eq 'ARRAY' ? [ @{$hashref->{$key}} ] | ||
1339 | 930 | : $hashref->{$key}; | ||
1340 | 931 | } | ||
1341 | 932 | $_ => $val_copy; | ||
1342 | 933 | } qw(opts short_opts defaults); | ||
1343 | 934 | |||
1344 | 935 | foreach my $scalar ( qw(got_opts) ) { | ||
1345 | 936 | $clone{$scalar} = $self->{$scalar}; | ||
1346 | 937 | } | ||
1347 | 938 | |||
1348 | 939 | return bless \%clone; | ||
1349 | 940 | } | ||
1350 | 941 | |||
1351 | 942 | sub _parse_size { | ||
1352 | 943 | my ( $self, $opt, $val ) = @_; | ||
1353 | 944 | |||
1354 | 945 | if ( lc($val || '') eq 'null' ) { | ||
1355 | 946 | PTDEBUG && _d('NULL size for', $opt->{long}); | ||
1356 | 947 | $opt->{value} = 'null'; | ||
1357 | 948 | return; | ||
1358 | 949 | } | ||
1359 | 950 | |||
1360 | 951 | my %factor_for = (k => 1_024, M => 1_048_576, G => 1_073_741_824); | ||
1361 | 952 | my ($pre, $num, $factor) = $val =~ m/^([+-])?(\d+)([kMG])?$/; | ||
1362 | 953 | if ( defined $num ) { | ||
1363 | 954 | if ( $factor ) { | ||
1364 | 955 | $num *= $factor_for{$factor}; | ||
1365 | 956 | PTDEBUG && _d('Setting option', $opt->{y}, | ||
1366 | 957 | 'to num', $num, '* factor', $factor); | ||
1367 | 958 | } | ||
1368 | 959 | $opt->{value} = ($pre || '') . $num; | ||
1369 | 960 | } | ||
1370 | 961 | else { | ||
1371 | 962 | $self->save_error("Invalid size for --$opt->{long}: $val"); | ||
1372 | 963 | } | ||
1373 | 964 | return; | ||
1374 | 965 | } | ||
1375 | 966 | |||
1376 | 967 | sub _parse_attribs { | ||
1377 | 968 | my ( $self, $option, $attribs ) = @_; | ||
1378 | 969 | my $types = $self->{types}; | ||
1379 | 970 | return $option | ||
1380 | 971 | . ($attribs->{'short form'} ? '|' . $attribs->{'short form'} : '' ) | ||
1381 | 972 | . ($attribs->{'negatable'} ? '!' : '' ) | ||
1382 | 973 | . ($attribs->{'cumulative'} ? '+' : '' ) | ||
1383 | 974 | . ($attribs->{'type'} ? '=' . $types->{$attribs->{type}} : '' ); | ||
1384 | 975 | } | ||
1385 | 976 | |||
1386 | 977 | sub _parse_synopsis { | ||
1387 | 978 | my ( $self, $file ) = @_; | ||
1388 | 979 | $file ||= $self->{file} || __FILE__; | ||
1389 | 980 | PTDEBUG && _d("Parsing SYNOPSIS in", $file); | ||
1390 | 981 | |||
1391 | 982 | local $INPUT_RECORD_SEPARATOR = ''; # read paragraphs | ||
1392 | 983 | open my $fh, "<", $file or die "Cannot open $file: $OS_ERROR"; | ||
1393 | 984 | my $para; | ||
1394 | 985 | 1 while defined($para = <$fh>) && $para !~ m/^=head1 SYNOPSIS/; | ||
1395 | 986 | die "$file does not contain a SYNOPSIS section" unless $para; | ||
1396 | 987 | my @synop; | ||
1397 | 988 | for ( 1..2 ) { # 1 for the usage, 2 for the description | ||
1398 | 989 | my $para = <$fh>; | ||
1399 | 990 | push @synop, $para; | ||
1400 | 991 | } | ||
1401 | 992 | close $fh; | ||
1402 | 993 | PTDEBUG && _d("Raw SYNOPSIS text:", @synop); | ||
1403 | 994 | my ($usage, $desc) = @synop; | ||
1404 | 995 | die "The SYNOPSIS section in $file is not formatted properly" | ||
1405 | 996 | unless $usage && $desc; | ||
1406 | 997 | |||
1407 | 998 | $usage =~ s/^\s*Usage:\s+(.+)/$1/; | ||
1408 | 999 | chomp $usage; | ||
1409 | 1000 | |||
1410 | 1001 | $desc =~ s/\n/ /g; | ||
1411 | 1002 | $desc =~ s/\s{2,}/ /g; | ||
1412 | 1003 | $desc =~ s/\. ([A-Z][a-z])/. $1/g; | ||
1413 | 1004 | $desc =~ s/\s+$//; | ||
1414 | 1005 | |||
1415 | 1006 | return ( | ||
1416 | 1007 | description => $desc, | ||
1417 | 1008 | usage => $usage, | ||
1418 | 1009 | ); | ||
1419 | 1010 | }; | ||
1420 | 1011 | |||
1421 | 1012 | sub _d { | ||
1422 | 1013 | my ($package, undef, $line) = caller 0; | ||
1423 | 1014 | @_ = map { (my $temp = $_) =~ s/\n/\n# /g; $temp; } | ||
1424 | 1015 | map { defined $_ ? $_ : 'undef' } | ||
1425 | 1016 | @_; | ||
1426 | 1017 | print STDERR "# $package:$line $PID ", join(' ', @_), "\n"; | ||
1427 | 1018 | } | ||
1428 | 1019 | |||
1429 | 1020 | if ( PTDEBUG ) { | ||
1430 | 1021 | print '# ', $^X, ' ', $], "\n"; | ||
1431 | 1022 | if ( my $uname = `uname -a` ) { | ||
1432 | 1023 | $uname =~ s/\s+/ /g; | ||
1433 | 1024 | print "# $uname\n"; | ||
1434 | 1025 | } | ||
1435 | 1026 | print '# Arguments: ', | ||
1436 | 1027 | join(' ', map { my $a = "_[$_]_"; $a =~ s/\n/\n# /g; $a; } @ARGV), "\n"; | ||
1437 | 1028 | } | ||
1438 | 1029 | |||
1439 | 1030 | 1; | ||
1440 | 1031 | } | ||
1441 | 1032 | # ########################################################################### | ||
1442 | 1033 | # End OptionParser package | ||
1443 | 1034 | # ########################################################################### | ||
1444 | 1035 | |||
1445 | 1036 | # ########################################################################### | ||
1446 | 1037 | # QueryParser package | ||
1447 | 1038 | # This package is a copy without comments from the original. The original | ||
1448 | 1039 | # with comments and its test file can be found in the Bazaar repository at, | ||
1449 | 1040 | # lib/QueryParser.pm | ||
1450 | 1041 | # t/lib/QueryParser.t | ||
1451 | 1042 | # See https://launchpad.net/percona-toolkit for more information. | ||
1452 | 1043 | # ########################################################################### | ||
1453 | 1044 | { | ||
1454 | 1045 | package QueryParser; | ||
1455 | 1046 | |||
1456 | 1047 | use strict; | ||
1457 | 1048 | use warnings FATAL => 'all'; | ||
1458 | 1049 | use English qw(-no_match_vars); | ||
1459 | 1050 | use constant PTDEBUG => $ENV{PTDEBUG} || 0; | ||
1460 | 1051 | |||
1461 | 1052 | our $tbl_ident = qr/(?:`[^`]+`|\w+)(?:\.(?:`[^`]+`|\w+))?/; | ||
1462 | 1053 | our $tbl_regex = qr{ | ||
1463 | 1054 | \b(?:FROM|JOIN|(?<!KEY\s)UPDATE|INTO) # Words that precede table names | ||
1464 | 1055 | \b\s* | ||
1465 | 1056 | \(? # Optional paren around tables | ||
1466 | 1057 | ($tbl_ident | ||
1467 | 1058 | (?: (?:\s+ (?:AS\s+)? \w+)?, \s*$tbl_ident )* | ||
1468 | 1059 | ) | ||
1469 | 1060 | }xio; | ||
1470 | 1061 | our $has_derived = qr{ | ||
1471 | 1062 | \b(?:FROM|JOIN|,) | ||
1472 | 1063 | \s*\(\s*SELECT | ||
1473 | 1064 | }xi; | ||
1474 | 1065 | |||
1475 | 1066 | our $data_def_stmts = qr/(?:CREATE|ALTER|TRUNCATE|DROP|RENAME)/i; | ||
1476 | 1067 | |||
1477 | 1068 | our $data_manip_stmts = qr/(?:INSERT|UPDATE|DELETE|REPLACE)/i; | ||
1478 | 1069 | |||
1479 | 1070 | sub new { | ||
1480 | 1071 | my ( $class ) = @_; | ||
1481 | 1072 | bless {}, $class; | ||
1482 | 1073 | } | ||
1483 | 1074 | |||
1484 | 1075 | sub get_tables { | ||
1485 | 1076 | my ( $self, $query ) = @_; | ||
1486 | 1077 | return unless $query; | ||
1487 | 1078 | PTDEBUG && _d('Getting tables for', $query); | ||
1488 | 1079 | |||
1489 | 1080 | my ( $ddl_stmt ) = $query =~ m/^\s*($data_def_stmts)\b/i; | ||
1490 | 1081 | if ( $ddl_stmt ) { | ||
1491 | 1082 | PTDEBUG && _d('Special table type:', $ddl_stmt); | ||
1492 | 1083 | $query =~ s/IF\s+(?:NOT\s+)?EXISTS//i; | ||
1493 | 1084 | if ( $query =~ m/$ddl_stmt DATABASE\b/i ) { | ||
1494 | 1085 | PTDEBUG && _d('Query alters a database, not a table'); | ||
1495 | 1086 | return (); | ||
1496 | 1087 | } | ||
1497 | 1088 | if ( $ddl_stmt =~ m/CREATE/i && $query =~ m/$ddl_stmt\b.+?\bSELECT\b/i ) { | ||
1498 | 1089 | my ($select) = $query =~ m/\b(SELECT\b.+)/is; | ||
1499 | 1090 | PTDEBUG && _d('CREATE TABLE ... SELECT:', $select); | ||
1500 | 1091 | return $self->get_tables($select); | ||
1501 | 1092 | } | ||
1502 | 1093 | my ($tbl) = $query =~ m/TABLE\s+($tbl_ident)(\s+.*)?/i; | ||
1503 | 1094 | PTDEBUG && _d('Matches table:', $tbl); | ||
1504 | 1095 | return ($tbl); | ||
1505 | 1096 | } | ||
1506 | 1097 | |||
1507 | 1098 | $query =~ s/ (?:LOW_PRIORITY|IGNORE|STRAIGHT_JOIN)//ig; | ||
1508 | 1099 | |||
1509 | 1100 | if ( $query =~ /^\s*LOCK TABLES/i ) { | ||
1510 | 1101 | PTDEBUG && _d('Special table type: LOCK TABLES'); | ||
1511 | 1102 | $query =~ s/^(\s*LOCK TABLES\s+)//; | ||
1512 | 1103 | $query =~ s/\s+(?:READ|WRITE|LOCAL)+\s*//g; | ||
1513 | 1104 | PTDEBUG && _d('Locked tables:', $query); | ||
1514 | 1105 | $query = "FROM $query"; | ||
1515 | 1106 | } | ||
1516 | 1107 | |||
1517 | 1108 | $query =~ s/\\["']//g; # quoted strings | ||
1518 | 1109 | $query =~ s/".*?"/?/sg; # quoted strings | ||
1519 | 1110 | $query =~ s/'.*?'/?/sg; # quoted strings | ||
1520 | 1111 | |||
1521 | 1112 | my @tables; | ||
1522 | 1113 | foreach my $tbls ( $query =~ m/$tbl_regex/gio ) { | ||
1523 | 1114 | PTDEBUG && _d('Match tables:', $tbls); | ||
1524 | 1115 | |||
1525 | 1116 | next if $tbls =~ m/\ASELECT\b/i; | ||
1526 | 1117 | |||
1527 | 1118 | foreach my $tbl ( split(',', $tbls) ) { | ||
1528 | 1119 | $tbl =~ s/\s*($tbl_ident)(\s+.*)?/$1/gio; | ||
1529 | 1120 | |||
1530 | 1121 | if ( $tbl !~ m/[a-zA-Z]/ ) { | ||
1531 | 1122 | PTDEBUG && _d('Skipping suspicious table name:', $tbl); | ||
1532 | 1123 | next; | ||
1533 | 1124 | } | ||
1534 | 1125 | |||
1535 | 1126 | push @tables, $tbl; | ||
1536 | 1127 | } | ||
1537 | 1128 | } | ||
1538 | 1129 | return @tables; | ||
1539 | 1130 | } | ||
1540 | 1131 | |||
1541 | 1132 | sub has_derived_table { | ||
1542 | 1133 | my ( $self, $query ) = @_; | ||
1543 | 1134 | my $match = $query =~ m/$has_derived/; | ||
1544 | 1135 | PTDEBUG && _d($query, 'has ' . ($match ? 'a' : 'no') . ' derived table'); | ||
1545 | 1136 | return $match; | ||
1546 | 1137 | } | ||
1547 | 1138 | |||
1548 | 1139 | sub get_aliases { | ||
1549 | 1140 | my ( $self, $query, $list ) = @_; | ||
1550 | 1141 | |||
1551 | 1142 | my $result = { | ||
1552 | 1143 | DATABASE => {}, | ||
1553 | 1144 | TABLE => {}, | ||
1554 | 1145 | }; | ||
1555 | 1146 | return $result unless $query; | ||
1556 | 1147 | |||
1557 | 1148 | $query =~ s/ (?:LOW_PRIORITY|IGNORE|STRAIGHT_JOIN)//ig; | ||
1558 | 1149 | |||
1559 | 1150 | $query =~ s/ (?:INNER|OUTER|CROSS|LEFT|RIGHT|NATURAL)//ig; | ||
1560 | 1151 | |||
1561 | 1152 | my @tbl_refs; | ||
1562 | 1153 | my ($tbl_refs, $from) = $query =~ m{ | ||
1563 | 1154 | ( | ||
1564 | 1155 | (FROM|INTO|UPDATE)\b\s* # Keyword before table refs | ||
1565 | 1156 | .+? # Table refs | ||
1566 | 1157 | ) | ||
1567 | 1158 | (?:\s+|\z) # If the query does not end with the table | ||
1568 | 1159 | (?:WHERE|ORDER|LIMIT|HAVING|SET|VALUES|\z) # Keyword after table refs | ||
1569 | 1160 | }ix; | ||
1570 | 1161 | |||
1571 | 1162 | if ( $tbl_refs ) { | ||
1572 | 1163 | |||
1573 | 1164 | if ( $query =~ m/^(?:INSERT|REPLACE)/i ) { | ||
1574 | 1165 | $tbl_refs =~ s/\([^\)]+\)\s*//; | ||
1575 | 1166 | } | ||
1576 | 1167 | |||
1577 | 1168 | PTDEBUG && _d('tbl refs:', $tbl_refs); | ||
1578 | 1169 | |||
1579 | 1170 | my $before_tbl = qr/(?:,|JOIN|\s|$from)+/i; | ||
1580 | 1171 | |||
1581 | 1172 | my $after_tbl = qr/(?:,|JOIN|ON|USING|\z)/i; | ||
1582 | 1173 | |||
1583 | 1174 | $tbl_refs =~ s/ = /=/g; | ||
1584 | 1175 | |||
1585 | 1176 | while ( | ||
1586 | 1177 | $tbl_refs =~ m{ | ||
1587 | 1178 | $before_tbl\b\s* | ||
1588 | 1179 | ( ($tbl_ident) (?:\s+ (?:AS\s+)? (\w+))? ) | ||
1589 | 1180 | \s*$after_tbl | ||
1590 | 1181 | }xgio ) | ||
1591 | 1182 | { | ||
1592 | 1183 | my ( $tbl_ref, $db_tbl, $alias ) = ($1, $2, $3); | ||
1593 | 1184 | PTDEBUG && _d('Match table:', $tbl_ref); | ||
1594 | 1185 | push @tbl_refs, $tbl_ref; | ||
1595 | 1186 | $alias = $self->trim_identifier($alias); | ||
1596 | 1187 | |||
1597 | 1188 | if ( $tbl_ref =~ m/^AS\s+\w+/i ) { | ||
1598 | 1189 | PTDEBUG && _d('Subquery', $tbl_ref); | ||
1599 | 1190 | $result->{TABLE}->{$alias} = undef; | ||
1600 | 1191 | next; | ||
1601 | 1192 | } | ||
1602 | 1193 | |||
1603 | 1194 | my ( $db, $tbl ) = $db_tbl =~ m/^(?:(.*?)\.)?(.*)/; | ||
1604 | 1195 | $db = $self->trim_identifier($db); | ||
1605 | 1196 | $tbl = $self->trim_identifier($tbl); | ||
1606 | 1197 | $result->{TABLE}->{$alias || $tbl} = $tbl; | ||
1607 | 1198 | $result->{DATABASE}->{$tbl} = $db if $db; | ||
1608 | 1199 | } | ||
1609 | 1200 | } | ||
1610 | 1201 | else { | ||
1611 | 1202 | PTDEBUG && _d("No tables ref in", $query); | ||
1612 | 1203 | } | ||
1613 | 1204 | |||
1614 | 1205 | if ( $list ) { | ||
1615 | 1206 | return \@tbl_refs; | ||
1616 | 1207 | } | ||
1617 | 1208 | else { | ||
1618 | 1209 | return $result; | ||
1619 | 1210 | } | ||
1620 | 1211 | } | ||
1621 | 1212 | |||
1622 | 1213 | sub split { | ||
1623 | 1214 | my ( $self, $query ) = @_; | ||
1624 | 1215 | return unless $query; | ||
1625 | 1216 | $query = $self->clean_query($query); | ||
1626 | 1217 | PTDEBUG && _d('Splitting', $query); | ||
1627 | 1218 | |||
1628 | 1219 | my $verbs = qr{SELECT|INSERT|UPDATE|DELETE|REPLACE|UNION|CREATE}i; | ||
1629 | 1220 | |||
1630 | 1221 | my @split_statements = grep { $_ } split(m/\b($verbs\b(?!(?:\s*\()))/io, $query); | ||
1631 | 1222 | |||
1632 | 1223 | my @statements; | ||
1633 | 1224 | if ( @split_statements == 1 ) { | ||
1634 | 1225 | push @statements, $query; | ||
1635 | 1226 | } | ||
1636 | 1227 | else { | ||
1637 | 1228 | for ( my $i = 0; $i <= $#split_statements; $i += 2 ) { | ||
1638 | 1229 | push @statements, $split_statements[$i].$split_statements[$i+1]; | ||
1639 | 1230 | |||
1640 | 1231 | if ( $statements[-2] && $statements[-2] =~ m/on duplicate key\s+$/i ) { | ||
1641 | 1232 | $statements[-2] .= pop @statements; | ||
1642 | 1233 | } | ||
1643 | 1234 | } | ||
1644 | 1235 | } | ||
1645 | 1236 | |||
1646 | 1237 | PTDEBUG && _d('statements:', map { $_ ? "<$_>" : 'none' } @statements); | ||
1647 | 1238 | return @statements; | ||
1648 | 1239 | } | ||
1649 | 1240 | |||
1650 | 1241 | sub clean_query { | ||
1651 | 1242 | my ( $self, $query ) = @_; | ||
1652 | 1243 | return unless $query; | ||
1653 | 1244 | $query =~ s!/\*.*?\*/! !g; # Remove /* comment blocks */ | ||
1654 | 1245 | $query =~ s/^\s+//; # Remove leading spaces | ||
1655 | 1246 | $query =~ s/\s+$//; # Remove trailing spaces | ||
1656 | 1247 | $query =~ s/\s{2,}/ /g; # Remove extra spaces | ||
1657 | 1248 | return $query; | ||
1658 | 1249 | } | ||
1659 | 1250 | |||
1660 | 1251 | sub split_subquery { | ||
1661 | 1252 | my ( $self, $query ) = @_; | ||
1662 | 1253 | return unless $query; | ||
1663 | 1254 | $query = $self->clean_query($query); | ||
1664 | 1255 | $query =~ s/;$//; | ||
1665 | 1256 | |||
1666 | 1257 | my @subqueries; | ||
1667 | 1258 | my $sqno = 0; # subquery number | ||
1668 | 1259 | my $pos = 0; | ||
1669 | 1260 | while ( $query =~ m/(\S+)(?:\s+|\Z)/g ) { | ||
1670 | 1261 | $pos = pos($query); | ||
1671 | 1262 | my $word = $1; | ||
1672 | 1263 | PTDEBUG && _d($word, $sqno); | ||
1673 | 1264 | if ( $word =~ m/^\(?SELECT\b/i ) { | ||
1674 | 1265 | my $start_pos = $pos - length($word) - 1; | ||
1675 | 1266 | if ( $start_pos ) { | ||
1676 | 1267 | $sqno++; | ||
1677 | 1268 | PTDEBUG && _d('Subquery', $sqno, 'starts at', $start_pos); | ||
1678 | 1269 | $subqueries[$sqno] = { | ||
1679 | 1270 | start_pos => $start_pos, | ||
1680 | 1271 | end_pos => 0, | ||
1681 | 1272 | len => 0, | ||
1682 | 1273 | words => [$word], | ||
1683 | 1274 | lp => 1, # left parentheses | ||
1684 | 1275 | rp => 0, # right parentheses | ||
1685 | 1276 | done => 0, | ||
1686 | 1277 | }; | ||
1687 | 1278 | } | ||
1688 | 1279 | else { | ||
1689 | 1280 | PTDEBUG && _d('Main SELECT at pos 0'); | ||
1690 | 1281 | } | ||
1691 | 1282 | } | ||
1692 | 1283 | else { | ||
1693 | 1284 | next unless $sqno; # next unless we're in a subquery | ||
1694 | 1285 | PTDEBUG && _d('In subquery', $sqno); | ||
1695 | 1286 | my $sq = $subqueries[$sqno]; | ||
1696 | 1287 | if ( $sq->{done} ) { | ||
1697 | 1288 | PTDEBUG && _d('This subquery is done; SQL is for', | ||
1698 | 1289 | ($sqno - 1 ? "subquery $sqno" : "the main SELECT")); | ||
1699 | 1290 | next; | ||
1700 | 1291 | } | ||
1701 | 1292 | push @{$sq->{words}}, $word; | ||
1702 | 1293 | my $lp = ($word =~ tr/\(//) || 0; | ||
1703 | 1294 | my $rp = ($word =~ tr/\)//) || 0; | ||
1704 | 1295 | PTDEBUG && _d('parentheses left', $lp, 'right', $rp); | ||
1705 | 1296 | if ( ($sq->{lp} + $lp) - ($sq->{rp} + $rp) == 0 ) { | ||
1706 | 1297 | my $end_pos = $pos - 1; | ||
1707 | 1298 | PTDEBUG && _d('Subquery', $sqno, 'ends at', $end_pos); | ||
1708 | 1299 | $sq->{end_pos} = $end_pos; | ||
1709 | 1300 | $sq->{len} = $end_pos - $sq->{start_pos}; | ||
1710 | 1301 | } | ||
1711 | 1302 | } | ||
1712 | 1303 | } | ||
1713 | 1304 | |||
1714 | 1305 | for my $i ( 1..$#subqueries ) { | ||
1715 | 1306 | my $sq = $subqueries[$i]; | ||
1716 | 1307 | next unless $sq; | ||
1717 | 1308 | $sq->{sql} = join(' ', @{$sq->{words}}); | ||
1718 | 1309 | substr $query, | ||
1719 | 1310 | $sq->{start_pos} + 1, # +1 for ( | ||
1720 | 1311 | $sq->{len} - 1, # -1 for ) | ||
1721 | 1312 | "__subquery_$i"; | ||
1722 | 1313 | } | ||
1723 | 1314 | |||
1724 | 1315 | return $query, map { $_->{sql} } grep { defined $_ } @subqueries; | ||
1725 | 1316 | } | ||
1726 | 1317 | |||
1727 | 1318 | sub query_type { | ||
1728 | 1319 | my ( $self, $query, $qr ) = @_; | ||
1729 | 1320 | my ($type, undef) = $qr->distill_verbs($query); | ||
1730 | 1321 | my $rw; | ||
1731 | 1322 | if ( $type =~ m/^SELECT\b/ ) { | ||
1732 | 1323 | $rw = 'read'; | ||
1733 | 1324 | } | ||
1734 | 1325 | elsif ( $type =~ m/^$data_manip_stmts\b/ | ||
1735 | 1326 | || $type =~ m/^$data_def_stmts\b/ ) { | ||
1736 | 1327 | $rw = 'write' | ||
1737 | 1328 | } | ||
1738 | 1329 | |||
1739 | 1330 | return { | ||
1740 | 1331 | type => $type, | ||
1741 | 1332 | rw => $rw, | ||
1742 | 1333 | } | ||
1743 | 1334 | } | ||
1744 | 1335 | |||
1745 | 1336 | sub get_columns { | ||
1746 | 1337 | my ( $self, $query ) = @_; | ||
1747 | 1338 | my $cols = []; | ||
1748 | 1339 | return $cols unless $query; | ||
1749 | 1340 | my $cols_def; | ||
1750 | 1341 | |||
1751 | 1342 | if ( $query =~ m/^SELECT/i ) { | ||
1752 | 1343 | $query =~ s/ | ||
1753 | 1344 | ^SELECT\s+ | ||
1754 | 1345 | (?:ALL | ||
1755 | 1346 | |DISTINCT | ||
1756 | 1347 | |DISTINCTROW | ||
1757 | 1348 | |HIGH_PRIORITY | ||
1758 | 1349 | |STRAIGHT_JOIN | ||
1759 | 1350 | |SQL_SMALL_RESULT | ||
1760 | 1351 | |SQL_BIG_RESULT | ||
1761 | 1352 | |SQL_BUFFER_RESULT | ||
1762 | 1353 | |SQL_CACHE | ||
1763 | 1354 | |SQL_NO_CACHE | ||
1764 | 1355 | |SQL_CALC_FOUND_ROWS | ||
1765 | 1356 | )\s+ | ||
1766 | 1357 | /SELECT /xgi; | ||
1767 | 1358 | ($cols_def) = $query =~ m/^SELECT\s+(.+?)\s+FROM/i; | ||
1768 | 1359 | } | ||
1769 | 1360 | elsif ( $query =~ m/^(?:INSERT|REPLACE)/i ) { | ||
1770 | 1361 | ($cols_def) = $query =~ m/\(([^\)]+)\)\s*VALUE/i; | ||
1771 | 1362 | } | ||
1772 | 1363 | |||
1773 | 1364 | PTDEBUG && _d('Columns:', $cols_def); | ||
1774 | 1365 | if ( $cols_def ) { | ||
1775 | 1366 | @$cols = split(',', $cols_def); | ||
1776 | 1367 | map { | ||
1777 | 1368 | my $col = $_; | ||
1778 | 1369 | $col = s/^\s+//g; | ||
1779 | 1370 | $col = s/\s+$//g; | ||
1780 | 1371 | $col; | ||
1781 | 1372 | } @$cols; | ||
1782 | 1373 | } | ||
1783 | 1374 | |||
1784 | 1375 | return $cols; | ||
1785 | 1376 | } | ||
1786 | 1377 | |||
1787 | 1378 | sub parse { | ||
1788 | 1379 | my ( $self, $query ) = @_; | ||
1789 | 1380 | return unless $query; | ||
1790 | 1381 | my $parsed = {}; | ||
1791 | 1382 | |||
1792 | 1383 | $query =~ s/\n/ /g; | ||
1793 | 1384 | $query = $self->clean_query($query); | ||
1794 | 1385 | |||
1795 | 1386 | $parsed->{query} = $query, | ||
1796 | 1387 | $parsed->{tables} = $self->get_aliases($query, 1); | ||
1797 | 1388 | $parsed->{columns} = $self->get_columns($query); | ||
1798 | 1389 | |||
1799 | 1390 | my ($type) = $query =~ m/^(\w+)/; | ||
1800 | 1391 | $parsed->{type} = lc $type; | ||
1801 | 1392 | |||
1802 | 1393 | |||
1803 | 1394 | $parsed->{sub_queries} = []; | ||
1804 | 1395 | |||
1805 | 1396 | return $parsed; | ||
1806 | 1397 | } | ||
1807 | 1398 | |||
1808 | 1399 | sub extract_tables { | ||
1809 | 1400 | my ( $self, %args ) = @_; | ||
1810 | 1401 | my $query = $args{query}; | ||
1811 | 1402 | my $default_db = $args{default_db}; | ||
1812 | 1403 | my $q = $self->{Quoter} || $args{Quoter}; | ||
1813 | 1404 | return unless $query; | ||
1814 | 1405 | PTDEBUG && _d('Extracting tables'); | ||
1815 | 1406 | my @tables; | ||
1816 | 1407 | my %seen; | ||
1817 | 1408 | foreach my $db_tbl ( $self->get_tables($query) ) { | ||
1818 | 1409 | next unless $db_tbl; | ||
1819 | 1410 | next if $seen{$db_tbl}++; # Unique-ify for issue 337. | ||
1820 | 1411 | my ( $db, $tbl ) = $q->split_unquote($db_tbl); | ||
1821 | 1412 | push @tables, [ $db || $default_db, $tbl ]; | ||
1822 | 1413 | } | ||
1823 | 1414 | return @tables; | ||
1824 | 1415 | } | ||
1825 | 1416 | |||
1826 | 1417 | sub trim_identifier { | ||
1827 | 1418 | my ($self, $str) = @_; | ||
1828 | 1419 | return unless defined $str; | ||
1829 | 1420 | $str =~ s/`//g; | ||
1830 | 1421 | $str =~ s/^\s+//; | ||
1831 | 1422 | $str =~ s/\s+$//; | ||
1832 | 1423 | return $str; | ||
1833 | 1424 | } | ||
1834 | 1425 | |||
1835 | 1426 | sub _d { | ||
1836 | 1427 | my ($package, undef, $line) = caller 0; | ||
1837 | 1428 | @_ = map { (my $temp = $_) =~ s/\n/\n# /g; $temp; } | ||
1838 | 1429 | map { defined $_ ? $_ : 'undef' } | ||
1839 | 1430 | @_; | ||
1840 | 1431 | print STDERR "# $package:$line $PID ", join(' ', @_), "\n"; | ||
1841 | 1432 | } | ||
1842 | 1433 | |||
1843 | 1434 | 1; | ||
1844 | 1435 | } | ||
1845 | 1436 | # ########################################################################### | ||
1846 | 1437 | # End QueryParser package | ||
1847 | 1438 | # ########################################################################### | ||
1848 | 1439 | |||
1849 | 1440 | # ########################################################################### | ||
1850 | 1441 | # QueryRewriter package | ||
1851 | 1442 | # This package is a copy without comments from the original. The original | ||
1852 | 1443 | # with comments and its test file can be found in the Bazaar repository at, | ||
1853 | 1444 | # lib/QueryRewriter.pm | ||
1854 | 1445 | # t/lib/QueryRewriter.t | ||
1855 | 1446 | # See https://launchpad.net/percona-toolkit for more information. | ||
1856 | 1447 | # ########################################################################### | ||
1857 | 1448 | { | ||
1858 | 1449 | package QueryRewriter; | ||
1859 | 1450 | |||
1860 | 1451 | use strict; | ||
1861 | 1452 | use warnings FATAL => 'all'; | ||
1862 | 1453 | use English qw(-no_match_vars); | ||
1863 | 1454 | use constant PTDEBUG => $ENV{PTDEBUG} || 0; | ||
1864 | 1455 | |||
1865 | 1456 | our $verbs = qr{^SHOW|^FLUSH|^COMMIT|^ROLLBACK|^BEGIN|SELECT|INSERT | ||
1866 | 1457 | |UPDATE|DELETE|REPLACE|^SET|UNION|^START|^LOCK}xi; | ||
1867 | 1458 | my $quote_re = qr/"(?:(?!(?<!\\)").)*"|'(?:(?!(?<!\\)').)*'/; # Costly! | ||
1868 | 1459 | my $bal; | ||
1869 | 1460 | $bal = qr/ | ||
1870 | 1461 | \( | ||
1871 | 1462 | (?: | ||
1872 | 1463 | (?> [^()]+ ) # Non-parens without backtracking | ||
1873 | 1464 | | | ||
1874 | 1465 | (??{ $bal }) # Group with matching parens | ||
1875 | 1466 | )* | ||
1876 | 1467 | \) | ||
1877 | 1468 | /x; | ||
1878 | 1469 | |||
1879 | 1470 | my $olc_re = qr/(?:--|#)[^'"\r\n]*(?=[\r\n]|\Z)/; # One-line comments | ||
1880 | 1471 | my $mlc_re = qr#/\*[^!].*?\*/#sm; # But not /*!version */ | ||
1881 | 1472 | my $vlc_re = qr#/\*.*?[0-9+].*?\*/#sm; # For SHOW + /*!version */ | ||
1882 | 1473 | my $vlc_rf = qr#^(SHOW).*?/\*![0-9+].*?\*/#sm; # Variation for SHOW | ||
1883 | 1474 | |||
1884 | 1475 | |||
1885 | 1476 | sub new { | ||
1886 | 1477 | my ( $class, %args ) = @_; | ||
1887 | 1478 | my $self = { %args }; | ||
1888 | 1479 | return bless $self, $class; | ||
1889 | 1480 | } | ||
1890 | 1481 | |||
1891 | 1482 | sub strip_comments { | ||
1892 | 1483 | my ( $self, $query ) = @_; | ||
1893 | 1484 | return unless $query; | ||
1894 | 1485 | $query =~ s/$olc_re//go; | ||
1895 | 1486 | $query =~ s/$mlc_re//go; | ||
1896 | 1487 | if ( $query =~ m/$vlc_rf/i ) { # contains show + version | ||
1897 | 1488 | $query =~ s/$vlc_re//go; | ||
1898 | 1489 | } | ||
1899 | 1490 | return $query; | ||
1900 | 1491 | } | ||
1901 | 1492 | |||
1902 | 1493 | sub shorten { | ||
1903 | 1494 | my ( $self, $query, $length ) = @_; | ||
1904 | 1495 | $query =~ s{ | ||
1905 | 1496 | \A( | ||
1906 | 1497 | (?:INSERT|REPLACE) | ||
1907 | 1498 | (?:\s+LOW_PRIORITY|DELAYED|HIGH_PRIORITY|IGNORE)? | ||
1908 | 1499 | (?:\s\w+)*\s+\S+\s+VALUES\s*\(.*?\) | ||
1909 | 1500 | ) | ||
1910 | 1501 | \s*,\s*\(.*?(ON\s+DUPLICATE|\Z)} | ||
1911 | 1502 | {$1 /*... omitted ...*/$2}xsi; | ||
1912 | 1503 | |||
1913 | 1504 | return $query unless $query =~ m/IN\s*\(\s*(?!select)/i; | ||
1914 | 1505 | |||
1915 | 1506 | my $last_length = 0; | ||
1916 | 1507 | my $query_length = length($query); | ||
1917 | 1508 | while ( | ||
1918 | 1509 | $length > 0 | ||
1919 | 1510 | && $query_length > $length | ||
1920 | 1511 | && $query_length < ( $last_length || $query_length + 1 ) | ||
1921 | 1512 | ) { | ||
1922 | 1513 | $last_length = $query_length; | ||
1923 | 1514 | $query =~ s{ | ||
1924 | 1515 | (\bIN\s*\() # The opening of an IN list | ||
1925 | 1516 | ([^\)]+) # Contents of the list, assuming no item contains paren | ||
1926 | 1517 | (?=\)) # Close of the list | ||
1927 | 1518 | } | ||
1928 | 1519 | { | ||
1929 | 1520 | $1 . __shorten($2) | ||
1930 | 1521 | }gexsi; | ||
1931 | 1522 | } | ||
1932 | 1523 | |||
1933 | 1524 | return $query; | ||
1934 | 1525 | } | ||
1935 | 1526 | |||
1936 | 1527 | sub __shorten { | ||
1937 | 1528 | my ( $snippet ) = @_; | ||
1938 | 1529 | my @vals = split(/,/, $snippet); | ||
1939 | 1530 | return $snippet unless @vals > 20; | ||
1940 | 1531 | my @keep = splice(@vals, 0, 20); # Remove and save the first 20 items | ||
1941 | 1532 | return | ||
1942 | 1533 | join(',', @keep) | ||
1943 | 1534 | . "/*... omitted " | ||
1944 | 1535 | . scalar(@vals) | ||
1945 | 1536 | . " items ...*/"; | ||
1946 | 1537 | } | ||
1947 | 1538 | |||
1948 | 1539 | sub fingerprint { | ||
1949 | 1540 | my ( $self, $query ) = @_; | ||
1950 | 1541 | |||
1951 | 1542 | $query =~ m#\ASELECT /\*!40001 SQL_NO_CACHE \*/ \* FROM `# # mysqldump query | ||
1952 | 1543 | && return 'mysqldump'; | ||
1953 | 1544 | $query =~ m#/\*\w+\.\w+:[0-9]/[0-9]\*/# # pt-table-checksum, etc query | ||
1954 | 1545 | && return 'percona-toolkit'; | ||
1955 | 1546 | $query =~ m/\Aadministrator command: / | ||
1956 | 1547 | && return $query; | ||
1957 | 1548 | $query =~ m/\A\s*(call\s+\S+)\(/i | ||
1958 | 1549 | && return lc($1); # Warning! $1 used, be careful. | ||
1959 | 1550 | if ( my ($beginning) = $query =~ m/\A((?:INSERT|REPLACE)(?: IGNORE)?\s+INTO.+?VALUES\s*\(.*?\))\s*,\s*\(/is ) { | ||
1960 | 1551 | $query = $beginning; # Shorten multi-value INSERT statements ASAP | ||
1961 | 1552 | } | ||
1962 | 1553 | |||
1963 | 1554 | $query =~ s/$olc_re//go; | ||
1964 | 1555 | $query =~ s/$mlc_re//go; | ||
1965 | 1556 | $query =~ s/\Ause \S+\Z/use ?/i # Abstract the DB in USE | ||
1966 | 1557 | && return $query; | ||
1967 | 1558 | |||
1968 | 1559 | $query =~ s/\\["']//g; # quoted strings | ||
1969 | 1560 | $query =~ s/".*?"/?/sg; # quoted strings | ||
1970 | 1561 | $query =~ s/'.*?'/?/sg; # quoted strings | ||
1971 | 1562 | |||
1972 | 1563 | if ( $self->{match_md5_checksums} ) { | ||
1973 | 1564 | $query =~ s/([._-])[a-f0-9]{32}/$1?/g; | ||
1974 | 1565 | } | ||
1975 | 1566 | |||
1976 | 1567 | if ( !$self->{match_embedded_numbers} ) { | ||
1977 | 1568 | $query =~ s/[0-9+-][0-9a-f.xb+-]*/?/g; | ||
1978 | 1569 | } | ||
1979 | 1570 | else { | ||
1980 | 1571 | $query =~ s/\b[0-9+-][0-9a-f.xb+-]*/?/g; | ||
1981 | 1572 | } | ||
1982 | 1573 | |||
1983 | 1574 | if ( $self->{match_md5_checksums} ) { | ||
1984 | 1575 | $query =~ s/[xb+-]\?/?/g; | ||
1985 | 1576 | } | ||
1986 | 1577 | else { | ||
1987 | 1578 | $query =~ s/[xb.+-]\?/?/g; | ||
1988 | 1579 | } | ||
1989 | 1580 | |||
1990 | 1581 | $query =~ s/\A\s+//; # Chop off leading whitespace | ||
1991 | 1582 | chomp $query; # Kill trailing whitespace | ||
1992 | 1583 | $query =~ tr[ \n\t\r\f][ ]s; # Collapse whitespace | ||
1993 | 1584 | $query = lc $query; | ||
1994 | 1585 | $query =~ s/\bnull\b/?/g; # Get rid of NULLs | ||
1995 | 1586 | $query =~ s{ # Collapse IN and VALUES lists | ||
1996 | 1587 | \b(in|values?)(?:[\s,]*\([\s?,]*\))+ | ||
1997 | 1588 | } | ||
1998 | 1589 | {$1(?+)}gx; | ||
1999 | 1590 | $query =~ s{ # Collapse UNION | ||
2000 | 1591 | \b(select\s.*?)(?:(\sunion(?:\sall)?)\s\1)+ | ||
2001 | 1592 | } | ||
2002 | 1593 | {$1 /*repeat$2*/}xg; | ||
2003 | 1594 | $query =~ s/\blimit \?(?:, ?\?| offset \?)?/limit ?/; # LIMIT | ||
2004 | 1595 | |||
2005 | 1596 | if ( $query =~ m/\bORDER BY /gi ) { # Find, anchor on ORDER BY clause | ||
2006 | 1597 | 1 while $query =~ s/\G(.+?)\s+ASC/$1/gi && pos $query; | ||
2007 | 1598 | } | ||
2008 | 1599 | |||
2009 | 1600 | return $query; | ||
2010 | 1601 | } | ||
2011 | 1602 | |||
2012 | 1603 | sub distill_verbs { | ||
2013 | 1604 | my ( $self, $query ) = @_; | ||
2014 | 1605 | |||
2015 | 1606 | $query =~ m/\A\s*call\s+(\S+)\(/i && return "CALL $1"; | ||
2016 | 1607 | $query =~ m/\A\s*use\s+/ && return "USE"; | ||
2017 | 1608 | $query =~ m/\A\s*UNLOCK TABLES/i && return "UNLOCK"; | ||
2018 | 1609 | $query =~ m/\A\s*xa\s+(\S+)/i && return "XA_$1"; | ||
2019 | 1610 | |||
2020 | 1611 | if ( $query =~ m/\Aadministrator command:/ ) { | ||
2021 | 1612 | $query =~ s/administrator command:/ADMIN/; | ||
2022 | 1613 | $query = uc $query; | ||
2023 | 1614 | return $query; | ||
2024 | 1615 | } | ||
2025 | 1616 | |||
2026 | 1617 | $query = $self->strip_comments($query); | ||
2027 | 1618 | |||
2028 | 1619 | if ( $query =~ m/\A\s*SHOW\s+/i ) { | ||
2029 | 1620 | PTDEBUG && _d($query); | ||
2030 | 1621 | |||
2031 | 1622 | $query = uc $query; | ||
2032 | 1623 | $query =~ s/\s+(?:GLOBAL|SESSION|FULL|STORAGE|ENGINE)\b/ /g; | ||
2033 | 1624 | $query =~ s/\s+COUNT[^)]+\)//g; | ||
2034 | 1625 | |||
2035 | 1626 | $query =~ s/\s+(?:FOR|FROM|LIKE|WHERE|LIMIT|IN)\b.+//ms; | ||
2036 | 1627 | |||
2037 | 1628 | $query =~ s/\A(SHOW(?:\s+\S+){1,2}).*\Z/$1/s; | ||
2038 | 1629 | $query =~ s/\s+/ /g; | ||
2039 | 1630 | PTDEBUG && _d($query); | ||
2040 | 1631 | return $query; | ||
2041 | 1632 | } | ||
2042 | 1633 | |||
2043 | 1634 | eval $QueryParser::data_def_stmts; | ||
2044 | 1635 | eval $QueryParser::tbl_ident; | ||
2045 | 1636 | my ( $dds ) = $query =~ /^\s*($QueryParser::data_def_stmts)\b/i; | ||
2046 | 1637 | if ( $dds) { | ||
2047 | 1638 | my ( $obj ) = $query =~ m/$dds.+(DATABASE|TABLE)\b/i; | ||
2048 | 1639 | $obj = uc $obj if $obj; | ||
2049 | 1640 | PTDEBUG && _d('Data def statment:', $dds, 'obj:', $obj); | ||
2050 | 1641 | my ($db_or_tbl) | ||
2051 | 1642 | = $query =~ m/(?:TABLE|DATABASE)\s+($QueryParser::tbl_ident)(\s+.*)?/i; | ||
2052 | 1643 | PTDEBUG && _d('Matches db or table:', $db_or_tbl); | ||
2053 | 1644 | return uc($dds . ($obj ? " $obj" : '')), $db_or_tbl; | ||
2054 | 1645 | } | ||
2055 | 1646 | |||
2056 | 1647 | my @verbs = $query =~ m/\b($verbs)\b/gio; | ||
2057 | 1648 | @verbs = do { | ||
2058 | 1649 | my $last = ''; | ||
2059 | 1650 | grep { my $pass = $_ ne $last; $last = $_; $pass } map { uc } @verbs; | ||
2060 | 1651 | }; | ||
2061 | 1652 | |||
2062 | 1653 | if ( ($verbs[0] || '') eq 'SELECT' && @verbs > 1 ) { | ||
2063 | 1654 | PTDEBUG && _d("False-positive verbs after SELECT:", @verbs[1..$#verbs]); | ||
2064 | 1655 | my $union = grep { $_ eq 'UNION' } @verbs; | ||
2065 | 1656 | @verbs = $union ? qw(SELECT UNION) : qw(SELECT); | ||
2066 | 1657 | } | ||
2067 | 1658 | |||
2068 | 1659 | my $verb_str = join(q{ }, @verbs); | ||
2069 | 1660 | return $verb_str; | ||
2070 | 1661 | } | ||
2071 | 1662 | |||
2072 | 1663 | sub __distill_tables { | ||
2073 | 1664 | my ( $self, $query, $table, %args ) = @_; | ||
2074 | 1665 | my $qp = $args{QueryParser} || $self->{QueryParser}; | ||
2075 | 1666 | die "I need a QueryParser argument" unless $qp; | ||
2076 | 1667 | |||
2077 | 1668 | my @tables = map { | ||
2078 | 1669 | $_ =~ s/`//g; | ||
2079 | 1670 | $_ =~ s/(_?)[0-9]+/$1?/g; | ||
2080 | 1671 | $_; | ||
2081 | 1672 | } grep { defined $_ } $qp->get_tables($query); | ||
2082 | 1673 | |||
2083 | 1674 | push @tables, $table if $table; | ||
2084 | 1675 | |||
2085 | 1676 | @tables = do { | ||
2086 | 1677 | my $last = ''; | ||
2087 | 1678 | grep { my $pass = $_ ne $last; $last = $_; $pass } @tables; | ||
2088 | 1679 | }; | ||
2089 | 1680 | |||
2090 | 1681 | return @tables; | ||
2091 | 1682 | } | ||
2092 | 1683 | |||
2093 | 1684 | sub distill { | ||
2094 | 1685 | my ( $self, $query, %args ) = @_; | ||
2095 | 1686 | |||
2096 | 1687 | if ( $args{generic} ) { | ||
2097 | 1688 | my ($cmd, $arg) = $query =~ m/^(\S+)\s+(\S+)/; | ||
2098 | 1689 | return '' unless $cmd; | ||
2099 | 1690 | $query = (uc $cmd) . ($arg ? " $arg" : ''); | ||
2100 | 1691 | } | ||
2101 | 1692 | else { | ||
2102 | 1693 | my ($verbs, $table) = $self->distill_verbs($query, %args); | ||
2103 | 1694 | |||
2104 | 1695 | if ( $verbs && $verbs =~ m/^SHOW/ ) { | ||
2105 | 1696 | my %alias_for = qw( | ||
2106 | 1697 | SCHEMA DATABASE | ||
2107 | 1698 | KEYS INDEX | ||
2108 | 1699 | INDEXES INDEX | ||
2109 | 1700 | ); | ||
2110 | 1701 | map { $verbs =~ s/$_/$alias_for{$_}/ } keys %alias_for; | ||
2111 | 1702 | $query = $verbs; | ||
2112 | 1703 | } | ||
2113 | 1704 | else { | ||
2114 | 1705 | my @tables = $self->__distill_tables($query, $table, %args); | ||
2115 | 1706 | $query = join(q{ }, $verbs, @tables); | ||
2116 | 1707 | } | ||
2117 | 1708 | } | ||
2118 | 1709 | |||
2119 | 1710 | if ( $args{trf} ) { | ||
2120 | 1711 | $query = $args{trf}->($query, %args); | ||
2121 | 1712 | } | ||
2122 | 1713 | |||
2123 | 1714 | return $query; | ||
2124 | 1715 | } | ||
2125 | 1716 | |||
2126 | 1717 | sub convert_to_select { | ||
2127 | 1718 | my ( $self, $query ) = @_; | ||
2128 | 1719 | return unless $query; | ||
2129 | 1720 | |||
2130 | 1721 | return if $query =~ m/=\s*\(\s*SELECT /i; | ||
2131 | 1722 | |||
2132 | 1723 | $query =~ s{ | ||
2133 | 1724 | \A.*? | ||
2134 | 1725 | update(?:\s+(?:low_priority|ignore))?\s+(.*?) | ||
2135 | 1726 | \s+set\b(.*?) | ||
2136 | 1727 | (?:\s*where\b(.*?))? | ||
2137 | 1728 | (limit\s*[0-9]+(?:\s*,\s*[0-9]+)?)? | ||
2138 | 1729 | \Z | ||
2139 | 1730 | } | ||
2140 | 1731 | {__update_to_select($1, $2, $3, $4)}exsi | ||
2141 | 1732 | || $query =~ s{ | ||
2142 | 1733 | \A.*? | ||
2143 | 1734 | (?:insert(?:\s+ignore)?|replace)\s+ | ||
2144 | 1735 | .*?\binto\b(.*?)\(([^\)]+)\)\s* | ||
2145 | 1736 | values?\s*(\(.*?\))\s* | ||
2146 | 1737 | (?:\blimit\b|on\s+duplicate\s+key.*)?\s* | ||
2147 | 1738 | \Z | ||
2148 | 1739 | } | ||
2149 | 1740 | {__insert_to_select($1, $2, $3)}exsi | ||
2150 | 1741 | || $query =~ s{ | ||
2151 | 1742 | \A.*? | ||
2152 | 1743 | (?:insert(?:\s+ignore)?|replace)\s+ | ||
2153 | 1744 | (?:.*?\binto)\b(.*?)\s* | ||
2154 | 1745 | set\s+(.*?)\s* | ||
2155 | 1746 | (?:\blimit\b|on\s+duplicate\s+key.*)?\s* | ||
2156 | 1747 | \Z | ||
2157 | 1748 | } | ||
2158 | 1749 | {__insert_to_select_with_set($1, $2)}exsi | ||
2159 | 1750 | || $query =~ s{ | ||
2160 | 1751 | \A.*? | ||
2161 | 1752 | delete\s+(.*?) | ||
2162 | 1753 | \bfrom\b(.*) | ||
2163 | 1754 | \Z | ||
2164 | 1755 | } | ||
2165 | 1756 | {__delete_to_select($1, $2)}exsi; | ||
2166 | 1757 | $query =~ s/\s*on\s+duplicate\s+key\s+update.*\Z//si; | ||
2167 | 1758 | $query =~ s/\A.*?(?=\bSELECT\s*\b)//ism; | ||
2168 | 1759 | return $query; | ||
2169 | 1760 | } | ||
2170 | 1761 | |||
2171 | 1762 | sub convert_select_list { | ||
2172 | 1763 | my ( $self, $query ) = @_; | ||
2173 | 1764 | $query =~ s{ | ||
2174 | 1765 | \A\s*select(.*?)\bfrom\b | ||
2175 | 1766 | } | ||
2176 | 1767 | {$1 =~ m/\*/ ? "select 1 from" : "select isnull(coalesce($1)) from"}exi; | ||
2177 | 1768 | return $query; | ||
2178 | 1769 | } | ||
2179 | 1770 | |||
2180 | 1771 | sub __delete_to_select { | ||
2181 | 1772 | my ( $delete, $join ) = @_; | ||
2182 | 1773 | if ( $join =~ m/\bjoin\b/ ) { | ||
2183 | 1774 | return "select 1 from $join"; | ||
2184 | 1775 | } | ||
2185 | 1776 | return "select * from $join"; | ||
2186 | 1777 | } | ||
2187 | 1778 | |||
2188 | 1779 | sub __insert_to_select { | ||
2189 | 1780 | my ( $tbl, $cols, $vals ) = @_; | ||
2190 | 1781 | PTDEBUG && _d('Args:', @_); | ||
2191 | 1782 | my @cols = split(/,/, $cols); | ||
2192 | 1783 | PTDEBUG && _d('Cols:', @cols); | ||
2193 | 1784 | $vals =~ s/^\(|\)$//g; # Strip leading/trailing parens | ||
2194 | 1785 | my @vals = $vals =~ m/($quote_re|[^,]*${bal}[^,]*|[^,]+)/g; | ||
2195 | 1786 | PTDEBUG && _d('Vals:', @vals); | ||
2196 | 1787 | if ( @cols == @vals ) { | ||
2197 | 1788 | return "select * from $tbl where " | ||
2198 | 1789 | . join(' and ', map { "$cols[$_]=$vals[$_]" } (0..$#cols)); | ||
2199 | 1790 | } | ||
2200 | 1791 | else { | ||
2201 | 1792 | return "select * from $tbl limit 1"; | ||
2202 | 1793 | } | ||
2203 | 1794 | } | ||
2204 | 1795 | |||
2205 | 1796 | sub __insert_to_select_with_set { | ||
2206 | 1797 | my ( $from, $set ) = @_; | ||
2207 | 1798 | $set =~ s/,/ and /g; | ||
2208 | 1799 | return "select * from $from where $set "; | ||
2209 | 1800 | } | ||
2210 | 1801 | |||
2211 | 1802 | sub __update_to_select { | ||
2212 | 1803 | my ( $from, $set, $where, $limit ) = @_; | ||
2213 | 1804 | return "select $set from $from " | ||
2214 | 1805 | . ( $where ? "where $where" : '' ) | ||
2215 | 1806 | . ( $limit ? " $limit " : '' ); | ||
2216 | 1807 | } | ||
2217 | 1808 | |||
2218 | 1809 | sub wrap_in_derived { | ||
2219 | 1810 | my ( $self, $query ) = @_; | ||
2220 | 1811 | return unless $query; | ||
2221 | 1812 | return $query =~ m/\A\s*select/i | ||
2222 | 1813 | ? "select 1 from ($query) as x limit 1" | ||
2223 | 1814 | : $query; | ||
2224 | 1815 | } | ||
2225 | 1816 | |||
2226 | 1817 | sub _d { | ||
2227 | 1818 | my ($package, undef, $line) = caller 0; | ||
2228 | 1819 | @_ = map { (my $temp = $_) =~ s/\n/\n# /g; $temp; } | ||
2229 | 1820 | map { defined $_ ? $_ : 'undef' } | ||
2230 | 1821 | @_; | ||
2231 | 1822 | print STDERR "# $package:$line $PID ", join(' ', @_), "\n"; | ||
2232 | 1823 | } | ||
2233 | 1824 | |||
2234 | 1825 | 1; | ||
2235 | 1826 | } | ||
2236 | 1827 | # ########################################################################### | ||
2237 | 1828 | # End QueryRewriter package | ||
2238 | 1829 | # ########################################################################### | ||
2239 | 1830 | |||
2240 | 1831 | # ########################################################################### | ||
2241 | 1832 | # This is a combination of modules and programs in one -- a runnable module. | ||
2242 | 1833 | # http://www.perl.com/pub/a/2006/07/13/lightning-articles.html?page=last | ||
2243 | 1834 | # Or, look it up in the Camel book on pages 642 and 643 in the 3rd edition. | ||
2244 | 1835 | # | ||
2245 | 1836 | # Check at the end of this package for the call to main() which actually runs | ||
2246 | 1837 | # the program. | ||
2247 | 1838 | # ########################################################################### | ||
2248 | 1839 | package pt_fingerprint; | ||
2249 | 1840 | |||
2250 | 1841 | use English qw(-no_match_vars); | ||
2251 | 1842 | use Data::Dumper; | ||
2252 | 1843 | $Data::Dumper::Indent = 1; | ||
2253 | 1844 | $OUTPUT_AUTOFLUSH = 1; | ||
2254 | 1845 | |||
2255 | 1846 | use constant MKDEBUG => $ENV{MKDEBUG} || 0; | ||
2256 | 1847 | |||
2257 | 1848 | sub main { | ||
2258 | 1849 | @ARGV = @_; # set global ARGV for this package | ||
2259 | 1850 | |||
2260 | 1851 | # ########################################################################## | ||
2261 | 1852 | # Get configuration information. | ||
2262 | 1853 | # ########################################################################## | ||
2263 | 1854 | my $o = new OptionParser(); | ||
2264 | 1855 | $o->get_specs(); | ||
2265 | 1856 | $o->get_opts(); | ||
2266 | 1857 | $o->usage_or_errors(); | ||
2267 | 1858 | |||
2268 | 1859 | my $qp = new QueryParser(); | ||
2269 | 1860 | my $qr = new QueryRewriter( | ||
2270 | 1861 | QueryParser => $qp, | ||
2271 | 1862 | match_md5_checksums => $o->get('match-md5-checksums'), | ||
2272 | 1863 | match_embedded_numbers => $o->get('match-embedded-numbers'), | ||
2273 | 1864 | ); | ||
2274 | 1865 | |||
2275 | 1866 | if ( $o->got('query') ) { | ||
2276 | 1867 | print $qr->fingerprint($o->get('query')), "\n"; | ||
2277 | 1868 | } | ||
2278 | 1869 | else { | ||
2279 | 1870 | local $INPUT_RECORD_SEPARATOR = ";\n"; | ||
2280 | 1871 | while ( <> ) { | ||
2281 | 1872 | my $query = $_; | ||
2282 | 1873 | chomp $query; | ||
2283 | 1874 | $query =~ s/^#.+$//mg; | ||
2284 | 1875 | $query =~ s/^\s+//; | ||
2285 | 1876 | next unless $query =~ m/^\w/; | ||
2286 | 1877 | print $qr->fingerprint($query), "\n"; | ||
2287 | 1878 | } | ||
2288 | 1879 | } | ||
2289 | 1880 | } | ||
2290 | 1881 | |||
2291 | 1882 | # ############################################################################ | ||
2292 | 1883 | # Run the program. | ||
2293 | 1884 | # ############################################################################ | ||
2294 | 1885 | if ( !caller ) { exit main(@ARGV); } | ||
2295 | 1886 | |||
2296 | 1887 | 1; # Because this is a module as well as a script. | ||
2297 | 1888 | |||
2298 | 1889 | # ############################################################################# | ||
2299 | 1890 | # Documentation. | ||
2300 | 1891 | # ############################################################################# | ||
2301 | 1892 | |||
2302 | 1893 | =pod | ||
2303 | 1894 | |||
2304 | 1895 | =head1 NAME | ||
2305 | 1896 | |||
2306 | 1897 | pt-fingerprint - Convert queries into fingerprints. | ||
2307 | 1898 | |||
2308 | 1899 | =head1 SYNOPSIS | ||
2309 | 1900 | |||
2310 | 1901 | Usage: pt-fingerprint [OPTIONS] [FILES] | ||
2311 | 1902 | |||
2312 | 1903 | pt-fingerprint converts queries into fingerprints. With the --query | ||
2313 | 1904 | option, converts the option's value into a fingerprint. With no options, treats | ||
2314 | 1905 | command-line arguments as FILEs and reads and converts semicolon-separated | ||
2315 | 1906 | queries from the FILEs. When FILE is -, it read standard input. | ||
2316 | 1907 | |||
2317 | 1908 | Convert a single query: | ||
2318 | 1909 | |||
2319 | 1910 | pt-fingerprint --query "select a, b, c from users where id = 500" | ||
2320 | 1911 | |||
2321 | 1912 | Convert a file full of queries: | ||
2322 | 1913 | |||
2323 | 1914 | pt-fingerprint /path/to/file.txt | ||
2324 | 1915 | |||
2325 | 1916 | =head1 RISKS | ||
2326 | 1917 | |||
2327 | 1918 | The following section is included to inform users about the potential risks, | ||
2328 | 1919 | whether known or unknown, of using this tool. The two main categories of risks | ||
2329 | 1920 | are those created by the nature of the tool (e.g. read-only tools vs. read-write | ||
2330 | 1921 | tools) and those created by bugs. | ||
2331 | 1922 | |||
2332 | 1923 | The pt-fingerprint tool simply reads data and transforms it, so risks are | ||
2333 | 1924 | minimal. | ||
2334 | 1925 | |||
2335 | 1926 | See also L<"BUGS"> for more information on filing bugs and getting help. | ||
2336 | 1927 | |||
2337 | 1928 | =head1 DESCRIPTION | ||
2338 | 1929 | |||
2339 | 1930 | A query fingerprint is the abstracted form of a query, which makes it possible | ||
2340 | 1931 | to group similar queries together. Abstracting a query removes literal values, | ||
2341 | 1932 | normalizes whitespace, and so on. For example, consider these two queries: | ||
2342 | 1933 | |||
2343 | 1934 | SELECT name, password FROM user WHERE id='12823'; | ||
2344 | 1935 | select name, password from user | ||
2345 | 1936 | where id=5; | ||
2346 | 1937 | |||
2347 | 1938 | Both of those queries will fingerprint to | ||
2348 | 1939 | |||
2349 | 1940 | select name, password from user where id=? | ||
2350 | 1941 | |||
2351 | 1942 | Once the query's fingerprint is known, we can then talk about a query as though | ||
2352 | 1943 | it represents all similar queries. | ||
2353 | 1944 | |||
2354 | 1945 | Query fingerprinting accommodates a great many special cases, which have proven | ||
2355 | 1946 | necessary in the real world. For example, an IN list with 5 literals is really | ||
2356 | 1947 | equivalent to one with 4 literals, so lists of literals are collapsed to a | ||
2357 | 1948 | single one. If you want to understand more about how and why all of these cases | ||
2358 | 1949 | are handled, please review the test cases in the Subversion repository. If you | ||
2359 | 1950 | find something that is not fingerprinted properly, please submit a bug report | ||
2360 | 1951 | with a reproducible test case. Here is a list of transformations during | ||
2361 | 1952 | fingerprinting, which might not be exhaustive: | ||
2362 | 1953 | |||
2363 | 1954 | =over | ||
2364 | 1955 | |||
2365 | 1956 | =item * | ||
2366 | 1957 | |||
2367 | 1958 | Group all SELECT queries from mysqldump together, even if they are against | ||
2368 | 1959 | different tables. Ditto for all of pt-table-checksum's checksum queries. | ||
2369 | 1960 | |||
2370 | 1961 | =item * | ||
2371 | 1962 | |||
2372 | 1963 | Shorten multi-value INSERT statements to a single VALUES() list. | ||
2373 | 1964 | |||
2374 | 1965 | =item * | ||
2375 | 1966 | |||
2376 | 1967 | Strip comments. | ||
2377 | 1968 | |||
2378 | 1969 | =item * | ||
2379 | 1970 | |||
2380 | 1971 | Abstract the databases in USE statements, so all USE statements are grouped | ||
2381 | 1972 | together. | ||
2382 | 1973 | |||
2383 | 1974 | =item * | ||
2384 | 1975 | |||
2385 | 1976 | Replace all literals, such as quoted strings. For efficiency, the code that | ||
2386 | 1977 | replaces literal numbers is somewhat non-selective, and might replace some | ||
2387 | 1978 | things as numbers when they really are not. Hexadecimal literals are also | ||
2388 | 1979 | replaced. NULL is treated as a literal. Numbers embedded in identifiers are | ||
2389 | 1980 | also replaced, so tables named similarly will be fingerprinted to the same | ||
2390 | 1981 | values (e.g. users_2009 and users_2010 will fingerprint identically). | ||
2391 | 1982 | |||
2392 | 1983 | =item * | ||
2393 | 1984 | |||
2394 | 1985 | Collapse all whitespace into a single space. | ||
2395 | 1986 | |||
2396 | 1987 | =item * | ||
2397 | 1988 | |||
2398 | 1989 | Lowercase the entire query. | ||
2399 | 1990 | |||
2400 | 1991 | =item * | ||
2401 | 1992 | |||
2402 | 1993 | Replace all literals inside of IN() and VALUES() lists with a single | ||
2403 | 1994 | placeholder, regardless of cardinality. | ||
2404 | 1995 | |||
2405 | 1996 | =item * | ||
2406 | 1997 | |||
2407 | 1998 | Collapse multiple identical UNION queries into a single one. | ||
2408 | 1999 | |||
2409 | 2000 | =back | ||
2410 | 2001 | |||
2411 | 2002 | =head1 OPTIONS | ||
2412 | 2003 | |||
2413 | 2004 | This tool accepts additional command-line arguments. Refer to the | ||
2414 | 2005 | L<"SYNOPSIS"> and usage information for details. | ||
2415 | 2006 | |||
2416 | 2007 | =over | ||
2417 | 2008 | |||
2418 | 2009 | =item --config | ||
2419 | 2010 | |||
2420 | 2011 | type: Array | ||
2421 | 2012 | |||
2422 | 2013 | Read this comma-separated list of config files; if specified, this must be the | ||
2423 | 2014 | first option on the command line. | ||
2424 | 2015 | |||
2425 | 2016 | =item --help | ||
2426 | 2017 | |||
2427 | 2018 | Show help and exit. | ||
2428 | 2019 | |||
2429 | 2020 | =item --match-embedded-numbers | ||
2430 | 2021 | |||
2431 | 2022 | Match numbers embedded in words and replace as single values. This option | ||
2432 | 2023 | causes the tool to be more careful about matching numbers so that words | ||
2433 | 2024 | with numbers, like C<catch22> are matched and replaced as a single C<?> | ||
2434 | 2025 | placeholder. Otherwise the default number matching pattern will replace | ||
2435 | 2026 | C<catch22> as C<catch?>. | ||
2436 | 2027 | |||
2437 | 2028 | This is helpful if database or table names contain numbers. | ||
2438 | 2029 | |||
2439 | 2030 | =item --match-md5-checksums | ||
2440 | 2031 | |||
2441 | 2032 | Match MD5 checksums and replace as single values. This option causes | ||
2442 | 2033 | the tool to be more careful about matching numbers so that MD5 checksums | ||
2443 | 2034 | like C<fbc5e685a5d3d45aa1d0347fdb7c4d35> are matched and replaced as a | ||
2444 | 2035 | single C<?> placeholder. Otherwise, the default number matching pattern will | ||
2445 | 2036 | replace C<fbc5e685a5d3d45aa1d0347fdb7c4d35> as C<fbc?>. | ||
2446 | 2037 | |||
2447 | 2038 | =item --query | ||
2448 | 2039 | |||
2449 | 2040 | type: string | ||
2450 | 2041 | |||
2451 | 2042 | The query to convert into a fingerprint. | ||
2452 | 2043 | |||
2453 | 2044 | =item --version | ||
2454 | 2045 | |||
2455 | 2046 | Show version and exit. | ||
2456 | 2047 | |||
2457 | 2048 | =back | ||
2458 | 2049 | |||
2459 | 2050 | =head1 ENVIRONMENT | ||
2460 | 2051 | |||
2461 | 2052 | The environment variable C<PTDEBUG> enables verbose debugging output to STDERR. | ||
2462 | 2053 | To enable debugging and capture all output to a file, run the tool like: | ||
2463 | 2054 | |||
2464 | 2055 | PTDEBUG=1 pt-fingerprint ... > FILE 2>&1 | ||
2465 | 2056 | |||
2466 | 2057 | Be careful: debugging output is voluminous and can generate several megabytes | ||
2467 | 2058 | of output. | ||
2468 | 2059 | |||
2469 | 2060 | =head1 SYSTEM REQUIREMENTS | ||
2470 | 2061 | |||
2471 | 2062 | You need Perl, DBI, DBD::mysql, and some core packages that ought to be | ||
2472 | 2063 | installed in any reasonably new version of Perl. | ||
2473 | 2064 | |||
2474 | 2065 | =head1 BUGS | ||
2475 | 2066 | |||
2476 | 2067 | For a list of known bugs, see L<http://www.percona.com/bugs/pt-fingerprint>. | ||
2477 | 2068 | |||
2478 | 2069 | Please report bugs at L<https://bugs.launchpad.net/percona-toolkit>. | ||
2479 | 2070 | Include the following information in your bug report: | ||
2480 | 2071 | |||
2481 | 2072 | =over | ||
2482 | 2073 | |||
2483 | 2074 | =item * Complete command-line used to run the tool | ||
2484 | 2075 | |||
2485 | 2076 | =item * Tool L<"--version"> | ||
2486 | 2077 | |||
2487 | 2078 | =item * MySQL version of all servers involved | ||
2488 | 2079 | |||
2489 | 2080 | =item * Output from the tool including STDERR | ||
2490 | 2081 | |||
2491 | 2082 | =item * Input files (log/dump/config files, etc.) | ||
2492 | 2083 | |||
2493 | 2084 | =back | ||
2494 | 2085 | |||
2495 | 2086 | If possible, include debugging output by running the tool with C<PTDEBUG>; | ||
2496 | 2087 | see L<"ENVIRONMENT">. | ||
2497 | 2088 | |||
2498 | 2089 | =head1 DOWNLOADING | ||
2499 | 2090 | |||
2500 | 2091 | Visit L<http://www.percona.com/software/percona-toolkit/> to download the | ||
2501 | 2092 | latest release of Percona Toolkit. Or, get the latest release from the | ||
2502 | 2093 | command line: | ||
2503 | 2094 | |||
2504 | 2095 | wget percona.com/get/percona-toolkit.tar.gz | ||
2505 | 2096 | |||
2506 | 2097 | wget percona.com/get/percona-toolkit.rpm | ||
2507 | 2098 | |||
2508 | 2099 | wget percona.com/get/percona-toolkit.deb | ||
2509 | 2100 | |||
2510 | 2101 | You can also get individual tools from the latest release: | ||
2511 | 2102 | |||
2512 | 2103 | wget percona.com/get/TOOL | ||
2513 | 2104 | |||
2514 | 2105 | Replace C<TOOL> with the name of any tool. | ||
2515 | 2106 | |||
2516 | 2107 | =head1 AUTHORS | ||
2517 | 2108 | |||
2518 | 2109 | Baron Schwartz and Daniel Nichter | ||
2519 | 2110 | |||
2520 | 2111 | =head1 ABOUT PERCONA TOOLKIT | ||
2521 | 2112 | |||
2522 | 2113 | This tool is part of Percona Toolkit, a collection of advanced command-line | ||
2523 | 2114 | tools developed by Percona for MySQL support and consulting. Percona Toolkit | ||
2524 | 2115 | was forked from two projects in June, 2011: Maatkit and Aspersa. Those | ||
2525 | 2116 | projects were created by Baron Schwartz and developed primarily by him and | ||
2526 | 2117 | Daniel Nichter, both of whom are employed by Percona. Visit | ||
2527 | 2118 | L<http://www.percona.com/software/> for more software developed by Percona. | ||
2528 | 2119 | |||
2529 | 2120 | =head1 COPYRIGHT, LICENSE, AND WARRANTY | ||
2530 | 2121 | |||
2531 | 2122 | This program is copyright 2011-2012 Percona Inc. | ||
2532 | 2123 | Feedback and improvements are welcome. | ||
2533 | 2124 | |||
2534 | 2125 | THIS PROGRAM IS PROVIDED "AS IS" AND WITHOUT ANY EXPRESS OR IMPLIED | ||
2535 | 2126 | WARRANTIES, INCLUDING, WITHOUT LIMITATION, THE IMPLIED WARRANTIES OF | ||
2536 | 2127 | MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE. | ||
2537 | 2128 | |||
2538 | 2129 | This program is free software; you can redistribute it and/or modify it under | ||
2539 | 2130 | the terms of the GNU General Public License as published by the Free Software | ||
2540 | 2131 | Foundation, version 2; OR the Perl Artistic License. On UNIX and similar | ||
2541 | 2132 | systems, you can issue `man perlgpl' or `man perlartistic' to read these | ||
2542 | 2133 | licenses. | ||
2543 | 2134 | |||
2544 | 2135 | You should have received a copy of the GNU General Public License along with | ||
2545 | 2136 | this program; if not, write to the Free Software Foundation, Inc., 59 Temple | ||
2546 | 2137 | Place, Suite 330, Boston, MA 02111-1307 USA. | ||
2547 | 2138 | |||
2548 | 2139 | =head1 VERSION | ||
2549 | 2140 | |||
2550 | 2141 | pt-fingerprint 2.0.0 | ||
2551 | 2142 | |||
2552 | 2143 | =cut | ||
2553 | 0 | 2144 | ||
2554 | === modified file 'bin/pt-index-usage' | |||
2555 | --- bin/pt-index-usage 2012-03-07 23:41:54 +0000 | |||
2556 | +++ bin/pt-index-usage 2012-03-31 16:07:24 +0000 | |||
2557 | @@ -2669,19 +2669,58 @@ | |||
2558 | 2669 | return bless $self, $class; | 2669 | return bless $self, $class; |
2559 | 2670 | } | 2670 | } |
2560 | 2671 | 2671 | ||
2561 | 2672 | sub get_create_table { | ||
2562 | 2673 | my ( $self, $dbh, $db, $tbl ) = @_; | ||
2563 | 2674 | die "I need a dbh parameter" unless $dbh; | ||
2564 | 2675 | die "I need a db parameter" unless $db; | ||
2565 | 2676 | die "I need a tbl parameter" unless $tbl; | ||
2566 | 2677 | my $q = $self->{Quoter}; | ||
2567 | 2678 | |||
2568 | 2679 | my $new_sql_mode | ||
2569 | 2680 | = '/*!40101 SET @OLD_SQL_MODE := @@SQL_MODE, ' | ||
2570 | 2681 | . q{@@SQL_MODE := REPLACE(REPLACE(@@SQL_MODE, 'ANSI_QUOTES', ''), ',,', ','), } | ||
2571 | 2682 | . '@OLD_QUOTE := @@SQL_QUOTE_SHOW_CREATE, ' | ||
2572 | 2683 | . '@@SQL_QUOTE_SHOW_CREATE := 1 */'; | ||
2573 | 2684 | |||
2574 | 2685 | my $old_sql_mode = '/*!40101 SET @@SQL_MODE := @OLD_SQL_MODE, ' | ||
2575 | 2686 | . '@@SQL_QUOTE_SHOW_CREATE := @OLD_QUOTE */'; | ||
2576 | 2687 | |||
2577 | 2688 | PTDEBUG && _d($new_sql_mode); | ||
2578 | 2689 | eval { $dbh->do($new_sql_mode); }; | ||
2579 | 2690 | PTDEBUG && $EVAL_ERROR && _d($EVAL_ERROR); | ||
2580 | 2691 | |||
2581 | 2692 | my $use_sql = 'USE ' . $q->quote($db); | ||
2582 | 2693 | PTDEBUG && _d($dbh, $use_sql); | ||
2583 | 2694 | $dbh->do($use_sql); | ||
2584 | 2695 | |||
2585 | 2696 | my $show_sql = "SHOW CREATE TABLE " . $q->quote($db, $tbl); | ||
2586 | 2697 | PTDEBUG && _d($show_sql); | ||
2587 | 2698 | my $href; | ||
2588 | 2699 | eval { $href = $dbh->selectrow_hashref($show_sql); }; | ||
2589 | 2700 | if ( $EVAL_ERROR ) { | ||
2590 | 2701 | PTDEBUG && _d($EVAL_ERROR); | ||
2591 | 2702 | |||
2592 | 2703 | PTDEBUG && _d($old_sql_mode); | ||
2593 | 2704 | $dbh->do($old_sql_mode); | ||
2594 | 2705 | |||
2595 | 2706 | return; | ||
2596 | 2707 | } | ||
2597 | 2708 | |||
2598 | 2709 | PTDEBUG && _d($old_sql_mode); | ||
2599 | 2710 | $dbh->do($old_sql_mode); | ||
2600 | 2711 | |||
2601 | 2712 | my ($key) = grep { m/create (?:table|view)/i } keys %$href; | ||
2602 | 2713 | if ( !$key ) { | ||
2603 | 2714 | die "Error: no 'Create Table' or 'Create View' in result set from " | ||
2604 | 2715 | . "$show_sql: " . Dumper($href); | ||
2605 | 2716 | } | ||
2606 | 2717 | |||
2607 | 2718 | return $href->{$key}; | ||
2608 | 2719 | } | ||
2609 | 2720 | |||
2610 | 2672 | sub parse { | 2721 | sub parse { |
2611 | 2673 | my ( $self, $ddl, $opts ) = @_; | 2722 | my ( $self, $ddl, $opts ) = @_; |
2612 | 2674 | return unless $ddl; | 2723 | return unless $ddl; |
2613 | 2675 | if ( ref $ddl eq 'ARRAY' ) { | ||
2614 | 2676 | if ( lc $ddl->[0] eq 'table' ) { | ||
2615 | 2677 | $ddl = $ddl->[1]; | ||
2616 | 2678 | } | ||
2617 | 2679 | else { | ||
2618 | 2680 | return { | ||
2619 | 2681 | engine => 'VIEW', | ||
2620 | 2682 | }; | ||
2621 | 2683 | } | ||
2622 | 2684 | } | ||
2623 | 2685 | 2724 | ||
2624 | 2686 | if ( $ddl !~ m/CREATE (?:TEMPORARY )?TABLE `/ ) { | 2725 | if ( $ddl !~ m/CREATE (?:TEMPORARY )?TABLE `/ ) { |
2625 | 2687 | die "Cannot parse table definition; is ANSI quoting " | 2726 | die "Cannot parse table definition; is ANSI quoting " |
2626 | @@ -2988,41 +3027,31 @@ | |||
2627 | 2988 | return $ddl; | 3027 | return $ddl; |
2628 | 2989 | } | 3028 | } |
2629 | 2990 | 3029 | ||
2665 | 2991 | sub remove_secondary_indexes { | 3030 | sub get_table_status { |
2666 | 2992 | my ( $self, $ddl ) = @_; | 3031 | my ( $self, $dbh, $db, $like ) = @_; |
2667 | 2993 | my $sec_indexes_ddl; | 3032 | my $q = $self->{Quoter}; |
2668 | 2994 | my $tbl_struct = $self->parse($ddl); | 3033 | my $sql = "SHOW TABLE STATUS FROM " . $q->quote($db); |
2669 | 2995 | 3034 | my @params; | |
2670 | 2996 | if ( ($tbl_struct->{engine} || '') =~ m/InnoDB/i ) { | 3035 | if ( $like ) { |
2671 | 2997 | my $clustered_key = $tbl_struct->{clustered_key}; | 3036 | $sql .= ' LIKE ?'; |
2672 | 2998 | $clustered_key ||= ''; | 3037 | push @params, $like; |
2673 | 2999 | 3038 | } | |
2674 | 3000 | my @sec_indexes = map { | 3039 | PTDEBUG && _d($sql, @params); |
2675 | 3001 | my $key_def = $_->{ddl}; | 3040 | my $sth = $dbh->prepare($sql); |
2676 | 3002 | $key_def =~ s/([\(\)])/\\$1/g; | 3041 | eval { $sth->execute(@params); }; |
2677 | 3003 | $ddl =~ s/\s+$key_def//i; | 3042 | if ($EVAL_ERROR) { |
2678 | 3004 | 3043 | PTDEBUG && _d($EVAL_ERROR); | |
2679 | 3005 | my $key_ddl = "ADD $_->{ddl}"; | 3044 | return; |
2680 | 3006 | $key_ddl .= ',' unless $key_ddl =~ m/,$/; | 3045 | } |
2681 | 3007 | $key_ddl; | 3046 | my @tables = @{$sth->fetchall_arrayref({})}; |
2682 | 3008 | } | 3047 | @tables = map { |
2683 | 3009 | grep { $_->{name} ne $clustered_key } | 3048 | my %tbl; # Make a copy with lowercased keys |
2684 | 3010 | values %{$tbl_struct->{keys}}; | 3049 | @tbl{ map { lc $_ } keys %$_ } = values %$_; |
2685 | 3011 | PTDEBUG && _d('Secondary indexes:', Dumper(\@sec_indexes)); | 3050 | $tbl{engine} ||= $tbl{type} || $tbl{comment}; |
2686 | 3012 | 3051 | delete $tbl{type}; | |
2687 | 3013 | if ( @sec_indexes ) { | 3052 | \%tbl; |
2688 | 3014 | $sec_indexes_ddl = join(' ', @sec_indexes); | 3053 | } @tables; |
2689 | 3015 | $sec_indexes_ddl =~ s/,$//; | 3054 | return @tables; |
2655 | 3016 | } | ||
2656 | 3017 | |||
2657 | 3018 | $ddl =~ s/,(\n\) )/$1/s; | ||
2658 | 3019 | } | ||
2659 | 3020 | else { | ||
2660 | 3021 | PTDEBUG && _d('Not removing secondary indexes from', | ||
2661 | 3022 | $tbl_struct->{engine}, 'table'); | ||
2662 | 3023 | } | ||
2663 | 3024 | |||
2664 | 3025 | return $ddl, $sec_indexes_ddl, $tbl_struct; | ||
2690 | 3026 | } | 3055 | } |
2691 | 3027 | 3056 | ||
2692 | 3028 | sub _d { | 3057 | sub _d { |
2693 | @@ -3912,7 +3941,7 @@ | |||
2694 | 3912 | 3941 | ||
2695 | 3913 | sub new { | 3942 | sub new { |
2696 | 3914 | my ( $class, %args ) = @_; | 3943 | my ( $class, %args ) = @_; |
2698 | 3915 | my @required_args = qw(OptionParser Quoter); | 3944 | my @required_args = qw(OptionParser TableParser Quoter); |
2699 | 3916 | foreach my $arg ( @required_args ) { | 3945 | foreach my $arg ( @required_args ) { |
2700 | 3917 | die "I need a $arg argument" unless $args{$arg}; | 3946 | die "I need a $arg argument" unless $args{$arg}; |
2701 | 3918 | } | 3947 | } |
2702 | @@ -3921,8 +3950,19 @@ | |||
2703 | 3921 | die "I need either a dbh or file_itr argument" | 3950 | die "I need either a dbh or file_itr argument" |
2704 | 3922 | if (!$dbh && !$file_itr) || ($dbh && $file_itr); | 3951 | if (!$dbh && !$file_itr) || ($dbh && $file_itr); |
2705 | 3923 | 3952 | ||
2706 | 3953 | my %resume; | ||
2707 | 3954 | if ( my $table = $args{resume} ) { | ||
2708 | 3955 | PTDEBUG && _d('Will resume from or after', $table); | ||
2709 | 3956 | my ($db, $tbl) = $args{Quoter}->split_unquote($table); | ||
2710 | 3957 | die "Resume table must be database-qualified: $table" | ||
2711 | 3958 | unless $db && $tbl; | ||
2712 | 3959 | $resume{db} = $db; | ||
2713 | 3960 | $resume{tbl} = $tbl; | ||
2714 | 3961 | } | ||
2715 | 3962 | |||
2716 | 3924 | my $self = { | 3963 | my $self = { |
2717 | 3925 | %args, | 3964 | %args, |
2718 | 3965 | resume => \%resume, | ||
2719 | 3926 | filters => _make_filters(%args), | 3966 | filters => _make_filters(%args), |
2720 | 3927 | }; | 3967 | }; |
2721 | 3928 | 3968 | ||
2722 | @@ -3983,9 +4023,19 @@ | |||
2723 | 3983 | return \%filters; | 4023 | return \%filters; |
2724 | 3984 | } | 4024 | } |
2725 | 3985 | 4025 | ||
2727 | 3986 | sub next_schema_object { | 4026 | sub next { |
2728 | 3987 | my ( $self ) = @_; | 4027 | my ( $self ) = @_; |
2729 | 3988 | 4028 | ||
2730 | 4029 | if ( !$self->{initialized} ) { | ||
2731 | 4030 | $self->{initialized} = 1; | ||
2732 | 4031 | if ( $self->{resume}->{tbl} | ||
2733 | 4032 | && !$self->table_is_allowed(@{$self->{resume}}{qw(db tbl)}) ) { | ||
2734 | 4033 | PTDEBUG && _d('Will resume after', | ||
2735 | 4034 | join('.', @{$self->{resume}}{qw(db tbl)})); | ||
2736 | 4035 | $self->{resume}->{after} = 1; | ||
2737 | 4036 | } | ||
2738 | 4037 | } | ||
2739 | 4038 | |||
2740 | 3989 | my $schema_obj; | 4039 | my $schema_obj; |
2741 | 3990 | if ( $self->{file_itr} ) { | 4040 | if ( $self->{file_itr} ) { |
2742 | 3991 | $schema_obj= $self->_iterate_files(); | 4041 | $schema_obj= $self->_iterate_files(); |
2743 | @@ -3995,24 +4045,18 @@ | |||
2744 | 3995 | } | 4045 | } |
2745 | 3996 | 4046 | ||
2746 | 3997 | if ( $schema_obj ) { | 4047 | if ( $schema_obj ) { |
2747 | 3998 | if ( $schema_obj->{ddl} && $self->{TableParser} ) { | ||
2748 | 3999 | $schema_obj->{tbl_struct} | ||
2749 | 4000 | = $self->{TableParser}->parse($schema_obj->{ddl}); | ||
2750 | 4001 | } | ||
2751 | 4002 | |||
2752 | 4003 | delete $schema_obj->{ddl} unless $self->{keep_ddl}; | ||
2753 | 4004 | |||
2754 | 4005 | if ( my $schema = $self->{Schema} ) { | 4048 | if ( my $schema = $self->{Schema} ) { |
2755 | 4006 | $schema->add_schema_object($schema_obj); | 4049 | $schema->add_schema_object($schema_obj); |
2756 | 4007 | } | 4050 | } |
2757 | 4051 | PTDEBUG && _d('Next schema object:', | ||
2758 | 4052 | $schema_obj->{db}, $schema_obj->{tbl}); | ||
2759 | 4008 | } | 4053 | } |
2760 | 4009 | 4054 | ||
2761 | 4010 | PTDEBUG && _d('Next schema object:', $schema_obj->{db}, $schema_obj->{tbl}); | ||
2762 | 4011 | return $schema_obj; | 4055 | return $schema_obj; |
2763 | 4012 | } | 4056 | } |
2764 | 4013 | 4057 | ||
2765 | 4014 | sub _iterate_files { | 4058 | sub _iterate_files { |
2767 | 4015 | my ( $self ) = @_; | 4059 | my ( $self ) = @_; |
2768 | 4016 | 4060 | ||
2769 | 4017 | if ( !$self->{fh} ) { | 4061 | if ( !$self->{fh} ) { |
2770 | 4018 | my ($fh, $file) = $self->{file_itr}->(); | 4062 | my ($fh, $file) = $self->{file_itr}->(); |
2771 | @@ -4033,7 +4077,8 @@ | |||
2772 | 4033 | my $db = $1; # XXX | 4077 | my $db = $1; # XXX |
2773 | 4034 | $db =~ s/^`//; # strip leading ` | 4078 | $db =~ s/^`//; # strip leading ` |
2774 | 4035 | $db =~ s/`$//; # and trailing ` | 4079 | $db =~ s/`$//; # and trailing ` |
2776 | 4036 | if ( $self->database_is_allowed($db) ) { | 4080 | if ( $self->database_is_allowed($db) |
2777 | 4081 | && $self->_resume_from_database($db) ) { | ||
2778 | 4037 | $self->{db} = $db; | 4082 | $self->{db} = $db; |
2779 | 4038 | } | 4083 | } |
2780 | 4039 | } | 4084 | } |
2781 | @@ -4046,21 +4091,22 @@ | |||
2782 | 4046 | my ($tbl) = $chunk =~ m/$tbl_name/; | 4091 | my ($tbl) = $chunk =~ m/$tbl_name/; |
2783 | 4047 | $tbl =~ s/^\s*`//; | 4092 | $tbl =~ s/^\s*`//; |
2784 | 4048 | $tbl =~ s/`\s*$//; | 4093 | $tbl =~ s/`\s*$//; |
2786 | 4049 | if ( $self->table_is_allowed($self->{db}, $tbl) ) { | 4094 | if ( $self->_resume_from_table($tbl) |
2787 | 4095 | && $self->table_is_allowed($self->{db}, $tbl) ) { | ||
2788 | 4050 | my ($ddl) = $chunk =~ m/^(?:$open_comment)?(CREATE TABLE.+?;)$/ms; | 4096 | my ($ddl) = $chunk =~ m/^(?:$open_comment)?(CREATE TABLE.+?;)$/ms; |
2789 | 4051 | if ( !$ddl ) { | 4097 | if ( !$ddl ) { |
2790 | 4052 | warn "Failed to parse CREATE TABLE from\n" . $chunk; | 4098 | warn "Failed to parse CREATE TABLE from\n" . $chunk; |
2791 | 4053 | next CHUNK; | 4099 | next CHUNK; |
2792 | 4054 | } | 4100 | } |
2793 | 4055 | $ddl =~ s/ \*\/;\Z/;/; # remove end of version comment | 4101 | $ddl =~ s/ \*\/;\Z/;/; # remove end of version comment |
2798 | 4056 | 4102 | my $tbl_struct = $self->{TableParser}->parse($ddl); | |
2799 | 4057 | my ($engine) = $ddl =~ m/\).*?(?:ENGINE|TYPE)=(\w+)/; | 4103 | if ( $self->engine_is_allowed($tbl_struct->{engine}) ) { |
2796 | 4058 | |||
2797 | 4059 | if ( !$engine || $self->engine_is_allowed($engine) ) { | ||
2800 | 4060 | return { | 4104 | return { |
2804 | 4061 | db => $self->{db}, | 4105 | db => $self->{db}, |
2805 | 4062 | tbl => $tbl, | 4106 | tbl => $tbl, |
2806 | 4063 | ddl => $ddl, | 4107 | name => $self->{Quoter}->quote($self->{db}, $tbl), |
2807 | 4108 | ddl => $ddl, | ||
2808 | 4109 | tbl_struct => $tbl_struct, | ||
2809 | 4064 | }; | 4110 | }; |
2810 | 4065 | } | 4111 | } |
2811 | 4066 | } | 4112 | } |
2812 | @@ -4077,6 +4123,7 @@ | |||
2813 | 4077 | sub _iterate_dbh { | 4123 | sub _iterate_dbh { |
2814 | 4078 | my ( $self ) = @_; | 4124 | my ( $self ) = @_; |
2815 | 4079 | my $q = $self->{Quoter}; | 4125 | my $q = $self->{Quoter}; |
2816 | 4126 | my $tp = $self->{TableParser}; | ||
2817 | 4080 | my $dbh = $self->{dbh}; | 4127 | my $dbh = $self->{dbh}; |
2818 | 4081 | PTDEBUG && _d('Getting next schema object from dbh', $dbh); | 4128 | PTDEBUG && _d('Getting next schema object from dbh', $dbh); |
2819 | 4082 | 4129 | ||
2820 | @@ -4090,7 +4137,9 @@ | |||
2821 | 4090 | } | 4137 | } |
2822 | 4091 | 4138 | ||
2823 | 4092 | if ( !$self->{db} ) { | 4139 | if ( !$self->{db} ) { |
2825 | 4093 | $self->{db} = shift @{$self->{dbs}}; | 4140 | do { |
2826 | 4141 | $self->{db} = shift @{$self->{dbs}}; | ||
2827 | 4142 | } until $self->_resume_from_database($self->{db}); | ||
2828 | 4094 | PTDEBUG && _d('Next database:', $self->{db}); | 4143 | PTDEBUG && _d('Next database:', $self->{db}); |
2829 | 4095 | return unless $self->{db}; | 4144 | return unless $self->{db}; |
2830 | 4096 | } | 4145 | } |
2831 | @@ -4103,8 +4152,9 @@ | |||
2832 | 4103 | } | 4152 | } |
2833 | 4104 | grep { | 4153 | grep { |
2834 | 4105 | my ($tbl, $type) = @$_; | 4154 | my ($tbl, $type) = @$_; |
2837 | 4106 | $self->table_is_allowed($self->{db}, $tbl) | 4155 | (!$type || ($type ne 'VIEW')) |
2838 | 4107 | && (!$type || ($type ne 'VIEW')); | 4156 | && $self->_resume_from_table($tbl) |
2839 | 4157 | && $self->table_is_allowed($self->{db}, $tbl); | ||
2840 | 4108 | } | 4158 | } |
2841 | 4109 | @{$dbh->selectall_arrayref($sql)}; | 4159 | @{$dbh->selectall_arrayref($sql)}; |
2842 | 4110 | PTDEBUG && _d('Found', scalar @tbls, 'tables in database', $self->{db}); | 4160 | PTDEBUG && _d('Found', scalar @tbls, 'tables in database', $self->{db}); |
2843 | @@ -4112,27 +4162,15 @@ | |||
2844 | 4112 | } | 4162 | } |
2845 | 4113 | 4163 | ||
2846 | 4114 | while ( my $tbl = shift @{$self->{tbls}} ) { | 4164 | while ( my $tbl = shift @{$self->{tbls}} ) { |
2864 | 4115 | my $engine; | 4165 | my $ddl = $tp->get_create_table($dbh, $self->{db}, $tbl); |
2865 | 4116 | if ( $self->{filters}->{'engines'} | 4166 | my $tbl_struct = $tp->parse($ddl); |
2866 | 4117 | || $self->{filters}->{'ignore-engines'} ) { | 4167 | if ( $self->engine_is_allowed($tbl_struct->{engine}) ) { |
2850 | 4118 | my $sql = "SHOW TABLE STATUS FROM " . $q->quote($self->{db}) | ||
2851 | 4119 | . " LIKE \'$tbl\'"; | ||
2852 | 4120 | PTDEBUG && _d($sql); | ||
2853 | 4121 | $engine = $dbh->selectrow_hashref($sql)->{engine}; | ||
2854 | 4122 | PTDEBUG && _d($tbl, 'uses', $engine, 'engine'); | ||
2855 | 4123 | } | ||
2856 | 4124 | |||
2857 | 4125 | |||
2858 | 4126 | if ( !$engine || $self->engine_is_allowed($engine) ) { | ||
2859 | 4127 | my $ddl; | ||
2860 | 4128 | if ( my $du = $self->{MySQLDump} ) { | ||
2861 | 4129 | $ddl = $du->get_create_table($dbh, $q, $self->{db}, $tbl)->[1]; | ||
2862 | 4130 | } | ||
2863 | 4131 | |||
2867 | 4132 | return { | 4168 | return { |
2871 | 4133 | db => $self->{db}, | 4169 | db => $self->{db}, |
2872 | 4134 | tbl => $tbl, | 4170 | tbl => $tbl, |
2873 | 4135 | ddl => $ddl, | 4171 | name => $q->quote($self->{db}, $tbl), |
2874 | 4172 | ddl => $ddl, | ||
2875 | 4173 | tbl_struct => $tbl_struct, | ||
2876 | 4136 | }; | 4174 | }; |
2877 | 4137 | } | 4175 | } |
2878 | 4138 | } | 4176 | } |
2879 | @@ -4193,6 +4231,10 @@ | |||
2880 | 4193 | 4231 | ||
2881 | 4194 | my $filter = $self->{filters}; | 4232 | my $filter = $self->{filters}; |
2882 | 4195 | 4233 | ||
2883 | 4234 | if ( $db eq 'mysql' && ($tbl eq 'general_log' || $tbl eq 'slow_log') ) { | ||
2884 | 4235 | return 0; | ||
2885 | 4236 | } | ||
2886 | 4237 | |||
2887 | 4196 | if ( $filter->{'ignore-tables'}->{$tbl} | 4238 | if ( $filter->{'ignore-tables'}->{$tbl} |
2888 | 4197 | && ($filter->{'ignore-tables'}->{$tbl} eq '*' | 4239 | && ($filter->{'ignore-tables'}->{$tbl} eq '*' |
2889 | 4198 | || $filter->{'ignore-tables'}->{$tbl} eq $db) ) { | 4240 | || $filter->{'ignore-tables'}->{$tbl} eq $db) ) { |
2890 | @@ -4232,7 +4274,11 @@ | |||
2891 | 4232 | 4274 | ||
2892 | 4233 | sub engine_is_allowed { | 4275 | sub engine_is_allowed { |
2893 | 4234 | my ( $self, $engine ) = @_; | 4276 | my ( $self, $engine ) = @_; |
2895 | 4235 | die "I need an engine argument" unless $engine; | 4277 | |
2896 | 4278 | if ( !$engine ) { | ||
2897 | 4279 | PTDEBUG && _d('No engine specified; allowing the table'); | ||
2898 | 4280 | return 1; | ||
2899 | 4281 | } | ||
2900 | 4236 | 4282 | ||
2901 | 4237 | $engine = lc $engine; | 4283 | $engine = lc $engine; |
2902 | 4238 | 4284 | ||
2903 | @@ -4252,6 +4298,40 @@ | |||
2904 | 4252 | return 1; | 4298 | return 1; |
2905 | 4253 | } | 4299 | } |
2906 | 4254 | 4300 | ||
2907 | 4301 | sub _resume_from_database { | ||
2908 | 4302 | my ($self, $db) = @_; | ||
2909 | 4303 | |||
2910 | 4304 | return 1 unless $self->{resume}->{db}; | ||
2911 | 4305 | |||
2912 | 4306 | if ( $db eq $self->{resume}->{db} ) { | ||
2913 | 4307 | PTDEBUG && _d('At resume db', $db); | ||
2914 | 4308 | delete $self->{resume}->{db}; | ||
2915 | 4309 | return 1; | ||
2916 | 4310 | } | ||
2917 | 4311 | |||
2918 | 4312 | return 0; | ||
2919 | 4313 | } | ||
2920 | 4314 | |||
2921 | 4315 | sub _resume_from_table { | ||
2922 | 4316 | my ($self, $tbl) = @_; | ||
2923 | 4317 | |||
2924 | 4318 | return 1 unless $self->{resume}->{tbl}; | ||
2925 | 4319 | |||
2926 | 4320 | if ( $tbl eq $self->{resume}->{tbl} ) { | ||
2927 | 4321 | if ( !$self->{resume}->{after} ) { | ||
2928 | 4322 | PTDEBUG && _d('Resuming from table', $tbl); | ||
2929 | 4323 | delete $self->{resume}->{tbl}; | ||
2930 | 4324 | return 1; | ||
2931 | 4325 | } | ||
2932 | 4326 | else { | ||
2933 | 4327 | PTDEBUG && _d('Resuming after table', $tbl); | ||
2934 | 4328 | delete $self->{resume}->{tbl}; | ||
2935 | 4329 | } | ||
2936 | 4330 | } | ||
2937 | 4331 | |||
2938 | 4332 | return 0; | ||
2939 | 4333 | } | ||
2940 | 4334 | |||
2941 | 4255 | sub _d { | 4335 | sub _d { |
2942 | 4256 | my ($package, undef, $line) = caller 0; | 4336 | my ($package, undef, $line) = caller 0; |
2943 | 4257 | @_ = map { (my $temp = $_) =~ s/\n/\n# /g; $temp; } | 4337 | @_ = map { (my $temp = $_) =~ s/\n/\n# /g; $temp; } |
2944 | @@ -5144,13 +5224,11 @@ | |||
2945 | 5144 | dbh => $dbh, | 5224 | dbh => $dbh, |
2946 | 5145 | OptionParser => $o, | 5225 | OptionParser => $o, |
2947 | 5146 | Quoter => $q, | 5226 | Quoter => $q, |
2948 | 5147 | MySQLDump => $du, | ||
2949 | 5148 | TableParser => $tp, | 5227 | TableParser => $tp, |
2950 | 5149 | Schema => $schema, | 5228 | Schema => $schema, |
2951 | 5150 | keep_ddl => 1, | ||
2952 | 5151 | ); | 5229 | ); |
2953 | 5152 | TALBE: | 5230 | TALBE: |
2955 | 5153 | while ( my $tbl = $schema_itr->next_schema_object() ) { | 5231 | while ( my $tbl = $schema_itr->next() ) { |
2956 | 5154 | eval { | 5232 | eval { |
2957 | 5155 | my ($indexes) = $tp->get_keys($tbl->{ddl}, {version => $version}); | 5233 | my ($indexes) = $tp->get_keys($tbl->{ddl}, {version => $version}); |
2958 | 5156 | $iu->add_indexes(%$tbl, indexes=>$indexes); | 5234 | $iu->add_indexes(%$tbl, indexes=>$indexes); |
2959 | 5157 | 5235 | ||
2960 | === modified file 'bin/pt-table-checksum' | |||
2961 | --- bin/pt-table-checksum 2012-03-30 16:10:23 +0000 | |||
2962 | +++ bin/pt-table-checksum 2012-03-31 16:07:24 +0000 | |||
2963 | @@ -3941,22 +3941,15 @@ | |||
2964 | 3941 | 3941 | ||
2965 | 3942 | sub get_row_estimate { | 3942 | sub get_row_estimate { |
2966 | 3943 | my (%args) = @_; | 3943 | my (%args) = @_; |
2969 | 3944 | my @required_args = qw(Cxn tbl OptionParser TableParser Quoter); | 3944 | my @required_args = qw(Cxn tbl); |
2970 | 3945 | my ($cxn, $tbl, $o, $tp, $q) = @args{@required_args}; | 3945 | my ($cxn, $tbl) = @args{@required_args}; |
2971 | 3946 | 3946 | ||
2985 | 3947 | if ( $args{where} ) { | 3947 | my $sql = "EXPLAIN SELECT * FROM $tbl->{name} " |
2986 | 3948 | PTDEBUG && _d('WHERE clause, using explain plan for row estimate'); | 3948 | . "WHERE " . ($args{where} || '1=1'); |
2987 | 3949 | my $table = $q->quote(@{$tbl}{qw(db tbl)}); | 3949 | PTDEBUG && _d($sql); |
2988 | 3950 | my $sql = "EXPLAIN SELECT * FROM $table WHERE $args{where}"; | 3950 | my $expl = $cxn->dbh()->selectrow_hashref($sql); |
2989 | 3951 | PTDEBUG && _d($sql); | 3951 | PTDEBUG && _d(Dumper($expl)); |
2990 | 3952 | my $expl = $cxn->dbh()->selectrow_hashref($sql); | 3952 | return ($expl->{rows} || 0), $expl->{key}; |
2978 | 3953 | PTDEBUG && _d(Dumper($expl)); | ||
2979 | 3954 | return ($expl->{rows} || 0), $expl->{key}; | ||
2980 | 3955 | } | ||
2981 | 3956 | else { | ||
2982 | 3957 | PTDEBUG && _d('No WHERE clause, using table status for row estimate'); | ||
2983 | 3958 | return $tbl->{tbl_status}->{rows} || 0; | ||
2984 | 3959 | } | ||
2991 | 3960 | } | 3953 | } |
2992 | 3961 | 3954 | ||
2993 | 3962 | sub _prepare_sths { | 3955 | sub _prepare_sths { |
2994 | @@ -4543,7 +4536,7 @@ | |||
2995 | 4543 | 4536 | ||
2996 | 4544 | sub new { | 4537 | sub new { |
2997 | 4545 | my ( $class, %args ) = @_; | 4538 | my ( $class, %args ) = @_; |
2999 | 4546 | my @required_args = qw(OptionParser Quoter); | 4539 | my @required_args = qw(OptionParser TableParser Quoter); |
3000 | 4547 | foreach my $arg ( @required_args ) { | 4540 | foreach my $arg ( @required_args ) { |
3001 | 4548 | die "I need a $arg argument" unless $args{$arg}; | 4541 | die "I need a $arg argument" unless $args{$arg}; |
3002 | 4549 | } | 4542 | } |
3003 | @@ -4647,25 +4640,18 @@ | |||
3004 | 4647 | } | 4640 | } |
3005 | 4648 | 4641 | ||
3006 | 4649 | if ( $schema_obj ) { | 4642 | if ( $schema_obj ) { |
3007 | 4650 | if ( $schema_obj->{ddl} && $self->{TableParser} ) { | ||
3008 | 4651 | $schema_obj->{tbl_struct} | ||
3009 | 4652 | = $self->{TableParser}->parse($schema_obj->{ddl}); | ||
3010 | 4653 | } | ||
3011 | 4654 | |||
3012 | 4655 | delete $schema_obj->{ddl} unless $self->{keep_ddl}; | ||
3013 | 4656 | delete $schema_obj->{tbl_status} unless $self->{keep_tbl_status}; | ||
3014 | 4657 | |||
3015 | 4658 | if ( my $schema = $self->{Schema} ) { | 4643 | if ( my $schema = $self->{Schema} ) { |
3016 | 4659 | $schema->add_schema_object($schema_obj); | 4644 | $schema->add_schema_object($schema_obj); |
3017 | 4660 | } | 4645 | } |
3019 | 4661 | PTDEBUG && _d('Next schema object:', $schema_obj->{db}, $schema_obj->{tbl}); | 4646 | PTDEBUG && _d('Next schema object:', |
3020 | 4647 | $schema_obj->{db}, $schema_obj->{tbl}); | ||
3021 | 4662 | } | 4648 | } |
3022 | 4663 | 4649 | ||
3023 | 4664 | return $schema_obj; | 4650 | return $schema_obj; |
3024 | 4665 | } | 4651 | } |
3025 | 4666 | 4652 | ||
3026 | 4667 | sub _iterate_files { | 4653 | sub _iterate_files { |
3028 | 4668 | my ( $self ) = @_; | 4654 | my ( $self ) = @_; |
3029 | 4669 | 4655 | ||
3030 | 4670 | if ( !$self->{fh} ) { | 4656 | if ( !$self->{fh} ) { |
3031 | 4671 | my ($fh, $file) = $self->{file_itr}->(); | 4657 | my ($fh, $file) = $self->{file_itr}->(); |
3032 | @@ -4708,14 +4694,14 @@ | |||
3033 | 4708 | next CHUNK; | 4694 | next CHUNK; |
3034 | 4709 | } | 4695 | } |
3035 | 4710 | $ddl =~ s/ \*\/;\Z/;/; # remove end of version comment | 4696 | $ddl =~ s/ \*\/;\Z/;/; # remove end of version comment |
3040 | 4711 | 4697 | my $tbl_struct = $self->{TableParser}->parse($ddl); | |
3041 | 4712 | my ($engine) = $ddl =~ m/\).*?(?:ENGINE|TYPE)=(\w+)/; | 4698 | if ( $self->engine_is_allowed($tbl_struct->{engine}) ) { |
3038 | 4713 | |||
3039 | 4714 | if ( !$engine || $self->engine_is_allowed($engine) ) { | ||
3042 | 4715 | return { | 4699 | return { |
3046 | 4716 | db => $self->{db}, | 4700 | db => $self->{db}, |
3047 | 4717 | tbl => $tbl, | 4701 | tbl => $tbl, |
3048 | 4718 | ddl => $ddl, | 4702 | name => $self->{Quoter}->quote($self->{db}, $tbl), |
3049 | 4703 | ddl => $ddl, | ||
3050 | 4704 | tbl_struct => $tbl_struct, | ||
3051 | 4719 | }; | 4705 | }; |
3052 | 4720 | } | 4706 | } |
3053 | 4721 | } | 4707 | } |
3054 | @@ -4732,6 +4718,7 @@ | |||
3055 | 4732 | sub _iterate_dbh { | 4718 | sub _iterate_dbh { |
3056 | 4733 | my ( $self ) = @_; | 4719 | my ( $self ) = @_; |
3057 | 4734 | my $q = $self->{Quoter}; | 4720 | my $q = $self->{Quoter}; |
3058 | 4721 | my $tp = $self->{TableParser}; | ||
3059 | 4735 | my $dbh = $self->{dbh}; | 4722 | my $dbh = $self->{dbh}; |
3060 | 4736 | PTDEBUG && _d('Getting next schema object from dbh', $dbh); | 4723 | PTDEBUG && _d('Getting next schema object from dbh', $dbh); |
3061 | 4737 | 4724 | ||
3062 | @@ -4770,30 +4757,15 @@ | |||
3063 | 4770 | } | 4757 | } |
3064 | 4771 | 4758 | ||
3065 | 4772 | while ( my $tbl = shift @{$self->{tbls}} ) { | 4759 | while ( my $tbl = shift @{$self->{tbls}} ) { |
3085 | 4773 | my $tbl_status; | 4760 | my $ddl = $tp->get_create_table($dbh, $self->{db}, $tbl); |
3086 | 4774 | if ( $self->{filters}->{'engines'} | 4761 | my $tbl_struct = $tp->parse($ddl); |
3087 | 4775 | || $self->{filters}->{'ignore-engines'} | 4762 | if ( $self->engine_is_allowed($tbl_struct->{engine}) ) { |
3069 | 4776 | || $self->{keep_tbl_status} ) | ||
3070 | 4777 | { | ||
3071 | 4778 | my $sql = "SHOW TABLE STATUS FROM " . $q->quote($self->{db}) | ||
3072 | 4779 | . " LIKE \'$tbl\'"; | ||
3073 | 4780 | PTDEBUG && _d($sql); | ||
3074 | 4781 | $tbl_status = $dbh->selectrow_hashref($sql); | ||
3075 | 4782 | PTDEBUG && _d(Dumper($tbl_status)); | ||
3076 | 4783 | } | ||
3077 | 4784 | |||
3078 | 4785 | if ( !$tbl_status | ||
3079 | 4786 | || $self->engine_is_allowed($tbl_status->{engine}) ) { | ||
3080 | 4787 | my $ddl; | ||
3081 | 4788 | if ( my $tp = $self->{TableParser} ) { | ||
3082 | 4789 | $ddl = $tp->get_create_table($dbh, $self->{db}, $tbl); | ||
3083 | 4790 | } | ||
3084 | 4791 | |||
3088 | 4792 | return { | 4763 | return { |
3089 | 4793 | db => $self->{db}, | 4764 | db => $self->{db}, |
3090 | 4794 | tbl => $tbl, | 4765 | tbl => $tbl, |
3091 | 4766 | name => $q->quote($self->{db}, $tbl), | ||
3092 | 4795 | ddl => $ddl, | 4767 | ddl => $ddl, |
3094 | 4796 | tbl_status => $tbl_status, | 4768 | tbl_struct => $tbl_struct, |
3095 | 4797 | }; | 4769 | }; |
3096 | 4798 | } | 4770 | } |
3097 | 4799 | } | 4771 | } |
3098 | @@ -4897,7 +4869,11 @@ | |||
3099 | 4897 | 4869 | ||
3100 | 4898 | sub engine_is_allowed { | 4870 | sub engine_is_allowed { |
3101 | 4899 | my ( $self, $engine ) = @_; | 4871 | my ( $self, $engine ) = @_; |
3103 | 4900 | die "I need an engine argument" unless $engine; | 4872 | |
3104 | 4873 | if ( !$engine ) { | ||
3105 | 4874 | PTDEBUG && _d('No engine specified; allowing the table'); | ||
3106 | 4875 | return 1; | ||
3107 | 4876 | } | ||
3108 | 4901 | 4877 | ||
3109 | 4902 | $engine = lc $engine; | 4878 | $engine = lc $engine; |
3110 | 4903 | 4879 | ||
3111 | @@ -6253,13 +6229,12 @@ | |||
3112 | 6253 | } | 6229 | } |
3113 | 6254 | 6230 | ||
3114 | 6255 | my $schema_iter = new SchemaIterator( | 6231 | my $schema_iter = new SchemaIterator( |
3122 | 6256 | dbh => $master_dbh, | 6232 | dbh => $master_dbh, |
3123 | 6257 | resume => $last_chunk ? $q->quote(@{$last_chunk}{qw(db tbl)}) | 6233 | resume => $last_chunk ? $q->quote(@{$last_chunk}{qw(db tbl)}) |
3124 | 6258 | : "", | 6234 | : "", |
3125 | 6259 | keep_tbl_status => 1, | 6235 | OptionParser => $o, |
3126 | 6260 | OptionParser => $o, | 6236 | TableParser => $tp, |
3127 | 6261 | TableParser => $tp, | 6237 | Quoter => $q, |
3121 | 6262 | Quoter => $q, | ||
3128 | 6263 | ); | 6238 | ); |
3129 | 6264 | 6239 | ||
3130 | 6265 | if ( $last_chunk && | 6240 | if ( $last_chunk && |
3131 | @@ -6334,13 +6309,13 @@ | |||
3132 | 6334 | my $chunk_size_limit = $o->get('chunk-size-limit'); | 6309 | my $chunk_size_limit = $o->get('chunk-size-limit'); |
3133 | 6335 | my @too_large; | 6310 | my @too_large; |
3134 | 6336 | foreach my $slave ( @$slaves ) { | 6311 | foreach my $slave ( @$slaves ) { |
3135 | 6312 | # get_row_estimate() returns (row_est, index), but | ||
3136 | 6313 | # we only need the row_est. Maybe in the future we'll | ||
3137 | 6314 | # care what index MySQL will use on a slave. | ||
3138 | 6337 | my ($n_rows) = NibbleIterator::get_row_estimate( | 6315 | my ($n_rows) = NibbleIterator::get_row_estimate( |
3145 | 6338 | Cxn => $slave, | 6316 | Cxn => $slave, |
3146 | 6339 | tbl => $tbl, | 6317 | tbl => $tbl, |
3147 | 6340 | where => $o->get('where') || "1=1", | 6318 | where => $o->get('where'), |
3142 | 6341 | OptionParser => $o, | ||
3143 | 6342 | TableParser => $tp, | ||
3144 | 6343 | Quoter => $q, | ||
3148 | 6344 | ); | 6319 | ); |
3149 | 6345 | PTDEBUG && _d('Table on', $slave->name(), | 6320 | PTDEBUG && _d('Table on', $slave->name(), |
3150 | 6346 | 'has', $n_rows, 'rows'); | 6321 | 'has', $n_rows, 'rows'); |
3151 | 6347 | 6322 | ||
3152 | === modified file 'bin/pt-table-sync' | |||
3153 | --- bin/pt-table-sync 2012-03-30 16:53:51 +0000 | |||
3154 | +++ bin/pt-table-sync 2012-03-31 16:07:24 +0000 | |||
3155 | @@ -959,7 +959,7 @@ | |||
3156 | 959 | $opt->{value} = ($pre || '') . $num; | 959 | $opt->{value} = ($pre || '') . $num; |
3157 | 960 | } | 960 | } |
3158 | 961 | else { | 961 | else { |
3160 | 962 | $self->save_error("Invalid size for --$opt->{long}"); | 962 | $self->save_error("Invalid size for --$opt->{long}: $val"); |
3161 | 963 | } | 963 | } |
3162 | 964 | return; | 964 | return; |
3163 | 965 | } | 965 | } |
3164 | @@ -1285,12 +1285,14 @@ | |||
3165 | 1285 | sub as_string { | 1285 | sub as_string { |
3166 | 1286 | my ( $self, $dsn, $props ) = @_; | 1286 | my ( $self, $dsn, $props ) = @_; |
3167 | 1287 | return $dsn unless ref $dsn; | 1287 | return $dsn unless ref $dsn; |
3169 | 1288 | my %allowed = $props ? map { $_=>1 } @$props : (); | 1288 | my @keys = $props ? @$props : sort keys %$dsn; |
3170 | 1289 | return join(',', | 1289 | return join(',', |
3175 | 1290 | map { "$_=" . ($_ eq 'p' ? '...' : $dsn->{$_}) } | 1290 | map { "$_=" . ($_ eq 'p' ? '...' : $dsn->{$_}) } |
3176 | 1291 | grep { defined $dsn->{$_} && $self->{opts}->{$_} } | 1291 | grep { |
3177 | 1292 | grep { !$props || $allowed{$_} } | 1292 | exists $self->{opts}->{$_} |
3178 | 1293 | sort keys %$dsn ); | 1293 | && exists $dsn->{$_} |
3179 | 1294 | && defined $dsn->{$_} | ||
3180 | 1295 | } @keys); | ||
3181 | 1294 | } | 1296 | } |
3182 | 1295 | 1297 | ||
3183 | 1296 | sub usage { | 1298 | sub usage { |
3184 | @@ -1741,19 +1743,58 @@ | |||
3185 | 1741 | return bless $self, $class; | 1743 | return bless $self, $class; |
3186 | 1742 | } | 1744 | } |
3187 | 1743 | 1745 | ||
3188 | 1746 | sub get_create_table { | ||
3189 | 1747 | my ( $self, $dbh, $db, $tbl ) = @_; | ||
3190 | 1748 | die "I need a dbh parameter" unless $dbh; | ||
3191 | 1749 | die "I need a db parameter" unless $db; | ||
3192 | 1750 | die "I need a tbl parameter" unless $tbl; | ||
3193 | 1751 | my $q = $self->{Quoter}; | ||
3194 | 1752 | |||
3195 | 1753 | my $new_sql_mode | ||
3196 | 1754 | = '/*!40101 SET @OLD_SQL_MODE := @@SQL_MODE, ' | ||
3197 | 1755 | . q{@@SQL_MODE := REPLACE(REPLACE(@@SQL_MODE, 'ANSI_QUOTES', ''), ',,', ','), } | ||
3198 | 1756 | . '@OLD_QUOTE := @@SQL_QUOTE_SHOW_CREATE, ' | ||
3199 | 1757 | . '@@SQL_QUOTE_SHOW_CREATE := 1 */'; | ||
3200 | 1758 | |||
3201 | 1759 | my $old_sql_mode = '/*!40101 SET @@SQL_MODE := @OLD_SQL_MODE, ' | ||
3202 | 1760 | . '@@SQL_QUOTE_SHOW_CREATE := @OLD_QUOTE */'; | ||
3203 | 1761 | |||
3204 | 1762 | PTDEBUG && _d($new_sql_mode); | ||
3205 | 1763 | eval { $dbh->do($new_sql_mode); }; | ||
3206 | 1764 | PTDEBUG && $EVAL_ERROR && _d($EVAL_ERROR); | ||
3207 | 1765 | |||
3208 | 1766 | my $use_sql = 'USE ' . $q->quote($db); | ||
3209 | 1767 | PTDEBUG && _d($dbh, $use_sql); | ||
3210 | 1768 | $dbh->do($use_sql); | ||
3211 | 1769 | |||
3212 | 1770 | my $show_sql = "SHOW CREATE TABLE " . $q->quote($db, $tbl); | ||
3213 | 1771 | PTDEBUG && _d($show_sql); | ||
3214 | 1772 | my $href; | ||
3215 | 1773 | eval { $href = $dbh->selectrow_hashref($show_sql); }; | ||
3216 | 1774 | if ( $EVAL_ERROR ) { | ||
3217 | 1775 | PTDEBUG && _d($EVAL_ERROR); | ||
3218 | 1776 | |||
3219 | 1777 | PTDEBUG && _d($old_sql_mode); | ||
3220 | 1778 | $dbh->do($old_sql_mode); | ||
3221 | 1779 | |||
3222 | 1780 | return; | ||
3223 | 1781 | } | ||
3224 | 1782 | |||
3225 | 1783 | PTDEBUG && _d($old_sql_mode); | ||
3226 | 1784 | $dbh->do($old_sql_mode); | ||
3227 | 1785 | |||
3228 | 1786 | my ($key) = grep { m/create (?:table|view)/i } keys %$href; | ||
3229 | 1787 | if ( !$key ) { | ||
3230 | 1788 | die "Error: no 'Create Table' or 'Create View' in result set from " | ||
3231 | 1789 | . "$show_sql: " . Dumper($href); | ||
3232 | 1790 | } | ||
3233 | 1791 | |||
3234 | 1792 | return $href->{$key}; | ||
3235 | 1793 | } | ||
3236 | 1794 | |||
3237 | 1744 | sub parse { | 1795 | sub parse { |
3238 | 1745 | my ( $self, $ddl, $opts ) = @_; | 1796 | my ( $self, $ddl, $opts ) = @_; |
3239 | 1746 | return unless $ddl; | 1797 | return unless $ddl; |
3240 | 1747 | if ( ref $ddl eq 'ARRAY' ) { | ||
3241 | 1748 | if ( lc $ddl->[0] eq 'table' ) { | ||
3242 | 1749 | $ddl = $ddl->[1]; | ||
3243 | 1750 | } | ||
3244 | 1751 | else { | ||
3245 | 1752 | return { | ||
3246 | 1753 | engine => 'VIEW', | ||
3247 | 1754 | }; | ||
3248 | 1755 | } | ||
3249 | 1756 | } | ||
3250 | 1757 | 1798 | ||
3251 | 1758 | if ( $ddl !~ m/CREATE (?:TEMPORARY )?TABLE `/ ) { | 1799 | if ( $ddl !~ m/CREATE (?:TEMPORARY )?TABLE `/ ) { |
3252 | 1759 | die "Cannot parse table definition; is ANSI quoting " | 1800 | die "Cannot parse table definition; is ANSI quoting " |
3253 | @@ -2060,41 +2101,31 @@ | |||
3254 | 2060 | return $ddl; | 2101 | return $ddl; |
3255 | 2061 | } | 2102 | } |
3256 | 2062 | 2103 | ||
3292 | 2063 | sub remove_secondary_indexes { | 2104 | sub get_table_status { |
3293 | 2064 | my ( $self, $ddl ) = @_; | 2105 | my ( $self, $dbh, $db, $like ) = @_; |
3294 | 2065 | my $sec_indexes_ddl; | 2106 | my $q = $self->{Quoter}; |
3295 | 2066 | my $tbl_struct = $self->parse($ddl); | 2107 | my $sql = "SHOW TABLE STATUS FROM " . $q->quote($db); |
3296 | 2067 | 2108 | my @params; | |
3297 | 2068 | if ( ($tbl_struct->{engine} || '') =~ m/InnoDB/i ) { | 2109 | if ( $like ) { |
3298 | 2069 | my $clustered_key = $tbl_struct->{clustered_key}; | 2110 | $sql .= ' LIKE ?'; |
3299 | 2070 | $clustered_key ||= ''; | 2111 | push @params, $like; |
3300 | 2071 | 2112 | } | |
3301 | 2072 | my @sec_indexes = map { | 2113 | PTDEBUG && _d($sql, @params); |
3302 | 2073 | my $key_def = $_->{ddl}; | 2114 | my $sth = $dbh->prepare($sql); |
3303 | 2074 | $key_def =~ s/([\(\)])/\\$1/g; | 2115 | eval { $sth->execute(@params); }; |
3304 | 2075 | $ddl =~ s/\s+$key_def//i; | 2116 | if ($EVAL_ERROR) { |
3305 | 2076 | 2117 | PTDEBUG && _d($EVAL_ERROR); | |
3306 | 2077 | my $key_ddl = "ADD $_->{ddl}"; | 2118 | return; |
3307 | 2078 | $key_ddl .= ',' unless $key_ddl =~ m/,$/; | 2119 | } |
3308 | 2079 | $key_ddl; | 2120 | my @tables = @{$sth->fetchall_arrayref({})}; |
3309 | 2080 | } | 2121 | @tables = map { |
3310 | 2081 | grep { $_->{name} ne $clustered_key } | 2122 | my %tbl; # Make a copy with lowercased keys |
3311 | 2082 | values %{$tbl_struct->{keys}}; | 2123 | @tbl{ map { lc $_ } keys %$_ } = values %$_; |
3312 | 2083 | PTDEBUG && _d('Secondary indexes:', Dumper(\@sec_indexes)); | 2124 | $tbl{engine} ||= $tbl{type} || $tbl{comment}; |
3313 | 2084 | 2125 | delete $tbl{type}; | |
3314 | 2085 | if ( @sec_indexes ) { | 2126 | \%tbl; |
3315 | 2086 | $sec_indexes_ddl = join(' ', @sec_indexes); | 2127 | } @tables; |
3316 | 2087 | $sec_indexes_ddl =~ s/,$//; | 2128 | return @tables; |
3282 | 2088 | } | ||
3283 | 2089 | |||
3284 | 2090 | $ddl =~ s/,(\n\) )/$1/s; | ||
3285 | 2091 | } | ||
3286 | 2092 | else { | ||
3287 | 2093 | PTDEBUG && _d('Not removing secondary indexes from', | ||
3288 | 2094 | $tbl_struct->{engine}, 'table'); | ||
3289 | 2095 | } | ||
3290 | 2096 | |||
3291 | 2097 | return $ddl, $sec_indexes_ddl, $tbl_struct; | ||
3317 | 2098 | } | 2129 | } |
3318 | 2099 | 2130 | ||
3319 | 2100 | sub _d { | 2131 | sub _d { |
3320 | @@ -5445,11 +5476,12 @@ | |||
3321 | 5445 | 5476 | ||
3322 | 5446 | eval { | 5477 | eval { |
3323 | 5447 | if ( my $timeout = $args{wait} ) { | 5478 | if ( my $timeout = $args{wait} ) { |
3325 | 5448 | my $wait = $args{wait_retry_args}->{wait} || 10; | 5479 | my $ms = $self->{MasterSlave}; |
3326 | 5449 | my $tries = $args{wait_retry_args}->{tries} || 3; | 5480 | my $tries = $args{wait_retry_args}->{tries} || 3; |
3327 | 5481 | my $wait; | ||
3328 | 5450 | $self->{Retry}->retry( | 5482 | $self->{Retry}->retry( |
3329 | 5451 | wait => sub { sleep $wait; }, | ||
3330 | 5452 | tries => $tries, | 5483 | tries => $tries, |
3331 | 5484 | wait => sub { sleep $args{wait_retry_args}->{wait} || 10 }, | ||
3332 | 5453 | try => sub { | 5485 | try => sub { |
3333 | 5454 | my ( %args ) = @_; | 5486 | my ( %args ) = @_; |
3334 | 5455 | 5487 | ||
3335 | @@ -5457,12 +5489,18 @@ | |||
3336 | 5457 | warn "Retrying MASTER_POS_WAIT() for --wait $timeout..."; | 5489 | warn "Retrying MASTER_POS_WAIT() for --wait $timeout..."; |
3337 | 5458 | } | 5490 | } |
3338 | 5459 | 5491 | ||
3341 | 5460 | my $ms = $self->{MasterSlave}; | 5492 | $wait = $ms->wait_for_master( |
3340 | 5461 | my $wait = $ms->wait_for_master( | ||
3342 | 5462 | master_status => $ms->get_master_status($src->{misc_dbh}), | 5493 | master_status => $ms->get_master_status($src->{misc_dbh}), |
3343 | 5463 | slave_dbh => $dst->{dbh}, | 5494 | slave_dbh => $dst->{dbh}, |
3344 | 5464 | timeout => $timeout, | 5495 | timeout => $timeout, |
3345 | 5465 | ); | 5496 | ); |
3346 | 5497 | if ( defined $wait->{result} && $wait->{result} != -1 ) { | ||
3347 | 5498 | return; # slave caught up | ||
3348 | 5499 | } | ||
3349 | 5500 | die; # call fail | ||
3350 | 5501 | }, | ||
3351 | 5502 | fail => sub { | ||
3352 | 5503 | my (%args) = @_; | ||
3353 | 5466 | if ( !defined $wait->{result} ) { | 5504 | if ( !defined $wait->{result} ) { |
3354 | 5467 | my $msg; | 5505 | my $msg; |
3355 | 5468 | if ( $wait->{waited} ) { | 5506 | if ( $wait->{waited} ) { |
3356 | @@ -5477,20 +5515,14 @@ | |||
3357 | 5477 | $msg .= " Sleeping $wait seconds then retrying " | 5515 | $msg .= " Sleeping $wait seconds then retrying " |
3358 | 5478 | . ($tries - $args{tryno}) . " more times."; | 5516 | . ($tries - $args{tryno}) . " more times."; |
3359 | 5479 | } | 5517 | } |
3362 | 5480 | warn $msg; | 5518 | warn "$msg\n"; |
3363 | 5481 | return; | 5519 | return 1; # call wait, call try |
3364 | 5482 | } | 5520 | } |
3365 | 5483 | elsif ( $wait->{result} == -1 ) { | 5521 | elsif ( $wait->{result} == -1 ) { |
3373 | 5484 | die "Slave did not catch up to its master after waiting " | 5522 | return 0; # call final_fail |
3367 | 5485 | . "$timeout seconds with MASTER_POS_WAIT. Try inceasing " | ||
3368 | 5486 | . "the --wait time, or disable this feature by specifying " | ||
3369 | 5487 | . "--wait 0."; | ||
3370 | 5488 | } | ||
3371 | 5489 | else { | ||
3372 | 5490 | return $result; # slave caught up | ||
3374 | 5491 | } | 5523 | } |
3375 | 5492 | }, | 5524 | }, |
3377 | 5493 | on_failure => sub { | 5525 | final_fail => sub { |
3378 | 5494 | die "Slave did not catch up to its master after $tries attempts " | 5526 | die "Slave did not catch up to its master after $tries attempts " |
3379 | 5495 | . "of waiting $timeout seconds with MASTER_POS_WAIT. " | 5527 | . "of waiting $timeout seconds with MASTER_POS_WAIT. " |
3380 | 5496 | . "Check that the slave is running, increase the --wait " | 5528 | . "Check that the slave is running, increase the --wait " |
3381 | @@ -5603,23 +5635,21 @@ | |||
3382 | 5603 | die "I need a $arg argument" unless defined $args{$arg}; | 5635 | die "I need a $arg argument" unless defined $args{$arg}; |
3383 | 5604 | } | 5636 | } |
3384 | 5605 | my ($tbl_struct, $index) = @args{@required_args}; | 5637 | my ($tbl_struct, $index) = @args{@required_args}; |
3386 | 5606 | my @cols = $args{cols} ? @{$args{cols}} : @{$tbl_struct->{cols}}; | 5638 | my @cols = $args{cols} ? @{$args{cols}} : @{$tbl_struct->{cols}}; |
3387 | 5607 | my $q = $self->{Quoter}; | 5639 | my $q = $self->{Quoter}; |
3388 | 5608 | 5640 | ||
3389 | 5609 | die "Index '$index' does not exist in table" | 5641 | die "Index '$index' does not exist in table" |
3390 | 5610 | unless exists $tbl_struct->{keys}->{$index}; | 5642 | unless exists $tbl_struct->{keys}->{$index}; |
3391 | 5643 | PTDEBUG && _d('Will ascend index', $index); | ||
3392 | 5611 | 5644 | ||
3393 | 5612 | my @asc_cols = @{$tbl_struct->{keys}->{$index}->{cols}}; | 5645 | my @asc_cols = @{$tbl_struct->{keys}->{$index}->{cols}}; |
3394 | 5613 | my @asc_slice; | ||
3395 | 5614 | |||
3396 | 5615 | @asc_cols = @{$tbl_struct->{keys}->{$index}->{cols}}; | ||
3397 | 5616 | PTDEBUG && _d('Will ascend index', $index); | ||
3398 | 5617 | PTDEBUG && _d('Will ascend columns', join(', ', @asc_cols)); | ||
3399 | 5618 | if ( $args{asc_first} ) { | 5646 | if ( $args{asc_first} ) { |
3400 | 5619 | @asc_cols = $asc_cols[0]; | 5647 | @asc_cols = $asc_cols[0]; |
3401 | 5620 | PTDEBUG && _d('Ascending only first column'); | 5648 | PTDEBUG && _d('Ascending only first column'); |
3402 | 5621 | } | 5649 | } |
3403 | 5650 | PTDEBUG && _d('Will ascend columns', join(', ', @asc_cols)); | ||
3404 | 5622 | 5651 | ||
3405 | 5652 | my @asc_slice; | ||
3406 | 5623 | my %col_posn = do { my $i = 0; map { $_ => $i++ } @cols }; | 5653 | my %col_posn = do { my $i = 0; map { $_ => $i++ } @cols }; |
3407 | 5624 | foreach my $col ( @asc_cols ) { | 5654 | foreach my $col ( @asc_cols ) { |
3408 | 5625 | if ( !exists $col_posn{$col} ) { | 5655 | if ( !exists $col_posn{$col} ) { |
3409 | @@ -6720,292 +6750,427 @@ | |||
3410 | 6720 | # ########################################################################### | 6750 | # ########################################################################### |
3411 | 6721 | 6751 | ||
3412 | 6722 | # ########################################################################### | 6752 | # ########################################################################### |
3415 | 6723 | # SchemaIterator r7141 | 6753 | # SchemaIterator package |
3416 | 6724 | # Don't update this package! | 6754 | # This package is a copy without comments from the original. The original |
3417 | 6755 | # with comments and its test file can be found in the Bazaar repository at, | ||
3418 | 6756 | # lib/SchemaIterator.pm | ||
3419 | 6757 | # t/lib/SchemaIterator.t | ||
3420 | 6758 | # See https://launchpad.net/percona-toolkit for more information. | ||
3421 | 6725 | # ########################################################################### | 6759 | # ########################################################################### |
3422 | 6760 | { | ||
3423 | 6726 | package SchemaIterator; | 6761 | package SchemaIterator; |
3424 | 6727 | 6762 | ||
3425 | 6728 | use strict; | 6763 | use strict; |
3426 | 6729 | use warnings FATAL => 'all'; | 6764 | use warnings FATAL => 'all'; |
3427 | 6730 | |||
3428 | 6731 | use English qw(-no_match_vars); | 6765 | use English qw(-no_match_vars); |
3429 | 6766 | use constant PTDEBUG => $ENV{PTDEBUG} || 0; | ||
3430 | 6767 | |||
3431 | 6732 | use Data::Dumper; | 6768 | use Data::Dumper; |
3432 | 6733 | $Data::Dumper::Indent = 1; | 6769 | $Data::Dumper::Indent = 1; |
3433 | 6734 | $Data::Dumper::Sortkeys = 1; | 6770 | $Data::Dumper::Sortkeys = 1; |
3434 | 6735 | $Data::Dumper::Quotekeys = 0; | 6771 | $Data::Dumper::Quotekeys = 0; |
3435 | 6736 | 6772 | ||
3437 | 6737 | use constant PTDEBUG => $ENV{PTDEBUG} || 0; | 6773 | my $open_comment = qr{/\*!\d{5} }; |
3438 | 6774 | my $tbl_name = qr{ | ||
3439 | 6775 | CREATE\s+ | ||
3440 | 6776 | (?:TEMPORARY\s+)? | ||
3441 | 6777 | TABLE\s+ | ||
3442 | 6778 | (?:IF NOT EXISTS\s+)? | ||
3443 | 6779 | ([^\(]+) | ||
3444 | 6780 | }x; | ||
3445 | 6781 | |||
3446 | 6738 | 6782 | ||
3447 | 6739 | sub new { | 6783 | sub new { |
3448 | 6740 | my ( $class, %args ) = @_; | 6784 | my ( $class, %args ) = @_; |
3450 | 6741 | foreach my $arg ( qw(Quoter) ) { | 6785 | my @required_args = qw(OptionParser TableParser Quoter); |
3451 | 6786 | foreach my $arg ( @required_args ) { | ||
3452 | 6742 | die "I need a $arg argument" unless $args{$arg}; | 6787 | die "I need a $arg argument" unless $args{$arg}; |
3453 | 6743 | } | 6788 | } |
3454 | 6789 | |||
3455 | 6790 | my ($file_itr, $dbh) = @args{qw(file_itr dbh)}; | ||
3456 | 6791 | die "I need either a dbh or file_itr argument" | ||
3457 | 6792 | if (!$dbh && !$file_itr) || ($dbh && $file_itr); | ||
3458 | 6793 | |||
3459 | 6794 | my %resume; | ||
3460 | 6795 | if ( my $table = $args{resume} ) { | ||
3461 | 6796 | PTDEBUG && _d('Will resume from or after', $table); | ||
3462 | 6797 | my ($db, $tbl) = $args{Quoter}->split_unquote($table); | ||
3463 | 6798 | die "Resume table must be database-qualified: $table" | ||
3464 | 6799 | unless $db && $tbl; | ||
3465 | 6800 | $resume{db} = $db; | ||
3466 | 6801 | $resume{tbl} = $tbl; | ||
3467 | 6802 | } | ||
3468 | 6803 | |||
3469 | 6744 | my $self = { | 6804 | my $self = { |
3470 | 6745 | %args, | 6805 | %args, |
3473 | 6746 | filter => undef, | 6806 | resume => \%resume, |
3474 | 6747 | dbs => [], | 6807 | filters => _make_filters(%args), |
3475 | 6748 | }; | 6808 | }; |
3476 | 6809 | |||
3477 | 6749 | return bless $self, $class; | 6810 | return bless $self, $class; |
3478 | 6750 | } | 6811 | } |
3479 | 6751 | 6812 | ||
3650 | 6752 | sub make_filter { | 6813 | sub _make_filters { |
3651 | 6753 | my ( $self, $o ) = @_; | 6814 | my ( %args ) = @_; |
3652 | 6754 | my @lines = ( | 6815 | my @required_args = qw(OptionParser Quoter); |
3483 | 6755 | 'sub {', | ||
3484 | 6756 | ' my ( $dbh, $db, $tbl ) = @_;', | ||
3485 | 6757 | ' my $engine = undef;', | ||
3486 | 6758 | ); | ||
3487 | 6759 | |||
3488 | 6760 | |||
3489 | 6761 | my @permit_dbs = _make_filter('unless', '$db', $o->get('databases')) | ||
3490 | 6762 | if $o->has('databases'); | ||
3491 | 6763 | my @reject_dbs = _make_filter('if', '$db', $o->get('ignore-databases')) | ||
3492 | 6764 | if $o->has('ignore-databases'); | ||
3493 | 6765 | my @dbs_regex; | ||
3494 | 6766 | if ( $o->has('databases-regex') && (my $p = $o->get('databases-regex')) ) { | ||
3495 | 6767 | push @dbs_regex, " return 0 unless \$db && (\$db =~ m/$p/o);"; | ||
3496 | 6768 | } | ||
3497 | 6769 | my @reject_dbs_regex; | ||
3498 | 6770 | if ( $o->has('ignore-databases-regex') | ||
3499 | 6771 | && (my $p = $o->get('ignore-databases-regex')) ) { | ||
3500 | 6772 | push @reject_dbs_regex, " return 0 if \$db && (\$db =~ m/$p/o);"; | ||
3501 | 6773 | } | ||
3502 | 6774 | if ( @permit_dbs || @reject_dbs || @dbs_regex || @reject_dbs_regex ) { | ||
3503 | 6775 | push @lines, | ||
3504 | 6776 | ' if ( $db ) {', | ||
3505 | 6777 | (@permit_dbs ? @permit_dbs : ()), | ||
3506 | 6778 | (@reject_dbs ? @reject_dbs : ()), | ||
3507 | 6779 | (@dbs_regex ? @dbs_regex : ()), | ||
3508 | 6780 | (@reject_dbs_regex ? @reject_dbs_regex : ()), | ||
3509 | 6781 | ' }'; | ||
3510 | 6782 | } | ||
3511 | 6783 | |||
3512 | 6784 | if ( $o->has('tables') || $o->has('ignore-tables') | ||
3513 | 6785 | || $o->has('ignore-tables-regex') ) { | ||
3514 | 6786 | |||
3515 | 6787 | my $have_qtbl = 0; | ||
3516 | 6788 | my $have_only_qtbls = 0; | ||
3517 | 6789 | my %qtbls; | ||
3518 | 6790 | |||
3519 | 6791 | my @permit_tbls; | ||
3520 | 6792 | my @permit_qtbls; | ||
3521 | 6793 | my %permit_qtbls; | ||
3522 | 6794 | if ( $o->get('tables') ) { | ||
3523 | 6795 | my %tbls; | ||
3524 | 6796 | map { | ||
3525 | 6797 | if ( $_ =~ m/\./ ) { | ||
3526 | 6798 | $permit_qtbls{$_} = 1; | ||
3527 | 6799 | } | ||
3528 | 6800 | else { | ||
3529 | 6801 | $tbls{$_} = 1; | ||
3530 | 6802 | } | ||
3531 | 6803 | } keys %{ $o->get('tables') }; | ||
3532 | 6804 | @permit_tbls = _make_filter('unless', '$tbl', \%tbls); | ||
3533 | 6805 | @permit_qtbls = _make_filter('unless', '$qtbl', \%permit_qtbls); | ||
3534 | 6806 | |||
3535 | 6807 | if ( @permit_qtbls ) { | ||
3536 | 6808 | push @lines, | ||
3537 | 6809 | ' my $qtbl = ($db ? "$db." : "") . ($tbl ? $tbl : "");'; | ||
3538 | 6810 | $have_qtbl = 1; | ||
3539 | 6811 | } | ||
3540 | 6812 | } | ||
3541 | 6813 | |||
3542 | 6814 | my @reject_tbls; | ||
3543 | 6815 | my @reject_qtbls; | ||
3544 | 6816 | my %reject_qtbls; | ||
3545 | 6817 | if ( $o->get('ignore-tables') ) { | ||
3546 | 6818 | my %tbls; | ||
3547 | 6819 | map { | ||
3548 | 6820 | if ( $_ =~ m/\./ ) { | ||
3549 | 6821 | $reject_qtbls{$_} = 1; | ||
3550 | 6822 | } | ||
3551 | 6823 | else { | ||
3552 | 6824 | $tbls{$_} = 1; | ||
3553 | 6825 | } | ||
3554 | 6826 | } keys %{ $o->get('ignore-tables') }; | ||
3555 | 6827 | @reject_tbls= _make_filter('if', '$tbl', \%tbls); | ||
3556 | 6828 | @reject_qtbls = _make_filter('if', '$qtbl', \%reject_qtbls); | ||
3557 | 6829 | |||
3558 | 6830 | if ( @reject_qtbls && !$have_qtbl ) { | ||
3559 | 6831 | push @lines, | ||
3560 | 6832 | ' my $qtbl = ($db ? "$db." : "") . ($tbl ? $tbl : "");'; | ||
3561 | 6833 | } | ||
3562 | 6834 | } | ||
3563 | 6835 | |||
3564 | 6836 | if ( keys %permit_qtbls && !@permit_dbs ) { | ||
3565 | 6837 | my $dbs = {}; | ||
3566 | 6838 | map { | ||
3567 | 6839 | my ($db, undef) = split(/\./, $_); | ||
3568 | 6840 | $dbs->{$db} = 1; | ||
3569 | 6841 | } keys %permit_qtbls; | ||
3570 | 6842 | PTDEBUG && _d('Adding restriction "--databases', | ||
3571 | 6843 | (join(',', keys %$dbs) . '"')); | ||
3572 | 6844 | if ( keys %$dbs ) { | ||
3573 | 6845 | $o->set('databases', $dbs); | ||
3574 | 6846 | return $self->make_filter($o); | ||
3575 | 6847 | } | ||
3576 | 6848 | } | ||
3577 | 6849 | |||
3578 | 6850 | my @tbls_regex; | ||
3579 | 6851 | if ( $o->has('tables-regex') && (my $p = $o->get('tables-regex')) ) { | ||
3580 | 6852 | push @tbls_regex, " return 0 unless \$tbl && (\$tbl =~ m/$p/o);"; | ||
3581 | 6853 | } | ||
3582 | 6854 | my @reject_tbls_regex; | ||
3583 | 6855 | if ( $o->has('ignore-tables-regex') | ||
3584 | 6856 | && (my $p = $o->get('ignore-tables-regex')) ) { | ||
3585 | 6857 | push @reject_tbls_regex, | ||
3586 | 6858 | " return 0 if \$tbl && (\$tbl =~ m/$p/o);"; | ||
3587 | 6859 | } | ||
3588 | 6860 | |||
3589 | 6861 | my @get_eng; | ||
3590 | 6862 | my @permit_engs; | ||
3591 | 6863 | my @reject_engs; | ||
3592 | 6864 | if ( ($o->has('engines') && $o->get('engines')) | ||
3593 | 6865 | || ($o->has('ignore-engines') && $o->get('ignore-engines')) ) { | ||
3594 | 6866 | push @get_eng, | ||
3595 | 6867 | ' my $sql = "SHOW TABLE STATUS "', | ||
3596 | 6868 | ' . ($db ? "FROM `$db`" : "")', | ||
3597 | 6869 | ' . " LIKE \'$tbl\'";', | ||
3598 | 6870 | ' PTDEBUG && _d($sql);', | ||
3599 | 6871 | ' eval {', | ||
3600 | 6872 | ' $engine = $dbh->selectrow_hashref($sql)->{engine};', | ||
3601 | 6873 | ' };', | ||
3602 | 6874 | ' PTDEBUG && $EVAL_ERROR && _d($EVAL_ERROR);', | ||
3603 | 6875 | ' PTDEBUG && _d($tbl, "uses engine", $engine);', | ||
3604 | 6876 | ' $engine = lc $engine if $engine;', | ||
3605 | 6877 | @permit_engs | ||
3606 | 6878 | = _make_filter('unless', '$engine', $o->get('engines'), 1); | ||
3607 | 6879 | @reject_engs | ||
3608 | 6880 | = _make_filter('if', '$engine', $o->get('ignore-engines'), 1) | ||
3609 | 6881 | } | ||
3610 | 6882 | |||
3611 | 6883 | if ( @permit_tbls || @permit_qtbls || @reject_tbls || @tbls_regex | ||
3612 | 6884 | || @reject_tbls_regex || @permit_engs || @reject_engs ) { | ||
3613 | 6885 | push @lines, | ||
3614 | 6886 | ' if ( $tbl ) {', | ||
3615 | 6887 | (@permit_tbls ? @permit_tbls : ()), | ||
3616 | 6888 | (@reject_tbls ? @reject_tbls : ()), | ||
3617 | 6889 | (@tbls_regex ? @tbls_regex : ()), | ||
3618 | 6890 | (@reject_tbls_regex ? @reject_tbls_regex : ()), | ||
3619 | 6891 | (@permit_qtbls ? @permit_qtbls : ()), | ||
3620 | 6892 | (@reject_qtbls ? @reject_qtbls : ()), | ||
3621 | 6893 | (@get_eng ? @get_eng : ()), | ||
3622 | 6894 | (@permit_engs ? @permit_engs : ()), | ||
3623 | 6895 | (@reject_engs ? @reject_engs : ()), | ||
3624 | 6896 | ' }'; | ||
3625 | 6897 | } | ||
3626 | 6898 | } | ||
3627 | 6899 | |||
3628 | 6900 | push @lines, | ||
3629 | 6901 | ' PTDEBUG && _d(\'Passes filters:\', $db, $tbl, $engine, $dbh);', | ||
3630 | 6902 | ' return 1;', '}'; | ||
3631 | 6903 | |||
3632 | 6904 | my $code = join("\n", @lines); | ||
3633 | 6905 | PTDEBUG && _d('filter sub:', $code); | ||
3634 | 6906 | my $filter_sub= eval $code | ||
3635 | 6907 | or die "Error compiling subroutine code:\n$code\n$EVAL_ERROR"; | ||
3636 | 6908 | |||
3637 | 6909 | return $filter_sub; | ||
3638 | 6910 | } | ||
3639 | 6911 | |||
3640 | 6912 | sub set_filter { | ||
3641 | 6913 | my ( $self, $filter_sub ) = @_; | ||
3642 | 6914 | $self->{filter} = $filter_sub; | ||
3643 | 6915 | PTDEBUG && _d('Set filter sub'); | ||
3644 | 6916 | return; | ||
3645 | 6917 | } | ||
3646 | 6918 | |||
3647 | 6919 | sub get_db_itr { | ||
3648 | 6920 | my ( $self, %args ) = @_; | ||
3649 | 6921 | my @required_args = qw(dbh); | ||
3653 | 6922 | foreach my $arg ( @required_args ) { | 6816 | foreach my $arg ( @required_args ) { |
3654 | 6923 | die "I need a $arg argument" unless $args{$arg}; | 6817 | die "I need a $arg argument" unless $args{$arg}; |
3655 | 6924 | } | 6818 | } |
3661 | 6925 | my ($dbh) = @args{@required_args}; | 6819 | my ($o, $q) = @args{@required_args}; |
3662 | 6926 | 6820 | ||
3663 | 6927 | my $filter = $self->{filter}; | 6821 | my %filters; |
3664 | 6928 | my @dbs; | 6822 | |
3665 | 6929 | eval { | 6823 | |
3666 | 6824 | my @simple_filters = qw( | ||
3667 | 6825 | databases tables engines | ||
3668 | 6826 | ignore-databases ignore-tables ignore-engines); | ||
3669 | 6827 | FILTER: | ||
3670 | 6828 | foreach my $filter ( @simple_filters ) { | ||
3671 | 6829 | if ( $o->has($filter) ) { | ||
3672 | 6830 | my $objs = $o->get($filter); | ||
3673 | 6831 | next FILTER unless $objs && scalar keys %$objs; | ||
3674 | 6832 | my $is_table = $filter =~ m/table/ ? 1 : 0; | ||
3675 | 6833 | foreach my $obj ( keys %$objs ) { | ||
3676 | 6834 | die "Undefined value for --$filter" unless $obj; | ||
3677 | 6835 | $obj = lc $obj; | ||
3678 | 6836 | if ( $is_table ) { | ||
3679 | 6837 | my ($db, $tbl) = $q->split_unquote($obj); | ||
3680 | 6838 | $db ||= '*'; | ||
3681 | 6839 | PTDEBUG && _d('Filter', $filter, 'value:', $db, $tbl); | ||
3682 | 6840 | $filters{$filter}->{$tbl} = $db; | ||
3683 | 6841 | } | ||
3684 | 6842 | else { # database | ||
3685 | 6843 | PTDEBUG && _d('Filter', $filter, 'value:', $obj); | ||
3686 | 6844 | $filters{$filter}->{$obj} = 1; | ||
3687 | 6845 | } | ||
3688 | 6846 | } | ||
3689 | 6847 | } | ||
3690 | 6848 | } | ||
3691 | 6849 | |||
3692 | 6850 | my @regex_filters = qw( | ||
3693 | 6851 | databases-regex tables-regex | ||
3694 | 6852 | ignore-databases-regex ignore-tables-regex); | ||
3695 | 6853 | REGEX_FILTER: | ||
3696 | 6854 | foreach my $filter ( @regex_filters ) { | ||
3697 | 6855 | if ( $o->has($filter) ) { | ||
3698 | 6856 | my $pat = $o->get($filter); | ||
3699 | 6857 | next REGEX_FILTER unless $pat; | ||
3700 | 6858 | $filters{$filter} = qr/$pat/; | ||
3701 | 6859 | PTDEBUG && _d('Filter', $filter, 'value:', $filters{$filter}); | ||
3702 | 6860 | } | ||
3703 | 6861 | } | ||
3704 | 6862 | |||
3705 | 6863 | PTDEBUG && _d('Schema object filters:', Dumper(\%filters)); | ||
3706 | 6864 | return \%filters; | ||
3707 | 6865 | } | ||
3708 | 6866 | |||
3709 | 6867 | sub next { | ||
3710 | 6868 | my ( $self ) = @_; | ||
3711 | 6869 | |||
3712 | 6870 | if ( !$self->{initialized} ) { | ||
3713 | 6871 | $self->{initialized} = 1; | ||
3714 | 6872 | if ( $self->{resume}->{tbl} | ||
3715 | 6873 | && !$self->table_is_allowed(@{$self->{resume}}{qw(db tbl)}) ) { | ||
3716 | 6874 | PTDEBUG && _d('Will resume after', | ||
3717 | 6875 | join('.', @{$self->{resume}}{qw(db tbl)})); | ||
3718 | 6876 | $self->{resume}->{after} = 1; | ||
3719 | 6877 | } | ||
3720 | 6878 | } | ||
3721 | 6879 | |||
3722 | 6880 | my $schema_obj; | ||
3723 | 6881 | if ( $self->{file_itr} ) { | ||
3724 | 6882 | $schema_obj= $self->_iterate_files(); | ||
3725 | 6883 | } | ||
3726 | 6884 | else { # dbh | ||
3727 | 6885 | $schema_obj= $self->_iterate_dbh(); | ||
3728 | 6886 | } | ||
3729 | 6887 | |||
3730 | 6888 | if ( $schema_obj ) { | ||
3731 | 6889 | if ( my $schema = $self->{Schema} ) { | ||
3732 | 6890 | $schema->add_schema_object($schema_obj); | ||
3733 | 6891 | } | ||
3734 | 6892 | PTDEBUG && _d('Next schema object:', | ||
3735 | 6893 | $schema_obj->{db}, $schema_obj->{tbl}); | ||
3736 | 6894 | } | ||
3737 | 6895 | |||
3738 | 6896 | return $schema_obj; | ||
3739 | 6897 | } | ||
3740 | 6898 | |||
3741 | 6899 | sub _iterate_files { | ||
3742 | 6900 | my ( $self ) = @_; | ||
3743 | 6901 | |||
3744 | 6902 | if ( !$self->{fh} ) { | ||
3745 | 6903 | my ($fh, $file) = $self->{file_itr}->(); | ||
3746 | 6904 | if ( !$fh ) { | ||
3747 | 6905 | PTDEBUG && _d('No more files to iterate'); | ||
3748 | 6906 | return; | ||
3749 | 6907 | } | ||
3750 | 6908 | $self->{fh} = $fh; | ||
3751 | 6909 | $self->{file} = $file; | ||
3752 | 6910 | } | ||
3753 | 6911 | my $fh = $self->{fh}; | ||
3754 | 6912 | PTDEBUG && _d('Getting next schema object from', $self->{file}); | ||
3755 | 6913 | |||
3756 | 6914 | local $INPUT_RECORD_SEPARATOR = ''; | ||
3757 | 6915 | CHUNK: | ||
3758 | 6916 | while (defined(my $chunk = <$fh>)) { | ||
3759 | 6917 | if ($chunk =~ m/Database: (\S+)/) { | ||
3760 | 6918 | my $db = $1; # XXX | ||
3761 | 6919 | $db =~ s/^`//; # strip leading ` | ||
3762 | 6920 | $db =~ s/`$//; # and trailing ` | ||
3763 | 6921 | if ( $self->database_is_allowed($db) | ||
3764 | 6922 | && $self->_resume_from_database($db) ) { | ||
3765 | 6923 | $self->{db} = $db; | ||
3766 | 6924 | } | ||
3767 | 6925 | } | ||
3768 | 6926 | elsif ($self->{db} && $chunk =~ m/CREATE TABLE/) { | ||
3769 | 6927 | if ($chunk =~ m/DROP VIEW IF EXISTS/) { | ||
3770 | 6928 | PTDEBUG && _d('Table is a VIEW, skipping'); | ||
3771 | 6929 | next CHUNK; | ||
3772 | 6930 | } | ||
3773 | 6931 | |||
3774 | 6932 | my ($tbl) = $chunk =~ m/$tbl_name/; | ||
3775 | 6933 | $tbl =~ s/^\s*`//; | ||
3776 | 6934 | $tbl =~ s/`\s*$//; | ||
3777 | 6935 | if ( $self->_resume_from_table($tbl) | ||
3778 | 6936 | && $self->table_is_allowed($self->{db}, $tbl) ) { | ||
3779 | 6937 | my ($ddl) = $chunk =~ m/^(?:$open_comment)?(CREATE TABLE.+?;)$/ms; | ||
3780 | 6938 | if ( !$ddl ) { | ||
3781 | 6939 | warn "Failed to parse CREATE TABLE from\n" . $chunk; | ||
3782 | 6940 | next CHUNK; | ||
3783 | 6941 | } | ||
3784 | 6942 | $ddl =~ s/ \*\/;\Z/;/; # remove end of version comment | ||
3785 | 6943 | my $tbl_struct = $self->{TableParser}->parse($ddl); | ||
3786 | 6944 | if ( $self->engine_is_allowed($tbl_struct->{engine}) ) { | ||
3787 | 6945 | return { | ||
3788 | 6946 | db => $self->{db}, | ||
3789 | 6947 | tbl => $tbl, | ||
3790 | 6948 | name => $self->{Quoter}->quote($self->{db}, $tbl), | ||
3791 | 6949 | ddl => $ddl, | ||
3792 | 6950 | tbl_struct => $tbl_struct, | ||
3793 | 6951 | }; | ||
3794 | 6952 | } | ||
3795 | 6953 | } | ||
3796 | 6954 | } | ||
3797 | 6955 | } # CHUNK | ||
3798 | 6956 | |||
3799 | 6957 | PTDEBUG && _d('No more schema objects in', $self->{file}); | ||
3800 | 6958 | close $self->{fh}; | ||
3801 | 6959 | $self->{fh} = undef; | ||
3802 | 6960 | |||
3803 | 6961 | return $self->_iterate_files(); | ||
3804 | 6962 | } | ||
3805 | 6963 | |||
3806 | 6964 | sub _iterate_dbh { | ||
3807 | 6965 | my ( $self ) = @_; | ||
3808 | 6966 | my $q = $self->{Quoter}; | ||
3809 | 6967 | my $tp = $self->{TableParser}; | ||
3810 | 6968 | my $dbh = $self->{dbh}; | ||
3811 | 6969 | PTDEBUG && _d('Getting next schema object from dbh', $dbh); | ||
3812 | 6970 | |||
3813 | 6971 | if ( !defined $self->{dbs} ) { | ||
3814 | 6930 | my $sql = 'SHOW DATABASES'; | 6972 | my $sql = 'SHOW DATABASES'; |
3815 | 6931 | PTDEBUG && _d($sql); | 6973 | PTDEBUG && _d($sql); |
3821 | 6932 | @dbs = grep { | 6974 | my @dbs = grep { $self->database_is_allowed($_) } |
3822 | 6933 | my $ok = $filter ? $filter->($dbh, $_, undef) : 1; | 6975 | @{$dbh->selectcol_arrayref($sql)}; |
3818 | 6934 | $ok = 0 if $_ =~ m/information_schema|performance_schema|lost\+found/; | ||
3819 | 6935 | $ok; | ||
3820 | 6936 | } @{ $dbh->selectcol_arrayref($sql) }; | ||
3823 | 6937 | PTDEBUG && _d('Found', scalar @dbs, 'databases'); | 6976 | PTDEBUG && _d('Found', scalar @dbs, 'databases'); |
3895 | 6938 | }; | 6977 | $self->{dbs} = \@dbs; |
3896 | 6939 | 6978 | } | |
3897 | 6940 | PTDEBUG && $EVAL_ERROR && _d($EVAL_ERROR); | 6979 | |
3898 | 6941 | my $iterator = sub { | 6980 | if ( !$self->{db} ) { |
3899 | 6942 | return shift @dbs; | 6981 | do { |
3900 | 6943 | }; | 6982 | $self->{db} = shift @{$self->{dbs}}; |
3901 | 6944 | 6983 | } until $self->_resume_from_database($self->{db}); | |
3902 | 6945 | if (wantarray) { | 6984 | PTDEBUG && _d('Next database:', $self->{db}); |
3903 | 6946 | return ($iterator, scalar @dbs); | 6985 | return unless $self->{db}; |
3904 | 6947 | } | 6986 | } |
3905 | 6948 | else { | 6987 | |
3906 | 6949 | return $iterator; | 6988 | if ( !defined $self->{tbls} ) { |
3907 | 6950 | } | 6989 | my $sql = 'SHOW /*!50002 FULL*/ TABLES FROM ' . $q->quote($self->{db}); |
3908 | 6951 | } | 6990 | PTDEBUG && _d($sql); |
3909 | 6952 | 6991 | my @tbls = map { | |
3910 | 6953 | sub get_tbl_itr { | 6992 | $_->[0]; # (tbl, type) |
3911 | 6954 | my ( $self, %args ) = @_; | 6993 | } |
3912 | 6955 | my @required_args = qw(dbh db); | 6994 | grep { |
3913 | 6956 | foreach my $arg ( @required_args ) { | 6995 | my ($tbl, $type) = @$_; |
3914 | 6957 | die "I need a $arg argument" unless $args{$arg}; | 6996 | (!$type || ($type ne 'VIEW')) |
3915 | 6958 | } | 6997 | && $self->_resume_from_table($tbl) |
3916 | 6959 | my ($dbh, $db, $views) = @args{@required_args, 'views'}; | 6998 | && $self->table_is_allowed($self->{db}, $tbl); |
3917 | 6960 | 6999 | } | |
3918 | 6961 | my $filter = $self->{filter}; | 7000 | @{$dbh->selectall_arrayref($sql)}; |
3919 | 6962 | my @tbls; | 7001 | PTDEBUG && _d('Found', scalar @tbls, 'tables in database', $self->{db}); |
3920 | 6963 | if ( $db ) { | 7002 | $self->{tbls} = \@tbls; |
3921 | 6964 | eval { | 7003 | } |
3922 | 6965 | my $sql = 'SHOW /*!50002 FULL*/ TABLES FROM ' | 7004 | |
3923 | 6966 | . $self->{Quoter}->quote($db); | 7005 | while ( my $tbl = shift @{$self->{tbls}} ) { |
3924 | 6967 | PTDEBUG && _d($sql); | 7006 | my $ddl = $tp->get_create_table($dbh, $self->{db}, $tbl); |
3925 | 6968 | @tbls = map { | 7007 | my $tbl_struct = $tp->parse($ddl); |
3926 | 6969 | $_->[0] | 7008 | if ( $self->engine_is_allowed($tbl_struct->{engine}) ) { |
3927 | 6970 | } | 7009 | return { |
3928 | 6971 | grep { | 7010 | db => $self->{db}, |
3929 | 6972 | my ($tbl, $type) = @$_; | 7011 | tbl => $tbl, |
3930 | 6973 | my $ok = $filter ? $filter->($dbh, $db, $tbl) : 1; | 7012 | name => $q->quote($self->{db}, $tbl), |
3931 | 6974 | if ( !$views ) { | 7013 | ddl => $ddl, |
3932 | 6975 | $ok = 0 if ($type || '') eq 'VIEW'; | 7014 | tbl_struct => $tbl_struct, |
3933 | 6976 | } | 7015 | }; |
3934 | 6977 | $ok; | 7016 | } |
3935 | 6978 | } | 7017 | } |
3936 | 6979 | @{ $dbh->selectall_arrayref($sql) }; | 7018 | |
3937 | 6980 | PTDEBUG && _d('Found', scalar @tbls, 'tables in', $db); | 7019 | PTDEBUG && _d('No more tables in database', $self->{db}); |
3938 | 6981 | }; | 7020 | $self->{db} = undef; |
3939 | 6982 | PTDEBUG && $EVAL_ERROR && _d($EVAL_ERROR); | 7021 | $self->{tbls} = undef; |
3940 | 6983 | } | 7022 | |
3941 | 6984 | else { | 7023 | return $self->_iterate_dbh(); |
3942 | 6985 | PTDEBUG && _d('No db given so no tables'); | 7024 | } |
3943 | 6986 | } | 7025 | |
3944 | 6987 | 7026 | sub database_is_allowed { | |
3945 | 6988 | my $iterator = sub { | 7027 | my ( $self, $db ) = @_; |
3946 | 6989 | return shift @tbls; | 7028 | die "I need a db argument" unless $db; |
3947 | 6990 | }; | 7029 | |
3948 | 6991 | 7030 | $db = lc $db; | |
3949 | 6992 | if ( wantarray ) { | 7031 | |
3950 | 6993 | return ($iterator, scalar @tbls); | 7032 | my $filter = $self->{filters}; |
3951 | 6994 | } | 7033 | |
3952 | 6995 | else { | 7034 | if ( $db =~ m/information_schema|performance_schema|lost\+found/ ) { |
3953 | 6996 | return $iterator; | 7035 | PTDEBUG && _d('Database', $db, 'is a system database, ignoring'); |
3954 | 6997 | } | 7036 | return 0; |
3955 | 6998 | } | 7037 | } |
3956 | 6999 | 7038 | ||
3957 | 7000 | sub _make_filter { | 7039 | if ( $self->{filters}->{'ignore-databases'}->{$db} ) { |
3958 | 7001 | my ( $cond, $var_name, $objs, $lc ) = @_; | 7040 | PTDEBUG && _d('Database', $db, 'is in --ignore-databases list'); |
3959 | 7002 | my @lines; | 7041 | return 0; |
3960 | 7003 | if ( scalar keys %$objs ) { | 7042 | } |
3961 | 7004 | my $test = join(' || ', | 7043 | |
3962 | 7005 | map { "$var_name eq '" . ($lc ? lc $_ : $_) ."'" } keys %$objs); | 7044 | if ( $filter->{'ignore-databases-regex'} |
3963 | 7006 | push @lines, " return 0 $cond $var_name && ($test);", | 7045 | && $db =~ $filter->{'ignore-databases-regex'} ) { |
3964 | 7007 | } | 7046 | PTDEBUG && _d('Database', $db, 'matches --ignore-databases-regex'); |
3965 | 7008 | return @lines; | 7047 | return 0; |
3966 | 7048 | } | ||
3967 | 7049 | |||
3968 | 7050 | if ( $filter->{'databases'} | ||
3969 | 7051 | && !$filter->{'databases'}->{$db} ) { | ||
3970 | 7052 | PTDEBUG && _d('Database', $db, 'is not in --databases list, ignoring'); | ||
3971 | 7053 | return 0; | ||
3972 | 7054 | } | ||
3973 | 7055 | |||
3974 | 7056 | if ( $filter->{'databases-regex'} | ||
3975 | 7057 | && $db !~ $filter->{'databases-regex'} ) { | ||
3976 | 7058 | PTDEBUG && _d('Database', $db, 'does not match --databases-regex, ignoring'); | ||
3977 | 7059 | return 0; | ||
3978 | 7060 | } | ||
3979 | 7061 | |||
3980 | 7062 | return 1; | ||
3981 | 7063 | } | ||
3982 | 7064 | |||
3983 | 7065 | sub table_is_allowed { | ||
3984 | 7066 | my ( $self, $db, $tbl ) = @_; | ||
3985 | 7067 | die "I need a db argument" unless $db; | ||
3986 | 7068 | die "I need a tbl argument" unless $tbl; | ||
3987 | 7069 | |||
3988 | 7070 | $db = lc $db; | ||
3989 | 7071 | $tbl = lc $tbl; | ||
3990 | 7072 | |||
3991 | 7073 | my $filter = $self->{filters}; | ||
3992 | 7074 | |||
3993 | 7075 | if ( $db eq 'mysql' && ($tbl eq 'general_log' || $tbl eq 'slow_log') ) { | ||
3994 | 7076 | return 0; | ||
3995 | 7077 | } | ||
3996 | 7078 | |||
3997 | 7079 | if ( $filter->{'ignore-tables'}->{$tbl} | ||
3998 | 7080 | && ($filter->{'ignore-tables'}->{$tbl} eq '*' | ||
3999 | 7081 | || $filter->{'ignore-tables'}->{$tbl} eq $db) ) { | ||
4000 | 7082 | PTDEBUG && _d('Table', $tbl, 'is in --ignore-tables list'); | ||
4001 | 7083 | return 0; | ||
4002 | 7084 | } | ||
4003 | 7085 | |||
4004 | 7086 | if ( $filter->{'ignore-tables-regex'} | ||
4005 | 7087 | && $tbl =~ $filter->{'ignore-tables-regex'} ) { | ||
4006 | 7088 | PTDEBUG && _d('Table', $tbl, 'matches --ignore-tables-regex'); | ||
4007 | 7089 | return 0; | ||
4008 | 7090 | } | ||
4009 | 7091 | |||
4010 | 7092 | if ( $filter->{'tables'} | ||
4011 | 7093 | && !$filter->{'tables'}->{$tbl} ) { | ||
4012 | 7094 | PTDEBUG && _d('Table', $tbl, 'is not in --tables list, ignoring'); | ||
4013 | 7095 | return 0; | ||
4014 | 7096 | } | ||
4015 | 7097 | |||
4016 | 7098 | if ( $filter->{'tables-regex'} | ||
4017 | 7099 | && $tbl !~ $filter->{'tables-regex'} ) { | ||
4018 | 7100 | PTDEBUG && _d('Table', $tbl, 'does not match --tables-regex, ignoring'); | ||
4019 | 7101 | return 0; | ||
4020 | 7102 | } | ||
4021 | 7103 | |||
4022 | 7104 | if ( $filter->{'tables'} | ||
4023 | 7105 | && $filter->{'tables'}->{$tbl} | ||
4024 | 7106 | && $filter->{'tables'}->{$tbl} ne '*' | ||
4025 | 7107 | && $filter->{'tables'}->{$tbl} ne $db ) { | ||
4026 | 7108 | PTDEBUG && _d('Table', $tbl, 'is only allowed in database', | ||
4027 | 7109 | $filter->{'tables'}->{$tbl}); | ||
4028 | 7110 | return 0; | ||
4029 | 7111 | } | ||
4030 | 7112 | |||
4031 | 7113 | return 1; | ||
4032 | 7114 | } | ||
4033 | 7115 | |||
4034 | 7116 | sub engine_is_allowed { | ||
4035 | 7117 | my ( $self, $engine ) = @_; | ||
4036 | 7118 | |||
4037 | 7119 | if ( !$engine ) { | ||
4038 | 7120 | PTDEBUG && _d('No engine specified; allowing the table'); | ||
4039 | 7121 | return 1; | ||
4040 | 7122 | } | ||
4041 | 7123 | |||
4042 | 7124 | $engine = lc $engine; | ||
4043 | 7125 | |||
4044 | 7126 | my $filter = $self->{filters}; | ||
4045 | 7127 | |||
4046 | 7128 | if ( $filter->{'ignore-engines'}->{$engine} ) { | ||
4047 | 7129 | PTDEBUG && _d('Engine', $engine, 'is in --ignore-databases list'); | ||
4048 | 7130 | return 0; | ||
4049 | 7131 | } | ||
4050 | 7132 | |||
4051 | 7133 | if ( $filter->{'engines'} | ||
4052 | 7134 | && !$filter->{'engines'}->{$engine} ) { | ||
4053 | 7135 | PTDEBUG && _d('Engine', $engine, 'is not in --engines list, ignoring'); | ||
4054 | 7136 | return 0; | ||
4055 | 7137 | } | ||
4056 | 7138 | |||
4057 | 7139 | return 1; | ||
4058 | 7140 | } | ||
4059 | 7141 | |||
4060 | 7142 | sub _resume_from_database { | ||
4061 | 7143 | my ($self, $db) = @_; | ||
4062 | 7144 | |||
4063 | 7145 | return 1 unless $self->{resume}->{db}; | ||
4064 | 7146 | |||
4065 | 7147 | if ( $db eq $self->{resume}->{db} ) { | ||
4066 | 7148 | PTDEBUG && _d('At resume db', $db); | ||
4067 | 7149 | delete $self->{resume}->{db}; | ||
4068 | 7150 | return 1; | ||
4069 | 7151 | } | ||
4070 | 7152 | |||
4071 | 7153 | return 0; | ||
4072 | 7154 | } | ||
4073 | 7155 | |||
4074 | 7156 | sub _resume_from_table { | ||
4075 | 7157 | my ($self, $tbl) = @_; | ||
4076 | 7158 | |||
4077 | 7159 | return 1 unless $self->{resume}->{tbl}; | ||
4078 | 7160 | |||
4079 | 7161 | if ( $tbl eq $self->{resume}->{tbl} ) { | ||
4080 | 7162 | if ( !$self->{resume}->{after} ) { | ||
4081 | 7163 | PTDEBUG && _d('Resuming from table', $tbl); | ||
4082 | 7164 | delete $self->{resume}->{tbl}; | ||
4083 | 7165 | return 1; | ||
4084 | 7166 | } | ||
4085 | 7167 | else { | ||
4086 | 7168 | PTDEBUG && _d('Resuming after table', $tbl); | ||
4087 | 7169 | delete $self->{resume}->{tbl}; | ||
4088 | 7170 | } | ||
4089 | 7171 | } | ||
4090 | 7172 | |||
4091 | 7173 | return 0; | ||
4092 | 7009 | } | 7174 | } |
4093 | 7010 | 7175 | ||
4094 | 7011 | sub _d { | 7176 | sub _d { |
4095 | @@ -7017,7 +7182,7 @@ | |||
4096 | 7017 | } | 7182 | } |
4097 | 7018 | 7183 | ||
4098 | 7019 | 1; | 7184 | 1; |
4100 | 7020 | 7185 | } | |
4101 | 7021 | # ########################################################################### | 7186 | # ########################################################################### |
4102 | 7022 | # End SchemaIterator package | 7187 | # End SchemaIterator package |
4103 | 7023 | # ########################################################################### | 7188 | # ########################################################################### |
4104 | @@ -7304,48 +7469,42 @@ | |||
4105 | 7304 | 7469 | ||
4106 | 7305 | sub retry { | 7470 | sub retry { |
4107 | 7306 | my ( $self, %args ) = @_; | 7471 | my ( $self, %args ) = @_; |
4109 | 7307 | my @required_args = qw(try wait); | 7472 | my @required_args = qw(try fail final_fail); |
4110 | 7308 | foreach my $arg ( @required_args ) { | 7473 | foreach my $arg ( @required_args ) { |
4111 | 7309 | die "I need a $arg argument" unless $args{$arg}; | 7474 | die "I need a $arg argument" unless $args{$arg}; |
4112 | 7310 | }; | 7475 | }; |
4114 | 7311 | my ($try, $wait) = @args{@required_args}; | 7476 | my ($try, $fail, $final_fail) = @args{@required_args}; |
4115 | 7477 | my $wait = $args{wait} || sub { sleep 1; }; | ||
4116 | 7312 | my $tries = $args{tries} || 3; | 7478 | my $tries = $args{tries} || 3; |
4117 | 7313 | 7479 | ||
4118 | 7480 | my $last_error; | ||
4119 | 7314 | my $tryno = 0; | 7481 | my $tryno = 0; |
4120 | 7482 | TRY: | ||
4121 | 7315 | while ( ++$tryno <= $tries ) { | 7483 | while ( ++$tryno <= $tries ) { |
4123 | 7316 | PTDEBUG && _d("Retry", $tryno, "of", $tries); | 7484 | PTDEBUG && _d("Try", $tryno, "of", $tries); |
4124 | 7317 | my $result; | 7485 | my $result; |
4125 | 7318 | eval { | 7486 | eval { |
4126 | 7319 | $result = $try->(tryno=>$tryno); | 7487 | $result = $try->(tryno=>$tryno); |
4127 | 7320 | }; | 7488 | }; |
4128 | 7489 | if ( $EVAL_ERROR ) { | ||
4129 | 7490 | PTDEBUG && _d("Try code failed:", $EVAL_ERROR); | ||
4130 | 7491 | $last_error = $EVAL_ERROR; | ||
4131 | 7321 | 7492 | ||
4133 | 7322 | if ( defined $result ) { | 7493 | if ( $tryno < $tries ) { # more retries |
4134 | 7494 | my $retry = $fail->(tryno=>$tryno, error=>$last_error); | ||
4135 | 7495 | last TRY unless $retry; | ||
4136 | 7496 | PTDEBUG && _d("Calling wait code"); | ||
4137 | 7497 | $wait->(tryno=>$tryno); | ||
4138 | 7498 | } | ||
4139 | 7499 | } | ||
4140 | 7500 | else { | ||
4141 | 7323 | PTDEBUG && _d("Try code succeeded"); | 7501 | PTDEBUG && _d("Try code succeeded"); |
4142 | 7324 | if ( my $on_success = $args{on_success} ) { | ||
4143 | 7325 | PTDEBUG && _d("Calling on_success code"); | ||
4144 | 7326 | $on_success->(tryno=>$tryno, result=>$result); | ||
4145 | 7327 | } | ||
4146 | 7328 | return $result; | 7502 | return $result; |
4147 | 7329 | } | 7503 | } |
4167 | 7330 | 7504 | } | |
4168 | 7331 | if ( $EVAL_ERROR ) { | 7505 | |
4169 | 7332 | PTDEBUG && _d("Try code died:", $EVAL_ERROR); | 7506 | PTDEBUG && _d('Try code did not succeed'); |
4170 | 7333 | die $EVAL_ERROR unless $args{retry_on_die}; | 7507 | return $final_fail->(error=>$last_error); |
4152 | 7334 | } | ||
4153 | 7335 | |||
4154 | 7336 | if ( $tryno < $tries ) { | ||
4155 | 7337 | PTDEBUG && _d("Try code failed, calling wait code"); | ||
4156 | 7338 | $wait->(tryno=>$tryno); | ||
4157 | 7339 | } | ||
4158 | 7340 | } | ||
4159 | 7341 | |||
4160 | 7342 | PTDEBUG && _d("Try code did not succeed"); | ||
4161 | 7343 | if ( my $on_failure = $args{on_failure} ) { | ||
4162 | 7344 | PTDEBUG && _d("Calling on_failure code"); | ||
4163 | 7345 | $on_failure->(); | ||
4164 | 7346 | } | ||
4165 | 7347 | |||
4166 | 7348 | return; | ||
4171 | 7349 | } | 7508 | } |
4172 | 7350 | 7509 | ||
4173 | 7351 | sub _d { | 7510 | sub _d { |
4174 | @@ -8006,27 +8165,20 @@ | |||
4175 | 8006 | tbl => undef, # set later | 8165 | tbl => undef, # set later |
4176 | 8007 | }; | 8166 | }; |
4177 | 8008 | 8167 | ||
4180 | 8009 | my $si = new SchemaIterator( | 8168 | my $schema_iter = new SchemaIterator( |
4181 | 8010 | Quoter => $args{Quoter}, | 8169 | dbh => $src->{dbh}, |
4182 | 8170 | OptionParser => $o, | ||
4183 | 8171 | TableParser => $args{TableParser}, | ||
4184 | 8172 | Quoter => $args{Quoter}, | ||
4185 | 8011 | ); | 8173 | ); |
4186 | 8012 | $si->set_filter($si->make_filter($o)); | ||
4187 | 8013 | 8174 | ||
4188 | 8014 | # Make a list of all dbs.tbls on the source. It's more efficient this | 8175 | # Make a list of all dbs.tbls on the source. It's more efficient this |
4189 | 8015 | # way because it avoids open/closing a dbh for each tbl and dsn, unless | 8176 | # way because it avoids open/closing a dbh for each tbl and dsn, unless |
4190 | 8016 | # we pre-opened the dsn. It would also cause confusing verbose output. | 8177 | # we pre-opened the dsn. It would also cause confusing verbose output. |
4191 | 8017 | my @dbs_tbls; | 8178 | my @dbs_tbls; |
4204 | 8018 | my $next_db = $si->get_db_itr(dbh => $src->{dbh}); | 8179 | while ( my $tbl = $schema_iter->next() ) { |
4205 | 8019 | while ( my $db = $next_db->() ) { | 8180 | PTDEBUG && _d('Got table', $tbl->{db}, $tbl->{tbl}); |
4206 | 8020 | PTDEBUG && _d('Getting tables from', $db); | 8181 | push @dbs_tbls, $tbl; |
4195 | 8021 | my $next_tbl = $si->get_tbl_itr( | ||
4196 | 8022 | dbh => $src->{dbh}, | ||
4197 | 8023 | db => $db, | ||
4198 | 8024 | views => 0, | ||
4199 | 8025 | ); | ||
4200 | 8026 | while ( my $tbl = $next_tbl->() ) { | ||
4201 | 8027 | PTDEBUG && _d('Got table', $tbl); | ||
4202 | 8028 | push @dbs_tbls, { db => $db, tbl => $tbl }; | ||
4203 | 8029 | } | ||
4207 | 8030 | } | 8182 | } |
4208 | 8031 | 8183 | ||
4209 | 8032 | my $exit_status = 0; | 8184 | my $exit_status = 0; |
4210 | @@ -8048,6 +8200,7 @@ | |||
4211 | 8048 | lock_server(src => $src, dst => $dst, %args); | 8200 | lock_server(src => $src, dst => $dst, %args); |
4212 | 8049 | 8201 | ||
4213 | 8050 | foreach my $db_tbl ( @dbs_tbls ) { | 8202 | foreach my $db_tbl ( @dbs_tbls ) { |
4214 | 8203 | $src->{tbl_struct} = $db_tbl->{tbl_struct}; | ||
4215 | 8051 | $src->{db} = $dst->{db} = $db_tbl->{db}; | 8204 | $src->{db} = $dst->{db} = $db_tbl->{db}; |
4216 | 8052 | $src->{tbl} = $dst->{tbl} = $db_tbl->{tbl}; | 8205 | $src->{tbl} = $dst->{tbl} = $db_tbl->{tbl}; |
4217 | 8053 | 8206 | ||
4218 | @@ -8194,8 +8347,9 @@ | |||
4219 | 8194 | $start_ts = get_server_time($src->{dbh}) if $o->get('verbose'); | 8347 | $start_ts = get_server_time($src->{dbh}) if $o->get('verbose'); |
4220 | 8195 | 8348 | ||
4221 | 8196 | # This will either die if there's a problem or return the tbl struct. | 8349 | # This will either die if there's a problem or return the tbl struct. |
4224 | 8197 | my $tbl_struct = ok_to_sync($src, $dst, %args); | 8350 | ok_to_sync($src, $dst, %args); |
4225 | 8198 | 8351 | my $tbl_struct = $src->{tbl_struct}; | |
4226 | 8352 | |||
4227 | 8199 | if ( my $diff = $args{diff} ) { | 8353 | if ( my $diff = $args{diff} ) { |
4228 | 8200 | PTDEBUG && _d('Converting checksum diff to WHERE:', Dumper($diff)); | 8354 | PTDEBUG && _d('Converting checksum diff to WHERE:', Dumper($diff)); |
4229 | 8201 | $args{where} = diff_where( | 8355 | $args{where} = diff_where( |
4230 | @@ -8568,35 +8722,30 @@ | |||
4231 | 8568 | } | 8722 | } |
4232 | 8569 | my ($src, $dst, $dp, $q, $vp, $tp, $du, $syncer, $o) = @args{@required_args}; | 8723 | my ($src, $dst, $dp, $q, $vp, $tp, $du, $syncer, $o) = @args{@required_args}; |
4233 | 8570 | 8724 | ||
4263 | 8571 | # First things first: check that the src and dst dbs and tbls exist. | 8725 | if ( !$src->{tbl_struct} ) { |
4264 | 8572 | # This can fail in cases like h=host,D=bad,t=also_bad (i.e. simple | 8726 | eval { |
4265 | 8573 | # user error). It can also fail when syncing all dbs/tbls with sync_all() | 8727 | $src->{ddl} = $tp->get_create_table( |
4266 | 8574 | # because the dst db/tbl is assumed to be the same as the src but | 8728 | $src->{dbh}, $src->{db}, $src->{tbl}); |
4267 | 8575 | # this isn't always the case. | 8729 | $src->{tbl_struct} = $tp->parse($src->{ddl}); |
4268 | 8576 | my $src_tbl_ddl; | 8730 | |
4269 | 8577 | eval { | 8731 | }; |
4270 | 8578 | # FYI: get_create_table() does USE db but doesn't eval it. | 8732 | if ( $EVAL_ERROR ) { |
4271 | 8579 | $src->{dbh}->do("USE `$src->{db}`"); | 8733 | die "Error getting table structure for $src->{db}.$src->{tbl} on " |
4272 | 8580 | $src_tbl_ddl = $du->get_create_table($src->{dbh}, $q, | 8734 | . $dp->as_string($src->{dsn}) . "$EVAL_ERROR\nEnsure that " |
4273 | 8581 | $src->{db}, $src->{tbl}); | 8735 | . "the table exists and is accessible.\n"; |
4274 | 8582 | }; | 8736 | } |
4275 | 8583 | die $EVAL_ERROR if $EVAL_ERROR; | 8737 | } |
4276 | 8584 | 8738 | ||
4277 | 8585 | my $dst_tbl_ddl; | 8739 | # Check that the dst has the table. |
4278 | 8586 | eval { | 8740 | my $dst_has_table = $tp->check_table( |
4279 | 8587 | # FYI: get_create_table() does USE db but doesn't eval it. | 8741 | dbh => $dst->{dbh}, |
4280 | 8588 | $dst->{dbh}->do("USE `$dst->{db}`"); | 8742 | db => $dst->{db}, |
4281 | 8589 | $dst_tbl_ddl = $du->get_create_table($dst->{dbh}, $q, | 8743 | tbl => $dst->{tbl}, |
4282 | 8590 | $dst->{db}, $dst->{tbl}); | 8744 | ); |
4283 | 8591 | }; | 8745 | if ( !$dst_has_table ) { |
4284 | 8592 | die $EVAL_ERROR if $EVAL_ERROR; | 8746 | die "Table $dst->{db}.$dst->{tbl} does not exist on " |
4285 | 8593 | 8747 | . $dp->as_string($dst->{dsn}) . "\n"; | |
4286 | 8594 | # This doesn't work at the moment when syncing different table names. | 8748 | } |
4258 | 8595 | # Check that src.db.tbl has the exact same schema as dst.db.tbl. | ||
4259 | 8596 | # if ( $o->get('check-schema') && ($src_tbl_ddl ne $dst_tbl_ddl) ) { | ||
4260 | 8597 | # die "Source and destination tables have different schemas"; | ||
4261 | 8598 | # } | ||
4262 | 8599 | my $tbl_struct = $tp->parse($src_tbl_ddl); | ||
4287 | 8600 | 8749 | ||
4288 | 8601 | # Check that the user has all the necessary privs on the tbls. | 8750 | # Check that the user has all the necessary privs on the tbls. |
4289 | 8602 | if ( $o->get('check-privileges') ) { | 8751 | if ( $o->get('check-privileges') ) { |
4290 | @@ -8629,7 +8778,7 @@ | |||
4291 | 8629 | } | 8778 | } |
4292 | 8630 | } | 8779 | } |
4293 | 8631 | 8780 | ||
4295 | 8632 | return $tbl_struct; | 8781 | return; |
4296 | 8633 | } | 8782 | } |
4297 | 8634 | 8783 | ||
4298 | 8635 | # Sub: filter_diffs | 8784 | # Sub: filter_diffs |
4299 | 8636 | 8785 | ||
4300 | === added file 'bin/pt-table-usage' | |||
4301 | --- bin/pt-table-usage 1970-01-01 00:00:00 +0000 | |||
4302 | +++ bin/pt-table-usage 2012-03-31 16:07:24 +0000 | |||
4303 | @@ -0,0 +1,7320 @@ | |||
4304 | 1 | #!/usr/bin/env perl | ||
4305 | 2 | |||
4306 | 3 | # This program is part of Percona Toolkit: http://www.percona.com/software/ | ||
4307 | 4 | # See "COPYRIGHT, LICENSE, AND WARRANTY" at the end of this file for legal | ||
4308 | 5 | # notices and disclaimers. | ||
4309 | 6 | |||
4310 | 7 | use strict; | ||
4311 | 8 | use warnings FATAL => 'all'; | ||
4312 | 9 | use constant PTDEBUG => $ENV{PTDEBUG} || 0; | ||
4313 | 10 | |||
4314 | 11 | # ########################################################################### | ||
4315 | 12 | # DSNParser package | ||
4316 | 13 | # This package is a copy without comments from the original. The original | ||
4317 | 14 | # with comments and its test file can be found in the Bazaar repository at, | ||
4318 | 15 | # lib/DSNParser.pm | ||
4319 | 16 | # t/lib/DSNParser.t | ||
4320 | 17 | # See https://launchpad.net/percona-toolkit for more information. | ||
4321 | 18 | # ########################################################################### | ||
4322 | 19 | { | ||
4323 | 20 | package DSNParser; | ||
4324 | 21 | |||
4325 | 22 | use strict; | ||
4326 | 23 | use warnings FATAL => 'all'; | ||
4327 | 24 | use English qw(-no_match_vars); | ||
4328 | 25 | use constant PTDEBUG => $ENV{PTDEBUG} || 0; | ||
4329 | 26 | |||
4330 | 27 | use Data::Dumper; | ||
4331 | 28 | $Data::Dumper::Indent = 0; | ||
4332 | 29 | $Data::Dumper::Quotekeys = 0; | ||
4333 | 30 | |||
4334 | 31 | eval { | ||
4335 | 32 | require DBI; | ||
4336 | 33 | }; | ||
4337 | 34 | my $have_dbi = $EVAL_ERROR ? 0 : 1; | ||
4338 | 35 | |||
4339 | 36 | sub new { | ||
4340 | 37 | my ( $class, %args ) = @_; | ||
4341 | 38 | foreach my $arg ( qw(opts) ) { | ||
4342 | 39 | die "I need a $arg argument" unless $args{$arg}; | ||
4343 | 40 | } | ||
4344 | 41 | my $self = { | ||
4345 | 42 | opts => {} # h, P, u, etc. Should come from DSN OPTIONS section in POD. | ||
4346 | 43 | }; | ||
4347 | 44 | foreach my $opt ( @{$args{opts}} ) { | ||
4348 | 45 | if ( !$opt->{key} || !$opt->{desc} ) { | ||
4349 | 46 | die "Invalid DSN option: ", Dumper($opt); | ||
4350 | 47 | } | ||
4351 | 48 | PTDEBUG && _d('DSN option:', | ||
4352 | 49 | join(', ', | ||
4353 | 50 | map { "$_=" . (defined $opt->{$_} ? ($opt->{$_} || '') : 'undef') } | ||
4354 | 51 | keys %$opt | ||
4355 | 52 | ) | ||
4356 | 53 | ); | ||
4357 | 54 | $self->{opts}->{$opt->{key}} = { | ||
4358 | 55 | dsn => $opt->{dsn}, | ||
4359 | 56 | desc => $opt->{desc}, | ||
4360 | 57 | copy => $opt->{copy} || 0, | ||
4361 | 58 | }; | ||
4362 | 59 | } | ||
4363 | 60 | return bless $self, $class; | ||
4364 | 61 | } | ||
4365 | 62 | |||
4366 | 63 | sub prop { | ||
4367 | 64 | my ( $self, $prop, $value ) = @_; | ||
4368 | 65 | if ( @_ > 2 ) { | ||
4369 | 66 | PTDEBUG && _d('Setting', $prop, 'property'); | ||
4370 | 67 | $self->{$prop} = $value; | ||
4371 | 68 | } | ||
4372 | 69 | return $self->{$prop}; | ||
4373 | 70 | } | ||
4374 | 71 | |||
4375 | 72 | sub parse { | ||
4376 | 73 | my ( $self, $dsn, $prev, $defaults ) = @_; | ||
4377 | 74 | if ( !$dsn ) { | ||
4378 | 75 | PTDEBUG && _d('No DSN to parse'); | ||
4379 | 76 | return; | ||
4380 | 77 | } | ||
4381 | 78 | PTDEBUG && _d('Parsing', $dsn); | ||
4382 | 79 | $prev ||= {}; | ||
4383 | 80 | $defaults ||= {}; | ||
4384 | 81 | my %given_props; | ||
4385 | 82 | my %final_props; | ||
4386 | 83 | my $opts = $self->{opts}; | ||
4387 | 84 | |||
4388 | 85 | foreach my $dsn_part ( split(/,/, $dsn) ) { | ||
4389 | 86 | if ( my ($prop_key, $prop_val) = $dsn_part =~ m/^(.)=(.*)$/ ) { | ||
4390 | 87 | $given_props{$prop_key} = $prop_val; | ||
4391 | 88 | } | ||
4392 | 89 | else { | ||
4393 | 90 | PTDEBUG && _d('Interpreting', $dsn_part, 'as h=', $dsn_part); | ||
4394 | 91 | $given_props{h} = $dsn_part; | ||
4395 | 92 | } | ||
4396 | 93 | } | ||
4397 | 94 | |||
4398 | 95 | foreach my $key ( keys %$opts ) { | ||
4399 | 96 | PTDEBUG && _d('Finding value for', $key); | ||
4400 | 97 | $final_props{$key} = $given_props{$key}; | ||
4401 | 98 | if ( !defined $final_props{$key} | ||
4402 | 99 | && defined $prev->{$key} && $opts->{$key}->{copy} ) | ||
4403 | 100 | { | ||
4404 | 101 | $final_props{$key} = $prev->{$key}; | ||
4405 | 102 | PTDEBUG && _d('Copying value for', $key, 'from previous DSN'); | ||
4406 | 103 | } | ||
4407 | 104 | if ( !defined $final_props{$key} ) { | ||
4408 | 105 | $final_props{$key} = $defaults->{$key}; | ||
4409 | 106 | PTDEBUG && _d('Copying value for', $key, 'from defaults'); | ||
4410 | 107 | } | ||
4411 | 108 | } | ||
4412 | 109 | |||
4413 | 110 | foreach my $key ( keys %given_props ) { | ||
4414 | 111 | die "Unknown DSN option '$key' in '$dsn'. For more details, " | ||
4415 | 112 | . "please use the --help option, or try 'perldoc $PROGRAM_NAME' " | ||
4416 | 113 | . "for complete documentation." | ||
4417 | 114 | unless exists $opts->{$key}; | ||
4418 | 115 | } | ||
4419 | 116 | if ( (my $required = $self->prop('required')) ) { | ||
4420 | 117 | foreach my $key ( keys %$required ) { | ||
4421 | 118 | die "Missing required DSN option '$key' in '$dsn'. For more details, " | ||
4422 | 119 | . "please use the --help option, or try 'perldoc $PROGRAM_NAME' " | ||
4423 | 120 | . "for complete documentation." | ||
4424 | 121 | unless $final_props{$key}; | ||
4425 | 122 | } | ||
4426 | 123 | } | ||
4427 | 124 | |||
4428 | 125 | return \%final_props; | ||
4429 | 126 | } | ||
4430 | 127 | |||
4431 | 128 | sub parse_options { | ||
4432 | 129 | my ( $self, $o ) = @_; | ||
4433 | 130 | die 'I need an OptionParser object' unless ref $o eq 'OptionParser'; | ||
4434 | 131 | my $dsn_string | ||
4435 | 132 | = join(',', | ||
4436 | 133 | map { "$_=".$o->get($_); } | ||
4437 | 134 | grep { $o->has($_) && $o->get($_) } | ||
4438 | 135 | keys %{$self->{opts}} | ||
4439 | 136 | ); | ||
4440 | 137 | PTDEBUG && _d('DSN string made from options:', $dsn_string); | ||
4441 | 138 | return $self->parse($dsn_string); | ||
4442 | 139 | } | ||
4443 | 140 | |||
4444 | 141 | sub as_string { | ||
4445 | 142 | my ( $self, $dsn, $props ) = @_; | ||
4446 | 143 | return $dsn unless ref $dsn; | ||
4447 | 144 | my @keys = $props ? @$props : sort keys %$dsn; | ||
4448 | 145 | return join(',', | ||
4449 | 146 | map { "$_=" . ($_ eq 'p' ? '...' : $dsn->{$_}) } | ||
4450 | 147 | grep { | ||
4451 | 148 | exists $self->{opts}->{$_} | ||
4452 | 149 | && exists $dsn->{$_} | ||
4453 | 150 | && defined $dsn->{$_} | ||
4454 | 151 | } @keys); | ||
4455 | 152 | } | ||
4456 | 153 | |||
4457 | 154 | sub usage { | ||
4458 | 155 | my ( $self ) = @_; | ||
4459 | 156 | my $usage | ||
4460 | 157 | = "DSN syntax is key=value[,key=value...] Allowable DSN keys:\n\n" | ||
4461 | 158 | . " KEY COPY MEANING\n" | ||
4462 | 159 | . " === ==== =============================================\n"; | ||
4463 | 160 | my %opts = %{$self->{opts}}; | ||
4464 | 161 | foreach my $key ( sort keys %opts ) { | ||
4465 | 162 | $usage .= " $key " | ||
4466 | 163 | . ($opts{$key}->{copy} ? 'yes ' : 'no ') | ||
4467 | 164 | . ($opts{$key}->{desc} || '[No description]') | ||
4468 | 165 | . "\n"; | ||
4469 | 166 | } | ||
4470 | 167 | $usage .= "\n If the DSN is a bareword, the word is treated as the 'h' key.\n"; | ||
4471 | 168 | return $usage; | ||
4472 | 169 | } | ||
4473 | 170 | |||
4474 | 171 | sub get_cxn_params { | ||
4475 | 172 | my ( $self, $info ) = @_; | ||
4476 | 173 | my $dsn; | ||
4477 | 174 | my %opts = %{$self->{opts}}; | ||
4478 | 175 | my $driver = $self->prop('dbidriver') || ''; | ||
4479 | 176 | if ( $driver eq 'Pg' ) { | ||
4480 | 177 | $dsn = 'DBI:Pg:dbname=' . ( $info->{D} || '' ) . ';' | ||
4481 | 178 | . join(';', map { "$opts{$_}->{dsn}=$info->{$_}" } | ||
4482 | 179 | grep { defined $info->{$_} } | ||
4483 | 180 | qw(h P)); | ||
4484 | 181 | } | ||
4485 | 182 | else { | ||
4486 | 183 | $dsn = 'DBI:mysql:' . ( $info->{D} || '' ) . ';' | ||
4487 | 184 | . join(';', map { "$opts{$_}->{dsn}=$info->{$_}" } | ||
4488 | 185 | grep { defined $info->{$_} } | ||
4489 | 186 | qw(F h P S A)) | ||
4490 | 187 | . ';mysql_read_default_group=client'; | ||
4491 | 188 | } | ||
4492 | 189 | PTDEBUG && _d($dsn); | ||
4493 | 190 | return ($dsn, $info->{u}, $info->{p}); | ||
4494 | 191 | } | ||
4495 | 192 | |||
4496 | 193 | sub fill_in_dsn { | ||
4497 | 194 | my ( $self, $dbh, $dsn ) = @_; | ||
4498 | 195 | my $vars = $dbh->selectall_hashref('SHOW VARIABLES', 'Variable_name'); | ||
4499 | 196 | my ($user, $db) = $dbh->selectrow_array('SELECT USER(), DATABASE()'); | ||
4500 | 197 | $user =~ s/@.*//; | ||
4501 | 198 | $dsn->{h} ||= $vars->{hostname}->{Value}; | ||
4502 | 199 | $dsn->{S} ||= $vars->{'socket'}->{Value}; | ||
4503 | 200 | $dsn->{P} ||= $vars->{port}->{Value}; | ||
4504 | 201 | $dsn->{u} ||= $user; | ||
4505 | 202 | $dsn->{D} ||= $db; | ||
4506 | 203 | } | ||
4507 | 204 | |||
4508 | 205 | sub get_dbh { | ||
4509 | 206 | my ( $self, $cxn_string, $user, $pass, $opts ) = @_; | ||
4510 | 207 | $opts ||= {}; | ||
4511 | 208 | my $defaults = { | ||
4512 | 209 | AutoCommit => 0, | ||
4513 | 210 | RaiseError => 1, | ||
4514 | 211 | PrintError => 0, | ||
4515 | 212 | ShowErrorStatement => 1, | ||
4516 | 213 | mysql_enable_utf8 => ($cxn_string =~ m/charset=utf8/i ? 1 : 0), | ||
4517 | 214 | }; | ||
4518 | 215 | @{$defaults}{ keys %$opts } = values %$opts; | ||
4519 | 216 | |||
4520 | 217 | if ( $opts->{mysql_use_result} ) { | ||
4521 | 218 | $defaults->{mysql_use_result} = 1; | ||
4522 | 219 | } | ||
4523 | 220 | |||
4524 | 221 | if ( !$have_dbi ) { | ||
4525 | 222 | die "Cannot connect to MySQL because the Perl DBI module is not " | ||
4526 | 223 | . "installed or not found. Run 'perl -MDBI' to see the directories " | ||
4527 | 224 | . "that Perl searches for DBI. If DBI is not installed, try:\n" | ||
4528 | 225 | . " Debian/Ubuntu apt-get install libdbi-perl\n" | ||
4529 | 226 | . " RHEL/CentOS yum install perl-DBI\n" | ||
4530 | 227 | . " OpenSolaris pgk install pkg:/SUNWpmdbi\n"; | ||
4531 | 228 | |||
4532 | 229 | } | ||
4533 | 230 | |||
4534 | 231 | my $dbh; | ||
4535 | 232 | my $tries = 2; | ||
4536 | 233 | while ( !$dbh && $tries-- ) { | ||
4537 | 234 | PTDEBUG && _d($cxn_string, ' ', $user, ' ', $pass, | ||
4538 | 235 | join(', ', map { "$_=>$defaults->{$_}" } keys %$defaults )); | ||
4539 | 236 | |||
4540 | 237 | eval { | ||
4541 | 238 | $dbh = DBI->connect($cxn_string, $user, $pass, $defaults); | ||
4542 | 239 | |||
4543 | 240 | if ( $cxn_string =~ m/mysql/i ) { | ||
4544 | 241 | my $sql; | ||
4545 | 242 | |||
4546 | 243 | $sql = 'SELECT @@SQL_MODE'; | ||
4547 | 244 | PTDEBUG && _d($dbh, $sql); | ||
4548 | 245 | my ($sql_mode) = $dbh->selectrow_array($sql); | ||
4549 | 246 | |||
4550 | 247 | $sql = 'SET @@SQL_QUOTE_SHOW_CREATE = 1' | ||
4551 | 248 | . '/*!40101, @@SQL_MODE=\'NO_AUTO_VALUE_ON_ZERO' | ||
4552 | 249 | . ($sql_mode ? ",$sql_mode" : '') | ||
4553 | 250 | . '\'*/'; | ||
4554 | 251 | PTDEBUG && _d($dbh, $sql); | ||
4555 | 252 | $dbh->do($sql); | ||
4556 | 253 | |||
4557 | 254 | if ( my ($charset) = $cxn_string =~ m/charset=(\w+)/ ) { | ||
4558 | 255 | $sql = "/*!40101 SET NAMES $charset*/"; | ||
4559 | 256 | PTDEBUG && _d($dbh, ':', $sql); | ||
4560 | 257 | $dbh->do($sql); | ||
4561 | 258 | PTDEBUG && _d('Enabling charset for STDOUT'); | ||
4562 | 259 | if ( $charset eq 'utf8' ) { | ||
4563 | 260 | binmode(STDOUT, ':utf8') | ||
4564 | 261 | or die "Can't binmode(STDOUT, ':utf8'): $OS_ERROR"; | ||
4565 | 262 | } | ||
4566 | 263 | else { | ||
4567 | 264 | binmode(STDOUT) or die "Can't binmode(STDOUT): $OS_ERROR"; | ||
4568 | 265 | } | ||
4569 | 266 | } | ||
4570 | 267 | |||
4571 | 268 | if ( $self->prop('set-vars') ) { | ||
4572 | 269 | $sql = "SET " . $self->prop('set-vars'); | ||
4573 | 270 | PTDEBUG && _d($dbh, ':', $sql); | ||
4574 | 271 | $dbh->do($sql); | ||
4575 | 272 | } | ||
4576 | 273 | } | ||
4577 | 274 | }; | ||
4578 | 275 | if ( !$dbh && $EVAL_ERROR ) { | ||
4579 | 276 | PTDEBUG && _d($EVAL_ERROR); | ||
4580 | 277 | if ( $EVAL_ERROR =~ m/not a compiled character set|character set utf8/ ) { | ||
4581 | 278 | PTDEBUG && _d('Going to try again without utf8 support'); | ||
4582 | 279 | delete $defaults->{mysql_enable_utf8}; | ||
4583 | 280 | } | ||
4584 | 281 | elsif ( $EVAL_ERROR =~ m/locate DBD\/mysql/i ) { | ||
4585 | 282 | die "Cannot connect to MySQL because the Perl DBD::mysql module is " | ||
4586 | 283 | . "not installed or not found. Run 'perl -MDBD::mysql' to see " | ||
4587 | 284 | . "the directories that Perl searches for DBD::mysql. If " | ||
4588 | 285 | . "DBD::mysql is not installed, try:\n" | ||
4589 | 286 | . " Debian/Ubuntu apt-get install libdbd-mysql-perl\n" | ||
4590 | 287 | . " RHEL/CentOS yum install perl-DBD-MySQL\n" | ||
4591 | 288 | . " OpenSolaris pgk install pkg:/SUNWapu13dbd-mysql\n"; | ||
4592 | 289 | } | ||
4593 | 290 | if ( !$tries ) { | ||
4594 | 291 | die $EVAL_ERROR; | ||
4595 | 292 | } | ||
4596 | 293 | } | ||
4597 | 294 | } | ||
4598 | 295 | |||
4599 | 296 | PTDEBUG && _d('DBH info: ', | ||
4600 | 297 | $dbh, | ||
4601 | 298 | Dumper($dbh->selectrow_hashref( | ||
4602 | 299 | 'SELECT DATABASE(), CONNECTION_ID(), VERSION()/*!50038 , @@hostname*/')), | ||
4603 | 300 | 'Connection info:', $dbh->{mysql_hostinfo}, | ||
4604 | 301 | 'Character set info:', Dumper($dbh->selectall_arrayref( | ||
4605 | 302 | 'SHOW VARIABLES LIKE "character_set%"', { Slice => {}})), | ||
4606 | 303 | '$DBD::mysql::VERSION:', $DBD::mysql::VERSION, | ||
4607 | 304 | '$DBI::VERSION:', $DBI::VERSION, | ||
4608 | 305 | ); | ||
4609 | 306 | |||
4610 | 307 | return $dbh; | ||
4611 | 308 | } | ||
4612 | 309 | |||
4613 | 310 | sub get_hostname { | ||
4614 | 311 | my ( $self, $dbh ) = @_; | ||
4615 | 312 | if ( my ($host) = ($dbh->{mysql_hostinfo} || '') =~ m/^(\w+) via/ ) { | ||
4616 | 313 | return $host; | ||
4617 | 314 | } | ||
4618 | 315 | my ( $hostname, $one ) = $dbh->selectrow_array( | ||
4619 | 316 | 'SELECT /*!50038 @@hostname, */ 1'); | ||
4620 | 317 | return $hostname; | ||
4621 | 318 | } | ||
4622 | 319 | |||
4623 | 320 | sub disconnect { | ||
4624 | 321 | my ( $self, $dbh ) = @_; | ||
4625 | 322 | PTDEBUG && $self->print_active_handles($dbh); | ||
4626 | 323 | $dbh->disconnect; | ||
4627 | 324 | } | ||
4628 | 325 | |||
4629 | 326 | sub print_active_handles { | ||
4630 | 327 | my ( $self, $thing, $level ) = @_; | ||
4631 | 328 | $level ||= 0; | ||
4632 | 329 | printf("# Active %sh: %s %s %s\n", ($thing->{Type} || 'undef'), "\t" x $level, | ||
4633 | 330 | $thing, (($thing->{Type} || '') eq 'st' ? $thing->{Statement} || '' : '')) | ||
4634 | 331 | or die "Cannot print: $OS_ERROR"; | ||
4635 | 332 | foreach my $handle ( grep {defined} @{ $thing->{ChildHandles} } ) { | ||
4636 | 333 | $self->print_active_handles( $handle, $level + 1 ); | ||
4637 | 334 | } | ||
4638 | 335 | } | ||
4639 | 336 | |||
4640 | 337 | sub copy { | ||
4641 | 338 | my ( $self, $dsn_1, $dsn_2, %args ) = @_; | ||
4642 | 339 | die 'I need a dsn_1 argument' unless $dsn_1; | ||
4643 | 340 | die 'I need a dsn_2 argument' unless $dsn_2; | ||
4644 | 341 | my %new_dsn = map { | ||
4645 | 342 | my $key = $_; | ||
4646 | 343 | my $val; | ||
4647 | 344 | if ( $args{overwrite} ) { | ||
4648 | 345 | $val = defined $dsn_1->{$key} ? $dsn_1->{$key} : $dsn_2->{$key}; | ||
4649 | 346 | } | ||
4650 | 347 | else { | ||
4651 | 348 | $val = defined $dsn_2->{$key} ? $dsn_2->{$key} : $dsn_1->{$key}; | ||
4652 | 349 | } | ||
4653 | 350 | $key => $val; | ||
4654 | 351 | } keys %{$self->{opts}}; | ||
4655 | 352 | return \%new_dsn; | ||
4656 | 353 | } | ||
4657 | 354 | |||
4658 | 355 | sub _d { | ||
4659 | 356 | my ($package, undef, $line) = caller 0; | ||
4660 | 357 | @_ = map { (my $temp = $_) =~ s/\n/\n# /g; $temp; } | ||
4661 | 358 | map { defined $_ ? $_ : 'undef' } | ||
4662 | 359 | @_; | ||
4663 | 360 | print STDERR "# $package:$line $PID ", join(' ', @_), "\n"; | ||
4664 | 361 | } | ||
4665 | 362 | |||
4666 | 363 | 1; | ||
4667 | 364 | } | ||
4668 | 365 | # ########################################################################### | ||
4669 | 366 | # End DSNParser package | ||
4670 | 367 | # ########################################################################### | ||
4671 | 368 | |||
4672 | 369 | # ########################################################################### | ||
4673 | 370 | # OptionParser package | ||
4674 | 371 | # This package is a copy without comments from the original. The original | ||
4675 | 372 | # with comments and its test file can be found in the Bazaar repository at, | ||
4676 | 373 | # lib/OptionParser.pm | ||
4677 | 374 | # t/lib/OptionParser.t | ||
4678 | 375 | # See https://launchpad.net/percona-toolkit for more information. | ||
4679 | 376 | # ########################################################################### | ||
4680 | 377 | { | ||
4681 | 378 | package OptionParser; | ||
4682 | 379 | |||
4683 | 380 | use strict; | ||
4684 | 381 | use warnings FATAL => 'all'; | ||
4685 | 382 | use English qw(-no_match_vars); | ||
4686 | 383 | use constant PTDEBUG => $ENV{PTDEBUG} || 0; | ||
4687 | 384 | |||
4688 | 385 | use List::Util qw(max); | ||
4689 | 386 | use Getopt::Long; | ||
4690 | 387 | |||
4691 | 388 | my $POD_link_re = '[LC]<"?([^">]+)"?>'; | ||
4692 | 389 | |||
4693 | 390 | sub new { | ||
4694 | 391 | my ( $class, %args ) = @_; | ||
4695 | 392 | my @required_args = qw(); | ||
4696 | 393 | foreach my $arg ( @required_args ) { | ||
4697 | 394 | die "I need a $arg argument" unless $args{$arg}; | ||
4698 | 395 | } | ||
4699 | 396 | |||
4700 | 397 | my ($program_name) = $PROGRAM_NAME =~ m/([.A-Za-z-]+)$/; | ||
4701 | 398 | $program_name ||= $PROGRAM_NAME; | ||
4702 | 399 | my $home = $ENV{HOME} || $ENV{HOMEPATH} || $ENV{USERPROFILE} || '.'; | ||
4703 | 400 | |||
4704 | 401 | my %attributes = ( | ||
4705 | 402 | 'type' => 1, | ||
4706 | 403 | 'short form' => 1, | ||
4707 | 404 | 'group' => 1, | ||
4708 | 405 | 'default' => 1, | ||
4709 | 406 | 'cumulative' => 1, | ||
4710 | 407 | 'negatable' => 1, | ||
4711 | 408 | ); | ||
4712 | 409 | |||
4713 | 410 | my $self = { | ||
4714 | 411 | head1 => 'OPTIONS', # These args are used internally | ||
4715 | 412 | skip_rules => 0, # to instantiate another Option- | ||
4716 | 413 | item => '--(.*)', # Parser obj that parses the | ||
4717 | 414 | attributes => \%attributes, # DSN OPTIONS section. Tools | ||
4718 | 415 | parse_attributes => \&_parse_attribs, # don't tinker with these args. | ||
4719 | 416 | |||
4720 | 417 | %args, | ||
4721 | 418 | |||
4722 | 419 | strict => 1, # disabled by a special rule | ||
4723 | 420 | program_name => $program_name, | ||
4724 | 421 | opts => {}, | ||
4725 | 422 | got_opts => 0, | ||
4726 | 423 | short_opts => {}, | ||
4727 | 424 | defaults => {}, | ||
4728 | 425 | groups => {}, | ||
4729 | 426 | allowed_groups => {}, | ||
4730 | 427 | errors => [], | ||
4731 | 428 | rules => [], # desc of rules for --help | ||
4732 | 429 | mutex => [], # rule: opts are mutually exclusive | ||
4733 | 430 | atleast1 => [], # rule: at least one opt is required | ||
4734 | 431 | disables => {}, # rule: opt disables other opts | ||
4735 | 432 | defaults_to => {}, # rule: opt defaults to value of other opt | ||
4736 | 433 | DSNParser => undef, | ||
4737 | 434 | default_files => [ | ||
4738 | 435 | "/etc/percona-toolkit/percona-toolkit.conf", | ||
4739 | 436 | "/etc/percona-toolkit/$program_name.conf", | ||
4740 | 437 | "$home/.percona-toolkit.conf", | ||
4741 | 438 | "$home/.$program_name.conf", | ||
4742 | 439 | ], | ||
4743 | 440 | types => { | ||
4744 | 441 | string => 's', # standard Getopt type | ||
4745 | 442 | int => 'i', # standard Getopt type | ||
4746 | 443 | float => 'f', # standard Getopt type | ||
4747 | 444 | Hash => 'H', # hash, formed from a comma-separated list | ||
4748 | 445 | hash => 'h', # hash as above, but only if a value is given | ||
4749 | 446 | Array => 'A', # array, similar to Hash | ||
4750 | 447 | array => 'a', # array, similar to hash | ||
4751 | 448 | DSN => 'd', # DSN | ||
4752 | 449 | size => 'z', # size with kMG suffix (powers of 2^10) | ||
4753 | 450 | time => 'm', # time, with an optional suffix of s/h/m/d | ||
4754 | 451 | }, | ||
4755 | 452 | }; | ||
4756 | 453 | |||
4757 | 454 | return bless $self, $class; | ||
4758 | 455 | } | ||
4759 | 456 | |||
4760 | 457 | sub get_specs { | ||
4761 | 458 | my ( $self, $file ) = @_; | ||
4762 | 459 | $file ||= $self->{file} || __FILE__; | ||
4763 | 460 | my @specs = $self->_pod_to_specs($file); | ||
4764 | 461 | $self->_parse_specs(@specs); | ||
4765 | 462 | |||
4766 | 463 | open my $fh, "<", $file or die "Cannot open $file: $OS_ERROR"; | ||
4767 | 464 | my $contents = do { local $/ = undef; <$fh> }; | ||
4768 | 465 | close $fh; | ||
4769 | 466 | if ( $contents =~ m/^=head1 DSN OPTIONS/m ) { | ||
4770 | 467 | PTDEBUG && _d('Parsing DSN OPTIONS'); | ||
4771 | 468 | my $dsn_attribs = { | ||
4772 | 469 | dsn => 1, | ||
4773 | 470 | copy => 1, | ||
4774 | 471 | }; | ||
4775 | 472 | my $parse_dsn_attribs = sub { | ||
4776 | 473 | my ( $self, $option, $attribs ) = @_; | ||
4777 | 474 | map { | ||
4778 | 475 | my $val = $attribs->{$_}; | ||
4779 | 476 | if ( $val ) { | ||
4780 | 477 | $val = $val eq 'yes' ? 1 | ||
4781 | 478 | : $val eq 'no' ? 0 | ||
4782 | 479 | : $val; | ||
4783 | 480 | $attribs->{$_} = $val; | ||
4784 | 481 | } | ||
4785 | 482 | } keys %$attribs; | ||
4786 | 483 | return { | ||
4787 | 484 | key => $option, | ||
4788 | 485 | %$attribs, | ||
4789 | 486 | }; | ||
4790 | 487 | }; | ||
4791 | 488 | my $dsn_o = new OptionParser( | ||
4792 | 489 | description => 'DSN OPTIONS', | ||
4793 | 490 | head1 => 'DSN OPTIONS', | ||
4794 | 491 | dsn => 0, # XXX don't infinitely recurse! | ||
4795 | 492 | item => '\* (.)', # key opts are a single character | ||
4796 | 493 | skip_rules => 1, # no rules before opts | ||
4797 | 494 | attributes => $dsn_attribs, | ||
4798 | 495 | parse_attributes => $parse_dsn_attribs, | ||
4799 | 496 | ); | ||
4800 | 497 | my @dsn_opts = map { | ||
4801 | 498 | my $opts = { | ||
4802 | 499 | key => $_->{spec}->{key}, | ||
4803 | 500 | dsn => $_->{spec}->{dsn}, | ||
4804 | 501 | copy => $_->{spec}->{copy}, | ||
4805 | 502 | desc => $_->{desc}, | ||
4806 | 503 | }; | ||
4807 | 504 | $opts; | ||
4808 | 505 | } $dsn_o->_pod_to_specs($file); | ||
4809 | 506 | $self->{DSNParser} = DSNParser->new(opts => \@dsn_opts); | ||
4810 | 507 | } | ||
4811 | 508 | |||
4812 | 509 | if ( $contents =~ m/^=head1 VERSION\n\n^(.+)$/m ) { | ||
4813 | 510 | $self->{version} = $1; | ||
4814 | 511 | PTDEBUG && _d($self->{version}); | ||
4815 | 512 | } | ||
4816 | 513 | |||
4817 | 514 | return; | ||
4818 | 515 | } | ||
4819 | 516 | |||
4820 | 517 | sub DSNParser { | ||
4821 | 518 | my ( $self ) = @_; | ||
4822 | 519 | return $self->{DSNParser}; | ||
4823 | 520 | }; | ||
4824 | 521 | |||
4825 | 522 | sub get_defaults_files { | ||
4826 | 523 | my ( $self ) = @_; | ||
4827 | 524 | return @{$self->{default_files}}; | ||
4828 | 525 | } | ||
4829 | 526 | |||
4830 | 527 | sub _pod_to_specs { | ||
4831 | 528 | my ( $self, $file ) = @_; | ||
4832 | 529 | $file ||= $self->{file} || __FILE__; | ||
4833 | 530 | open my $fh, '<', $file or die "Cannot open $file: $OS_ERROR"; | ||
4834 | 531 | |||
4835 | 532 | my @specs = (); | ||
4836 | 533 | my @rules = (); | ||
4837 | 534 | my $para; | ||
4838 | 535 | |||
4839 | 536 | local $INPUT_RECORD_SEPARATOR = ''; | ||
4840 | 537 | while ( $para = <$fh> ) { | ||
4841 | 538 | next unless $para =~ m/^=head1 $self->{head1}/; | ||
4842 | 539 | last; | ||
4843 | 540 | } | ||
4844 | 541 | |||
4845 | 542 | while ( $para = <$fh> ) { | ||
4846 | 543 | last if $para =~ m/^=over/; | ||
4847 | 544 | next if $self->{skip_rules}; | ||
4848 | 545 | chomp $para; | ||
4849 | 546 | $para =~ s/\s+/ /g; | ||
4850 | 547 | $para =~ s/$POD_link_re/$1/go; | ||
4851 | 548 | PTDEBUG && _d('Option rule:', $para); | ||
4852 | 549 | push @rules, $para; | ||
4853 | 550 | } | ||
4854 | 551 | |||
4855 | 552 | die "POD has no $self->{head1} section" unless $para; | ||
4856 | 553 | |||
4857 | 554 | do { | ||
4858 | 555 | if ( my ($option) = $para =~ m/^=item $self->{item}/ ) { | ||
4859 | 556 | chomp $para; | ||
4860 | 557 | PTDEBUG && _d($para); | ||
4861 | 558 | my %attribs; | ||
4862 | 559 | |||
4863 | 560 | $para = <$fh>; # read next paragraph, possibly attributes | ||
4864 | 561 | |||
4865 | 562 | if ( $para =~ m/: / ) { # attributes | ||
4866 | 563 | $para =~ s/\s+\Z//g; | ||
4867 | 564 | %attribs = map { | ||
4868 | 565 | my ( $attrib, $val) = split(/: /, $_); | ||
4869 | 566 | die "Unrecognized attribute for --$option: $attrib" | ||
4870 | 567 | unless $self->{attributes}->{$attrib}; | ||
4871 | 568 | ($attrib, $val); | ||
4872 | 569 | } split(/; /, $para); | ||
4873 | 570 | if ( $attribs{'short form'} ) { | ||
4874 | 571 | $attribs{'short form'} =~ s/-//; | ||
4875 | 572 | } | ||
4876 | 573 | $para = <$fh>; # read next paragraph, probably short help desc | ||
4877 | 574 | } | ||
4878 | 575 | else { | ||
4879 | 576 | PTDEBUG && _d('Option has no attributes'); | ||
4880 | 577 | } | ||
4881 | 578 | |||
4882 | 579 | $para =~ s/\s+\Z//g; | ||
4883 | 580 | $para =~ s/\s+/ /g; | ||
4884 | 581 | $para =~ s/$POD_link_re/$1/go; | ||
4885 | 582 | |||
4886 | 583 | $para =~ s/\.(?:\n.*| [A-Z].*|\Z)//s; | ||
4887 | 584 | PTDEBUG && _d('Short help:', $para); | ||
4888 | 585 | |||
4889 | 586 | die "No description after option spec $option" if $para =~ m/^=item/; | ||
4890 | 587 | |||
4891 | 588 | if ( my ($base_option) = $option =~ m/^\[no\](.*)/ ) { | ||
4892 | 589 | $option = $base_option; | ||
4893 | 590 | $attribs{'negatable'} = 1; | ||
4894 | 591 | } | ||
4895 | 592 | |||
4896 | 593 | push @specs, { | ||
4897 | 594 | spec => $self->{parse_attributes}->($self, $option, \%attribs), | ||
4898 | 595 | desc => $para | ||
4899 | 596 | . (defined $attribs{default} ? " (default $attribs{default})" : ''), | ||
4900 | 597 | group => ($attribs{'group'} ? $attribs{'group'} : 'default'), | ||
4901 | 598 | }; | ||
4902 | 599 | } | ||
4903 | 600 | while ( $para = <$fh> ) { | ||
4904 | 601 | last unless $para; | ||
4905 | 602 | if ( $para =~ m/^=head1/ ) { | ||
4906 | 603 | $para = undef; # Can't 'last' out of a do {} block. | ||
4907 | 604 | last; | ||
4908 | 605 | } | ||
4909 | 606 | last if $para =~ m/^=item /; | ||
4910 | 607 | } | ||
4911 | 608 | } while ( $para ); | ||
4912 | 609 | |||
4913 | 610 | die "No valid specs in $self->{head1}" unless @specs; | ||
4914 | 611 | |||
4915 | 612 | close $fh; | ||
4916 | 613 | return @specs, @rules; | ||
4917 | 614 | } | ||
4918 | 615 | |||
4919 | 616 | sub _parse_specs { | ||
4920 | 617 | my ( $self, @specs ) = @_; | ||
4921 | 618 | my %disables; # special rule that requires deferred checking | ||
4922 | 619 | |||
4923 | 620 | foreach my $opt ( @specs ) { | ||
4924 | 621 | if ( ref $opt ) { # It's an option spec, not a rule. | ||
4925 | 622 | PTDEBUG && _d('Parsing opt spec:', | ||
4926 | 623 | map { ($_, '=>', $opt->{$_}) } keys %$opt); | ||
4927 | 624 | |||
4928 | 625 | my ( $long, $short ) = $opt->{spec} =~ m/^([\w-]+)(?:\|([^!+=]*))?/; | ||
4929 | 626 | if ( !$long ) { | ||
4930 | 627 | die "Cannot parse long option from spec $opt->{spec}"; | ||
4931 | 628 | } | ||
4932 | 629 | $opt->{long} = $long; | ||
4933 | 630 | |||
4934 | 631 | die "Duplicate long option --$long" if exists $self->{opts}->{$long}; | ||
4935 | 632 | $self->{opts}->{$long} = $opt; | ||
4936 | 633 | |||
4937 | 634 | if ( length $long == 1 ) { | ||
4938 | 635 | PTDEBUG && _d('Long opt', $long, 'looks like short opt'); | ||
4939 | 636 | $self->{short_opts}->{$long} = $long; | ||
4940 | 637 | } | ||
4941 | 638 | |||
4942 | 639 | if ( $short ) { | ||
4943 | 640 | die "Duplicate short option -$short" | ||
4944 | 641 | if exists $self->{short_opts}->{$short}; | ||
4945 | 642 | $self->{short_opts}->{$short} = $long; | ||
4946 | 643 | $opt->{short} = $short; | ||
4947 | 644 | } | ||
4948 | 645 | else { | ||
4949 | 646 | $opt->{short} = undef; | ||
4950 | 647 | } | ||
4951 | 648 | |||
4952 | 649 | $opt->{is_negatable} = $opt->{spec} =~ m/!/ ? 1 : 0; | ||
4953 | 650 | $opt->{is_cumulative} = $opt->{spec} =~ m/\+/ ? 1 : 0; | ||
4954 | 651 | $opt->{is_required} = $opt->{desc} =~ m/required/ ? 1 : 0; | ||
4955 | 652 | |||
4956 | 653 | $opt->{group} ||= 'default'; | ||
4957 | 654 | $self->{groups}->{ $opt->{group} }->{$long} = 1; | ||
4958 | 655 | |||
4959 | 656 | $opt->{value} = undef; | ||
4960 | 657 | $opt->{got} = 0; | ||
4961 | 658 | |||
4962 | 659 | my ( $type ) = $opt->{spec} =~ m/=(.)/; | ||
4963 | 660 | $opt->{type} = $type; | ||
4964 | 661 | PTDEBUG && _d($long, 'type:', $type); | ||
4965 | 662 | |||
4966 | 663 | |||
4967 | 664 | $opt->{spec} =~ s/=./=s/ if ( $type && $type =~ m/[HhAadzm]/ ); | ||
4968 | 665 | |||
4969 | 666 | if ( (my ($def) = $opt->{desc} =~ m/default\b(?: ([^)]+))?/) ) { | ||
4970 | 667 | $self->{defaults}->{$long} = defined $def ? $def : 1; | ||
4971 | 668 | PTDEBUG && _d($long, 'default:', $def); | ||
4972 | 669 | } | ||
4973 | 670 | |||
4974 | 671 | if ( $long eq 'config' ) { | ||
4975 | 672 | $self->{defaults}->{$long} = join(',', $self->get_defaults_files()); | ||
4976 | 673 | } | ||
4977 | 674 | |||
4978 | 675 | if ( (my ($dis) = $opt->{desc} =~ m/(disables .*)/) ) { | ||
4979 | 676 | $disables{$long} = $dis; | ||
4980 | 677 | PTDEBUG && _d('Deferring check of disables rule for', $opt, $dis); | ||
4981 | 678 | } | ||
4982 | 679 | |||
4983 | 680 | $self->{opts}->{$long} = $opt; | ||
4984 | 681 | } | ||
4985 | 682 | else { # It's an option rule, not a spec. | ||
4986 | 683 | PTDEBUG && _d('Parsing rule:', $opt); | ||
4987 | 684 | push @{$self->{rules}}, $opt; | ||
4988 | 685 | my @participants = $self->_get_participants($opt); | ||
4989 | 686 | my $rule_ok = 0; | ||
4990 | 687 | |||
4991 | 688 | if ( $opt =~ m/mutually exclusive|one and only one/ ) { | ||
4992 | 689 | $rule_ok = 1; | ||
4993 | 690 | push @{$self->{mutex}}, \@participants; | ||
4994 | 691 | PTDEBUG && _d(@participants, 'are mutually exclusive'); | ||
4995 | 692 | } | ||
4996 | 693 | if ( $opt =~ m/at least one|one and only one/ ) { | ||
4997 | 694 | $rule_ok = 1; | ||
4998 | 695 | push @{$self->{atleast1}}, \@participants; | ||
4999 | 696 | PTDEBUG && _d(@participants, 'require at least one'); | ||
5000 | 697 | } |
The diff has been truncated for viewing.