FIX Handle polymorphic relationships that use Owner instead of Parent

silverstripe · Jan 9, 2024 · 9bfb731 · 9bfb731
1 parent b1a1d4b
commit 9bfb731
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/ORM/RelatedData/StandardRelatedDataService.php b/src/ORM/RelatedData/StandardRelatedDataService.php
@@ -313,9 +313,10 @@ private function fetchReverseHasOneResults(DataObject $record, string $class, st
         $tableName = $this->dataObjectSchema->tableName($class);
         $where = sprintf('"%s" = %u', $componentIDField, $record->ID);
 
-        // Polymorphic
-        if ($componentIDField === 'ParentID' && isset($dbFields['ParentClass'])) {
-            $where .= sprintf(' AND "ParentClass" = %s', $this->prepareClassNameLiteral(get_class($record)));
+        // Polymorphic - if $component is "Parent" or "Owner" then usually it will be polymorphic
+        $isPolymorphic = DataObject::getSchema()->hasOneComponent($class, $component) === DataObject::class;
+        if ($isPolymorphic) {
+            $where .= sprintf(' AND "' . $component . 'Class" = %s', $this->prepareClassNameLiteral(get_class($record)));
         }
 
         // Example SQL: