lib/DBIx/Class/Storage/DBI/SQLite.pm

   1 package DBIx::Class::Storage::DBI::SQLite;
   2
   3 use strict;
   4 use warnings;
   5 use POSIX 'strftime';
   6 use File::Copy;
   7 use File::Spec;
   8
   9 use base qw/DBIx::Class::Storage::DBI::MultiDistinctEmulation/;
  10
  11 sub _dbh_last_insert_id {
  12   my ($self, $dbh, $source, $col) = @_;
  13   $dbh->func('last_insert_rowid');
  14 }
  15
  16 sub backup
  17 {
  18   my ($self, $dir) = @_;
  19   $dir ||= './';
  20
  21   ## Where is the db file?
  22   my $dsn = $self->connect_info()->[0];
  23
  24   my $dbname = $1 if($dsn =~ /dbname=([^;]+)/);
  25   if(!$dbname)
  26   {
  27     $dbname = $1 if($dsn =~ /^dbi:SQLite:(.+)$/i);
  28   }
  29   $self->throw_exception("Cannot determine name of SQLite db file")
  30     if(!$dbname || !-f $dbname);
  31
  32 #  print "Found database: $dbname\n";
  33 #  my $dbfile = file($dbname);
  34   my ($vol, $dbdir, $file) = File::Spec->splitpath($dbname);
  35 #  my $file = $dbfile->basename();
  36   $file = strftime("%Y-%m-%d-%H_%M_%S", localtime()) . $file;
  37   $file = "B$file" while(-f $file);
  38
  39   mkdir($dir) unless -f $dir;
  40   my $backupfile = File::Spec->catfile($dir, $file);
  41
  42   my $res = copy($dbname, $backupfile);
  43   $self->throw_exception("Backup failed! ($!)") if(!$res);
  44
  45   return $backupfile;
  46 }
  47
  48 sub disconnect {
  49
  50   # As described in this node http://www.perlmonks.org/?node_id=666210
  51   # there seems to be no sane way to ->disconnect a SQLite database with
  52   # cached statement handles. As per mst we just zap the cache and
  53   # proceed as normal.
  54
  55   my $self = shift;
  56   if ($self->connected) {
  57     $self->_dbh->{CachedKids} = {};
  58     $self->next::method (@_);
  59   }
  60 }
  61
  62
  63 1;
  64
  65 =head1 NAME
  66
  67 DBIx::Class::Storage::DBI::SQLite - Automatic primary key class for SQLite
  68
  69 =head1 SYNOPSIS
  70
  71   # In your table classes
  72   __PACKAGE__->load_components(qw/PK::Auto Core/);
  73   __PACKAGE__->set_primary_key('id');
  74
  75 =head1 DESCRIPTION
  76
  77 This class implements autoincrements for SQLite.
  78
  79 =head1 AUTHORS
  80
  81 Matt S. Trout <mst@shadowcatsystems.co.uk>
  82
  83 =head1 LICENSE
  84
  85 You may distribute this code under the same terms as Perl itself.
  86
  87 =cut