diff --git a/lib/RippingCluster/Database.class.php b/lib/RippingCluster/Database.class.php index 9ff1b14..49f0b13 100644 --- a/lib/RippingCluster/Database.class.php +++ b/lib/RippingCluster/Database.class.php @@ -32,11 +32,20 @@ class RippingCluster_Database { $this->dbh = null; } - public function selectAssoc($sql, $key_col, $value_col) { + public function selectAssoc($sql, $key_col, $value_cols) { $results = array(); foreach ($this->dbh->query($sql) as $row) { - $results[$row[$key_col]] = $row[$value_col]; + if (is_array($value_cols)) { + $values = array(); + foreach ($value_cols as $value_col) { + $values[$value_col] = $row[$value_col]; + } + + $results[$row[$key_col]] = $values; + } else { + $results[$row[$key_col]] = $row[$value_col]; + } } return $results;