105 lines
3.7 KiB
Java
105 lines
3.7 KiB
Java
|
package cn.bunny.service.impl;
|
|||
|
|
|
|||
|
|
import cn.bunny.core.vms.VmsCodeGeneratorService;
|
|||
|
|
import cn.bunny.core.vms.VmsZipService;
|
|||
|
|
import cn.bunny.domain.dto.VmsArgumentDto;
|
|||
|
|
import cn.bunny.domain.vo.GeneratorVo;
|
|||
|
|
import cn.bunny.domain.vo.VmsPathVo;
|
|||
|
|
import cn.bunny.service.VmsService;
|
|||
|
|
import cn.bunny.utils.ResourceFileUtil;
|
|||
|
|
import lombok.RequiredArgsConstructor;
|
|||
|
|
import org.springframework.http.HttpHeaders;
|
|||
|
|
import org.springframework.http.HttpStatus;
|
|||
|
|
import org.springframework.http.ResponseEntity;
|
|||
|
|
import org.springframework.stereotype.Service;
|
|||
|
|
|
|||
|
|
import java.util.List;
|
|||
|
|
import java.util.Map;
|
|||
|
|
import java.util.UUID;
|
|||
|
|
import java.util.stream.Collectors;
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* VMS服务主实现类，负责协调各子服务完成代码生成、资源管理和打包下载功能
|
|||
|
|
*/
|
|||
|
|
@Service
|
|||
|
|
@RequiredArgsConstructor
|
|||
|
|
public class VmsServiceImpl implements VmsService {
|
|||
|
|
|
|||
|
|
private final VmsCodeGeneratorService codeGeneratorService;
|
|||
|
|
private final VmsZipService zipService;
|
|||
|
|
|
|||
|
|
@Override
|
|||
|
|
public Map<String, List<GeneratorVo>> generator(VmsArgumentDto dto) {
|
|||
|
|
return codeGeneratorService.generateCode(dto);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 获取VMS资源文件路径列表并按类型分组
|
|||
|
|
*
|
|||
|
|
* @return 按类型分组的VMS路径Map，key为类型，value为对应类型的VMS路径列表
|
|||
|
|
* @throws RuntimeException 当获取资源路径失败时抛出
|
|||
|
|
*/
|
|||
|
|
public Map<String, List<VmsPathVo>> vmsResourcePathList() {
|
|||
|
|
try {
|
|||
|
|
// 1. 获取vms目录下所有相对路径文件列表
|
|||
|
|
List<String> vmsRelativeFiles = ResourceFileUtil.getRelativeFiles("vms");
|
|||
|
|
|
|||
|
|
// 2. 处理文件路径并分组
|
|||
|
|
return vmsRelativeFiles.stream()
|
|||
|
|
.map(this::convertToVmsPathVo) // 转换为VO对象
|
|||
|
|
.collect(Collectors.groupingBy(VmsPathVo::getType)); // 按类型分组
|
|||
|
|
} catch (Exception e) {
|
|||
|
|
throw new RuntimeException("Failed to get VMS resource paths: " + e.getMessage(), e);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 将文件路径字符串转换为VmsPathVo对象
|
|||
|
|
*
|
|||
|
|
* @param vmFile 文件相对路径字符串
|
|||
|
|
* @return 转换后的VmsPathVo对象
|
|||
|
|
*/
|
|||
|
|
private VmsPathVo convertToVmsPathVo(String vmFile) {
|
|||
|
|
// 分割文件路径
|
|||
|
|
String[] filepathList = vmFile.split("/");
|
|||
|
|
|
|||
|
|
// 获取文件名（不含扩展名）
|
|||
|
|
String filename = filepathList[filepathList.length - 1].replace(".vm", "");
|
|||
|
|
|
|||
|
|
/*
|
|||
|
|
生成前端可用的唯一DOM元素ID
|
|||
|
|
格式: "id-" + 无横线的UUID (例如: "id-550e8400e29b41d4a716446655440000")
|
|||
|
|
|
|||
|
|
用途:
|
|||
|
|
1. 用于关联label标签和input元素的for属性
|
|||
|
|
2. 确保列表项在前端有唯一标识
|
|||
|
|
*/
|
|||
|
|
String id = "id-" + UUID.randomUUID().toString().replace("-", "");
|
|||
|
|
|
|||
|
|
return VmsPathVo.builder()
|
|||
|
|
.id(id)
|
|||
|
|
.name(vmFile)
|
|||
|
|
.label(filename)
|
|||
|
|
.type(filepathList[0]) // 使用路径的第一部分作为类型
|
|||
|
|
.build();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
@Override
|
|||
|
|
public ResponseEntity<byte[]> downloadByZip(VmsArgumentDto dto) {
|
|||
|
|
// 创建ZIP文件
|
|||
|
|
byte[] zipBytes = zipService.createZipFile(dto);
|
|||
|
|
|
|||
|
|
// 下载文件名称
|
|||
|
|
String uuid = UUID.randomUUID().toString().split("-")[0];
|
|||
|
|
String generateZipFilename = "code-" + uuid + ".zip";
|
|||
|
|
|
|||
|
|
// 设置响应头
|
|||
|
|
HttpHeaders headers = new HttpHeaders();
|
|||
|
|
headers.add("Content-Disposition", "attachment; filename=" + generateZipFilename);
|
|||
|
|
headers.add("Cache-Control", "no-cache, no-store, must-revalidate");
|
|||
|
|
headers.add("Pragma", "no-cache");
|
|||
|
|
headers.add("Expires", "0");
|
|||
|
|
|
|||
|
|
return new ResponseEntity<>(zipBytes, headers, HttpStatus.OK);
|
|||
|
|
}
|
|||
|
|
}
|