Throw exception if get_column returns multiple columns

tigron · Feb 23, 2021 · f77b0fe · f77b0fe
1 parent 6203ea5
commit f77b0fe
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/lib/Skeleton/Database/Driver/Mysqli/Proxy.php b/lib/Skeleton/Database/Driver/Mysqli/Proxy.php
@@ -418,6 +418,12 @@ public function get_column($query, $params = []) {
 		$statement->execute();
 		$result = $statement->fetch_assoc();
 
+		if (count($result) == 0) {
+			return [];
+		} elseif (count($result[0]) != 1) {
+			throw new \Exception('Resultset has more than 1 column');
+		}
+
 		$col = [];
 		foreach ($result as $row) {
 			$col[] = array_shift($row);

diff --git a/lib/Skeleton/Database/Driver/Pdo/Proxy.php b/lib/Skeleton/Database/Driver/Pdo/Proxy.php
@@ -502,6 +502,12 @@ public function get_column($query, $params = []) {
 		$statement->execute();
 		$result = $statement->fetch_assoc();
 
+		if (count($result) == 0) {
+			return [];
+		} elseif (count($result[0]) != 1) {
+			throw new \Exception('Resultset has more than 1 column');
+		}
+
 		$col = [];
 		foreach ($result as $row) {
 			$col[] = array_shift($row);