feat: added support for custom datalist to results

WPP-Public · Sep 18, 2023 · 5781d68 · 5781d68
1 parent 79ff3c0
commit 5781d68
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/src/ResultList.php b/src/ResultList.php
@@ -221,7 +221,15 @@ public function toArray()
                         return end($parts);
                     }, $documentIds);
 
-                    foreach (DataObject::get($class)->byIDs($ids) as $record) {
+                    $sng = singleton($class);
+
+                    if ($sng->hasMethod('getDataListToIndex')) {
+                        $list = $sng->getDataListToIndex();
+                    } else {
+                        $list = $sng::get();
+                    }
+
+                    foreach ($list->byIDs($ids) as $record) {
                         $retrieved[$class][$record->ID] = $record;
                     }
                 }