Merge 'trunk' into 'sybase_mssql'
[dbsrgits/DBIx-Class.git] / lib / DBIx / Class / Storage / DBI / Oracle / Generic.pm
CommitLineData
18360aed 1package DBIx::Class::Storage::DBI::Oracle::Generic;
e21dfd6a 2# -*- mode: cperl; cperl-indent-level: 2 -*-
18360aed 3
4use strict;
5use warnings;
6
7137528d 7=head1 NAME
8
92bc2a19 9DBIx::Class::Storage::DBI::Oracle::Generic - Automatic primary key class for Oracle
7137528d 10
11=head1 SYNOPSIS
12
13 # In your table classes
14 __PACKAGE__->load_components(qw/PK::Auto Core/);
2e46b6eb 15 __PACKAGE__->add_columns({ id => { sequence => 'mysequence', auto_nextval => 1 } });
7137528d 16 __PACKAGE__->set_primary_key('id');
17 __PACKAGE__->sequence('mysequence');
18
19=head1 DESCRIPTION
20
21This class implements autoincrements for Oracle.
22
23=head1 METHODS
24
25=cut
26
18360aed 27use Carp::Clan qw/^DBIx::Class/;
28
29use base qw/DBIx::Class::Storage::DBI::MultiDistinctEmulation/;
30
31# __PACKAGE__->load_components(qw/PK::Auto/);
32
33sub _dbh_last_insert_id {
2e46b6eb 34 my ($self, $dbh, $source, @columns) = @_;
35 my @ids = ();
36 foreach my $col (@columns) {
37 my $seq = ($source->column_info($col)->{sequence} ||= $self->get_autoinc_seq($source,$col));
38 my $id = $self->_sequence_fetch( 'currval', $seq );
39 push @ids, $id;
40 }
41 return @ids;
18360aed 42}
43
44sub _dbh_get_autoinc_seq {
45 my ($self, $dbh, $source, $col) = @_;
46
47 # look up the correct sequence automatically
48 my $sql = q{
49 SELECT trigger_body FROM ALL_TRIGGERS t
50 WHERE t.table_name = ?
51 AND t.triggering_event = 'INSERT'
52 AND t.status = 'ENABLED'
53 };
54
55 # trigger_body is a LONG
56 $dbh->{LongReadLen} = 64 * 1024 if ($dbh->{LongReadLen} < 64 * 1024);
57
cb464582 58 my $sth;
59
60 # check for fully-qualified name (eg. SCHEMA.TABLENAME)
61 if ( my ( $schema, $table ) = $source->name =~ /(\w+)\.(\w+)/ ) {
62 $sql = q{
63 SELECT trigger_body FROM ALL_TRIGGERS t
64 WHERE t.owner = ? AND t.table_name = ?
65 AND t.triggering_event = 'INSERT'
66 AND t.status = 'ENABLED'
67 };
68 $sth = $dbh->prepare($sql);
69 $sth->execute( uc($schema), uc($table) );
70 }
71 else {
72 $sth = $dbh->prepare($sql);
73 $sth->execute( uc( $source->name ) );
74 }
18360aed 75 while (my ($insert_trigger) = $sth->fetchrow_array) {
76 return uc($1) if $insert_trigger =~ m!(\w+)\.nextval!i; # col name goes here???
77 }
66cab05c 78 $self->throw_exception("Unable to find a sequence INSERT trigger on table '" . $source->name . "'.");
18360aed 79}
80
2e46b6eb 81sub _sequence_fetch {
82 my ( $self, $type, $seq ) = @_;
83 my ($id) = $self->dbh->selectrow_array("SELECT ${seq}.${type} FROM DUAL");
84 return $id;
85}
86
20f27051 87=head2 connected
88
89Returns true if we have an open (and working) database connection, false if it is not (yet)
90open (or does not work). (Executes a simple SELECT to make sure it works.)
91
0f0abc97 92The reason this is needed is that L<DBD::Oracle>'s ping() does not do a real
93OCIPing but just gets the server version, which doesn't help if someone killed
94your session.
95
20f27051 96=cut
97
c2481821 98sub connected {
99 my $self = shift;
7ba7a57d 100
c2d7baef 101 if (not $self->SUPER::connected(@_)) {
102 return 0;
103 }
104 else {
7ba7a57d 105 my $dbh = $self->_dbh;
106
7ba7a57d 107 local $dbh->{RaiseError} = 1;
c2d7baef 108
7ba7a57d 109 eval {
c2d7baef 110 my $ping_sth = $dbh->prepare_cached("select 1 from dual");
7ba7a57d 111 $ping_sth->execute;
112 $ping_sth->finish;
113 };
114
c2d7baef 115 return $@ ? 0 : 1;
c2481821 116 }
c2481821 117}
118
d789fa99 119sub _dbh_execute {
120 my $self = shift;
121 my ($dbh, $op, $extra_bind, $ident, $bind_attributes, @args) = @_;
122
123 my $wantarray = wantarray;
d789fa99 124
c2d7baef 125 my (@res, $exception, $retried);
126
0f0abc97 127 RETRY: {
128 do {
129 eval {
130 if ($wantarray) {
131 @res = $self->SUPER::_dbh_execute(@_);
132 } else {
133 $res[0] = $self->SUPER::_dbh_execute(@_);
134 }
135 };
136 $exception = $@;
137 if ($exception =~ /ORA-01003/) {
138 # ORA-01003: no statement parsed (someone changed the table somehow,
139 # invalidating your cursor.)
140 my ($sql, $bind) = $self->_prep_for_execute($op, $extra_bind, $ident, \@args);
141 delete $dbh->{CachedKids}{$sql};
d789fa99 142 } else {
0f0abc97 143 last RETRY;
d789fa99 144 }
0f0abc97 145 } while (not $retried++);
146 }
d789fa99 147
148 $self->throw_exception($exception) if $exception;
149
150 wantarray ? @res : $res[0]
151}
152
7137528d 153=head2 get_autoinc_seq
154
155Returns the sequence name for an autoincrement column
156
157=cut
158
18360aed 159sub get_autoinc_seq {
160 my ($self, $source, $col) = @_;
161
373940e1 162 $self->dbh_do('_dbh_get_autoinc_seq', $source, $col);
18360aed 163}
164
7137528d 165=head2 columns_info_for
166
167This wraps the superclass version of this method to force table
168names to uppercase
169
170=cut
171
18360aed 172sub columns_info_for {
173 my ($self, $table) = @_;
174
175 $self->next::method(uc($table));
176}
177
8f7e044c 178=head2 datetime_parser_type
179
180This sets the proper DateTime::Format module for use with
181L<DBIx::Class::InflateColumn::DateTime>.
182
183=cut
184
185sub datetime_parser_type { return "DateTime::Format::Oracle"; }
186
281719d2 187sub _svp_begin {
188 my ($self, $name) = @_;
189
190 $self->dbh->do("SAVEPOINT $name");
191}
192
193# Oracle automatically releases a savepoint when you start another one with the
194# same name.
195sub _svp_release { 1 }
196
197sub _svp_rollback {
198 my ($self, $name) = @_;
199
200 $self->dbh->do("ROLLBACK TO SAVEPOINT $name")
201}
202
18360aed 203=head1 AUTHORS
204
205Andy Grundman <andy@hybridized.org>
206
207Scott Connelly <scottsweep@yahoo.com>
208
209=head1 LICENSE
210
211You may distribute this code under the same terms as Perl itself.
212
213=cut
7137528d 214
2151;