diff --git a/vue/src/api/tool/gen.js b/vue/src/api/tool/gen.js
index 45069278..c44638e0 100644
--- a/vue/src/api/tool/gen.js
+++ b/vue/src/api/tool/gen.js
@@ -74,3 +74,11 @@ export function synchDb(tableName) {
     method: 'get'
   })
 }
+
+// 查询数据源列表
+export function listDataSource() {
+  return request({
+      url: '/tool/gen/getDataNames',
+      method: 'get',
+  });
+}
\ No newline at end of file
diff --git a/vue/src/views/tool/gen/importTable.vue b/vue/src/views/tool/gen/importTable.vue
index 3ea9532b..27a0c586 100644
--- a/vue/src/views/tool/gen/importTable.vue
+++ b/vue/src/views/tool/gen/importTable.vue
@@ -2,6 +2,11 @@
   <!-- 导入表 -->
   <el-dialog title="导入表" :visible.sync="visible" width="800px" top="5vh" append-to-body>
     <el-form :model="queryParams" ref="queryForm" size="small" :inline="true">
+      <el-form-item label="数据源" prop="dataName">
+                <el-select v-model="queryParams.dataName" size="small" placeholder="请选择数据源" style="width: 100%" :clearable="true">
+                    <el-option v-for="(item, index) in dataSources" :key="index" :label="item" :value="item"></el-option>
+                </el-select>
+            </el-form-item>
       <el-form-item label="表名称" prop="tableName">
         <el-input
           v-model="queryParams.tableName"
@@ -47,7 +52,7 @@
 </template>
 
 <script>
-import { listDbTable, importTable } from "@/api/tool/gen";
+import { listDbTable, importTable, listDataSource } from '@/api/tool/gen';
 export default {
   data() {
     return {
@@ -59,12 +64,14 @@ export default {
       total: 0,
       // 表数据
       dbTableList: [],
+      dataSources: [],
       // 查询参数
       queryParams: {
         pageNum: 1,
         pageSize: 10,
         tableName: undefined,
-        tableComment: undefined
+        tableComment: undefined,
+        dataName: 'master',
       }
     };
   },
@@ -90,6 +97,14 @@ export default {
         }
       });
     },
+     //获取数据源
+     handleDataSource() {
+            this.loading = true;
+            listDataSource().then((response) => {
+                this.dataSources = response.data;
+                this.loading = false;
+            });
+        },
     /** 搜索按钮操作 */
     handleQuery() {
       this.queryParams.pageNum = 1;