Fixed up PK::Auto::* to use result_source, added connection, connect and clone method...

[dbsrgits/DBIx-Class.git] / lib / DBIx / Class / PK / Auto / MSSQL.pm

diff --git a/lib/DBIx/Class/PK/Auto/MSSQL.pm b/lib/DBIx/Class/PK/Auto/MSSQL.pm
index 3ae6287..4619998 100644 (file)
--- a/lib/DBIx/Class/PK/Auto/MSSQL.pm
+++ b/lib/DBIx/Class/PK/Auto/MSSQL.pm
@@ -8,7 +8,8 @@ use base qw/DBIx::Class/;
 __PACKAGE__->load_components(qw/PK::Auto/);\r
 \r
 sub last_insert_id {\r
-  my( $id ) = $_[0]->storage->dbh->selectrow_array( 'SELECT @@IDENTITY' );\r
+  my( $id ) = $_[0]->result_source->storage->dbh->selectrow_array(\r
+                                                    'SELECT @@IDENTITY' );\r
   return $id;\r
 }\r
 \r