add column variable

krzysztofzylka · Mar 24, 2023 · b70d655 · b70d655
1 parent 28eeb50
commit b70d655
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/Trait/TableSelect.php b/src/Trait/TableSelect.php
@@ -22,8 +22,8 @@ trait TableSelect {
      * @throws SelectException
      * @throws TableException
      */
-    public function find(null|array $condition = null, ?string $orderBy = null) : array {
-        $sql = 'SELECT ' . $this->prepareColumnListForSql() . ' FROM `' . $this->getName() . '` ' . implode(', ', $this->prepareBindData());
+    public function find(null|array $condition = null, ?array $columns = null, ?string $orderBy = null) : array {
+        $sql = 'SELECT ' . ($columns ?? $this->prepareColumnListForSql()) . ' FROM `' . $this->getName() . '` ' . implode(', ', $this->prepareBindData());
 
         if (!is_null($condition)) {
             $sql .= ' WHERE ' . (new Where())->getPrepareConditions($condition);
@@ -55,8 +55,8 @@ public function find(null|array $condition = null, ?string $orderBy = null) : ar
      * @return array
      * @throws SelectException
      */
-    public function findAll(null|array $condition = null, ?string $orderBy = null, ?string $limit = null, ?string $groupBy = null) : array {
-        $sql = 'SELECT ' . $this->prepareColumnListForSql() . ' FROM `' . $this->getName() . '` ' . implode(', ', $this->prepareBindData());
+    public function findAll(null|array $condition = null, ?array $columns = null, ?string $orderBy = null, ?string $limit = null, ?string $groupBy = null) : array {
+        $sql = 'SELECT ' . ($columns ?? $this->prepareColumnListForSql()) . ' FROM `' . $this->getName() . '` ' . implode(', ', $this->prepareBindData());
 
         if (!is_null($condition)) {
             $sql .= ' WHERE ' . (new Where())->getPrepareConditions($condition);