From 3ecd8c548355c20312fbd8bb156ec886191c79ea Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BC=80=E6=BA=90=E6=B5=B7=E5=93=A5?= <fuhai999@gmail.com>
Date: Sat, 29 Jul 2023 17:40:10 +0800
Subject: [PATCH] feat: add more QueryWrapper.select() method

---
 .../com/mybatisflex/core/query/QueryWrapper.java   | 14 ++++++++++++++
 .../core/query/QueryWrapperAdapter.java            |  6 ++++++
 2 files changed, 20 insertions(+)

diff --git a/mybatis-flex-core/src/main/java/com/mybatisflex/core/query/QueryWrapper.java b/mybatis-flex-core/src/main/java/com/mybatisflex/core/query/QueryWrapper.java
index f70009d4..bc82cb23 100644
--- a/mybatis-flex-core/src/main/java/com/mybatisflex/core/query/QueryWrapper.java
+++ b/mybatis-flex-core/src/main/java/com/mybatisflex/core/query/QueryWrapper.java
@@ -117,6 +117,20 @@ public class QueryWrapper extends BaseQueryWrapper<QueryWrapper> {
         return this;
     }
 
+    public QueryWrapper select(QueryColumn[] queryColumns, QueryColumn... queryColumns2) {
+        for (QueryColumn column : queryColumns) {
+            if (column != null) {
+                addSelectColumn(column);
+            }
+        }
+        for (QueryColumn column : queryColumns2) {
+            if (column != null) {
+                addSelectColumn(column);
+            }
+        }
+        return this;
+    }
+
     public QueryWrapper from(TableDef... tableDefs) {
         for (TableDef tableDef : tableDefs) {
             from(new QueryTable(tableDef));
diff --git a/mybatis-flex-core/src/main/java/com/mybatisflex/core/query/QueryWrapperAdapter.java b/mybatis-flex-core/src/main/java/com/mybatisflex/core/query/QueryWrapperAdapter.java
index a4f397af..8d859f92 100644
--- a/mybatis-flex-core/src/main/java/com/mybatisflex/core/query/QueryWrapperAdapter.java
+++ b/mybatis-flex-core/src/main/java/com/mybatisflex/core/query/QueryWrapperAdapter.java
@@ -83,6 +83,12 @@ public class QueryWrapperAdapter<R extends QueryWrapperAdapter<R>> extends Query
         return (R) this;
     }
 
+    @Override
+    public R select(QueryColumn[] queryColumns, QueryColumn... queryColumns2) {
+        super.select(queryColumns,queryColumns2);
+        return (R) this;
+    }
+
     @Override
     public R from(TableDef... tableDefs) {
         super.from(tableDefs);