percona/percona-toolkit
1
0
Fork 0
mirror of https://github.com/percona/percona-toolkit.git synced 2025-09-02 02:34:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Have ResultWriter.pm save the current db.

Browse Source
This commit is contained in:
Daniel Nichter
2013-02-20 09:37:43 -07:00
parent 144b4794b9
commit a4d771e991
9 changed files with 50 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
lib/ResultWriter.pm
View File

@@ -42,6 +42,18 @@ has 'pretty' => (
default => 0,
);
has 'default_database' => (
is => 'rw',
isa => 'Maybe[Str]',
required => 0,
);
has 'current_database' => (
is => 'rw',
isa => 'Maybe[Str]',
required => 0,
);
has '_query_fh' => (
is => 'rw',
isa => 'Maybe[FileHandle]',
@@ -92,10 +104,18 @@ sub BUILDARGS {
sub save {
my ($self, %args) = @_;
my $host = $args{host};
my $event = $args{event};
my $results = $args{results};
# Save the query.
my $current_db = $self->current_database;
my $db = $event->{db} || $event->{Schema} || $self->default_database;
if ( $db && (!$current_db || $current_db ne $db) ) {
PTDEBUG && _d('New current db:', $db);
print { $self->_query_fh } "use `$db`;\n";
$self->current_database($db);
}
print { $self->_query_fh } $event->{arg}, "\n##\n";
if ( my $error = $results->{error} ) {