/[SQLSession]/trunk/lib/SQLSession/Action/DoSQL.pm
This is repository of my old source code which isn't updated any more. Go to git.rot13.org for current projects!
ViewVC logotype

Diff of /trunk/lib/SQLSession/Action/DoSQL.pm

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 13 by dpavlin, Sun Dec 3 13:28:48 2006 UTC revision 99 by dpavlin, Fri Feb 16 22:20:41 2007 UTC
# Line 12  use base qw/SQLSession::Action Jifty::Ac Line 12  use base qw/SQLSession::Action Jifty::Ac
12    
13  use SQLSession::Model::DatabaseCollection;  use SQLSession::Model::DatabaseCollection;
14  use DBI;  use DBI;
15    use Time::HiRes qw/time/;
16    
17  use Data::Dump qw/dump/;  use Data::Dump qw/dump/;
18    
# Line 21  use Jifty::Action schema { Line 22  use Jifty::Action schema {
22          param sql_query =>          param sql_query =>
23                  label is 'SQL',                  label is 'SQL',
24                  render as 'textarea',                  render as 'textarea',
                 is mandatory,  
25                  ajax validates,                  ajax validates,
26                  ajax canonicalizes;                  ajax canonicalizes,
27                    is mandatory;
28                    
29          param database =>          param on_database =>
30                  label is 'Database',                  label is 'Database',
31                  render as 'Select',                  render as 'Select',
32  #               available are qw( test foo bar );  #               available are qw( test foo bar );
# Line 34  use Jifty::Action schema { Line 35  use Jifty::Action schema {
35                          $dbs->unlimit;                          $dbs->unlimit;
36                          [{                          [{
37                                  display_from => 'name',                                  display_from => 'name',
38                                  value_from => 'name',                                  value_from => 'id',
39                                  collection => $dbs,                                  collection => $dbs,
40                          }];                          }];
41                  };                  };
# Line 61  sub validate_sql_query { Line 62  sub validate_sql_query {
62          }          }
63  }  }
64    
65  use SQL::ReservedWords;  use SQLSession::Utils;
66    
67  sub canonicalize_sql_query {  sub canonicalize_sql_query {
68          my $self = shift;          my $self = shift;
69          my $sql = shift;          my $sql = shift;
70    
71          my @words = SQL::ReservedWords->words;          $sql = Jifty->app_class('Utils')->reformat_sql( $sql );
72    
73          my $w = join('|', @words);          Jifty->log->debug("cannonicaze SQL: $sql");
   
         warn "canonicalize SQL: $sql";  
   
         $sql =~ s/\b($w)\b/uc($1)/egis;  
74    
75          return $sql;          return $sql;
76  }  }
# Line 89  sub take_action { Line 86  sub take_action {
86            
87      # Custom action code      # Custom action code
88    
89          my $database = $self->argument_value('database');          my $on_database = $self->argument_value('on_database') || $self->form_value('on_database');
90          warn "database: $database\n";          warn "on_database: $on_database\n";
91    
92          my $dbs = SQLSession::Model::DatabaseCollection->new;          my $db = SQLSession::Model::Database->new;
93          $dbs->limit( column => 'name', value => $database );          $db->load_by_cols( id => $on_database ) ||
94          my $db = $dbs->first || $self->result->error("Can't find database $database");                  $self->result->error("Can't find database $on_database");
95    
96          my $dbh = DBI->connect( $db->dsn, $db->login, $db->passwd ) ||          my $dbh = DBI->connect( $db->dsn, $db->login, $db->passwd ) ||
97                  $self->result->error("Can't connect to " . $db->dsn . " as " . $db->login . "<br/>" . $DBI::errstr) && return 0;                  $self->result->error("Can't connect to " . $db->dsn . " as " . $db->login . "<br/>" . $DBI::errstr) && return 0;
98    
99            $dbh->do("SET client_encoding = UTF8") if ($db->dsn =~ /Pg/);
100    
101          my $sql = $self->argument_value('sql_query');          my $sql = $self->argument_value('sql_query');
102    
103          warn "SQL: $sql\n";          warn "SQL: $sql\n";
104    
105            my $t = time();
106    
107          my $sth = $dbh->prepare( $sql ) ||          my $sth = $dbh->prepare( $sql ) ||
108                  $self->result->error( $dbh->errstr() ) && return 0;                  $self->result->error( $dbh->errstr() ) && return 0;
109    
110          $sth->execute() ||          $sth->execute() ||
111                  $self->result->error( $sth->errstr() ) && return 0;                  $self->result->error( $sth->errstr() ) && return 0;
112                    
113      $self->result->message('Query produced ' . (          $self->result->message('Query produced ' . (
114                  $sth->rows == 0 ?       'no results' :                  $sth->rows == 0 ?       'no results' :
115                  $sth->rows == 1 ?       'single row' :                  $sth->rows == 1 ?       'single row' :
116                                                          $sth->rows . ' rows'                  sprintf("%d rows in %.2fs", $sth->rows, time() - $t)
117          )) if ($sth->rows >= 0);          )) if ($sth->rows >= 0);
118    
119          $self->result->content( sth => $sth );          $self->result->content( sth => $sth );
Legend:
Removed from v.13  
changed lines
  Added in v.99
  ViewVC Help
Powered by ViewVC 1.1.26