tada! the new worm
[working/Evergreen.git] / Open-ILS / src / perlmods / OpenILS / Application / Storage / Driver / Pg.pm
1
2 { # The driver package itself just needs a db_Main method (or db_Slaves if
3   #Class::DBI::Replication is in use) for Class::DBI to call.
4   #
5   # Any other fixups can go in here too... Also, the drivers should subclass the
6   # DBI driver that they are wrapping, or provide a 'quote()' method that calls
7   # the DBD::xxx::quote() method on FTI's behalf.
8   #
9   # The dirver MUST be a subclass of Class::DBI(::Replication) and
10   # OpenILS::Application::Storage.
11   #-------------------------------------------------------------------------------
12         package OpenILS::Application::Storage::Driver::Pg;
13         use OpenILS::Application::Storage::Driver::Pg::cdbi;
14         use OpenILS::Application::Storage::Driver::Pg::fts;
15         use OpenILS::Application::Storage::Driver::Pg::storage;
16         use OpenILS::Application::Storage::Driver::Pg::dbi;
17         use Class::DBI;
18         use base qw/Class::DBI OpenILS::Application::Storage/;
19         use DBI;
20         use OpenSRF::EX qw/:try/;
21         use OpenSRF::DomainObject::oilsResponse;
22         use OpenSRF::Utils::Logger qw/:level/;
23         my $log = 'OpenSRF::Utils::Logger';
24
25         __PACKAGE__->set_sql( retrieve_limited => 'SELECT * FROM __TABLE__ ORDER BY id LIMIT ?' );
26         __PACKAGE__->set_sql( copy_start => 'COPY %s (%s) FROM STDIN;' );
27         __PACKAGE__->set_sql( copy_end => '\.' );
28
29         my $master_db;
30         my @slave_dbs;
31         my $_db_params;
32         sub child_init {
33                 my $self = shift;
34                 $_db_params = shift;
35
36                 $log->debug("Running child_init inside ".__PACKAGE__, INTERNAL);
37
38                 $_db_params = [ $_db_params ] unless (ref($_db_params) eq 'ARRAY');
39
40                 my %attrs = (   %{$self->_default_attributes},
41                                 RootClass => 'DBIx::ContextualFetch',
42                                 ShowErrorStatement => 1,
43                                 RaiseError => 1,
44                                 AutoCommit => 1,
45                                 PrintError => 1,
46                                 Taint => 1,
47                                 #TraceLevel => "1|SQL",
48                                 pg_enable_utf8 => 1,
49                                 pg_server_prepare => 0,
50                                 FetchHashKeyName => 'NAME_lc',
51                                 ChopBlanks => 1,
52                 );
53
54                 my $master = shift @$_db_params;
55                 $log->debug("Attmpting to connet to $$master{db} at $$master{host}", INFO);
56
57                 try {
58                         $master_db = DBI->connect("dbi:Pg:host=$$master{host};dbname=$$master{db}",$$master{user},$$master{pw}, \%attrs) ||
59                                 throw OpenSRF::EX::ERROR ("Couldn't connect to $$master{db} on $$master{host} as $$master{user}!!");
60                 } catch Error with {
61                         my $e = shift;
62                         $log->debug("Error connecting to database:\n\t$e\n\t$DBI::errstr", ERROR);
63                         throw $e;
64                 };
65
66                 $log->debug("Connected to MASTER db $$master{db} at $$master{host}", INFO);
67                 
68                 $master_db->do("SET NAMES '$$master{client_encoding}';") if ($$master{client_encoding});
69
70                 for my $db (@$_db_params) {
71                         push @slave_dbs, DBI->connect("dbi:Pg:host=$$db{host};dbname=$$db{db}",$$db{user},$$db{pw}, \%attrs);
72                         $slave_dbs[-1]->do("SET NAMES '$$db{client_encoding}';") if ($$master{client_encoding});
73
74                         $log->debug("Connected to MASTER db '$$master{db} at $$master{host}", INFO);
75                 }
76
77                 $log->debug("All is well on the western front", INTERNAL);
78         }
79
80         sub db_Main {
81                 my $self = shift;
82                 return $master_db if ($self->current_xact_session);
83                 return $master_db unless (@slave_dbs);
84                 return ($master_db, @slave_dbs)[rand(scalar(@slave_dbs))];
85         }
86
87         sub quote {
88                 my $self = shift;
89                 return $self->db_Main->quote(@_)
90         }
91
92 #       sub tsearch2_trigger {
93 #               my $self = shift;
94 #               return unless ($self->value);
95 #               $self->index_vector(
96 #                       $self->db_Slaves->selectrow_array(
97 #                               "SELECT to_tsvector('default',?);",
98 #                               {},
99 #                               $self->value
100 #                       )
101 #               );
102 #       }
103
104         my $_xact_session;
105
106         sub current_xact_session {
107                 my $self = shift;
108                 if (defined($_xact_session)) {
109                         return $_xact_session;
110                 }
111                 return undef;
112         }
113
114         sub current_xact_is_auto {
115                 my $self = shift;
116                 my $auto = shift;
117                 if (defined($_xact_session) and ref($_xact_session)) {
118                         if (defined $auto) {
119                                 $_xact_session->session_data(autocommit => $auto);
120                         }
121                         return $_xact_session->session_data('autocommit'); 
122                 }
123         }
124
125         sub current_xact_id {
126                 my $self = shift;
127                 if (defined($_xact_session) and ref($_xact_session)) {
128                         return $_xact_session->session_id;
129                 }
130                 return undef;
131         }
132
133         sub set_xact_session {
134                 my $self = shift;
135                 my $ses = shift;
136                 if (!defined($ses)) {
137                         return undef;
138                 }
139                 $_xact_session = $ses;
140                 return $_xact_session;
141         }
142
143         sub unset_xact_session {
144                 my $self = shift;
145                 my $ses = $_xact_session;
146                 undef $_xact_session;
147                 return $ses;
148         }
149
150 }
151
152 1;