diff --git a/pigx-visual/pigx-codegen/src/main/java/com/pig4cloud/pigx/codegen/service/impl/GenFormConfServiceImpl.java b/pigx-visual/pigx-codegen/src/main/java/com/pig4cloud/pigx/codegen/service/impl/GenFormConfServiceImpl.java
index 48c36d38b..9b7a545d2 100755
--- a/pigx-visual/pigx-codegen/src/main/java/com/pig4cloud/pigx/codegen/service/impl/GenFormConfServiceImpl.java
+++ b/pigx-visual/pigx-codegen/src/main/java/com/pig4cloud/pigx/codegen/service/impl/GenFormConfServiceImpl.java
@@ -65,7 +65,6 @@ public class GenFormConfServiceImpl extends ServiceImpl<GenFormConfMapper, GenFo
 	@Override
 	@SneakyThrows
 	public String getForm(Integer dsId, String tableName) {
-		DynamicDataSourceContextHolder.setDataSourceType(dsId);
 		GenFormConf form = getOne(Wrappers.<GenFormConf>lambdaQuery()
 				.eq(GenFormConf::getTableName, tableName)
 				.orderByDesc(GenFormConf::getCreateTime), false);
@@ -74,6 +73,7 @@ public class GenFormConfServiceImpl extends ServiceImpl<GenFormConfMapper, GenFo
 			return form.getFormInfo();
 		}
 
+		DynamicDataSourceContextHolder.setDataSourceType(dsId);
 		List<Map<String, String>> columns = generatorMapper.queryColumns(tableName);
 		//设置velocity资源加载器
 		Properties prop = new Properties();
diff --git a/pigx-visual/pigx-codegen/src/main/resources/template/index.vue.vm b/pigx-visual/pigx-codegen/src/main/resources/template/index.vue.vm
index 91ab08216..47fd26051 100755
--- a/pigx-visual/pigx-codegen/src/main/resources/template/index.vue.vm
+++ b/pigx-visual/pigx-codegen/src/main/resources/template/index.vue.vm
@@ -87,22 +87,13 @@
                 }).then(function () {
                     return delObj(row.$pk.lowerAttrName)
                 }).then(data => {
-                    _this.tableData.splice(index, 1)
-                    _this.$message({
-                        showClose: true,
-                        message: '删除成功',
-                        type: 'success'
-                    })
+                    _this.$message.success('删除成功')
                     this.getList(this.page)
                 })
             },
             handleUpdate: function (row, index, done,loading) {
                 putObj(row).then(data => {
-                    this.$message({
-                        showClose: true,
-                        message: '修改成功',
-                        type: 'success'
-                    })
+                    this.$message.success('修改成功')
                     done()
                     this.getList(this.page)
                 }).catch(() => {
@@ -111,11 +102,7 @@
             },
             handleSave: function (row, done,loading) {
                 addObj(row).then(data => {
-                    this.$message({
-                        showClose: true,
-                        message: '添加成功',
-                        type: 'success'
-                    })
+                    this.$message.success('添加成功')
                     done()
                     this.getList(this.page)
                 }).catch(() => {
@@ -131,7 +118,4 @@
             }
         }
     }
-</script>
-
-<style lang="scss" scoped>
-</style>
+</script>
\ No newline at end of file