I was wrong about 2d12a809 - the crash is real
[dbsrgits/DBIx-Class.git] / lib / DBIx / Class / Storage / DBI / IdentityInsert.pm
1 package DBIx::Class::Storage::DBI::IdentityInsert;
2
3 use strict;
4 use warnings;
5 use base 'DBIx::Class::Storage::DBI';
6 use mro 'c3';
7
8 use namespace::clean;
9
10 =head1 NAME
11
12 DBIx::Class::Storage::DBI::IdentityInsert - Storage Component for Sybase ASE and
13 MSSQL for Identity Inserts / Updates
14
15 =head1 DESCRIPTION
16
17 This is a storage component for Sybase ASE
18 (L<DBIx::Class::Storage::DBI::Sybase::ASE>) and Microsoft SQL Server
19 (L<DBIx::Class::Storage::DBI::MSSQL>) to support identity inserts, that is
20 inserts of explicit values into C<IDENTITY> columns.
21
22 This is done by wrapping C<INSERT> operations in a pair of table identity
23 toggles like:
24
25   SET IDENTITY_INSERT $table ON
26   $sql
27   SET IDENTITY_INSERT $table OFF
28
29 =cut
30
31 # SET IDENTITY_X only works as part of a statement scope. We can not
32 # $dbh->do the $sql and the wrapping set()s individualy. Hence the
33 # sql mangling. The newlines are important.
34 sub _prep_for_execute {
35   my $self = shift;
36
37   return $self->next::method(@_) unless $self->_autoinc_supplied_for_op;
38
39   my ($op, $ident) = @_;
40
41   my $table = $self->sql_maker->_quote($ident->name);
42   $op = uc $op;
43
44   my ($sql, $bind) = $self->next::method(@_);
45
46   return (<<EOS, $bind);
47 SET IDENTITY_$op $table ON
48 $sql
49 SET IDENTITY_$op $table OFF
50 EOS
51
52 }
53
54 =head1 AUTHOR
55
56 See L<DBIx::Class/AUTHOR> and L<DBIx::Class/CONTRIBUTORS>.
57
58 =head1 LICENSE
59
60 You may distribute this code under the same terms as Perl itself.
61
62 =cut
63
64 1;