support HRI'd resultsets

frioux · Nov 3, 2014 · 9d15337 · 9d15337
1 parent e63de95
commit 9d15337
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 1 deletion.
diff --git a/Changes b/Changes
@@ -1,6 +1,7 @@
 Revision history for {{$dist->name}}
 
 {{$NEXT}}
+ - Transparently support resultsets using HRI (Wes Malone)
 
 0.001001  2013-05-31 23:00:29 CST6CDT
  - Remove unneeded dep on parent.pm

diff --git a/lib/Web/Util/ExtPaging.pm b/lib/Web/Util/ExtPaging.pm
@@ -23,7 +23,9 @@ sub ext_paginate {
    }
 
    return ext_parcel(
-      [map $_->$method, $resultset->all],
+      $resultset->result_class->isa('DBIx::Class::ResultClass::HashRefInflator') ?
+         [$resultset->all] :
+         [map $_->$method, $resultset->all],
       $resultset->is_paged
          ? ($resultset->pager->total_entries)
          : (),

diff --git a/t/basic.t b/t/basic.t
@@ -71,6 +71,28 @@ my $rs = $schema->resultset('Stations')->search(undef, {
    }, 'ext_paginate with coderef correctly builds structure';
 }
 
+{
+   my $rs = $schema->resultset('Stations')->search(undef, {
+      columns => ['id'],
+      order_by => 'id',
+      rows => 3,
+      page => 1,
+      result_class => 'DBIx::Class::ResultClass::HashRefInflator',
+   });
+   my $data = ext_paginate($rs);
+
+   cmp_deeply $data, {
+       total => 9,
+       data=> set({
+          id => 1,
+       },{
+          id => 2,
+       },{
+          id => 3,
+       })
+   }, 'ext_paginate passes through HRI resultset';
+}
+
 {
    my $data = ext_parcel(
       [ map +{ id => $_->id }, $rs->all ],