/[cvs]/nfo/perl/libs/Data/Storage/Handler/DBI.pm
ViewVC logotype

Diff of /nfo/perl/libs/Data/Storage/Handler/DBI.pm

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

revision 1.3 by joko, Sun Nov 17 06:34:39 2002 UTC revision 1.14 by joko, Wed Apr 9 06:06:04 2003 UTC
# Line 3  Line 3 
3  #  $Id$  #  $Id$
4  #  #
5  #  $Log$  #  $Log$
6    #  Revision 1.14  2003/04/09 06:06:04  joko
7    #  sendQuery now is capable of doing 'SELECT'- or 'INSERT'-queries
8    #
9    #  Revision 1.13  2003/04/08 23:06:45  joko
10    #  renamed core database helper functions
11    #
12    #  Revision 1.12  2003/01/30 22:28:21  joko
13    #  + implemented new concrete methods
14    #
15    #  Revision 1.11  2002/12/19 16:31:05  joko
16    #  + sub dropDb
17    #  + sub rebuildDb
18    #
19    #  Revision 1.10  2002/12/15 02:02:22  joko
20    #  + fixed logging-message
21    #
22    #  Revision 1.9  2002/12/05 07:58:20  joko
23    #  + now using Tie::SecureHash as a base for the COREHANDLE
24    #  + former public COREHANDLE becomes private _COREHANDLE now
25    #
26    #  Revision 1.8  2002/12/01 22:20:43  joko
27    #  + sub createDb (from Storage.pm)
28    #
29    #  Revision 1.7  2002/12/01 07:09:09  joko
30    #  + sub getListFiltered (dummy redirecting to getListUnfiltered)
31    #
32    #  Revision 1.6  2002/12/01 04:46:01  joko
33    #  + sub eraseAll
34    #
35    #  Revision 1.5  2002/11/29 05:00:26  joko
36    #  + sub getListUnfiltered
37    #  + sub sendQuery
38    #
39    #  Revision 1.4  2002/11/17 08:46:42  jonen
40    #  + wrapped eval around DBI->connect to prevent deaths
41    #
42  #  Revision 1.3  2002/11/17 06:34:39  joko  #  Revision 1.3  2002/11/17 06:34:39  joko
43  #  + locator metadata can now be reached via ->{locator}  #  + locator metadata can now be reached via ->{locator}
44  #  - sub hash2sql now taken from libdb  #  - sub hash2sql now taken from libdb
# Line 24  use warnings; Line 60  use warnings;
60    
61  use base ("Data::Storage::Handler::Abstract");  use base ("Data::Storage::Handler::Abstract");
62    
63    
64  use DBI;  use DBI;
65  use Data::Dumper;  use Data::Dumper;
66  use libdb;  use shortcuts::db qw( hash2sql dsn2dbname );
67    use Data::Storage::Result::DBI;
68    
69    
70  # get logger instance  # get logger instance
71  my $logger = Log::Dispatch::Config->instance;  my $logger = Log::Dispatch::Config->instance;
72    
73    
74  our $metainfo = {  sub getMetaInfo {
75    'disconnectMethod' => 'disconnect',    my $self = shift;
76  };    $logger->debug( __PACKAGE__ . "->getMetaInfo()"  );
77      return {
78        'disconnectMethod' => 'disconnect',
79      };
80    }
81    
82  sub connect {  sub connect {
83    
# Line 52  sub connect { Line 95  sub connect {
95                    
96          #use Data::Dumper; print Dumper($self->{dbi});          #use Data::Dumper; print Dumper($self->{dbi});
97                    
98          $self->{COREHANDLE} = DBI->connect( $dsn, '', '', $self->{locator}->{dbi} );          eval {
99          if (!$self->{COREHANDLE}) {            $self->{_COREHANDLE} = DBI->connect( $dsn, '', '', $self->{locator}->{dbi} );
100            $logger->warning( __PACKAGE__ . "->connect failed: " . DBI::errstr );            if (!$self->{_COREHANDLE}) {
101            return;              $logger->warning( __PACKAGE__ . "->connect failed: " . DBI::errstr );
102          }              return;
103              }
104            };
105            $logger->warning( __PACKAGE__ . "->connect failed: " . $@ ) if $@;
106    
107        }        }
108        $self->configureCOREHANDLE();        $self->configureCOREHANDLE();
109    
# Line 72  sub configureCOREHANDLE { Line 119  sub configureCOREHANDLE {
119    
120    $logger->debug( __PACKAGE__ . "->configureCOREHANDLE" );    $logger->debug( __PACKAGE__ . "->configureCOREHANDLE" );
121    
122    # apply configured modifications    return if !$self->{_COREHANDLE};
123    
124      # apply configured modifications to DBI-handle
125      if (exists $self->{locator}->{dbi}->{trace_level} && exists $self->{locator}->{dbi}->{trace_file}) {      if (exists $self->{locator}->{dbi}->{trace_level} && exists $self->{locator}->{dbi}->{trace_file}) {
126        $self->{COREHANDLE}->trace($self->{locator}->{dbi}->{trace_level}, $self->{locator}->{dbi}->{trace_file});        $self->{_COREHANDLE}->trace($self->{locator}->{dbi}->{trace_level}, $self->{locator}->{dbi}->{trace_file});
127      }      }
128      if (exists $self->{locator}->{dbi}->{RaiseError}) {      if (exists $self->{locator}->{dbi}->{RaiseError}) {
129        $self->{COREHANDLE}->{RaiseError} = $self->{locator}->{dbi}->{RaiseError};        $self->{_COREHANDLE}->{RaiseError} = $self->{locator}->{dbi}->{RaiseError};
130      }      }
131      if (exists $self->{locator}->{dbi}->{PrintError}) {      if (exists $self->{locator}->{dbi}->{PrintError}) {
132        $self->{COREHANDLE}->{PrintError} = $self->{locator}->{dbi}->{PrintError};        $self->{_COREHANDLE}->{PrintError} = $self->{locator}->{dbi}->{PrintError};
133      }      }
134      if (exists $self->{locator}->{dbi}->{HandleError}) {      if (exists $self->{locator}->{dbi}->{HandleError}) {
135        $self->{COREHANDLE}->{HandleError} = $self->{locator}->{dbi}->{HandleError};        $self->{_COREHANDLE}->{HandleError} = $self->{locator}->{dbi}->{HandleError};
136      }      }
137    
138  }  }
# Line 94  sub _sendSql { Line 143  sub _sendSql {
143        
144    # two-level handling for implicit connect:    # two-level handling for implicit connect:
145    # if there's no corehandle ...    # if there's no corehandle ...
146    if (!$self->{COREHANDLE}) {    if (!$self->{_COREHANDLE}) {
147      # ... try to connect, but ...      # ... try to connect, but ...
148      $self->connect();      $self->connect();
149      # ... if this still fails, there's something wrong probably, so we won't continue      # ... if this still fails, there's something wrong probably, so we won't continue
150      if (!$self->{COREHANDLE}) {      if (!$self->{_COREHANDLE}) {
151        return;        return;
152      }      }
153    }    }
154        
155    #print "prepare sql: $sql\n";    #print "prepare sql: $sql\n";
156        
157    my $sth = $self->{COREHANDLE}->prepare($sql);    my $sth = $self->{_COREHANDLE}->prepare($sql);
158    $sth->execute();    $sth->execute();
159    return $sth;    return $sth;
160  }  }
# Line 113  sub _sendSql { Line 162  sub _sendSql {
162  sub sendCommand {  sub sendCommand {
163    my $self = shift;    my $self = shift;
164    my $command = shift;    my $command = shift;
165      # TODO: when tracing: yes, do actually log this
166    #$logger->debug( __PACKAGE__ . "->sendCommand( command $command )" );    #$logger->debug( __PACKAGE__ . "->sendCommand( command $command )" );
167    my $cmdHandle = $self->_sendSql($command);    my $cmdHandle = $self->_sendSql($command);
168    my $result = Data::Storage::Result::DBI->new( RESULTHANDLE => $cmdHandle );    my $result = Data::Storage::Result::DBI->new( RESULTHANDLE => $cmdHandle );
169    return $result;    return $result;
170  }  }
171    
172  sub quoteSql {  sub getChildNodes {
173    my $self = shift;    my $self = shift;
174    my $string = shift;    my @nodes;
175    if ($string) {    $logger->debug( __PACKAGE__ . "->getChildNodes()" );
176      $string =~ s/'/\\'/g;    my $locator = $self->{locator};
177      #print Dumper($locator); exit;
178      if (my $result = $self->sendCommand( 'SHOW TABLES;' ) ) {
179        my $dbname = dsn2dbname($self->{locator}->{dbi}->{dsn});
180        my $key = "Tables_in_$dbname";
181        while ( my $row = $result->getNextEntry() ) {
182          push @nodes, $row->{$key};
183        }
184      }
185      return \@nodes;
186    }
187    
188    sub getListUnfiltered {
189      my $self = shift;
190      my $nodename = shift;
191      my @list;
192      $logger->debug( __PACKAGE__ . "->getListUnfiltered( nodename => '" . $nodename . "' )" );
193      # get list of rows from rdbms by table name
194      my $result = $self->sendCommand("SELECT * FROM $nodename");
195      while ( my $row = $result->getNextEntry() ) {
196        push @list, $row;
197    }    }
198    return $string;    return \@list;
199  }  }
200    
201    sub sendQuery {
202      my $self = shift;
203      my $query = shift;
204    
205  sub getChildNodes {    $logger->debug( __PACKAGE__ . "->sendQuery" );
206    
207      my $action = $query->{options}->{action};
208      $action ||= 'load';
209    
210      if (my $guid = $query->{options}->{GUID}) {
211        $query->{criterias} = [ { key => 'guid', op => 'eq', val => $guid } ];
212      }
213    
214      # check criterias (load & save)
215      #my $sql = "SELECT cs FROM $self->{metainfo}->{$descent}->{node} WHERE $self->{metainfo}->{$descent}->{IdentProvider}->{arg}='$self->{entry}->{source}->{ident}';";
216      #my $result = $self->{metainfo}->{$descent}->{storage}->sendCommand($sql);
217      my @crits;
218      foreach (@{$query->{criterias}}) {
219        my $op = '';
220        $op = '=' if lc $_->{op} eq 'eq';
221        push @crits, "$_->{key}$op'$_->{val}'";
222      }
223      # HACK: this is hardcoded ;(    expand possibilities!
224      my $crit = join(' AND ', @crits);
225      
226      # check subnodes (load)
227      my $subnodes = {};
228      map { $subnodes->{$_}++ } @{$query->{subnodes}};
229      
230      # check payload (save)
231      #print Dumper($query->{payload});
232      #$subnodes ||= $query->{payload};
233      
234      # dispatch action
235      if ($action eq 'load') {
236        my $sql = hash2sql($query->{node}, $subnodes, 'SELECT', $crit);
237        return $self->sendCommand($sql);
238      } elsif ($action eq 'save') {
239        my $sql = hash2sql($query->{node}, $query->{payload}, 'UPDATE', $crit);
240        $self->sendCommand($sql);
241      }
242    
243    }
244    
245    sub eraseAll {
246    my $self = shift;    my $self = shift;
247    my @nodes;    my $classname = shift;
248      $logger->debug( __PACKAGE__ . "->eraseAll" );
249      my $sql = "DELETE FROM $classname";
250      $self->sendCommand($sql);
251    }
252    
253    $logger->debug( __PACKAGE__ . "->getChildNodes()" );  # TODO: actually implement the filtering functionality using $this->sendQuery
254    sub getListFiltered {
255      my $self = shift;
256      my $nodename = shift;
257      return $self->getListUnfiltered($nodename);
258    }
259    
260    if (my $result = $self->sendCommand( 'SHOW TABLES;' ) ) {  # TODO: do this via a parametrized "$self->connect(<connect just to database server - don't select database>)"
261      my $dbname = getDbNameByDsn($self->{locator}->{dbi}->{dsn});  sub createDb {
262      my $key = "Tables_in_$dbname";    
263      while ( my $row = $result->_getNextEntry() ) {    my $self = shift;
264        push @nodes, $row->{$key};    
265      # get dsn from Data::Storage::Locator instance
266      my $dsn = $self->{locator}->{dbi}->{dsn};
267    
268      $logger->debug( __PACKAGE__ .  "->createDb( dsn $dsn )" );
269    
270      # remove database setting from dsn-string
271      $dsn =~ s/database=(.+?);//;
272      
273      # remember extracted database name to know what actually to create right now
274      my $database_name = $1;
275    
276      # flag to indicate goodness
277      my $ok;
278      
279      # connect to database server - don't select/use any specific database
280      #if ( my $dbh = DBI->connect($dsn, '', '', { PrintError => 0 } ) ) {
281      if ( my $dbh = DBI->connect($dsn, '', '', $self->{locator}->{dbi} ) ) {
282    
283        if ($database_name) {
284          if ($dbh->do("CREATE DATABASE $database_name")) {
285            $ok = 1;
286          }
287      }      }
288    
289        $dbh->disconnect();
290    
291    }    }
292        
293    return \@nodes;    return $ok;
294      
295    }
296    
297    sub getCOREHANDLE2 {
298      my $self = shift;
299      return $self->{_COREHANDLE};
300  }  }
301    
302    sub dropDb {
303      my $self = shift;
304      my $dsn = $self->{locator}->{dbi}->{dsn};
305    
306      $logger->debug( __PACKAGE__ .  "->dropDb( dsn $dsn )" );
307    
308      $dsn =~ s/database=(.+?);//;
309      my $database_name = $1;
310    
311  package Data::Storage::Result::DBI;    my $ok;
312      
313      if ( my $dbh = DBI->connect($dsn, '', '', {
314                                                          PrintError => 0,
315                                                          } ) ) {
316        if ($database_name) {
317          if ($dbh->do("DROP DATABASE $database_name;")) {
318            $ok = 1;
319          }
320        }
321    
322  use strict;      $dbh->disconnect();
 use warnings;  
323    
324  use base ("Data::Storage::Result");    }
325      
326      return $ok;
327    }
328    
329  sub DESTROY {  sub rebuildDb {
330    my $self = shift;    my $self = shift;
331    #$logger->debug( __PACKAGE__ . "->" . "DESTROY" );    $logger->info( __PACKAGE__ . "->rebuildDb()" );
332    $self->{RESULTHANDLE} && $self->{RESULTHANDLE}->finish();    my @results;
333    
334      # sum up results (bool (0/1)) in array
335      #push @results, $self->retreatSchema();
336      push @results, $self->dropDb();
337      push @results, $self->createDb();
338      #push @results, $self->deploySchema();
339    
340      # scan array for "bad ones"
341      my $res = 1;
342      map {
343        $res = 0 if (!$_);
344      } @results;
345      
346      return $res;
347  }  }
348    
349  sub _getNextEntry {  sub testAvailability {
350    my $self = shift;    my $self = shift;
351    $self->{RESULTHANDLE} && return $self->{RESULTHANDLE}->fetchrow_hashref;    my $status = $self->testDsn();
352      $self->{locator}->{status}->{available} = $status;
353      return $status;
354  }  }
355    
356    sub testDsn {
357      my $self = shift;
358      my $dsn = $self->{locator}->{dbi}->{dsn};
359      my $result;
360      if ( my $dbh = DBI->connect($dsn, '', '', {
361                                                          PrintError => 0,
362                                                          } ) ) {
363        
364        # TODO: REVIEW
365        $dbh->disconnect();
366        
367        return 1;
368      } else {
369        $logger->warning( __PACKAGE__ .  "[$self->{locator}->{type}]" . "->testDsn(): " . "DBI-error: " . $DBI::errstr );
370      }
371    }
372    
 1;  
373    1;
374    __END__
Legend:
Removed from v.1.3  
changed lines
  Added in v.1.14
MailToCvsAdmin">MailToCvsAdmin
 ViewVC Help
Powered by ViewVC 1.1.26 RSS 2.0 feed