C::M::DBIC::Schema - minor cleanups

[catagits/Catalyst-Model-DBIC-Schema.git] / lib / Catalyst / Model / DBIC / Schema.pm

diff --git a/lib/Catalyst/Model/DBIC/Schema.pm b/lib/Catalyst/Model/DBIC/Schema.pm
index 41a8fd0..e39ffcf 100644 (file)
--- a/lib/Catalyst/Model/DBIC/Schema.pm
+++ b/lib/Catalyst/Model/DBIC/Schema.pm
@@ -14,7 +14,9 @@ use Scalar::Util 'reftype';
 use MooseX::ClassAttribute;
 use Moose::Autobox;
 
-use Catalyst::Model::DBIC::Schema::Types qw/ConnectInfo SchemaClass/;
+use Catalyst::Model::DBIC::Schema::Types
+    qw/ConnectInfo SchemaClass CursorClass/;
+
 use MooseX::Types::Moose qw/ArrayRef Str ClassName/;
 
 use namespace::clean -except => 'meta';
@@ -409,6 +411,13 @@ has 'model_name' => (is => 'ro', isa => Str, default => sub {
 
 has 'roles' => (is => 'ro', isa => ArrayRef|Str);
 
+has '_default_cursor_class' => (
+    is => 'ro',
+    isa => CursorClass,
+    default => 'DBIx::Class::Storage::DBI::Cursor',
+    coerce => 1
+);
+
 sub BUILD {
     my $self = shift;
     my $class = ref $self;
@@ -503,6 +512,15 @@ sub _install_rs_models {
     }
 }
 
+sub _reset_cursor_class {
+    my $self = shift;
+
+    if ($self->storage->can('cursor_class')) {
+       $self->storage->cursor_class($self->_default_cursor_class)
+           if $self->storage->cursor_class ne $self->_default_cursor_class;
+    }
+}
+
 __PACKAGE__->meta->make_immutable;
 
 =head1 SEE ALSO