zhaowenyu
2025-11-12 0f5b969ddd52ca43ef003867b5a6749d74b7d71d
ard-route/src/main/java/com/ard/route/service/impl/ArdRouteFolderServiceImpl.java
@@ -1,5 +1,6 @@
package com.ard.route.service.impl;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
@@ -8,7 +9,9 @@
import com.ard.common.utils.DateUtils;
import com.ard.common.utils.uuid.IdUtils;
import com.ard.route.domain.route.ArdRouteFile;
import com.ard.route.domain.route.ArdRouteInformation;
import com.ard.route.mapper.ArdRouteFileMapper;
import com.ard.route.mapper.ArdRouteInformationMapper;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import com.ard.route.mapper.ArdRouteFolderMapper;
@@ -29,93 +32,105 @@
    private ArdRouteFolderMapper ardRouteFolderMapper;
    @Autowired
    private ArdRouteFileMapper ardRouteFileMapper;
    private ArdRouteInformationMapper ardRouteInformationMapper;
    /**
     * 查询文件夹
     * 查询分组
     * 
     * @param id 文件夹主键
     * @return 文件夹
     * @param id 分组主键
     * @return 分组
     */
    @Override
    public ArdRouteFolder selectArdRouteFolderById(String id) {
        ArdRouteFolder ardRouteFolder = ardRouteFolderMapper.selectArdRouteFolderById(id);
        ardRouteFolder.setArdRouteFileLs(getArdRouteFiles(id));
        ardRouteFolder.setArdRouteInformationLs(ardRouteInformationMapper.selectArdRouteInformationByFolderId(id));
        return ardRouteFolder;
    }
    /**
     * 查询文件夹列表
     * 查询分组列表
     * 
     * @param ardRouteFolder 文件夹
     * @return 文件夹
     * @param ardRouteFolder 分组
     * @return 分组
     */
    @Override
    @DataScope(deptAlias = "d", userAlias = "u")
    public List<ArdRouteFolder> selectArdRouteFolderList(ArdRouteFolder ardRouteFolder) {
        List<ArdRouteFolder> list = ardRouteFolderMapper.selectArdRouteFolderList(ardRouteFolder);
        for (ArdRouteFolder routeFolder : list) {
            routeFolder.setArdRouteFileLs(getArdRouteFiles(routeFolder.getId()));
            routeFolder.setArdRouteInformationLs(ardRouteInformationMapper.selectArdRouteInformationByFolderId(routeFolder.getId()));
        }
        return list;
    }
    /**
     * 查询文件列表
     */
    private List<ArdRouteFile> getArdRouteFiles(String folderId) {
        ArdRouteFile ardRouteFile = new ArdRouteFile();
        ardRouteFile.setFolderId(folderId);
        return ardRouteFileMapper.selectArdRouteFileList(ardRouteFile);
    }
    /**
     * 新增文件夹
     * 新增分组
     * 
     * @param ardRouteFolder 文件夹
     * @param ardRouteFolder 分组
     * @return 结果
     */
    @Override
    public int insertArdRouteFolder(ArdRouteFolder ardRouteFolder) {
        ardRouteFolder.setId(IdUtils.simpleUUID());
        ardRouteFolder.setCreateTime(DateUtils.getNowDate());
        // 获取当前数据条数
        int count = ardRouteFolderMapper.selectArdRouteFolderList(null).size();
        ardRouteFolder.setSerialNumber(count + 1);
        return ardRouteFolderMapper.insertArdRouteFolder(ardRouteFolder);
    }
    /**
     * 修改文件夹
     * 修改分组
     * 
     * @param ardRouteFolder 文件夹
     * @param ardRouteFolder 分组
     * @return 结果
     */
    @Override
    public int updateArdRouteFolder(ArdRouteFolder ardRouteFolder) {
        // 不能修改默认分组
        if ("0".equals(ardRouteFolder.getId())) {
            throw new RuntimeException("不能修改默认分组");
        }
        ardRouteFolder.setUpdateTime(DateUtils.getNowDate());
        return ardRouteFolderMapper.updateArdRouteFolder(ardRouteFolder);
    }
    /**
     * 批量删除文件夹
     * 批量删除分组
     * 
     * @param ids 需要删除的文件夹主键
     * @param ids 需要删除的分组主键
     * @return 结果
     */
    @Override
    @Transactional(rollbackFor = Exception.class)
    public Map<String, Integer> deleteArdRouteFolderByIds(String[] ids) {
        // 检查是否包含默认分组,不能删除默认分组
        for (String id : ids) {
            if ("0".equals(id)) {
                throw new RuntimeException("不能删除默认分组");
            }
        }
        Map<String, Integer> result = new HashMap<>();
        // 删除文件夹的同时也删除文件夹下的所有文件
        int fileAdd = ardRouteFileMapper.deleteRouteFileByFolderIds(ids);
        result.put("file", fileAdd);
        int folderCount = ardRouteFolderMapper.deleteArdRouteFolderByIds(ids);
        result.put("folder", folderCount);
        return result;
        try {
            // 先删除航线信息
            int informationCount = ardRouteInformationMapper.deleteRouteInformationByFolderIds(ids);
            result.put("informationCount", informationCount);
            // 再删除分组
            int folderCount = ardRouteFolderMapper.deleteArdRouteFolderByIds(ids);
            result.put("folderCount", folderCount);
            return result;
        } catch (Exception e) {
            throw e; // 抛出异常让事务回滚
        }
    }
    /**
     * 删除文件夹信息
     * 删除分组信息
     * 
     * @param id 文件夹主键
     * @param id 分组主键
     * @return 结果
     */
    @Override