- Move search examples under a new =head2; additional examples go under a =head3

[dbsrgits/DBIx-Class-Historic.git] / lib / DBIx / Class / ResultSourceInstance.pm

diff --git a/lib/DBIx/Class/ResultSourceInstance.pm b/lib/DBIx/Class/ResultSourceInstance.pm
index ebe53c3..e9fd560 100644 (file)
--- a/lib/DBIx/Class/ResultSourceInstance.pm
+++ b/lib/DBIx/Class/ResultSourceInstance.pm
@@ -10,8 +10,11 @@ sub resultset_class { shift->result_source_instance->resultset_class(@_) }
 
 sub add_columns {
   my ($class, @cols) = @_;
-  $class->result_source_instance->add_columns(@cols);
-  $class->_mk_column_accessors(@cols);
+  my $source = $class->result_source_instance;
+  $source->add_columns(@cols);
+  foreach my $c (grep { !ref } @cols) {
+    $class->register_column($c => $source->column_info($c));
+  }
 }
 
 sub _select_columns {
@@ -37,7 +40,10 @@ sub set_primary_key { shift->result_source_instance->set_primary_key(@_); }
 sub primary_columns { shift->result_source_instance->primary_columns(@_); }
 
 sub add_relationship {
-  shift->result_source_instance->add_relationship(@_);
+  my ($class, $rel, @rest) = @_;
+  my $source = $class->result_source_instance;
+  $source->add_relationship($rel => @rest);
+  $class->register_relationship($rel => $source->relationship_info($rel));
 }
 
 sub relationships {
@@ -48,6 +54,4 @@ sub relationship_info {
   shift->result_source_instance->relationship_info(@_);
 }
 
-sub result_source { shift->result_source_instance(@_); }
-
 1;