From 9272a0b954da58731d54855175d61f2f15093913 Mon Sep 17 00:00:00 2001
From: junjie <junjie.xia@fit2cloud.com>
Date: Tue, 23 Feb 2021 23:34:52 +0800
Subject: [PATCH] =?UTF-8?q?feat(=E6=95=B0=E6=8D=AE=E9=9B=86):=20=E6=95=B0?=
 =?UTF-8?q?=E6=8D=AE=E5=BA=93=E8=A1=A8=EF=BC=9Amysql=E7=9B=B4=E8=BF=9E?=
 =?UTF-8?q?=E6=A8=A1=E5=BC=8F=EF=BC=8C=E8=A1=A8=E5=AD=97=E6=AE=B5=20?=
 =?UTF-8?q?=E7=B1=BB=E5=9E=8B=20=E6=95=B0=E6=8D=AE=20=E8=AF=BB=E5=8F=96?=
 =?UTF-8?q?=E5=B1=95=E7=A4=BA?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../io/dataease/base/domain/DatasetTable.java |  2 +
 .../base/mapper/DatasetTableMapper.java       |  6 ++
 .../base/mapper/DatasetTableMapper.xml        | 67 ++++++++++++++++++-
 .../dataset/DataSetTableController.java       | 11 +++
 .../dto/dataset/DataTableInfoDTO.java         | 15 +++++
 .../service/dataset/DataSetTableService.java  | 54 ++++++++++++++-
 .../db/migration/V9__dataset_tables.sql       |  1 +
 .../dataset/data/TabDataPreview.vue           | 26 ++++---
 .../components/dataset/data/ViewTable.vue     | 28 +++++++-
 9 files changed, 188 insertions(+), 22 deletions(-)
 create mode 100644 backend/src/main/java/io/dataease/dto/dataset/DataTableInfoDTO.java

diff --git a/backend/src/main/java/io/dataease/base/domain/DatasetTable.java b/backend/src/main/java/io/dataease/base/domain/DatasetTable.java
index 1d9a38ef03..a9d66c4f6d 100644
--- a/backend/src/main/java/io/dataease/base/domain/DatasetTable.java
+++ b/backend/src/main/java/io/dataease/base/domain/DatasetTable.java
@@ -19,5 +19,7 @@ public class DatasetTable implements Serializable {
 
     private Long createTime;
 
+    private String info;
+
     private static final long serialVersionUID = 1L;
 }
\ No newline at end of file
diff --git a/backend/src/main/java/io/dataease/base/mapper/DatasetTableMapper.java b/backend/src/main/java/io/dataease/base/mapper/DatasetTableMapper.java
index e99df7a37d..41e195161c 100644
--- a/backend/src/main/java/io/dataease/base/mapper/DatasetTableMapper.java
+++ b/backend/src/main/java/io/dataease/base/mapper/DatasetTableMapper.java
@@ -16,15 +16,21 @@ public interface DatasetTableMapper {
 
     int insertSelective(DatasetTable record);
 
+    List<DatasetTable> selectByExampleWithBLOBs(DatasetTableExample example);
+
     List<DatasetTable> selectByExample(DatasetTableExample example);
 
     DatasetTable selectByPrimaryKey(String id);
 
     int updateByExampleSelective(@Param("record") DatasetTable record, @Param("example") DatasetTableExample example);
 
+    int updateByExampleWithBLOBs(@Param("record") DatasetTable record, @Param("example") DatasetTableExample example);
+
     int updateByExample(@Param("record") DatasetTable record, @Param("example") DatasetTableExample example);
 
     int updateByPrimaryKeySelective(DatasetTable record);
 
+    int updateByPrimaryKeyWithBLOBs(DatasetTable record);
+
     int updateByPrimaryKey(DatasetTable record);
 }
\ No newline at end of file
diff --git a/backend/src/main/java/io/dataease/base/mapper/DatasetTableMapper.xml b/backend/src/main/java/io/dataease/base/mapper/DatasetTableMapper.xml
index 1a6c56e7af..bec94d3931 100644
--- a/backend/src/main/java/io/dataease/base/mapper/DatasetTableMapper.xml
+++ b/backend/src/main/java/io/dataease/base/mapper/DatasetTableMapper.xml
@@ -10,6 +10,9 @@
     <result column="create_by" jdbcType="VARCHAR" property="createBy" />
     <result column="create_time" jdbcType="BIGINT" property="createTime" />
   </resultMap>
+  <resultMap extends="BaseResultMap" id="ResultMapWithBLOBs" type="io.dataease.base.domain.DatasetTable">
+    <result column="info" jdbcType="LONGVARCHAR" property="info" />
+  </resultMap>
   <sql id="Example_Where_Clause">
     <where>
       <foreach collection="oredCriteria" item="criteria" separator="or">
@@ -71,6 +74,25 @@
   <sql id="Base_Column_List">
     id, `name`, scene_id, data_source_id, `type`, create_by, create_time
   </sql>
+  <sql id="Blob_Column_List">
+    info
+  </sql>
+  <select id="selectByExampleWithBLOBs" parameterType="io.dataease.base.domain.DatasetTableExample" resultMap="ResultMapWithBLOBs">
+    select
+    <if test="distinct">
+      distinct
+    </if>
+    <include refid="Base_Column_List" />
+    ,
+    <include refid="Blob_Column_List" />
+    from dataset_table
+    <if test="_parameter != null">
+      <include refid="Example_Where_Clause" />
+    </if>
+    <if test="orderByClause != null">
+      order by ${orderByClause}
+    </if>
+  </select>
   <select id="selectByExample" parameterType="io.dataease.base.domain.DatasetTableExample" resultMap="BaseResultMap">
     select
     <if test="distinct">
@@ -85,9 +107,11 @@
       order by ${orderByClause}
     </if>
   </select>
-  <select id="selectByPrimaryKey" parameterType="java.lang.String" resultMap="BaseResultMap">
+  <select id="selectByPrimaryKey" parameterType="java.lang.String" resultMap="ResultMapWithBLOBs">
     select 
     <include refid="Base_Column_List" />
+    ,
+    <include refid="Blob_Column_List" />
     from dataset_table
     where id = #{id,jdbcType=VARCHAR}
   </select>
@@ -104,10 +128,10 @@
   <insert id="insert" parameterType="io.dataease.base.domain.DatasetTable">
     insert into dataset_table (id, `name`, scene_id, 
       data_source_id, `type`, create_by, 
-      create_time)
+      create_time, info)
     values (#{id,jdbcType=VARCHAR}, #{name,jdbcType=VARCHAR}, #{sceneId,jdbcType=VARCHAR}, 
       #{dataSourceId,jdbcType=VARCHAR}, #{type,jdbcType=VARCHAR}, #{createBy,jdbcType=VARCHAR}, 
-      #{createTime,jdbcType=BIGINT})
+      #{createTime,jdbcType=BIGINT}, #{info,jdbcType=LONGVARCHAR})
   </insert>
   <insert id="insertSelective" parameterType="io.dataease.base.domain.DatasetTable">
     insert into dataset_table
@@ -133,6 +157,9 @@
       <if test="createTime != null">
         create_time,
       </if>
+      <if test="info != null">
+        info,
+      </if>
     </trim>
     <trim prefix="values (" suffix=")" suffixOverrides=",">
       <if test="id != null">
@@ -156,6 +183,9 @@
       <if test="createTime != null">
         #{createTime,jdbcType=BIGINT},
       </if>
+      <if test="info != null">
+        #{info,jdbcType=LONGVARCHAR},
+      </if>
     </trim>
   </insert>
   <select id="countByExample" parameterType="io.dataease.base.domain.DatasetTableExample" resultType="java.lang.Long">
@@ -188,11 +218,28 @@
       <if test="record.createTime != null">
         create_time = #{record.createTime,jdbcType=BIGINT},
       </if>
+      <if test="record.info != null">
+        info = #{record.info,jdbcType=LONGVARCHAR},
+      </if>
     </set>
     <if test="_parameter != null">
       <include refid="Update_By_Example_Where_Clause" />
     </if>
   </update>
+  <update id="updateByExampleWithBLOBs" parameterType="map">
+    update dataset_table
+    set id = #{record.id,jdbcType=VARCHAR},
+      `name` = #{record.name,jdbcType=VARCHAR},
+      scene_id = #{record.sceneId,jdbcType=VARCHAR},
+      data_source_id = #{record.dataSourceId,jdbcType=VARCHAR},
+      `type` = #{record.type,jdbcType=VARCHAR},
+      create_by = #{record.createBy,jdbcType=VARCHAR},
+      create_time = #{record.createTime,jdbcType=BIGINT},
+      info = #{record.info,jdbcType=LONGVARCHAR}
+    <if test="_parameter != null">
+      <include refid="Update_By_Example_Where_Clause" />
+    </if>
+  </update>
   <update id="updateByExample" parameterType="map">
     update dataset_table
     set id = #{record.id,jdbcType=VARCHAR},
@@ -227,9 +274,23 @@
       <if test="createTime != null">
         create_time = #{createTime,jdbcType=BIGINT},
       </if>
+      <if test="info != null">
+        info = #{info,jdbcType=LONGVARCHAR},
+      </if>
     </set>
     where id = #{id,jdbcType=VARCHAR}
   </update>
+  <update id="updateByPrimaryKeyWithBLOBs" parameterType="io.dataease.base.domain.DatasetTable">
+    update dataset_table
+    set `name` = #{name,jdbcType=VARCHAR},
+      scene_id = #{sceneId,jdbcType=VARCHAR},
+      data_source_id = #{dataSourceId,jdbcType=VARCHAR},
+      `type` = #{type,jdbcType=VARCHAR},
+      create_by = #{createBy,jdbcType=VARCHAR},
+      create_time = #{createTime,jdbcType=BIGINT},
+      info = #{info,jdbcType=LONGVARCHAR}
+    where id = #{id,jdbcType=VARCHAR}
+  </update>
   <update id="updateByPrimaryKey" parameterType="io.dataease.base.domain.DatasetTable">
     update dataset_table
     set `name` = #{name,jdbcType=VARCHAR},
diff --git a/backend/src/main/java/io/dataease/controller/dataset/DataSetTableController.java b/backend/src/main/java/io/dataease/controller/dataset/DataSetTableController.java
index bf401cbc8d..32df6747e0 100644
--- a/backend/src/main/java/io/dataease/controller/dataset/DataSetTableController.java
+++ b/backend/src/main/java/io/dataease/controller/dataset/DataSetTableController.java
@@ -2,6 +2,7 @@ package io.dataease.controller.dataset;
 
 import io.dataease.base.domain.DatasetTable;
 import io.dataease.controller.request.dataset.DataSetTableRequest;
+import io.dataease.datasource.dto.TableFiled;
 import io.dataease.service.dataset.DataSetTableService;
 import org.springframework.web.bind.annotation.*;
 
@@ -42,4 +43,14 @@ public class DataSetTableController {
     public DatasetTable get(@PathVariable String id) {
         return dataSetTableService.get(id);
     }
+
+    @PostMapping("getFields")
+    public List<TableFiled> getFields(@RequestBody DataSetTableRequest dataSetTableRequest) throws Exception {
+        return dataSetTableService.getFields(dataSetTableRequest);
+    }
+
+    @PostMapping("getData")
+    public List<Object> getData(@RequestBody DataSetTableRequest dataSetTableRequest) throws Exception {
+        return dataSetTableService.getData(dataSetTableRequest);
+    }
 }
diff --git a/backend/src/main/java/io/dataease/dto/dataset/DataTableInfoDTO.java b/backend/src/main/java/io/dataease/dto/dataset/DataTableInfoDTO.java
new file mode 100644
index 0000000000..1cb2bb4427
--- /dev/null
+++ b/backend/src/main/java/io/dataease/dto/dataset/DataTableInfoDTO.java
@@ -0,0 +1,15 @@
+package io.dataease.dto.dataset;
+
+import lombok.Getter;
+import lombok.Setter;
+
+/**
+ * @Author gin
+ * @Date 2021/2/23 8:47 下午
+ */
+@Setter
+@Getter
+public class DataTableInfoDTO {
+    private String table;
+    private String sql;
+}
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 1fc10c9cc6..fbc8f7aa0f 100644
--- a/backend/src/main/java/io/dataease/service/dataset/DataSetTableService.java
+++ b/backend/src/main/java/io/dataease/service/dataset/DataSetTableService.java
@@ -1,11 +1,20 @@
 package io.dataease.service.dataset;
 
+import com.alibaba.fastjson.JSONArray;
+import com.alibaba.fastjson.JSONObject;
 import com.alibaba.nacos.common.util.UuidUtils;
+import com.google.gson.Gson;
 import io.dataease.base.domain.DatasetTable;
 import io.dataease.base.domain.DatasetTableExample;
+import io.dataease.base.domain.Datasource;
 import io.dataease.base.mapper.DatasetTableMapper;
-import io.dataease.commons.utils.BeanUtils;
+import io.dataease.base.mapper.DatasourceMapper;
 import io.dataease.controller.request.dataset.DataSetTableRequest;
+import io.dataease.datasource.dto.TableFiled;
+import io.dataease.datasource.provider.DatasourceProvider;
+import io.dataease.datasource.provider.ProviderFactory;
+import io.dataease.datasource.request.DatasourceRequest;
+import io.dataease.dto.dataset.DataTableInfoDTO;
 import org.apache.commons.lang3.StringUtils;
 import org.springframework.stereotype.Service;
 
@@ -20,6 +29,8 @@ import java.util.List;
 public class DataSetTableService {
     @Resource
     private DatasetTableMapper datasetTableMapper;
+    @Resource
+    private DatasourceMapper datasourceMapper;
 
     public void batchInsert(List<DatasetTable> datasetTable) {
         for (DatasetTable table : datasetTable) {
@@ -31,9 +42,14 @@ public class DataSetTableService {
         if (StringUtils.isEmpty(datasetTable.getId())) {
             datasetTable.setId(UuidUtils.generateUuid());
             datasetTable.setCreateTime(System.currentTimeMillis());
+            DataTableInfoDTO dataTableInfoDTO = new DataTableInfoDTO();
+            if (StringUtils.equalsIgnoreCase("db", datasetTable.getType())) {
+                dataTableInfoDTO.setTable(datasetTable.getName());
+            }
+            datasetTable.setInfo(new Gson().toJson(dataTableInfoDTO));
             datasetTableMapper.insert(datasetTable);
         } else {
-            datasetTableMapper.updateByPrimaryKey(datasetTable);
+            datasetTableMapper.updateByPrimaryKeyWithBLOBs(datasetTable);
         }
         return datasetTable;
     }
@@ -48,10 +64,42 @@ public class DataSetTableService {
         if (StringUtils.isNotEmpty(dataSetTableRequest.getSort())) {
             datasetTableExample.setOrderByClause(dataSetTableRequest.getSort());
         }
-        return datasetTableMapper.selectByExample(datasetTableExample);
+        return datasetTableMapper.selectByExampleWithBLOBs(datasetTableExample);
     }
 
     public DatasetTable get(String id) {
         return datasetTableMapper.selectByPrimaryKey(id);
     }
+
+    public List<TableFiled> getFields(DataSetTableRequest dataSetTableRequest) throws Exception {
+        Datasource ds = datasourceMapper.selectByPrimaryKey(dataSetTableRequest.getDataSourceId());
+        DatasourceProvider datasourceProvider = ProviderFactory.getProvider(ds.getType());
+        DatasourceRequest datasourceRequest = new DatasourceRequest();
+        datasourceRequest.setDatasource(ds);
+        datasourceRequest.setTable(new Gson().fromJson(dataSetTableRequest.getInfo(), DataTableInfoDTO.class).getTable());
+        return datasourceProvider.getTableFileds(datasourceRequest);
+    }
+
+    public List<Object> getData(DataSetTableRequest dataSetTableRequest) throws Exception {
+        Datasource ds = datasourceMapper.selectByPrimaryKey(dataSetTableRequest.getDataSourceId());
+        DatasourceProvider datasourceProvider = ProviderFactory.getProvider(ds.getType());
+        DatasourceRequest datasourceRequest = new DatasourceRequest();
+        datasourceRequest.setDatasource(ds);
+        String table = new Gson().fromJson(dataSetTableRequest.getInfo(), DataTableInfoDTO.class).getTable();
+        datasourceRequest.setQuery("SELECT * FROM " + table + " LIMIT 0,100;");
+
+        List<TableFiled> fields = getFields(dataSetTableRequest);
+        List<String[]> data = datasourceProvider.getData(datasourceRequest);
+
+        JSONArray jsonArray = new JSONArray();
+        data.forEach(ele -> {
+            JSONObject jsonObject = new JSONObject();
+            for (int i = 0; i < ele.length; i++) {
+                jsonObject.put(fields.get(i).getFieldName(), ele[i]);
+            }
+            jsonArray.add(jsonObject);
+        });
+
+        return jsonArray;
+    }
 }
diff --git a/backend/src/main/resources/db/migration/V9__dataset_tables.sql b/backend/src/main/resources/db/migration/V9__dataset_tables.sql
index b5983e1b8d..f69e8d94ac 100644
--- a/backend/src/main/resources/db/migration/V9__dataset_tables.sql
+++ b/backend/src/main/resources/db/migration/V9__dataset_tables.sql
@@ -4,6 +4,7 @@ CREATE TABLE IF NOT EXISTS `dataset_table` (
   `scene_id` varchar(50) NOT NULL COMMENT '场景ID',
   `data_source_id` varchar(50) NOT NULL COMMENT '数据源ID',
   `type` varchar(50) COMMENT 'db,sql,excel,custom',
+  `info` longtext COMMENT '表原始信息',
   `create_by` varchar(50)  COMMENT '创建人ID',
   `create_time` bigint(13) COMMENT '创建时间',
   PRIMARY KEY (`id`)
diff --git a/frontend/src/business/components/dataset/data/TabDataPreview.vue b/frontend/src/business/components/dataset/data/TabDataPreview.vue
index 480d362da0..8a8e52eb13 100644
--- a/frontend/src/business/components/dataset/data/TabDataPreview.vue
+++ b/frontend/src/business/components/dataset/data/TabDataPreview.vue
@@ -2,22 +2,15 @@
 <el-col>
   <el-table
     size="mini"
-    :data="tableData"
+    :data="data"
     border
-    style="width: 100%">
+    style="width: 100%;">
     <el-table-column
-      prop="date"
-      label="日期"
-      width="180">
-    </el-table-column>
-    <el-table-column
-      prop="name"
-      label="姓名"
-      width="180">
-    </el-table-column>
-    <el-table-column
-      prop="address"
-      label="地址">
+      width="180px"
+      v-for="field in fields"
+      :key="field.fieldName"
+      :prop="field.fieldName"
+      :label="field.fieldName">
     </el-table-column>
   </el-table>
 </el-col>
@@ -25,6 +18,11 @@
 
 <script>
 export default {
+  props: {
+    table: Object,
+    fields: Array,
+    data: Array
+  },
   name: "TabDataPreview",
   data() {
     return {
diff --git a/frontend/src/business/components/dataset/data/ViewTable.vue b/frontend/src/business/components/dataset/data/ViewTable.vue
index 051831bd82..079a34c887 100644
--- a/frontend/src/business/components/dataset/data/ViewTable.vue
+++ b/frontend/src/business/components/dataset/data/ViewTable.vue
@@ -19,7 +19,7 @@
 
     <el-tabs v-model="tabActive">
       <el-tab-pane :label="$t('dataset.data_preview')" name="dataPreview">
-        <tab-data-preview/>
+        <tab-data-preview :table="table" :fields="fields" :data="data"/>
       </el-tab-pane>
       <el-tab-pane label="tab2" name="tab2">
         tab2
@@ -46,7 +46,9 @@ export default {
       table: {
         name: ''
       },
-      tabActive: 'dataPreview'
+      fields: [],
+      data: [],
+      tabActive: 'dataPreview',
     }
   },
   computed: {
@@ -66,13 +68,35 @@ export default {
   },
   methods: {
     initTable(id) {
+      console.log(id);
       if (id !== null) {
+        this.data = [];
         this.$post('/dataset/table/get/' + id, null, response => {
           this.table = response.data;
+          this.initTableFields();
         })
       }
     },
 
+    initTableFields() {
+      if (this.table.id) {
+        this.$post('/dataset/table/getFields', this.table, response => {
+          // console.log(response.data);
+          this.fields = response.data;
+          this.initTableData();
+        });
+      }
+    },
+    initTableData() {
+      if (this.table.id) {
+        this.$post('/dataset/table/getData', this.table, response => {
+          // console.log(response.data);
+          this.data = response.data;
+          console.log(this.data)
+        });
+      }
+    },
+
     resetTable() {
       this.table = {
         name: ''