commit
e6a5ed9b4f
@ -0,0 +1,65 @@ |
||||
package com.keyware.shandan.bianmu.controller; |
||||
|
||||
import cn.hutool.core.io.FileUtil; |
||||
import com.keyware.shandan.bianmu.export.ExportCache; |
||||
import com.keyware.shandan.bianmu.export.ExportProgress; |
||||
import com.keyware.shandan.bianmu.service.MetadataService; |
||||
import com.keyware.shandan.common.entity.Result; |
||||
import com.keyware.shandan.common.util.FileDownload; |
||||
import org.springframework.beans.factory.annotation.Autowired; |
||||
import org.springframework.security.core.Authentication; |
||||
import org.springframework.web.bind.annotation.*; |
||||
import javax.servlet.http.HttpServletResponse; |
||||
import java.io.File; |
||||
|
||||
/** |
||||
* 组合搜索前端控制器 |
||||
* |
||||
* @author Administrator |
||||
* @since 2021/7/1 |
||||
*/ |
||||
@RestController |
||||
@RequestMapping("/search") |
||||
public class SearchController { |
||||
@Autowired |
||||
private MetadataService metadataService; |
||||
/** |
||||
* 目录数据导出-查询任务状态 |
||||
* |
||||
* @param exportId 导出ID |
||||
* @return 导出ID |
||||
*/ |
||||
@GetMapping("/export/status/{exportId}") |
||||
public Result<Object> exportStatus(@PathVariable String exportId, Authentication auth) { |
||||
String userId = auth.getPrincipal().toString(); |
||||
ExportProgress export = ExportCache.getCache(userId, exportId); |
||||
if (export == null) { |
||||
return Result.of(null, false, "导出任务不存在"); |
||||
} |
||||
if (export.isError()) { |
||||
ExportCache.popCache(userId, exportId); |
||||
} |
||||
return Result.of(export.getProgress(), !export.isError()); |
||||
} |
||||
|
||||
/** |
||||
* 目录数据导出-下载 |
||||
* |
||||
* @param response HttpServletResponse |
||||
* @param exportId 导出ID |
||||
* @return - |
||||
*/ |
||||
@GetMapping("/export/download/{exportId}") |
||||
public String exportDownload(HttpServletResponse response, @PathVariable String exportId, Authentication auth) { |
||||
String userId = auth.getPrincipal().toString(); |
||||
ExportProgress export = ExportCache.popCache(userId, exportId); |
||||
if (export == null) { |
||||
return "错误：导出任务不存在"; |
||||
} |
||||
String path = export.getDownloadPath(); |
||||
String downloadName = FileUtil.getName(path); |
||||
File file = new File(path); |
||||
return FileDownload.download(response, file, downloadName, true); |
||||
} |
||||
|
||||
} |
||||
@ -0,0 +1,10 @@ |
||||
package com.keyware.shandan.common.service; |
||||
|
||||
/** |
||||
* @author YaoJz |
||||
* @description |
||||
* @date 2024/4/16 15:28 |
||||
*/ |
||||
public class ExportServiceImpl implements IExportService{ |
||||
|
||||
} |
||||
@ -0,0 +1,9 @@ |
||||
package com.keyware.shandan.common.service; |
||||
|
||||
/** |
||||
* @author YaoJz |
||||
* @description |
||||
* @date 2024/4/16 15:27 |
||||
*/ |
||||
public interface IExportService { |
||||
} |
||||
@ -0,0 +1,191 @@ |
||||
package com.keyware.shandan.bianmu.export; |
||||
|
||||
import cn.hutool.core.collection.CollUtil; |
||||
import cn.hutool.core.date.DateUtil; |
||||
import cn.hutool.core.io.FileUtil; |
||||
import cn.hutool.core.util.ZipUtil; |
||||
import cn.hutool.http.HttpUtil; |
||||
import cn.hutool.poi.excel.BigExcelWriter; |
||||
import cn.hutool.poi.excel.ExcelUtil; |
||||
import com.alibaba.fastjson.JSONArray; |
||||
import com.alibaba.fastjson.JSONObject; |
||||
import com.keyware.shandan.bianmu.entity.DirectoryResource; |
||||
import com.keyware.shandan.bianmu.entity.MetadataBasicVo; |
||||
import com.keyware.shandan.bianmu.service.MetadataService; |
||||
import com.keyware.shandan.common.util.StreamUtil; |
||||
import com.keyware.shandan.common.util.StringUtils; |
||||
import com.keyware.shandan.frame.config.component.AppContext; |
||||
import com.keyware.shandan.system.entity.SysFile; |
||||
import com.keyware.shandan.system.service.SysFileService; |
||||
import com.keyware.shandan.system.utils.SysSettingUtil; |
||||
|
||||
import java.io.File; |
||||
import java.util.ArrayList; |
||||
import java.util.Date; |
||||
import java.util.List; |
||||
import java.util.Map; |
||||
import java.util.stream.Collectors; |
||||
|
||||
/** |
||||
* @author YaoJz |
||||
* @description |
||||
* @date 2024/4/15 16:40 |
||||
*/ |
||||
public class FileSearchExport extends ExportProgress { |
||||
private final List<DirectoryResource> dataList; |
||||
private final SysFileService fileService; |
||||
private final MetadataService metadataService; |
||||
private final String outputDir = "目录导出_" + DateUtil.format(new Date(), "yyyyMMddHHmmss"); |
||||
private final String outputPath = tempDir + File.separator + outputDir; |
||||
private final String zipName = outputPath + ".zip"; |
||||
|
||||
public FileSearchExport(String userId, List<DirectoryResource> dataList) { |
||||
super(userId); |
||||
this.dataList = dataList; |
||||
this.fileService = AppContext.getContext().getBean(SysFileService.class); |
||||
this.metadataService = AppContext.getContext().getBean(MetadataService.class); |
||||
} |
||||
|
||||
@Override |
||||
public void run() { |
||||
try { |
||||
this.setTitle("正在生成临时数据文件"); |
||||
dataList.forEach(resource -> { |
||||
if ("file".equals(resource.getResourceType())) { |
||||
exportFile(resource); |
||||
} else { |
||||
exportMetadata(resource); |
||||
} |
||||
this.autoAddStep(); |
||||
}); |
||||
this.setTitle("正在打包数据文件"); |
||||
ZipUtil.zip(outputPath, zipName, true); |
||||
this.autoAddStep(); |
||||
this.setDone(); |
||||
this.setTitle("数据包准备完毕，开始下载"); |
||||
} catch (Exception e) { |
||||
throw new RuntimeException("导出数据在打包时出错", e); |
||||
} finally { |
||||
delete(outputPath); |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* 导出文件 |
||||
* |
||||
* @param resource 数据资源 |
||||
*/ |
||||
private void exportFile(DirectoryResource resource) { |
||||
File serverFile = new File(createResourceDir(resource.getDirectoryPath())); |
||||
SysFile file = fileService.getById(resource.getId()); |
||||
String fileDownloadUrl = SysSettingUtil.getSysSetting("BIANMU").getSysInternalAddress() + file.getWebUri(); |
||||
try { |
||||
HttpUtil.downloadFile(fileDownloadUrl, serverFile); |
||||
} catch (Exception e) { |
||||
List<String> lines = new ArrayList<>(); |
||||
String dirPath = serverFile.getParentFile().getPath(); |
||||
String name = "(文件读取错误)" + serverFile.getName(); |
||||
FileUtil.writeLines(lines, new File(dirPath, name), "utf-8"); |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* 导出数据表 |
||||
* |
||||
* @param resource 数据资源 |
||||
*/ |
||||
private void exportMetadata(DirectoryResource resource) { |
||||
File serverFile = new File(createResourceDir(resource.getDirectoryPath())); |
||||
String path = serverFile.getPath() + ".xlsx"; |
||||
MetadataBasicVo metadata = null; |
||||
JSONArray colsArray = null; |
||||
List<Map<String, Object>> datas = null; |
||||
|
||||
try { |
||||
metadata = metadataService.getById(resource.getId()); |
||||
colsArray = metadataService.getColumns(metadata.getId()); |
||||
datas = metadataService.getDynamicData(metadata); |
||||
} catch (Exception e) { |
||||
colsArray = new JSONArray(); |
||||
JSONObject defaultJson = new JSONObject(); |
||||
defaultJson.put("comment", "数据查询失败，无法连接到指定的数据库"); |
||||
colsArray.add(defaultJson); |
||||
datas = new ArrayList<>(); |
||||
|
||||
File temp = new File(path); |
||||
String dirPath = temp.getParentFile().getPath(); |
||||
path = dirPath + File.separator + "(数据读取错误)" + temp.getName(); |
||||
} |
||||
|
||||
// 所有行的集合
|
||||
List<List<String>> rowList = new ArrayList<>(); |
||||
// 字段列名集合
|
||||
List<String> colNameList = new ArrayList<>(); |
||||
// 字段列名注释集合，当做表格的第一行
|
||||
List<String> columns = StreamUtil.as(colsArray).map(json -> { |
||||
JSONObject col = (JSONObject) json; |
||||
String comment = col.getString("comment"); |
||||
String colName = col.getString("columnName"); |
||||
colNameList.add(colName); |
||||
return colName + (StringUtils.hasText(comment) ? "[" + comment + "]" : ""); |
||||
}).toList(); |
||||
|
||||
// 添加第一行
|
||||
rowList.add(columns); |
||||
|
||||
// 遍历数据
|
||||
datas.forEach(data -> { |
||||
// 遍历字段列名，获取对应数据并拼接成列的集合
|
||||
List<String> cells = colNameList.stream().map(col -> { |
||||
Object value = data.get(col); |
||||
if (value == null) { |
||||
return ""; |
||||
} |
||||
return String.valueOf(value); |
||||
}).collect(Collectors.toList()); |
||||
rowList.add(cells); |
||||
}); |
||||
|
||||
List<List<String>> rows = CollUtil.newArrayList(rowList); |
||||
//通过工具类创建writer
|
||||
BigExcelWriter writer = ExcelUtil.getBigWriter(path); |
||||
//一次性写出内容
|
||||
writer.write(rows); |
||||
//关闭writer，释放内存
|
||||
writer.close(); |
||||
} |
||||
|
||||
/** |
||||
* 根据资源路径创建目录 |
||||
* |
||||
* @param resourcePath 资源路径 |
||||
* @return 资源的的绝对路径 |
||||
*/ |
||||
private String createResourceDir(String resourcePath) { |
||||
String path = outputPath + File.separator + resourcePath; |
||||
File file = new File(path); |
||||
File parent = file.getParentFile(); |
||||
if (!parent.exists()) { |
||||
parent.mkdirs(); |
||||
} |
||||
return path; |
||||
} |
||||
|
||||
private void delete(String path) { |
||||
File file = new File(path); |
||||
if (file.exists()) { |
||||
FileUtil.del(file); |
||||
} |
||||
} |
||||
|
||||
@Override |
||||
public long getStepTotal() { |
||||
return dataList.size() + 1; |
||||
} |
||||
|
||||
@Override |
||||
public String getDownloadPath() { |
||||
return zipName; |
||||
} |
||||
|
||||
} |
||||
Reference in new issue