浏览代码

!18 fix 代码生成器编辑问题
Merge pull request !18 from 疯狂的狮子li/dev

疯狂的狮子li 4 年之前
父节点
当前提交
db43f678d8
共有 1 个文件被更改,包括 10 次插入1 次删除
  1. 10 1
      ruoyi-generator/src/main/java/com/ruoyi/generator/service/GenTableServiceImpl.java

+ 10 - 1
ruoyi-generator/src/main/java/com/ruoyi/generator/service/GenTableServiceImpl.java

@@ -6,6 +6,7 @@ import cn.hutool.core.util.StrUtil;
 import com.alibaba.fastjson.JSON;
 import com.alibaba.fastjson.JSONObject;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.ruoyi.common.constant.Constants;
 import com.ruoyi.common.constant.GenConstants;
@@ -122,7 +123,15 @@ public class GenTableServiceImpl extends ServiceImpl<GenTableMapper, GenTable> i
         int row = baseMapper.updateById(genTable);
         if (row > 0) {
             for (GenTableColumn cenTableColumn : genTable.getColumns()) {
-                genTableColumnMapper.updateById(cenTableColumn);
+                genTableColumnMapper.update(cenTableColumn,
+                        new LambdaUpdateWrapper<GenTableColumn>()
+                                .set(cenTableColumn.getIsPk() == null, GenTableColumn::getIsPk, null)
+                                .set(cenTableColumn.getIsIncrement() == null, GenTableColumn::getIsIncrement, null)
+                                .set(cenTableColumn.getIsInsert() == null, GenTableColumn::getIsInsert, null)
+                                .set(cenTableColumn.getIsEdit() == null, GenTableColumn::getIsEdit, null)
+                                .set(cenTableColumn.getIsList() == null, GenTableColumn::getIsList, null)
+                                .set(cenTableColumn.getIsQuery() == null, GenTableColumn::getIsQuery, null)
+                                .eq(GenTableColumn::getColumnId,cenTableColumn.getColumnId()));
             }
         }
     }