兰宝车间质量管理系统
blame | 历史 | 补丁 | 提交 | 提交对比 | 忽略空白
疯狂的狮子li
2022-07-05 0d8510b8b37c08092958e01f2f06f079343d38fd 
 ruoyi-generator/src/main/java/com/ruoyi/generator/service/GenTableServiceImpl.java


@@ -18,6 +18,7 @@


import com.ruoyi.common.exception.ServiceException;


import com.ruoyi.common.helper.LoginHelper;


import com.ruoyi.common.utils.JsonUtils;


import com.ruoyi.common.utils.StreamUtils;


import com.ruoyi.common.utils.StringUtils;


import com.ruoyi.common.utils.file.FileUtils;


import com.ruoyi.generator.domain.GenTable;


@@ -33,6 +34,7 @@


import org.apache.velocity.VelocityContext;


import org.apache.velocity.app.Velocity;


import org.springframework.stereotype.Service;


import org.springframework.transaction.annotation.Transactional;




import java.io.ByteArrayOutputStream;


import java.io.File;


@@ -40,8 +42,6 @@


import java.io.StringWriter;


import java.nio.charset.StandardCharsets;


import java.util.*;


import java.util.function.Function;


import java.util.stream.Collectors;


import java.util.zip.ZipEntry;


import java.util.zip.ZipOutputStream;




@@ -135,6 +135,7 @@


     * @param genTable 业务信息


     * @return 结果


     */


    @Transactional(rollbackFor = Exception.class)


    @Override


    public void updateGenTable(GenTable genTable) {


        String options = JsonUtils.toJsonString(genTable.getParams());


@@ -153,6 +154,7 @@


     * @param tableIds 需要删除的数据ID


     * @return 结果


     */


    @Transactional(rollbackFor = Exception.class)


    @Override


    public void deleteGenTableByIds(Long[] tableIds) {


        List<Long> ids = Arrays.asList(tableIds);


@@ -165,6 +167,7 @@


     *


     * @param tableList 导入表列表


     */


    @Transactional(rollbackFor = Exception.class)


    @Override


    public void importGenTable(List<GenTable> tableList) {


        String operName = LoginHelper.getUsername();


@@ -284,17 +287,18 @@


     *


     * @param tableName 表名称


     */


    @Transactional(rollbackFor = Exception.class)


    @Override


    public void synchDb(String tableName) {


        GenTable table = baseMapper.selectGenTableByName(tableName);


        List<GenTableColumn> tableColumns = table.getColumns();


        Map<String, GenTableColumn> tableColumnMap = tableColumns.stream().collect(Collectors.toMap(GenTableColumn::getColumnName, Function.identity()));


        Map<String, GenTableColumn> tableColumnMap = StreamUtils.toIdentityMap(tableColumns, GenTableColumn::getColumnName);




        List<GenTableColumn> dbTableColumns = genTableColumnMapper.selectDbTableColumnsByName(tableName);


        if (CollUtil.isEmpty(dbTableColumns)) {


            throw new ServiceException("同步数据失败，原表结构不存在");


        }


        List<String> dbTableColumnNames = dbTableColumns.stream().map(GenTableColumn::getColumnName).collect(Collectors.toList());


        List<String> dbTableColumnNames = StreamUtils.toList(dbTableColumns, GenTableColumn::getColumnName);




        List<GenTableColumn> saveColumns = new ArrayList<>();


        dbTableColumns.forEach(column -> {


@@ -322,10 +326,9 @@


        if (CollUtil.isNotEmpty(saveColumns)) {


            genTableColumnMapper.insertBatch(saveColumns);


        }




        List<GenTableColumn> delColumns = tableColumns.stream().filter(column -> !dbTableColumnNames.contains(column.getColumnName())).collect(Collectors.toList());


        List<GenTableColumn> delColumns = StreamUtils.filter(tableColumns, column -> !dbTableColumnNames.contains(column.getColumnName()));


        if (CollUtil.isNotEmpty(delColumns)) {


            List<Long> ids = delColumns.stream().map(GenTableColumn::getColumnId).collect(Collectors.toList());


            List<Long> ids = StreamUtils.toList(delColumns, GenTableColumn::getColumnId);


            genTableColumnMapper.deleteBatchIds(ids);


        }


    }