X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=lib%2FDBIx%2FClass%2FStorage%2FDBI%2FOracle%2FGeneric.pm;h=448c203fde1a4e06b07f017544f45c8e095f1a18;hb=b7b18f328d214290d4ae84dd94125d72db88db38;hp=808e20f08562e06a7deff49c88425761663823d4;hpb=2e46b6eb92524b5b39b37791c122a1e2c2ee90b9;p=dbsrgits%2FDBIx-Class.git diff --git a/lib/DBIx/Class/Storage/DBI/Oracle/Generic.pm b/lib/DBIx/Class/Storage/DBI/Oracle/Generic.pm index 808e20f..448c203 100644 --- a/lib/DBIx/Class/Storage/DBI/Oracle/Generic.pm +++ b/lib/DBIx/Class/Storage/DBI/Oracle/Generic.pm @@ -1,34 +1,34 @@ package DBIx::Class::Storage::DBI::Oracle::Generic; -# -*- mode: cperl; cperl-indent-level: 2 -*- use strict; use warnings; +use Scope::Guard (); +use Context::Preserve (); =head1 NAME -DBIx::Class::Storage::DBI::Oracle - Automatic primary key class for Oracle +DBIx::Class::Storage::DBI::Oracle::Generic - Oracle Support for DBIx::Class =head1 SYNOPSIS - # In your table classes - __PACKAGE__->load_components(qw/PK::Auto Core/); + # In your result (table) classes + use base 'DBIx::Class::Core'; __PACKAGE__->add_columns({ id => { sequence => 'mysequence', auto_nextval => 1 } }); __PACKAGE__->set_primary_key('id'); __PACKAGE__->sequence('mysequence'); =head1 DESCRIPTION -This class implements autoincrements for Oracle. +This class implements base Oracle support. The subclass +L is for C<(+)> joins in Oracle +versions before 9. =head1 METHODS =cut -use Carp::Clan qw/^DBIx::Class/; - -use base qw/DBIx::Class::Storage::DBI::MultiDistinctEmulation/; - -# __PACKAGE__->load_components(qw/PK::Auto/); +use base qw/DBIx::Class::Storage::DBI/; +use mro 'c3'; sub _dbh_last_insert_id { my ($self, $dbh, $source, @columns) = @_; @@ -53,40 +53,91 @@ sub _dbh_get_autoinc_seq { }; # trigger_body is a LONG - $dbh->{LongReadLen} = 64 * 1024 if ($dbh->{LongReadLen} < 64 * 1024); + local $dbh->{LongReadLen} = 64 * 1024 if ($dbh->{LongReadLen} < 64 * 1024); + + my $sth; + + my $source_name; + if ( ref $source->name ne 'SCALAR' ) { + $source_name = $source->name; + } + else { + $source_name = ${$source->name}; + } - my $sth = $dbh->prepare($sql); - $sth->execute( uc($source->name) ); + # check for fully-qualified name (eg. SCHEMA.TABLENAME) + if ( my ( $schema, $table ) = $source_name =~ /(\w+)\.(\w+)/ ) { + $sql = q{ + SELECT trigger_body FROM ALL_TRIGGERS t + WHERE t.owner = ? AND t.table_name = ? + AND t.triggering_event = 'INSERT' + AND t.status = 'ENABLED' + }; + $sth = $dbh->prepare($sql); + $sth->execute( uc($schema), uc($table) ); + } + else { + $sth = $dbh->prepare($sql); + $sth->execute( uc( $source_name ) ); + } while (my ($insert_trigger) = $sth->fetchrow_array) { return uc($1) if $insert_trigger =~ m!(\w+)\.nextval!i; # col name goes here??? } $self->throw_exception("Unable to find a sequence INSERT trigger on table '" . $source->name . "'."); } -=head2 insert +sub _sequence_fetch { + my ( $self, $type, $seq ) = @_; + my ($id) = $self->_get_dbh->selectrow_array("SELECT ${seq}.${type} FROM DUAL"); + return $id; +} + +sub _ping { + my $self = shift; -Fetch nextval from sequence and handle insert statement. + my $dbh = $self->_dbh or return 0; -=cut + local $dbh->{RaiseError} = 1; -sub insert { - my ( $self, $source, $to_insert ) = @_; - foreach my $col ( $source->columns ) { - if ( !defined $to_insert->{$col} ) { - my $col_info = $source->column_info($col); + eval { + $dbh->do("select 1 from dual"); + }; + + return $@ ? 0 : 1; +} - if ( $col_info->{auto_nextval} ) { - $to_insert->{$col} = $self->_sequence_fetch( 'nextval', $col_info->{sequence} || $self->_dbh_get_autoinc_seq($self->dbh, $source) ); +sub _dbh_execute { + my $self = shift; + my ($dbh, $op, $extra_bind, $ident, $bind_attributes, @args) = @_; + + my $wantarray = wantarray; + + my (@res, $exception, $retried); + + RETRY: { + do { + eval { + if ($wantarray) { + @res = $self->next::method(@_); + } else { + $res[0] = $self->next::method(@_); + } + }; + $exception = $@; + if ($exception =~ /ORA-01003/) { + # ORA-01003: no statement parsed (someone changed the table somehow, + # invalidating your cursor.) + my ($sql, $bind) = $self->_prep_for_execute($op, $extra_bind, $ident, \@args); + delete $dbh->{CachedKids}{$sql}; + } else { + last RETRY; } - } + } while (not $retried++); } - $self->next::method( $source, $to_insert ); -} -sub _sequence_fetch { - my ( $self, $type, $seq ) = @_; - my ($id) = $self->dbh->selectrow_array("SELECT ${seq}.${type} FROM DUAL"); - return $id; + $self->throw_exception($exception) if $exception; + + wantarray ? @res : $res[0] } =head2 get_autoinc_seq @@ -97,8 +148,8 @@ Returns the sequence name for an autoincrement column sub get_autoinc_seq { my ($self, $source, $col) = @_; - - $self->dbh_do($self->can('_dbh_get_autoinc_seq'), $source, $col); + + $self->dbh_do('_dbh_get_autoinc_seq', $source, $col); } =head2 columns_info_for @@ -123,11 +174,192 @@ L. sub datetime_parser_type { return "DateTime::Format::Oracle"; } -=head1 AUTHORS +=head2 connect_call_datetime_setup + +Used as: + + on_connect_call => 'datetime_setup' + +In L to set the session nls date, and +timestamp values for use with L and the +necessary environment variables for L, which is used +by it. + +Maximum allowable precision is used, unless the environment variables have +already been set. + +These are the defaults used: + + $ENV{NLS_DATE_FORMAT} ||= 'YYYY-MM-DD HH24:MI:SS'; + $ENV{NLS_TIMESTAMP_FORMAT} ||= 'YYYY-MM-DD HH24:MI:SS.FF'; + $ENV{NLS_TIMESTAMP_TZ_FORMAT} ||= 'YYYY-MM-DD HH24:MI:SS.FF TZHTZM'; + +To get more than second precision with L +for your timestamps, use something like this: + + use Time::HiRes 'time'; + my $ts = DateTime->from_epoch(epoch => time); + +=cut + +sub connect_call_datetime_setup { + my $self = shift; + + my $date_format = $ENV{NLS_DATE_FORMAT} ||= 'YYYY-MM-DD HH24:MI:SS'; + my $timestamp_format = $ENV{NLS_TIMESTAMP_FORMAT} ||= + 'YYYY-MM-DD HH24:MI:SS.FF'; + my $timestamp_tz_format = $ENV{NLS_TIMESTAMP_TZ_FORMAT} ||= + 'YYYY-MM-DD HH24:MI:SS.FF TZHTZM'; + + $self->_do_query( + "alter session set nls_date_format = '$date_format'" + ); + $self->_do_query( + "alter session set nls_timestamp_format = '$timestamp_format'" + ); + $self->_do_query( + "alter session set nls_timestamp_tz_format='$timestamp_tz_format'" + ); +} + +=head2 source_bind_attributes + +Handle LOB types in Oracle. Under a certain size (4k?), you can get away +with the driver assuming your input is the deprecated LONG type if you +encode it as a hex string. That ain't gonna fly at larger values, where +you'll discover you have to do what this does. + +This method had to be overridden because we need to set ora_field to the +actual column, and that isn't passed to the call (provided by Storage) to +bind_attribute_by_data_type. + +According to L, the ora_field isn't always necessary, but +adding it doesn't hurt, and will save your bacon if you're modifying a +table with more than one LOB column. + +=cut + +sub source_bind_attributes +{ + require DBD::Oracle; + my $self = shift; + my($source) = @_; + + my %bind_attributes; + + foreach my $column ($source->columns) { + my $data_type = $source->column_info($column)->{data_type} || ''; + next unless $data_type; + + my %column_bind_attrs = $self->bind_attribute_by_data_type($data_type); + + if ($data_type =~ /^[BC]LOB$/i) { + if ($DBD::Oracle::VERSION eq '1.23') { + $self->throw_exception( +"BLOB/CLOB support in DBD::Oracle == 1.23 is broken, use an earlier or later ". +"version.\n\nSee: https://rt.cpan.org/Public/Bug/Display.html?id=46016\n" + ); + } + + $column_bind_attrs{'ora_type'} = uc($data_type) eq 'CLOB' + ? DBD::Oracle::ORA_CLOB() + : DBD::Oracle::ORA_BLOB() + ; + $column_bind_attrs{'ora_field'} = $column; + } + + $bind_attributes{$column} = \%column_bind_attrs; + } + + return \%bind_attributes; +} + +sub _svp_begin { + my ($self, $name) = @_; + $self->_get_dbh->do("SAVEPOINT $name"); +} + +# Oracle automatically releases a savepoint when you start another one with the +# same name. +sub _svp_release { 1 } + +sub _svp_rollback { + my ($self, $name) = @_; + $self->_get_dbh->do("ROLLBACK TO SAVEPOINT $name") +} + +=head2 relname_to_table_alias + +L uses L names as table aliases in +queries. + +Unfortunately, Oracle doesn't support identifiers over 30 chars in length, so +the L name is shortened and appended with half of an +MD5 hash. + +See L. + +=cut + +sub relname_to_table_alias { + my $self = shift; + my ($relname, $join_count) = @_; + + my $alias = $self->next::method(@_); + + return $alias if length($alias) <= 30; + + # get a base64 md5 of the alias with join_count + require Digest::MD5; + my $ctx = Digest::MD5->new; + $ctx->add($alias); + my $md5 = $ctx->b64digest; + + # remove alignment mark just in case + $md5 =~ s/=*\z//; + + # truncate and prepend to truncated relname without vowels + (my $devoweled = $relname) =~ s/[aeiou]//g; + my $shortened = substr($devoweled, 0, 18); + + my $new_alias = + $shortened . '_' . substr($md5, 0, 30 - length($shortened) - 1); + + return $new_alias; +} + +=head2 with_deferred_fk_checks + +Runs a coderef between: + + alter session set constraints = deferred + ... + alter session set constraints = immediate + +to defer foreign key checks. + +Constraints must be declared C for this to work. + +=cut + +sub with_deferred_fk_checks { + my ($self, $sub) = @_; + + my $txn_scope_guard = $self->txn_scope_guard; + + $self->_do_query('alter session set constraints = deferred'); + + my $sg = Scope::Guard->new(sub { + $self->_do_query('alter session set constraints = immediate'); + }); + + return Context::Preserve::preserve_context(sub { $sub->() }, + after => sub { $txn_scope_guard->commit }); +} -Andy Grundman +=head1 AUTHOR -Scott Connelly +See L. =head1 LICENSE