DataSources: MergedDataSource - support class subtypes

inteve · Dec 5, 2023 · d5a9bc3 · d5a9bc3
1 parent d246c5f
commit d5a9bc3
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/DataSources/MergedDataSource.php b/src/DataSources/MergedDataSource.php
@@ -32,9 +32,16 @@ public function getRowId($row)
 			if (is_array($row)) {
 				throw new \Inteve\DataGrid\InvalidArgumentException(self::class . ' requires only objects, array given.');
 			}
+
 			$class = get_class($row);
 
 			if (!isset($this->dataSources[$class])) {
+				foreach ($this->dataSources as $dataSourceClass => $dataSource) {
+					if ($row instanceof $dataSourceClass) {
+						return $dataSource->getRowId($row);
+					}
+				}
+
 				throw new \Inteve\DataGrid\InvalidArgumentException('Unknow row type, there is no data source for it.');
 			}