Open-ILS/src/perlmods/OpenILS/Application/Storage.pm

   1 package OpenILS::Application::Storage;
   2 use OpenSRF::Application;
   3 use base qw/OpenSRF::Application/;
   4
   5 use OpenSRF::EX qw/:try/;
   6 use OpenSRF::Utils::Logger qw/:level/;
   7
   8 # Pull this in so we can adjust it's @ISA
   9 use OpenILS::Application::Storage::CDBI;
  10
  11 use OpenILS::Application::Storage::FTS;
  12
  13 # Suck in the method publishing modules
  14 use OpenILS::Application::Storage::Publisher;
  15
  16 # the easy way to get to the logger...
  17 my $log = "OpenSRF::Utils::Logger";
  18
  19 sub DESTROY {};
  20
  21 sub initialize {
  22
  23         my $conf = OpenSRF::Utils::SettingsClient->new;
  24
  25         $log->debug('Initializing ' . __PACKAGE__ . '...', DEBUG);
  26
  27         my $driver = "OpenILS::Application::Storage::Driver::".
  28                 $conf->config_value( apps => 'open-ils.storage' => app_settings => databases => 'driver');
  29
  30
  31         $log->debug("Attempting to load $driver ...", DEBUG);
  32
  33         eval "use $driver;";
  34         throw OpenILS::EX::PANIC ( "Can't load $driver!  :  $@" ) if ($@);
  35
  36         $log->debug("$driver loaded successfully", DEBUG);
  37
  38         @OpenILS::Application::Storage::CDBI::ISA = ( $driver );
  39 }
  40
  41 sub child_init {
  42
  43         my $conf = OpenSRF::Utils::SettingsClient->new;
  44
  45         $log->debug('Running child_init for ' . __PACKAGE__ . '...', DEBUG);
  46
  47         OpenILS::Application::Storage::CDBI->child_init(
  48                 $conf->config_value( apps => 'open-ils.storage' => app_settings => databases => 'database')
  49         );
  50
  51         return 1 if (OpenILS::Application::Storage::CDBI->db_Main());
  52         return 0;
  53 }
  54
  55 sub begin_xaction {
  56         my $self = shift;
  57         my $client = shift;
  58
  59         my $dbh = OpenILS::Application::Storage::CDBI->db_Main;
  60
  61         $client->session->registger_callback( disconnect => sub { $dbh->commit; } )
  62                 if ($self->api_name =~ /autocommit$/o);
  63
  64         $client->session->registger_callback( death => sub { $dbh->rollback; } );
  65
  66         $dbh->begin_work;
  67         return 1;
  68 }
  69 __PACKAGE__->register_method(
  70         method          => 'begin_xaction',
  71         api_name        => 'open-ils.storage.transaction.begin',
  72         api_level       => 1,
  73         argc            => 0,
  74 );
  75 __PACKAGE__->register_method(
  76         method          => 'begin_xaction',
  77         api_name        => 'open-ils.storage.transaction.begin.autocommit',
  78         api_level       => 1,
  79         argc            => 0,
  80 );
  81
  82 sub commit_xaction {
  83         my $self = shift;
  84         my $client = shift;
  85
  86         try {
  87                 $client->session->session_data('dbh')->commit;
  88         } catch Error with {
  89                 $client->session->session_data('dbh')->rollback;
  90         };
  91 }
  92 __PACKAGE__->register_method(
  93         method          => 'commit_xaction',
  94         api_name        => 'open-ils.storage.transaction.commit',
  95 );
  96
  97
  98 sub rollback_xaction {
  99         my $self = shift;
 100         my $client = shift;
 101
 102         $client->session->session_data('dbh')->rollback;
 103 }
 104 __PACKAGE__->register_method(
 105         method          => 'rollback_xaction',
 106         api_name        => 'open-ils.storage.transaction.rollback',
 107 );
 108
 109
 110 sub _cdbi2Hash {
 111         my $self = shift;
 112         my $obj = shift;
 113         return { map { ( $_ => $obj->$_ ) } ($obj->columns('All')) };
 114 }
 115
 116 sub _cdbi_list2AoH {
 117         my $self = shift;
 118         my @objs = @_;
 119         return [ map { $self->_cdbi2Hash($_) } @objs ];
 120 }
 121
 122 1;