X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=lib%2FDBIx%2FClass%2FStorage%2FDebug%2FPrettyPrint.pm;h=59ffe420e65dfcf1e4a2ae24ccc29f680ce7d4cd;hb=2924f8b3ce58b66565f0e733b9daeba416cbf74a;hp=94fcadc66430f40b38559fa334d3e954f107cf89;hpb=a2b743ce05a705f3795376649b2511601a92ca55;p=dbsrgits%2FSQL-Abstract.git

diff --git a/lib/DBIx/Class/Storage/Debug/PrettyPrint.pm b/lib/DBIx/Class/Storage/Debug/PrettyPrint.pm
index 94fcadc..59ffe42 100644
--- a/lib/DBIx/Class/Storage/Debug/PrettyPrint.pm
+++ b/lib/DBIx/Class/Storage/Debug/PrettyPrint.pm
@@ -11,25 +11,33 @@ __PACKAGE__->mk_group_accessors( simple => '_sqlat' );
 __PACKAGE__->mk_group_accessors( simple => '_clear_line_str' );
 __PACKAGE__->mk_group_accessors( simple => '_executing_str' );
 __PACKAGE__->mk_group_accessors( simple => '_show_progress' );
+__PACKAGE__->mk_group_accessors( simple => '_last_sql' );
+__PACKAGE__->mk_group_accessors( simple => 'squash_repeats' );
 
 sub new {
    my $class = shift;
    my $args  = shift;
 
    my $clear_line = $args->{clear_line} || "\r";
-   my $executing  = $args->{executing}  || eval { require Term::ANSIColor } ? do {
-       my $c = \&Term::ANSIColor::color;
-       $c->('blink white on_black') . 'EXECUTING...' . $c->('reset');;
-   } : 'EXECUTING...';
-   my $show_progress = defined $args->{show_progress} ? $args->{show_progress} : 1;
-
+   my $executing  = $args->{executing}  || (
+      eval { require Term::ANSIColor } ? do {
+          my $c = \&Term::ANSIColor::color;
+          $c->('blink white on_black') . 'EXECUTING...' . $c->('reset');
+      } : 'EXECUTING...'
+   );
+   my $show_progress = $args->{show_progress};
+
+   my $squash_repeats = $args->{squash_repeats};
    my $sqlat = SQL::Abstract::Tree->new($args);
    my $self = $class->next::method(@_);
    $self->_clear_line_str($clear_line);
    $self->_executing_str($executing);
    $self->_show_progress($show_progress);
 
+   $self->squash_repeats($squash_repeats);
+
    $self->_sqlat($sqlat);
+   $self->_last_sql('');
 
    return $self
 }
@@ -42,18 +50,25 @@ sub print {
   my ($lw, $lr);
   ($lw, $string, $lr) = $string =~ /^(\s*)(.+?)(\s*)$/s;
 
-  return if defined $bindargs && defined $bindargs->[0] &&
-    $bindargs->[0] eq q('__BULK_INSERT__');
+  local $self->_sqlat->{fill_in_placeholders} = 0 if defined $bindargs
+    && defined $bindargs->[0] && $bindargs->[0] eq q('__BULK_INSERT__');
 
   my $use_placeholders = !!$self->_sqlat->fill_in_placeholders;
 
   # DBIC pre-quotes bindargs
   $bindargs = [map { s/^'//; s/'$//; $_ } @{$bindargs}] if $use_placeholders;
 
-  my $formatted = $self->_sqlat->format($string, $bindargs);
-
-  $formatted = "$formatted: " . join ', ', @{$bindargs}
-     unless $use_placeholders;
+  my $sqlat = $self->_sqlat;
+  my $formatted;
+  if ($self->squash_repeats && $self->_last_sql eq $string) {
+     my ( $l, $r ) = @{ $sqlat->placeholder_surround };
+     $formatted = '... : ' . join(', ', map "$l$_$r", @$bindargs)
+  } else {
+     $self->_last_sql($string);
+     $formatted = $sqlat->format($string, $bindargs);
+     $formatted = "$formatted : " . join ', ', @{$bindargs}
+        unless $use_placeholders;
+  }
 
   $self->next::method("$lw$formatted$lr", @_);
 }
@@ -84,25 +99,33 @@ sub query_end {
 
 =head1 SYNOPSIS
 
- package MyApp::Schema;
+ DBIC_TRACE_PROFILE=~/dbic.json perl -Ilib ./foo.pl
+
+Where dbic.json contains:
 
- use parent 'DBIx::Class::Schema';
+ {
+   "profile":"console",
+   "show_progress":1,
+   "squash_repeats":1
+ }
 
- use DBIx::Class::Storage::Debug::PrettyPrint;
+=head1 METHODS
 
- __PACKAGE__->load_namespaces;
+=head2 new
 
  my $pp = DBIx::Class::Storage::Debug::PrettyPrint->new({
-   profile => 'console',
+   show_progress  => 1,        # tries it's best to make it clear that a SQL
+                               # statement is still running
+   executing      => '...',    # the string that is added to the end of SQL
+                               # if show_progress is on.  You probably don't
+                               # need to set this
+   clear_line     => '\r^[[J', # the string used to erase the string added
+                               # to SQL if show_progress is on.  Again, the
+                               # default is probably good enough.
+
+   squash_repeats => 1,        # set to true to make repeated SQL queries
+                               # be ellided and only show the new bind params
+   # any other args are passed through directly to SQL::Abstract::Tree
  });
 
- sub connection {
-   my $self = shift;
-
-   my $ret = $self->next::method(@_);
-
-   $self->storage->debugobj($pp);
-
-   $ret
- }