|
|
@@ -0,0 +1,294 @@
|
|
|
+package org.dromara.business.service.impl;
|
|
|
+
|
|
|
+import cn.idev.excel.EasyExcel;
|
|
|
+import cn.idev.excel.context.AnalysisContext;
|
|
|
+import cn.idev.excel.event.AnalysisEventListener;
|
|
|
+import cn.idev.excel.read.listener.ReadListener;
|
|
|
+import org.dromara.business.domain.FileUploadTask;
|
|
|
+import org.dromara.business.domain.bo.FileUploadTaskBo;
|
|
|
+import org.dromara.business.domain.vo.BlindLevelsImportVo;
|
|
|
+import org.dromara.business.domain.vo.BlindLevelsVo;
|
|
|
+import org.dromara.business.domain.vo.FileUploadTaskVo;
|
|
|
+import org.dromara.business.mapper.FileUploadTaskMapper;
|
|
|
+import org.dromara.business.service.IFileUploadTaskService;
|
|
|
+import org.dromara.common.core.domain.R;
|
|
|
+import org.dromara.common.core.utils.MapstructUtils;
|
|
|
+import org.dromara.common.core.utils.StringUtils;
|
|
|
+import org.dromara.common.mybatis.core.page.TableDataInfo;
|
|
|
+import org.dromara.common.mybatis.core.page.PageQuery;
|
|
|
+import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
|
|
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
|
|
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
|
|
+import lombok.RequiredArgsConstructor;
|
|
|
+import lombok.extern.slf4j.Slf4j;
|
|
|
+import org.dromara.system.domain.vo.SysOssVo;
|
|
|
+import org.dromara.system.service.ISysOssService;
|
|
|
+import org.springframework.beans.BeanUtils;
|
|
|
+import org.springframework.stereotype.Service;
|
|
|
+import org.springframework.web.multipart.MultipartFile;
|
|
|
+import java.io.InputStream;
|
|
|
+import java.util.*;
|
|
|
+
|
|
|
+/**
|
|
|
+ * 【请填写功能名称】Service业务层处理
|
|
|
+ *
|
|
|
+ * @author Lion Li
|
|
|
+ * @date 2025-06-27
|
|
|
+ */
|
|
|
+@Slf4j
|
|
|
+@RequiredArgsConstructor
|
|
|
+@Service
|
|
|
+public class FileUploadTaskServiceImpl implements IFileUploadTaskService {
|
|
|
+
|
|
|
+ private final FileUploadTaskMapper baseMapper;
|
|
|
+
|
|
|
+
|
|
|
+ private final ISysOssService ossService;
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 查询【请填写功能名称】
|
|
|
+ *
|
|
|
+ * @param id 主键
|
|
|
+ * @return 【请填写功能名称】
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public FileUploadTaskVo queryById(Long id){
|
|
|
+ return baseMapper.selectVoById(id);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 分页查询【请填写功能名称】列表
|
|
|
+ *
|
|
|
+ * @param bo 查询条件
|
|
|
+ * @param pageQuery 分页参数
|
|
|
+ * @return 【请填写功能名称】分页列表
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public TableDataInfo<FileUploadTaskVo> queryPageList(FileUploadTaskBo bo, PageQuery pageQuery) {
|
|
|
+ LambdaQueryWrapper<FileUploadTask> lqw = buildQueryWrapper(bo);
|
|
|
+ Page<FileUploadTaskVo> result = baseMapper.selectVoPage(pageQuery.build(), lqw);
|
|
|
+ return TableDataInfo.build(result);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 查询符合条件的【请填写功能名称】列表
|
|
|
+ *
|
|
|
+ * @param bo 查询条件
|
|
|
+ * @return 【请填写功能名称】列表
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public List<FileUploadTaskVo> queryList(FileUploadTaskBo bo) {
|
|
|
+ LambdaQueryWrapper<FileUploadTask> lqw = buildQueryWrapper(bo);
|
|
|
+ return baseMapper.selectVoList(lqw);
|
|
|
+ }
|
|
|
+
|
|
|
+ private LambdaQueryWrapper<FileUploadTask> buildQueryWrapper(FileUploadTaskBo bo) {
|
|
|
+ Map<String, Object> params = bo.getParams();
|
|
|
+ LambdaQueryWrapper<FileUploadTask> lqw = Wrappers.lambdaQuery();
|
|
|
+ lqw.orderByAsc(FileUploadTask::getId);
|
|
|
+ lqw.eq(StringUtils.isNotBlank(bo.getTaskId()), FileUploadTask::getTaskId, bo.getTaskId());
|
|
|
+ lqw.like(StringUtils.isNotBlank(bo.getFileName()), FileUploadTask::getFileName, bo.getFileName());
|
|
|
+ lqw.eq(StringUtils.isNotBlank(bo.getErrorMessage()), FileUploadTask::getErrorMessage, bo.getErrorMessage());
|
|
|
+ lqw.eq(StringUtils.isNotBlank(bo.getStatus()), FileUploadTask::getStatus, bo.getStatus());
|
|
|
+ lqw.eq(bo.getUploadTime() != null, FileUploadTask::getUploadTime, bo.getUploadTime());
|
|
|
+ lqw.eq(bo.getFinishTime() != null, FileUploadTask::getFinishTime, bo.getFinishTime());
|
|
|
+ lqw.eq(bo.getUserId() != null, FileUploadTask::getUserId, bo.getUserId());
|
|
|
+ lqw.eq(StringUtils.isNotBlank(bo.getFileUrl()), FileUploadTask::getFileUrl, bo.getFileUrl());
|
|
|
+ lqw.eq(bo.getRetryCount() != null, FileUploadTask::getRetryCount, bo.getRetryCount());
|
|
|
+ lqw.eq(bo.getMaxRetry() != null, FileUploadTask::getMaxRetry, bo.getMaxRetry());
|
|
|
+ lqw.eq(bo.getFileSize() != null, FileUploadTask::getFileSize, bo.getFileSize());
|
|
|
+ lqw.eq(bo.getTotalRows() != null, FileUploadTask::getTotalRows, bo.getTotalRows());
|
|
|
+ lqw.eq(bo.getSuccessRows() != null, FileUploadTask::getSuccessRows, bo.getSuccessRows());
|
|
|
+ lqw.eq(bo.getFailedRows() != null, FileUploadTask::getFailedRows, bo.getFailedRows());
|
|
|
+ return lqw;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 新增【请填写功能名称】
|
|
|
+ *
|
|
|
+ * @param bo 【请填写功能名称】
|
|
|
+ * @return 是否新增成功
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public Boolean insertByBo(FileUploadTaskBo bo) {
|
|
|
+ FileUploadTask add = MapstructUtils.convert(bo, FileUploadTask.class);
|
|
|
+ validEntityBeforeSave(add);
|
|
|
+ boolean flag = baseMapper.insert(add) > 0;
|
|
|
+ if (flag) {
|
|
|
+ bo.setId(add.getId());
|
|
|
+ }
|
|
|
+ return flag;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 修改【请填写功能名称】
|
|
|
+ *
|
|
|
+ * @param bo 【请填写功能名称】
|
|
|
+ * @return 是否修改成功
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public Boolean updateByBo(FileUploadTaskBo bo) {
|
|
|
+ FileUploadTask update = MapstructUtils.convert(bo, FileUploadTask.class);
|
|
|
+ validEntityBeforeSave(update);
|
|
|
+ return baseMapper.updateById(update) > 0;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 保存前的数据校验
|
|
|
+ */
|
|
|
+ private void validEntityBeforeSave(FileUploadTask entity){
|
|
|
+ //TODO 做一些数据校验,如唯一约束
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 校验并批量删除【请填写功能名称】信息
|
|
|
+ *
|
|
|
+ * @param ids 待删除的主键集合
|
|
|
+ * @param isValid 是否进行有效性校验
|
|
|
+ * @return 是否删除成功
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public Boolean deleteWithValidByIds(Collection<Long> ids, Boolean isValid) {
|
|
|
+ if(isValid){
|
|
|
+ //TODO 做一些业务上的校验,判断是否需要校验
|
|
|
+ }
|
|
|
+ return baseMapper.deleteByIds(ids) > 0;
|
|
|
+ }
|
|
|
+
|
|
|
+ @Override
|
|
|
+ public String handleFileUpload(MultipartFile file, Long userId) {
|
|
|
+ String taskId = UUID.randomUUID().toString();
|
|
|
+
|
|
|
+ FileUploadTask task = new FileUploadTask();
|
|
|
+ task.setId(System.currentTimeMillis());
|
|
|
+ task.setTaskId(taskId);
|
|
|
+ task.setFileName(file.getOriginalFilename());
|
|
|
+ task.setStatus("UPLOADING");
|
|
|
+ task.setUserId(userId);
|
|
|
+ task.setUploadTime(new Date());
|
|
|
+ task.setRetryCount(0);
|
|
|
+ task.setMaxRetry(3);
|
|
|
+ baseMapper.insertFileUploadTask(task);
|
|
|
+
|
|
|
+ // 使用 CompletableFuture 实现异步上传
|
|
|
+ /* CompletableFuture.runAsync(() -> processUpload(taskId, file));*/
|
|
|
+
|
|
|
+ processUpload(taskId, file);
|
|
|
+
|
|
|
+ return taskId;
|
|
|
+
|
|
|
+ }
|
|
|
+
|
|
|
+ @Override
|
|
|
+ public FileUploadTaskVo selectFileUploadTaskInfoByTaskId(String taskId) {
|
|
|
+ return baseMapper.selectFileUploadTaskInfoByTaskId(taskId);
|
|
|
+ }
|
|
|
+
|
|
|
+ @Override
|
|
|
+ public R<Void> retryUpload(String taskId, Long userId) {
|
|
|
+ FileUploadTaskVo task = baseMapper.selectFileUploadTaskInfoByTaskId(taskId);
|
|
|
+ if (task== null) {
|
|
|
+ return R.fail("任务不存在");
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ if (!"FAILED".equals(task.getStatus())) {
|
|
|
+ return R.fail("只有失败的任务才可以重试一下");
|
|
|
+ }
|
|
|
+
|
|
|
+ if (task.getRetryCount() >= task.getMaxRetry()) {
|
|
|
+ return R.fail("重试次数超出");
|
|
|
+ }
|
|
|
+
|
|
|
+ // 获取原始文件内容(假设已缓存或可从 OSS 下载) todo 需要加逻辑
|
|
|
+ MultipartFile file = null; // 这里需要你根据实际情况获取原始文件流
|
|
|
+ if (file == null) {
|
|
|
+ return R.fail("无法获取原始文件进行重试");
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ FileUploadTask task2 = new FileUploadTask();
|
|
|
+ BeanUtils.copyProperties(task, task2);
|
|
|
+ task2.setStatus("UPLOADING");
|
|
|
+ task2.setRetryCount(task.getRetryCount() + 1);
|
|
|
+ task2.setUploadTime(new Date());
|
|
|
+ task2.setFinishTime(null);
|
|
|
+ task2.setErrorMessage(null);
|
|
|
+ baseMapper.updateFileUploadTaskById(task2);
|
|
|
+
|
|
|
+ //TODO 做文件校验
|
|
|
+
|
|
|
+
|
|
|
+ // 异步执行重传任务
|
|
|
+ /* CompletableFuture.runAsync(() -> processUpload(taskId, file));*/
|
|
|
+ processUpload(taskId, file);
|
|
|
+ return R.ok();
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ private void processUpload(String taskId, MultipartFile file) {
|
|
|
+ try {
|
|
|
+ // 1. 上传文件到 OSS
|
|
|
+ SysOssVo sysOssVo = ossService.upload(file);
|
|
|
+ String fileUrl=sysOssVo.getUrl();
|
|
|
+ // 2. 解析 Excel 并入库
|
|
|
+ List<BlindLevelsImportVo> dataList = parseExcel(file.getInputStream());
|
|
|
+ // 3. 更新成功状态
|
|
|
+ updateTaskSuccess(taskId, fileUrl, sysOssVo.getOssId(),dataList.size(), 0);
|
|
|
+
|
|
|
+ } catch (Exception e) {
|
|
|
+ updateTaskFailed(taskId, e.getMessage());
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ // 解析 Excel 方法
|
|
|
+ private List<BlindLevelsImportVo> parseExcel(InputStream inputStream) {
|
|
|
+ List<BlindLevelsImportVo> list = new ArrayList<>();
|
|
|
+ EasyExcel.read(inputStream, BlindLevelsImportVo.class, new AnalysisEventListener<BlindLevelsImportVo>() {
|
|
|
+ @Override
|
|
|
+ public void invoke(BlindLevelsImportVo data, AnalysisContext context) {
|
|
|
+ list.add(data);
|
|
|
+ }
|
|
|
+
|
|
|
+ @Override
|
|
|
+ public void doAfterAllAnalysed(AnalysisContext context) {
|
|
|
+ // 解析完成
|
|
|
+ }
|
|
|
+ }).sheet().doRead();
|
|
|
+
|
|
|
+ return list;
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+ private void updateTaskSuccess(String taskId, String fileUrl,Long ossId,int total, int success) {
|
|
|
+ FileUploadTaskVo fileUploadTaskVo = baseMapper.selectFileUploadTaskInfoByTaskId(taskId);
|
|
|
+ FileUploadTask task = new FileUploadTask();
|
|
|
+ BeanUtils.copyProperties(fileUploadTaskVo, task);
|
|
|
+ task.setStatus("SUCCESS");
|
|
|
+ task.setFileUrl(fileUrl);
|
|
|
+ task.setFinishTime(new Date());
|
|
|
+ task.setTotalRows(total);
|
|
|
+ task.setSuccessRows(success);
|
|
|
+ task.setFailedRows(total - success);
|
|
|
+ task.setOssId(ossId);
|
|
|
+ baseMapper.updateFileUploadTaskById(task);
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ private void updateTaskFailed(String taskId, String errorMessage) {
|
|
|
+ FileUploadTaskVo fileUploadTaskVo = baseMapper.selectFileUploadTaskInfoByTaskId(taskId);
|
|
|
+ FileUploadTask task = new FileUploadTask();
|
|
|
+ BeanUtils.copyProperties(fileUploadTaskVo, task);
|
|
|
+ task.setStatus("FAILED");
|
|
|
+ task.setErrorMessage(errorMessage);
|
|
|
+ task.setId(fileUploadTaskVo.getId());
|
|
|
+ baseMapper.updateFileUploadTaskById(task);
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+}
|