From dd8e622958bdac4b3c5cc6fbd34562f2087021b0 Mon Sep 17 00:00:00 2001
From: junjie <junjie.xia@fit2cloud.com>
Date: Thu, 10 Jun 2021 10:48:05 +0800
Subject: [PATCH] =?UTF-8?q?feat(=E6=95=B0=E6=8D=AE=E9=9B=86):=E8=87=AA?=
 =?UTF-8?q?=E5=AE=9A=E4=B9=89=E6=95=B0=E6=8D=AE=E9=9B=86=E4=B8=8D=E8=83=BD?=
 =?UTF-8?q?=E5=86=8D=E5=88=B6=E4=BD=9C=E8=87=AA=E5=AE=9A=E4=B9=89=E6=95=B0?=
 =?UTF-8?q?=E6=8D=AE=E9=9B=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../io/dataease/base/mapper/ext/ExtDataSetTableMapper.xml | 6 ++++++
 .../controller/request/dataset/DataSetTableRequest.java   | 1 +
 .../io/dataease/service/dataset/DataSetTableService.java  | 1 +
 frontend/src/views/dataset/add/AddCustom.vue              | 5 +++--
 .../src/views/dataset/common/DatasetGroupSelector.vue     | 8 +++++++-
 5 files changed, 18 insertions(+), 3 deletions(-)

diff --git a/backend/src/main/java/io/dataease/base/mapper/ext/ExtDataSetTableMapper.xml b/backend/src/main/java/io/dataease/base/mapper/ext/ExtDataSetTableMapper.xml
index 8b51a258e0..1f1d6ad214 100644
--- a/backend/src/main/java/io/dataease/base/mapper/ext/ExtDataSetTableMapper.xml
+++ b/backend/src/main/java/io/dataease/base/mapper/ext/ExtDataSetTableMapper.xml
@@ -21,6 +21,12 @@
             <if test="mode != null">
                 and mode = #{mode,jdbcType=INTEGER}
             </if>
+            <if test="typeFilter != null">
+                and type in
+                <foreach collection="typeFilter" item="item" open="(" separator="," close=")">
+                    #{item,jdbcType=INTEGER}
+                </foreach>
+            </if>
         </where>
         <if test="sort != null">
             order by ${sort}
diff --git a/backend/src/main/java/io/dataease/controller/request/dataset/DataSetTableRequest.java b/backend/src/main/java/io/dataease/controller/request/dataset/DataSetTableRequest.java
index ac4119d714..a71b592fc4 100644
--- a/backend/src/main/java/io/dataease/controller/request/dataset/DataSetTableRequest.java
+++ b/backend/src/main/java/io/dataease/controller/request/dataset/DataSetTableRequest.java
@@ -19,4 +19,5 @@ public class DataSetTableRequest extends DatasetTable {
     private String userId;
     private Integer editType;
     private Boolean isRename;
+    private List<String> typeFilter;
 }
diff --git a/backend/src/main/java/io/dataease/service/dataset/DataSetTableService.java b/backend/src/main/java/io/dataease/service/dataset/DataSetTableService.java
index 0b6bbbeb47..0c822736fd 100644
--- a/backend/src/main/java/io/dataease/service/dataset/DataSetTableService.java
+++ b/backend/src/main/java/io/dataease/service/dataset/DataSetTableService.java
@@ -176,6 +176,7 @@ public class DataSetTableService {
 
     public List<DataSetTableDTO> list(DataSetTableRequest dataSetTableRequest) {
         dataSetTableRequest.setUserId(String.valueOf(AuthUtils.getUser().getUserId()));
+        dataSetTableRequest.setTypeFilter(dataSetTableRequest.getTypeFilter());
         return extDataSetTableMapper.search(dataSetTableRequest);
     }
 
diff --git a/frontend/src/views/dataset/add/AddCustom.vue b/frontend/src/views/dataset/add/AddCustom.vue
index c30c3ed23b..512b4029c0 100644
--- a/frontend/src/views/dataset/add/AddCustom.vue
+++ b/frontend/src/views/dataset/add/AddCustom.vue
@@ -23,7 +23,7 @@
     </el-row>
     <el-col style="display: flex;flex-direction: row">
       <el-col class="panel-height" style="width: 220px;border-right:solid 1px #dcdfe6;border-top:solid 1px #dcdfe6;padding-right: 15px;overflow-y: auto;">
-        <dataset-group-selector :table="table" :mode="1" :checked-list="checkedList" :union-data="unionData" @getTable="getTable" />
+        <dataset-group-selector :custom-type="customType" :table="table" :mode="1" :checked-list="checkedList" :union-data="unionData" @getTable="getTable" />
       </el-col>
       <el-col class="panel-height" style="width: 235px;border-top:solid 1px #dcdfe6;padding: 0 15px;overflow-y: auto;">
         <dataset-custom-field :table="table" :checked-list="checkedList" @getChecked="getChecked" />
@@ -83,7 +83,8 @@ export default {
       unionData: [],
       height: 500,
       data: [],
-      fields: []
+      fields: [],
+      customType: ['db', 'sql', 'excel']
     }
   },
   watch: {
diff --git a/frontend/src/views/dataset/common/DatasetGroupSelector.vue b/frontend/src/views/dataset/common/DatasetGroupSelector.vue
index e44c6f0792..572974e556 100644
--- a/frontend/src/views/dataset/common/DatasetGroupSelector.vue
+++ b/frontend/src/views/dataset/common/DatasetGroupSelector.vue
@@ -108,6 +108,11 @@ import { isKettleRunning, post } from '@/api/dataset/dataset'
 export default {
   name: 'DatasetGroupSelector',
   props: {
+    customType: {
+      type: Array,
+      required: false,
+      default: null
+    },
     mode: {
       type: Number,
       required: false,
@@ -226,7 +231,8 @@ export default {
         post('/dataset/table/list', {
           sort: 'type asc,create_time desc,name asc',
           sceneId: this.currGroup.id,
-          mode: this.mode < 0 ? null : this.mode
+          mode: this.mode < 0 ? null : this.mode,
+          typeFilter: this.customType ? this.customType : null
         }, false).then(response => {
           this.tables = response.data
           for (let i = 0; i < this.tables.length; i++) {