8 Commits

Author SHA1 Message Date
dk
55a74815b2 feat(ai-report): 新增MinIO文件存储服务
- 添加AiReportFileConst和AiReportFileExtensionConst常量类
- 实现AiReportMinioStorageService存储服务
- 添加AiReportMinioProperties配置类
- 集成MinIO依赖到ai-report-common模块
- 配置应用yml中的ai-report桶设置
- 更新报告模型控制器使用MinIO存储服务
- 修改报告模型文件存储组件集成MinIO
- 添加相关单元测试验证功能
2026-07-14 16:38:11 +08:00
aac21e5f73 Merge remote-tracking branch 'origin/main' 2026-07-14 08:37:23 +08:00
6659df649a 添加了单独的长闪报文 2026-07-14 08:37:12 +08:00
dk
9b71dfdce3 Merge remote-tracking branch 'origin/main' 2026-07-13 22:00:49 +08:00
dk
46ae986db2 feat(storage): 集成 MinIO 存储系统并支持用户签名功能
- 添加 MinIO 配置和依赖项,包括 minio、okhttp、okio 和 kotlin 库
- 创建通用文件服务接口和实现,支持文件上传到 MinIO 并记录元数据
- 添加用户签名服务接口和数据库表结构,支持签名文件管理
- 修改用户服务以支持签名文件上传,在新增和更新用户时可上传签名文件
- 实现文件预览服务测试用例的注释调整
- 创建 SysFile 和 SysUserSignature 数据库实体及对应常量类
- 实现 MinIO 存储组件,提供文件上传、下载和删除功能
- 在用户实体中添加签名相关字段用于展示签名信息
2026-07-13 22:00:39 +08:00
周宇 蔡
6fc0d58fe4 fix(config): 修正配置文件中的命名规范和路径引用
- 将XML模板中的零序电流不平衡字段名从G_VZSEQ_UNBAN/VZSEQ_UNBAN更正为G_IZSEQ_UNBAN/IZSEQ_UNBAN
- 将默认规则文件路径从template/默认规则.txt统一更改为template/DefaultRule.txt
- 更新配置类和映射服务中对默认规则文件的引用路径
2026-07-13 10:48:56 +08:00
周宇 蔡
42911e35bb Merge branch 'dev-czy' 2026-07-10 11:20:58 +08:00
周宇 蔡
cbafe85c4a fix(mms-mapping): 修复谐波数据映射配置和偏移量计算问题
- 修改JiangSu_Config1.xml中谐波电流相关指标的命名规则
- 修改默认规则文件中相电压间谐波含有率序列的描述名称
- 添加相电压间谐波含有率序列实时数据值配置
- 更新JsonToXmlConversionService中遥测数据处理逻辑,将urcb相关标识修改为brcb
- 新增calculateOffset方法用于计算指标偏移量
- 实现adjustPathWithOffset方法根据offset调整路径中的数值格式
- 添加offset字段到MetricInfo类中用于存储偏移信息
- 在规则匹配过程中传递offset值并在生成Value节点时应用偏移调整
2026-07-10 11:20:31 +08:00
62 changed files with 10220 additions and 1393 deletions

View File

@@ -0,0 +1,35 @@
<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0"
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
<modelVersion>4.0.0</modelVersion>
<parent>
<groupId>com.njcn.gather</groupId>
<artifactId>ai-report</artifactId>
<version>1.0.0</version>
</parent>
<artifactId>ai-report-common</artifactId>
<packaging>jar</packaging>
<dependencies>
<dependency>
<groupId>com.njcn</groupId>
<artifactId>njcn-common</artifactId>
<version>0.0.1</version>
</dependency>
<dependency>
<groupId>com.njcn</groupId>
<artifactId>spingboot2.3.12</artifactId>
<version>2.3.12</version>
</dependency>
<dependency>
<groupId>io.minio</groupId>
<artifactId>minio</artifactId>
<version>8.4.3</version>
</dependency>
</dependencies>
</project>

View File

@@ -0,0 +1,36 @@
package com.njcn.gather.aireport.common.config;
import cn.hutool.core.util.StrUtil;
import com.njcn.common.pojo.enums.response.CommonResponseEnum;
import com.njcn.common.pojo.exception.BusinessException;
import lombok.Data;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.stereotype.Component;
import java.util.LinkedHashMap;
import java.util.Map;
/**
* ai-report MinIO 配置。
*/
@Data
@Component
@ConfigurationProperties(prefix = "storage.minio")
public class AiReportMinioProperties {
private String endpoint;
private String accessKey;
private String secretKey;
private Map<String, String> buckets = new LinkedHashMap<String, String>();
public String requireBucket(String bucketKey) {
String bucketName = buckets.get(bucketKey);
if (StrUtil.isBlank(bucketName)) {
throw new BusinessException(CommonResponseEnum.FAIL, "MinIO桶配置缺失" + bucketKey);
}
return bucketName;
}
}

View File

@@ -0,0 +1,16 @@
package com.njcn.gather.aireport.common.constant;
/**
* ai-report 文件常量。
*/
public final class AiReportFileConst {
private AiReportFileConst() {
}
public static final String BUCKET_KEY_AI_REPORT = "ai-report";
public static final String BIZ_DIR_REPORT_MODEL = "report-model";
public static final String BIZ_DIR_TEST_DEVICE = "test-device";
public static final String BIZ_DIR_TEST_REPORT = "test-report";
}

View File

@@ -0,0 +1,27 @@
package com.njcn.gather.aireport.common.constant;
import java.util.Arrays;
import java.util.Collections;
import java.util.LinkedHashSet;
import java.util.Set;
/**
* ai-report 文件扩展名白名单。
*/
public final class AiReportFileExtensionConst {
private AiReportFileExtensionConst() {
}
public static final Set<String> ALLOWED_EXTENSIONS = unmodifiable("docx", "xlsx", "pdf");
public static final Set<String> WORD_EXTENSIONS = unmodifiable("docx");
public static final Set<String> EXCEL_EXTENSIONS = unmodifiable("xlsx");
public static final Set<String> PDF_EXTENSIONS = unmodifiable("pdf");
private static Set<String> unmodifiable(String... extensions) {
return Collections.unmodifiableSet(new LinkedHashSet<String>(Arrays.asList(extensions)));
}
}

View File

@@ -0,0 +1,384 @@
package com.njcn.gather.aireport.common.storage;
import cn.hutool.core.util.StrUtil;
import com.njcn.common.pojo.enums.response.CommonResponseEnum;
import com.njcn.common.pojo.exception.BusinessException;
import com.njcn.gather.aireport.common.config.AiReportMinioProperties;
import com.njcn.gather.aireport.common.constant.AiReportFileConst;
import io.minio.BucketExistsArgs;
import io.minio.GetObjectArgs;
import io.minio.GetObjectResponse;
import io.minio.ListObjectsArgs;
import io.minio.MakeBucketArgs;
import io.minio.MinioClient;
import io.minio.PutObjectArgs;
import io.minio.RemoveObjectArgs;
import io.minio.Result;
import io.minio.StatObjectArgs;
import io.minio.errors.ErrorResponseException;
import io.minio.messages.Item;
import lombok.Getter;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.http.MediaType;
import org.springframework.http.MediaTypeFactory;
import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.ArrayList;
import java.util.List;
import java.util.Locale;
import java.util.UUID;
/**
* ai-report MinIO 存储服务。
*/
@Slf4j
@Component
@RequiredArgsConstructor
public class AiReportMinioStorageService {
private final AiReportMinioProperties properties;
public StoredFile saveMultipartFile(String bizDir, String ownerId, MultipartFile file, boolean keepOriginalName) {
MultipartFile normalizedFile = requireFile(file);
String originalFileName = normalizeFileName(normalizedFile.getOriginalFilename());
String targetFileName = keepOriginalName ? sanitizeFileName(originalFileName) : buildStoredFileName(originalFileName);
String objectName = buildObjectName(bizDir, ownerId, targetFileName);
String bucketName = requireBucket();
String contentType = resolveContentType(normalizedFile.getContentType(), originalFileName);
try (InputStream inputStream = normalizedFile.getInputStream()) {
putObject(bucketName, objectName, inputStream, normalizedFile.getSize(), contentType);
return new StoredFile(bucketName, objectName, targetFileName, originalFileName, contentType, normalizedFile.getSize());
} catch (BusinessException exception) {
throw exception;
} catch (Exception exception) {
log.error("save multipart file to minio failed, objectName={}", objectName, exception);
throw new BusinessException(CommonResponseEnum.FAIL, "保存文件失败");
}
}
public StoredFile saveInputStream(String bizDir, String ownerId, String originalFileName, InputStream inputStream,
long size, String contentType, boolean keepOriginalName) {
String normalizedFileName = normalizeFileName(originalFileName);
String targetFileName = keepOriginalName ? sanitizeFileName(normalizedFileName) : buildStoredFileName(normalizedFileName);
String objectName = buildObjectName(bizDir, ownerId, targetFileName);
String bucketName = requireBucket();
try {
putObject(bucketName, objectName, inputStream, size, resolveContentType(contentType, normalizedFileName));
return new StoredFile(bucketName, objectName, targetFileName, normalizedFileName,
resolveContentType(contentType, normalizedFileName), size);
} catch (BusinessException exception) {
throw exception;
} catch (Exception exception) {
log.error("save input stream to minio failed, objectName={}", objectName, exception);
throw new BusinessException(CommonResponseEnum.FAIL, "保存文件失败");
}
}
public StoredFile saveBytes(String bizDir, String ownerId, String originalFileName, byte[] content,
String contentType, boolean keepOriginalName) {
byte[] fileContent = content == null ? new byte[0] : content;
return saveInputStream(bizDir, ownerId, originalFileName, new ByteArrayInputStream(fileContent),
fileContent.length, contentType, keepOriginalName);
}
public StoredObject getObject(String objectName) {
if (StrUtil.isBlank(objectName)) {
throw new BusinessException(CommonResponseEnum.FAIL, "文件不存在");
}
String bucketName = requireBucket();
try {
GetObjectResponse response = buildClient().getObject(GetObjectArgs.builder()
.bucket(bucketName)
.object(objectName)
.build());
return new StoredObject(bucketName, objectName, response);
} catch (BusinessException exception) {
throw exception;
} catch (Exception exception) {
log.error("read file from minio failed, objectName={}", objectName, exception);
throw new BusinessException(CommonResponseEnum.FAIL, "读取文件失败");
}
}
public byte[] readBytes(String objectName) {
try (StoredObject storedObject = getObject(objectName);
InputStream inputStream = storedObject.getInputStream();
ByteArrayOutputStream outputStream = new ByteArrayOutputStream()) {
byte[] buffer = new byte[8192];
int length;
while ((length = inputStream.read(buffer)) != -1) {
outputStream.write(buffer, 0, length);
}
return outputStream.toByteArray();
} catch (BusinessException exception) {
throw exception;
} catch (Exception exception) {
log.error("read file bytes from minio failed, objectName={}", objectName, exception);
throw new BusinessException(CommonResponseEnum.FAIL, "读取文件失败");
}
}
public Path downloadToTempFile(String objectName, String fileName) {
String normalizedFileName = normalizeFileName(fileName);
String extension = resolveExtension(normalizedFileName);
String suffix = StrUtil.isBlank(extension) ? ".tmp" : "." + extension;
try (StoredObject storedObject = getObject(objectName);
InputStream inputStream = storedObject.getInputStream()) {
Path tempFile = Files.createTempFile("ai-report-", suffix);
Files.copy(inputStream, tempFile, java.nio.file.StandardCopyOption.REPLACE_EXISTING);
return tempFile;
} catch (BusinessException exception) {
throw exception;
} catch (Exception exception) {
log.error("download minio object to temp file failed, objectName={}", objectName, exception);
throw new BusinessException(CommonResponseEnum.FAIL, "读取文件失败");
}
}
public List<String> listFileNames(String prefix) {
List<String> objectNames = listObjectNames(prefix);
List<String> result = new ArrayList<String>();
for (String objectName : objectNames) {
int slashIndex = objectName.lastIndexOf('/');
result.add(slashIndex >= 0 ? objectName.substring(slashIndex + 1) : objectName);
}
return result;
}
public List<String> listObjectNames(String prefix) {
List<String> result = new ArrayList<String>();
if (StrUtil.isBlank(prefix)) {
return result;
}
try {
Iterable<Result<Item>> iterable = buildClient().listObjects(ListObjectsArgs.builder()
.bucket(requireBucket())
.prefix(trimPath(prefix) + "/")
.recursive(true)
.build());
for (Result<Item> itemResult : iterable) {
Item item = itemResult.get();
if (item == null || item.isDir()) {
continue;
}
String objectName = item.objectName();
if (StrUtil.isBlank(objectName)) {
continue;
}
result.add(objectName);
}
return result;
} catch (Exception exception) {
log.error("list minio objects failed, prefix={}", prefix, exception);
throw new BusinessException(CommonResponseEnum.FAIL, "读取文件失败");
}
}
public void deleteQuietly(String objectName) {
if (StrUtil.isBlank(objectName)) {
return;
}
try {
buildClient().removeObject(RemoveObjectArgs.builder()
.bucket(requireBucket())
.object(objectName)
.build());
} catch (Exception exception) {
log.warn("delete file from minio failed, objectName={}", objectName, exception);
}
}
public boolean exists(String objectName) {
if (StrUtil.isBlank(objectName)) {
return false;
}
try {
buildClient().statObject(StatObjectArgs.builder()
.bucket(requireBucket())
.object(objectName)
.build());
return true;
} catch (ErrorResponseException exception) {
String errorCode = exception.errorResponse() == null ? null : exception.errorResponse().code();
if ("NoSuchKey".equals(errorCode) || "NoSuchObject".equals(errorCode) || "NoSuchBucket".equals(errorCode)) {
return false;
}
log.error("stat minio object failed, objectName={}", objectName, exception);
throw new BusinessException(CommonResponseEnum.FAIL, "读取文件失败");
} catch (Exception exception) {
log.error("stat minio object failed, objectName={}", objectName, exception);
throw new BusinessException(CommonResponseEnum.FAIL, "读取文件失败");
}
}
public void deleteTempFileQuietly(Path tempFile) {
if (tempFile == null) {
return;
}
try {
Files.deleteIfExists(tempFile);
} catch (IOException ignored) {
// 临时文件清理失败不影响主流程。
}
}
private void putObject(String bucketName, String objectName, InputStream inputStream, long size, String contentType) throws Exception {
ensureBucket(buildClient(), bucketName);
buildClient().putObject(PutObjectArgs.builder()
.bucket(bucketName)
.object(objectName)
.stream(inputStream, size, -1)
.contentType(contentType)
.build());
}
private MultipartFile requireFile(MultipartFile file) {
if (file == null || file.isEmpty()) {
throw new BusinessException(CommonResponseEnum.FAIL, "上传文件不能为空");
}
return file;
}
private String requireBucket() {
validateConfig();
return properties.requireBucket(AiReportFileConst.BUCKET_KEY_AI_REPORT);
}
private void validateConfig() {
if (StrUtil.hasBlank(properties.getEndpoint(), properties.getAccessKey(), properties.getSecretKey())) {
throw new BusinessException(CommonResponseEnum.FAIL, "MinIO配置不完整");
}
}
private MinioClient buildClient() {
return MinioClient.builder()
.endpoint(properties.getEndpoint())
.credentials(properties.getAccessKey(), properties.getSecretKey())
.build();
}
private void ensureBucket(MinioClient client, String bucketName) throws Exception {
boolean exists = client.bucketExists(BucketExistsArgs.builder().bucket(bucketName).build());
if (!exists) {
client.makeBucket(MakeBucketArgs.builder().bucket(bucketName).build());
}
}
private String buildObjectName(String bizDir, String ownerId, String fileName) {
StringBuilder builder = new StringBuilder();
builder.append(trimPath(bizDir));
if (StrUtil.isNotBlank(ownerId)) {
builder.append("/").append(trimPath(ownerId));
}
builder.append("/").append(fileName);
return builder.toString();
}
private String buildStoredFileName(String originalFileName) {
return UUID.randomUUID().toString().replace("-", "") + "-" + sanitizeFileName(originalFileName);
}
private String sanitizeFileName(String originalFileName) {
String fileName = normalizeFileName(originalFileName);
int slashIndex = Math.max(fileName.lastIndexOf('/'), fileName.lastIndexOf('\\'));
String normalizedFileName = slashIndex >= 0 ? fileName.substring(slashIndex + 1) : fileName;
return normalizedFileName.replaceAll("[\\\\/:*?\"<>|]", "_");
}
private String normalizeFileName(String originalFileName) {
String fileName = StrUtil.trimToEmpty(originalFileName);
if (StrUtil.isBlank(fileName)) {
throw new BusinessException(CommonResponseEnum.FAIL, "文件名不能为空");
}
return fileName;
}
private String trimPath(String value) {
String normalized = StrUtil.trimToEmpty(value).replace("\\", "/");
while (normalized.startsWith("/")) {
normalized = normalized.substring(1);
}
while (normalized.endsWith("/")) {
normalized = normalized.substring(0, normalized.length() - 1);
}
if (StrUtil.isBlank(normalized)) {
throw new BusinessException(CommonResponseEnum.FAIL, "对象目录配置不合法");
}
return normalized;
}
private String resolveExtension(String fileName) {
int dotIndex = fileName.lastIndexOf('.');
if (dotIndex < 0 || dotIndex == fileName.length() - 1) {
return "";
}
return fileName.substring(dotIndex + 1).toLowerCase(Locale.ENGLISH);
}
private String resolveContentType(String contentType, String fileName) {
if (StrUtil.isNotBlank(contentType)) {
return contentType;
}
return MediaTypeFactory.getMediaType(fileName).map(MediaType::toString)
.orElse(MediaType.APPLICATION_OCTET_STREAM_VALUE);
}
@Getter
public static class StoredFile {
private final String bucketName;
private final String objectName;
private final String storedFileName;
private final String originalFileName;
private final String contentType;
private final long fileSize;
public StoredFile(String bucketName, String objectName, String storedFileName, String originalFileName,
String contentType, long fileSize) {
this.bucketName = bucketName;
this.objectName = objectName;
this.storedFileName = storedFileName;
this.originalFileName = originalFileName;
this.contentType = contentType;
this.fileSize = fileSize;
}
}
public static class StoredObject implements AutoCloseable {
private final String bucketName;
private final String objectName;
private final GetObjectResponse inputStream;
public StoredObject(String bucketName, String objectName, GetObjectResponse inputStream) {
this.bucketName = bucketName;
this.objectName = objectName;
this.inputStream = inputStream;
}
public String getBucketName() {
return bucketName;
}
public String getObjectName() {
return objectName;
}
public GetObjectResponse getInputStream() {
return inputStream;
}
@Override
public void close() throws Exception {
if (inputStream != null) {
inputStream.close();
}
}
}
}

View File

@@ -16,6 +16,7 @@
<description>AI report capability aggregator.</description>
<modules>
<module>ai-report-common</module>
<module>report-model</module>
<module>client-unit</module>
<module>test-device</module>

View File

@@ -14,6 +14,12 @@
<packaging>jar</packaging>
<dependencies>
<dependency>
<groupId>com.njcn.gather</groupId>
<artifactId>ai-report-common</artifactId>
<version>1.0.0</version>
</dependency>
<dependency>
<groupId>com.njcn.gather</groupId>
<artifactId>comservice</artifactId>

View File

@@ -1,115 +1,84 @@
package com.njcn.gather.aireport.reportmodel.component;
import org.springframework.beans.factory.annotation.Value;
import cn.hutool.core.util.StrUtil;
import com.njcn.common.pojo.enums.response.CommonResponseEnum;
import com.njcn.common.pojo.exception.BusinessException;
import com.njcn.gather.aireport.common.constant.AiReportFileConst;
import com.njcn.gather.aireport.common.constant.AiReportFileExtensionConst;
import com.njcn.gather.aireport.common.storage.AiReportMinioStorageService;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;
import java.io.InputStream;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.nio.file.StandardCopyOption;
import java.time.LocalDate;
import java.time.format.DateTimeFormatter;
import java.util.UUID;
import java.util.Locale;
/**
* Local storage for uploaded report model template files.
* 报告模板文件 MinIO 存储服务。
*/
@Component
@RequiredArgsConstructor
public class ReportModelFileStorageService {
private static final String DEFAULT_STORAGE_DIR = "data/report-model";
private static final String REPORT_MODEL_STORAGE_DIR = "report-model";
private static final DateTimeFormatter DATE_DIR_FORMATTER = DateTimeFormatter.ofPattern("yyyyMMdd");
@Value("${files.path:}")
private String filesPath;
private final AiReportMinioStorageService minioStorageService;
public StoredFile save(MultipartFile templateFile) {
if (templateFile == null || templateFile.isEmpty()) {
throw new IllegalArgumentException("模板文件不能为空");
}
try {
Path storageRoot = resolveStorageRoot();
String dateDir = LocalDate.now().format(DATE_DIR_FORMATTER);
Path storageDir = storageRoot.resolve(dateDir).normalize();
if (!storageDir.startsWith(storageRoot)) {
throw new IllegalArgumentException("模板文件存储路径不合法");
}
Files.createDirectories(storageDir);
String sanitizedFileName = sanitizeFileName(templateFile.getOriginalFilename());
String storedFileName = UUID.randomUUID().toString().replace("-", "") + "-" + sanitizedFileName;
Path target = storageDir.resolve(storedFileName).normalize();
ensurePathInRoot(storageRoot, target);
try (InputStream inputStream = templateFile.getInputStream()) {
Files.copy(inputStream, target, StandardCopyOption.REPLACE_EXISTING);
}
String storedPath = storageRoot.relativize(target).toString().replace('\\', '/');
return new StoredFile(storedPath, sanitizedFileName);
} catch (Exception exception) {
throw new IllegalArgumentException("保存模板文件失败: " + exception.getMessage(), exception);
}
validateTemplateFile(templateFile);
AiReportMinioStorageService.StoredFile storedFile = minioStorageService.saveMultipartFile(
AiReportFileConst.BIZ_DIR_REPORT_MODEL, null, templateFile, false);
return new StoredFile(storedFile.getObjectName(), storedFile.getOriginalFileName());
}
public Path resolveDownloadPath(String storedPath) {
if (trimToNull(storedPath) == null) {
throw new IllegalArgumentException("模板文件路径不能为空");
}
Path storageRoot = resolveStorageRoot();
Path target = storageRoot.resolve(storedPath).normalize();
ensurePathInRoot(storageRoot, target);
return target;
public AiReportMinioStorageService.StoredObject getObject(String storedPath) {
return minioStorageService.getObject(storedPath);
}
public byte[] readBytes(String storedPath) {
return minioStorageService.readBytes(storedPath);
}
public Path downloadToTempFile(String storedPath, String fileName) {
return minioStorageService.downloadToTempFile(storedPath, fileName);
}
public void deleteQuietly(String storedPath) {
try {
if (trimToNull(storedPath) != null) {
Files.deleteIfExists(resolveDownloadPath(storedPath));
}
} catch (Exception ignored) {
// Cleanup failure must not hide the original database error.
}
minioStorageService.deleteQuietly(storedPath);
}
public void deleteTempFileQuietly(Path tempFile) {
minioStorageService.deleteTempFileQuietly(tempFile);
}
public String resolveDownloadFileName(String storedPath) {
Path fileName = Paths.get(storedPath).getFileName();
if (fileName == null) {
return "report-model";
if (StrUtil.isBlank(storedPath)) {
return "report-model.docx";
}
String name = fileName.toString();
int separatorIndex = name.indexOf('-');
return separatorIndex >= 0 && separatorIndex + 1 < name.length() ? name.substring(separatorIndex + 1) : name;
int slashIndex = storedPath.lastIndexOf('/');
String fileName = slashIndex >= 0 ? storedPath.substring(slashIndex + 1) : storedPath;
int separatorIndex = fileName.indexOf('-');
return separatorIndex >= 0 && separatorIndex + 1 < fileName.length() ? fileName.substring(separatorIndex + 1) : fileName;
}
private Path resolveStorageRoot() {
String configuredPath = trimToNull(filesPath);
Path path = configuredPath == null ? Paths.get(DEFAULT_STORAGE_DIR) : Paths.get(configuredPath, REPORT_MODEL_STORAGE_DIR);
return path.toAbsolutePath().normalize();
}
private void ensurePathInRoot(Path storageRoot, Path target) {
if (!target.startsWith(storageRoot)) {
throw new IllegalArgumentException("模板文件路径不合法");
private void validateTemplateFile(MultipartFile templateFile) {
if (templateFile == null || templateFile.isEmpty()) {
throw new IllegalArgumentException("模板文件不能为空");
}
String fileName = templateFile.getOriginalFilename();
if (StrUtil.isBlank(fileName)) {
throw new IllegalArgumentException("模板文件名不能为空");
}
String extension = resolveExtension(fileName);
if (!AiReportFileExtensionConst.WORD_EXTENSIONS.contains(extension)) {
throw new IllegalArgumentException("模板文件仅支持.docx格式");
}
}
private String sanitizeFileName(String originalFilename) {
String fileName = trimToNull(originalFilename);
if (fileName == null) {
return "unnamed";
private String resolveExtension(String fileName) {
int dotIndex = fileName.lastIndexOf('.');
if (dotIndex < 0 || dotIndex == fileName.length() - 1) {
return "";
}
String normalizedName = Paths.get(fileName).getFileName().toString();
String sanitized = normalizedName.replaceAll("[\\\\/:*?\"<>|]", "_");
return trimToNull(sanitized) == null ? "unnamed" : sanitized;
}
private String trimToNull(String value) {
if (value == null) {
return null;
}
String text = value.trim();
return text.isEmpty() ? null : text;
return fileName.substring(dotIndex + 1).toLowerCase(Locale.ENGLISH);
}
public static class StoredFile {

View File

@@ -7,6 +7,7 @@ import com.njcn.common.pojo.enums.common.LogEnum;
import com.njcn.common.pojo.enums.response.CommonResponseEnum;
import com.njcn.common.pojo.exception.BusinessException;
import com.njcn.common.pojo.response.HttpResult;
import com.njcn.gather.aireport.common.storage.AiReportMinioStorageService;
import com.njcn.gather.aireport.reportmodel.component.ReportModelFileStorageService;
import com.njcn.gather.aireport.reportmodel.pojo.param.ReportModelParam;
import com.njcn.gather.aireport.reportmodel.pojo.po.ReportModelPO;
@@ -33,10 +34,8 @@ import org.springframework.web.bind.annotation.RequestPart;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;
import java.io.InputStream;
import java.net.URLEncoder;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.List;
@@ -110,21 +109,14 @@ public class ReportModelController extends BaseController {
@GetMapping("/{id}/download")
public ResponseEntity<InputStreamResource> download(@PathVariable("id") String id) {
ReportModelPO model = reportModelService.requireNormal(id);
Path filePath = reportModelFileStorageService.resolveDownloadPath(model.getPath());
if (!Files.exists(filePath) || !Files.isRegularFile(filePath)) {
throw new BusinessException(CommonResponseEnum.FAIL, "模板文件不存在");
}
try {
String fileName = model.getFileName();
if (fileName == null || fileName.trim().isEmpty()) {
fileName = reportModelFileStorageService.resolveDownloadFileName(model.getPath());
}
String fileName = resolveFileName(model);
String encodedFileName = encodeFileName(fileName);
InputStream inputStream = Files.newInputStream(filePath);
AiReportMinioStorageService.StoredObject storedObject = reportModelFileStorageService.getObject(model.getPath());
return ResponseEntity.ok()
.contentType(MediaType.APPLICATION_OCTET_STREAM)
.header(HttpHeaders.CONTENT_DISPOSITION, "attachment; filename*=UTF-8''" + encodedFileName)
.body(new InputStreamResource(inputStream));
.body(new InputStreamResource(storedObject.getInputStream()));
} catch (Exception exception) {
throw new BusinessException(CommonResponseEnum.FAIL, "读取模板文件失败");
}
@@ -136,21 +128,19 @@ public class ReportModelController extends BaseController {
@GetMapping("/{id}/preview")
public ResponseEntity<byte[]> preview(@PathVariable("id") String id) {
ReportModelPO model = reportModelService.requireNormal(id);
Path filePath = reportModelFileStorageService.resolveDownloadPath(model.getPath());
if (!Files.exists(filePath) || !Files.isRegularFile(filePath)) {
throw new BusinessException(CommonResponseEnum.FAIL, "模板文件不存在");
String fileName = resolveFileName(model);
Path tempFile = reportModelFileStorageService.downloadToTempFile(model.getPath(), fileName);
try {
FilePreviewService.PreviewContent previewContent = filePreviewService.preview(tempFile, fileName);
return ResponseEntity.ok()
.header(HttpHeaders.CONTENT_DISPOSITION,
"inline; filename*=UTF-8''" + encodeFileName(fileName))
.header(HttpHeaders.CACHE_CONTROL, "no-cache, no-store, must-revalidate")
.contentType(MediaType.parseMediaType(previewContent.getContentType()))
.body(previewContent.getContent());
} finally {
reportModelFileStorageService.deleteTempFileQuietly(tempFile);
}
String fileName = model.getFileName();
if (fileName == null || fileName.trim().isEmpty()) {
fileName = reportModelFileStorageService.resolveDownloadFileName(model.getPath());
}
FilePreviewService.PreviewContent previewContent = filePreviewService.preview(filePath, fileName);
return ResponseEntity.ok()
.header(HttpHeaders.CONTENT_DISPOSITION,
"inline; filename*=UTF-8''" + encodeFileName(fileName))
.header(HttpHeaders.CACHE_CONTROL, "no-cache, no-store, must-revalidate")
.contentType(MediaType.parseMediaType(previewContent.getContentType()))
.body(previewContent.getContent());
}
@OperateInfo(info = LogEnum.BUSINESS_COMMON, operateType = OperateType.UPDATE)
@@ -170,4 +160,12 @@ public class ReportModelController extends BaseController {
throw new BusinessException(CommonResponseEnum.FAIL, "编码模板文件名失败");
}
}
private String resolveFileName(ReportModelPO model) {
String fileName = model.getFileName();
if (fileName == null || fileName.trim().isEmpty()) {
return reportModelFileStorageService.resolveDownloadFileName(model.getPath());
}
return fileName;
}
}

View File

@@ -107,6 +107,7 @@ public class ReportModelServiceImpl extends ServiceImpl<ReportModelMapper, Repor
}
String name = resolveName(param.getName());
checkNameUnique(name, model.getId());
String oldStoredPath = model.getPath();
String newStoredPath = null;
if (templateFile != null && !templateFile.isEmpty()) {
ReportModelFileStorageService.StoredFile storedFile = reportModelFileStorageService.save(templateFile);
@@ -118,7 +119,12 @@ public class ReportModelServiceImpl extends ServiceImpl<ReportModelMapper, Repor
model.setName(name);
model.setUpdateBy(resolveCurrentUserId());
model.setUpdateTime(LocalDateTime.now());
return this.updateById(model);
boolean updated = this.updateById(model);
if (updated && StrUtil.isNotBlank(newStoredPath) && StrUtil.isNotBlank(oldStoredPath)
&& !StrUtil.equals(oldStoredPath, newStoredPath)) {
reportModelFileStorageService.deleteQuietly(oldStoredPath);
}
return updated;
} catch (RuntimeException exception) {
reportModelFileStorageService.deleteQuietly(newStoredPath);
throw exception;
@@ -135,7 +141,11 @@ public class ReportModelServiceImpl extends ServiceImpl<ReportModelMapper, Repor
if (validIds.isEmpty()) {
throw new BusinessException(CommonResponseEnum.FAIL, "模板ID不能为空");
}
return this.lambdaUpdate()
List<ReportModelPO> models = this.lambdaQuery()
.in(ReportModelPO::getId, validIds)
.eq(ReportModelPO::getStatus, ReportModelConst.STATUS_NORMAL)
.list();
boolean deleted = this.lambdaUpdate()
.set(ReportModelPO::getStatus, ReportModelConst.STATUS_DELETED)
.set(ReportModelPO::getState, ReportModelConst.STATE_DELETED)
.set(ReportModelPO::getUpdateBy, resolveCurrentUserId())
@@ -143,6 +153,12 @@ public class ReportModelServiceImpl extends ServiceImpl<ReportModelMapper, Repor
.in(ReportModelPO::getId, validIds)
.eq(ReportModelPO::getStatus, ReportModelConst.STATUS_NORMAL)
.update();
if (deleted) {
for (ReportModelPO model : models) {
reportModelFileStorageService.deleteQuietly(model.getPath());
}
}
return deleted;
}
@Override

View File

@@ -1,39 +1,33 @@
package com.njcn.gather.aireport.reportmodel.component;
import com.njcn.gather.aireport.common.storage.AiReportMinioStorageService;
import org.junit.Assert;
import org.junit.Test;
import org.mockito.Mockito;
import org.springframework.mock.web.MockMultipartFile;
import org.springframework.test.util.ReflectionTestUtils;
import java.nio.file.Files;
import java.nio.file.Path;
public class ReportModelFileStorageServiceTest {
@Test
public void saveSanitizesFileNameAndReturnsRelativePath() throws Exception {
Path filesRoot = Files.createTempDirectory("files-root");
ReportModelFileStorageService service = new ReportModelFileStorageService();
ReflectionTestUtils.setField(service, "filesPath", filesRoot.toString());
public void saveShouldDelegateToMinioStorage() {
AiReportMinioStorageService minioStorageService = Mockito.mock(AiReportMinioStorageService.class);
ReportModelFileStorageService service = new ReportModelFileStorageService(minioStorageService);
Mockito.when(minioStorageService.saveMultipartFile(Mockito.anyString(), Mockito.isNull(),
Mockito.any(MockMultipartFile.class), Mockito.eq(false)))
.thenReturn(new AiReportMinioStorageService.StoredFile("cn-tool-ai-report",
"report-model/template-id-a_b.docx", "template-id-a_b.docx", "a_b.docx",
"application/octet-stream", 7L));
ReportModelFileStorageService.StoredFile storedFile = service.save(new MockMultipartFile("templateFile", "../a:b.docx",
"application/octet-stream", "content".getBytes("UTF-8")));
String storedPath = storedFile.getPath();
ReportModelFileStorageService.StoredFile storedFile = service.save(new MockMultipartFile(
"templateFile", "../a:b.docx", "application/octet-stream", "content".getBytes()));
Assert.assertFalse(storedPath.contains(".."));
Assert.assertTrue(storedPath.endsWith("-a_b.docx"));
Assert.assertEquals("report-model/template-id-a_b.docx", storedFile.getPath());
Assert.assertEquals("a_b.docx", storedFile.getFileName());
Assert.assertTrue(service.resolveDownloadPath(storedPath)
.startsWith(filesRoot.resolve("report-model").toAbsolutePath().normalize()));
Assert.assertTrue(Files.exists(service.resolveDownloadPath(storedPath)));
}
@Test(expected = IllegalArgumentException.class)
public void resolveDownloadPathRejectsPathOutsideStorageRoot() throws Exception {
Path filesRoot = Files.createTempDirectory("files-root");
ReportModelFileStorageService service = new ReportModelFileStorageService();
ReflectionTestUtils.setField(service, "filesPath", filesRoot.toString());
service.resolveDownloadPath("../outside.docx");
public void saveShouldRejectNonDocxFile() {
ReportModelFileStorageService service = new ReportModelFileStorageService(Mockito.mock(AiReportMinioStorageService.class));
service.save(new MockMultipartFile("templateFile", "template.pdf", "application/pdf", new byte[]{1}));
}
}

View File

@@ -14,6 +14,12 @@
<packaging>jar</packaging>
<dependencies>
<dependency>
<groupId>com.njcn.gather</groupId>
<artifactId>ai-report-common</artifactId>
<version>1.0.0</version>
</dependency>
<dependency>
<groupId>com.njcn</groupId>
<artifactId>njcn-common</artifactId>

View File

@@ -3,98 +3,65 @@ package com.njcn.gather.aireport.testdevice.component;
import cn.hutool.core.util.StrUtil;
import com.njcn.common.pojo.enums.response.CommonResponseEnum;
import com.njcn.common.pojo.exception.BusinessException;
import org.springframework.beans.factory.annotation.Value;
import com.njcn.gather.aireport.common.constant.AiReportFileConst;
import com.njcn.gather.aireport.common.constant.AiReportFileExtensionConst;
import com.njcn.gather.aireport.common.storage.AiReportMinioStorageService;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.nio.file.StandardCopyOption;
import java.time.LocalDate;
import java.time.format.DateTimeFormatter;
import java.util.UUID;
import java.util.Locale;
@Component
@RequiredArgsConstructor
public class TestDeviceReportStorageService {
private static final String DEFAULT_STORAGE_DIR = "data/test-device-report";
private static final String TEST_DEVICE_REPORT_STORAGE_DIR = "test-device-report";
private static final DateTimeFormatter DATE_DIR_FORMATTER = DateTimeFormatter.ofPattern("yyyyMMdd");
private static final long MAX_REPORT_SIZE = 10L * 1024L * 1024L;
private static final int MAX_ORIGINAL_NAME_LENGTH = 80;
@Value("${files.path:}")
private String filesPath;
private final AiReportMinioStorageService minioStorageService;
public String save(MultipartFile reportFile) {
if (reportFile == null || reportFile.isEmpty()) {
throw new BusinessException(CommonResponseEnum.FAIL, "校准报告不能为空");
}
validatePdf(reportFile.getOriginalFilename(), reportFile.getSize());
try (InputStream inputStream = reportFile.getInputStream()) {
return savePdf(reportFile.getOriginalFilename(), inputStream, reportFile.getSize());
} catch (IOException exception) {
throw new BusinessException(CommonResponseEnum.FAIL, "保存校准报告失败");
}
return minioStorageService.saveMultipartFile(
AiReportFileConst.BIZ_DIR_TEST_DEVICE, null, reportFile, false).getObjectName();
}
public String savePdf(String originalFilename, InputStream inputStream, long size) {
validatePdf(originalFilename, size);
try {
Path storageRoot = resolveStorageRoot();
String dateDir = LocalDate.now().format(DATE_DIR_FORMATTER);
Path storageDir = storageRoot.resolve(dateDir).normalize();
ensurePathInRoot(storageRoot, storageDir);
Files.createDirectories(storageDir);
String fileName = UUID.randomUUID().toString().replace("-", "") + "-" + sanitizeFileName(originalFilename);
Path target = storageDir.resolve(fileName).normalize();
ensurePathInRoot(storageRoot, target);
copyWithSizeLimit(inputStream, target);
return storageRoot.relativize(target).toString().replace('\\', '/');
} catch (BusinessException exception) {
throw exception;
} catch (Exception exception) {
throw new BusinessException(CommonResponseEnum.FAIL, "保存校准报告失败");
}
return minioStorageService.saveInputStream(AiReportFileConst.BIZ_DIR_TEST_DEVICE, null,
trimFileName(originalFilename), inputStream, size, "application/pdf", false).getObjectName();
}
public Path resolveDownloadPath(String storedPath) {
if (StrUtil.isBlank(storedPath)) {
throw new BusinessException(CommonResponseEnum.FAIL, "校准报告路径不能为空");
}
Path storageRoot = resolveStorageRoot();
Path target = storageRoot.resolve(storedPath).normalize();
ensurePathInRoot(storageRoot, target);
return target;
public AiReportMinioStorageService.StoredObject getObject(String storedPath) {
return minioStorageService.getObject(storedPath);
}
public String resolveDownloadFileName(String storedPath) {
Path fileName = Paths.get(storedPath).getFileName();
if (fileName == null) {
if (StrUtil.isBlank(storedPath)) {
return "校准报告.pdf";
}
String name = fileName.toString();
int separatorIndex = name.indexOf('-');
return separatorIndex >= 0 && separatorIndex + 1 < name.length() ? name.substring(separatorIndex + 1) : name;
int slashIndex = storedPath.lastIndexOf('/');
String fileName = slashIndex >= 0 ? storedPath.substring(slashIndex + 1) : storedPath;
int separatorIndex = fileName.indexOf('-');
return separatorIndex >= 0 && separatorIndex + 1 < fileName.length() ? fileName.substring(separatorIndex + 1) : fileName;
}
public void deleteQuietly(String storedPath) {
try {
if (StrUtil.isNotBlank(storedPath)) {
Files.deleteIfExists(resolveDownloadPath(storedPath));
}
} catch (Exception ignored) {
// 文件清理失败不能覆盖原始业务结果。
}
minioStorageService.deleteQuietly(storedPath);
}
private void validatePdf(String originalFilename, long size) {
if (StrUtil.isBlank(originalFilename) || !originalFilename.toLowerCase().endsWith(".pdf")) {
String fileName = trimFileName(originalFilename);
if (StrUtil.isBlank(fileName)) {
throw new BusinessException(CommonResponseEnum.FAIL, "校准报告文件名不能为空");
}
String extension = resolveExtension(fileName);
if (!AiReportFileExtensionConst.PDF_EXTENSIONS.contains(extension)) {
throw new BusinessException(CommonResponseEnum.FAIL, "校准报告仅支持PDF文件");
}
if (size > MAX_REPORT_SIZE) {
@@ -102,48 +69,27 @@ public class TestDeviceReportStorageService {
}
}
private void copyWithSizeLimit(InputStream inputStream, Path target) throws IOException {
Path tempFile = Files.createTempFile(target.getParent(), "upload-", ".tmp");
long bytes = 0L;
byte[] buffer = new byte[8192];
try {
int length;
try (InputStream source = inputStream;
OutputStream targetStream = Files.newOutputStream(tempFile)) {
while ((length = source.read(buffer)) != -1) {
bytes += length;
if (bytes > MAX_REPORT_SIZE) {
throw new BusinessException(CommonResponseEnum.FAIL, "校准报告不能超过10MB");
}
targetStream.write(buffer, 0, length);
}
}
Files.move(tempFile, target, StandardCopyOption.REPLACE_EXISTING);
} finally {
Files.deleteIfExists(tempFile);
private String trimFileName(String originalFilename) {
String fileName = originalFilename == null ? null : originalFilename.trim();
if (StrUtil.isBlank(fileName)) {
return fileName;
}
}
private Path resolveStorageRoot() {
String configuredPath = StrUtil.isBlank(filesPath) ? null : filesPath.trim();
Path path = configuredPath == null ? Paths.get(DEFAULT_STORAGE_DIR) : Paths.get(configuredPath, TEST_DEVICE_REPORT_STORAGE_DIR);
return path.toAbsolutePath().normalize();
}
private void ensurePathInRoot(Path storageRoot, Path target) {
if (!target.startsWith(storageRoot)) {
throw new BusinessException(CommonResponseEnum.FAIL, "校准报告路径不合法");
}
}
private String sanitizeFileName(String originalFilename) {
String fileName = Paths.get(originalFilename).getFileName().toString();
String sanitized = fileName.replaceAll("[\\\\/:*?\"<>|]", "_");
int slashIndex = Math.max(fileName.lastIndexOf('/'), fileName.lastIndexOf('\\'));
String normalizedName = slashIndex >= 0 ? fileName.substring(slashIndex + 1) : fileName;
String sanitized = normalizedName.replaceAll("[\\\\/:*?\"<>|]", "_");
if (sanitized.length() > MAX_ORIGINAL_NAME_LENGTH) {
String suffix = sanitized.toLowerCase().endsWith(".pdf") ? ".pdf" : "";
String suffix = sanitized.toLowerCase(Locale.ENGLISH).endsWith(".pdf") ? ".pdf" : "";
int keepLength = MAX_ORIGINAL_NAME_LENGTH - suffix.length();
sanitized = sanitized.substring(0, keepLength) + suffix;
}
return StrUtil.isBlank(sanitized) ? "report.pdf" : sanitized;
return sanitized;
}
private String resolveExtension(String fileName) {
int dotIndex = fileName.lastIndexOf('.');
if (dotIndex < 0 || dotIndex == fileName.length() - 1) {
return "";
}
return fileName.substring(dotIndex + 1).toLowerCase(Locale.ENGLISH);
}
}

View File

@@ -7,6 +7,7 @@ import com.njcn.common.pojo.enums.common.LogEnum;
import com.njcn.common.pojo.enums.response.CommonResponseEnum;
import com.njcn.common.pojo.exception.BusinessException;
import com.njcn.common.pojo.response.HttpResult;
import com.njcn.gather.aireport.common.storage.AiReportMinioStorageService;
import com.njcn.gather.aireport.testdevice.component.TestDeviceReportStorageService;
import com.njcn.gather.aireport.testdevice.pojo.param.TestDeviceParam;
import com.njcn.gather.aireport.testdevice.pojo.po.TestDevicePO;
@@ -34,10 +35,7 @@ import org.springframework.web.bind.annotation.RequestPart;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;
import java.io.InputStream;
import java.net.URLEncoder;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.List;
@Api(tags = "检测设备管理")
@@ -145,19 +143,15 @@ public class TestDeviceController extends BaseController {
if (device == null || device.getValidityReport() == null || device.getValidityReport().trim().isEmpty()) {
throw new BusinessException(CommonResponseEnum.FAIL, "校准报告文件不存在");
}
Path filePath = reportStorageService.resolveDownloadPath(device.getValidityReport());
if (!Files.exists(filePath) || !Files.isRegularFile(filePath)) {
throw new BusinessException(CommonResponseEnum.FAIL, "校准报告文件不存在");
}
try {
String fileName = reportStorageService.resolveDownloadFileName(device.getValidityReport());
String encodedFileName = URLEncoder.encode(fileName, "UTF-8").replace("+", "%20");
InputStream inputStream = Files.newInputStream(filePath);
AiReportMinioStorageService.StoredObject storedObject = reportStorageService.getObject(device.getValidityReport());
return ResponseEntity.ok()
.contentType(preview ? MediaType.APPLICATION_PDF : MediaType.APPLICATION_OCTET_STREAM)
.header(HttpHeaders.CONTENT_DISPOSITION,
(preview ? "inline" : "attachment") + "; filename*=UTF-8''" + encodedFileName)
.body(new InputStreamResource(inputStream));
.body(new InputStreamResource(storedObject.getInputStream()));
} catch (Exception exception) {
throw new BusinessException(CommonResponseEnum.FAIL, "读取校准报告失败");
}

View File

@@ -163,13 +163,23 @@ public class TestDeviceServiceImpl extends ServiceImpl<TestDeviceMapper, TestDev
if (validIds.isEmpty()) {
throw new BusinessException(CommonResponseEnum.FAIL, "检测设备ID不能为空");
}
return this.lambdaUpdate()
List<TestDevicePO> devices = this.lambdaQuery()
.in(TestDevicePO::getId, validIds)
.eq(TestDevicePO::getStatus, TestDeviceConst.STATUS_NORMAL)
.list();
boolean deleted = this.lambdaUpdate()
.set(TestDevicePO::getStatus, TestDeviceConst.STATUS_DELETED)
.set(TestDevicePO::getUpdateBy, resolveCurrentUserId())
.set(TestDevicePO::getUpdateTime, LocalDateTime.now())
.in(TestDevicePO::getId, validIds)
.eq(TestDevicePO::getStatus, TestDeviceConst.STATUS_NORMAL)
.update();
if (deleted) {
for (TestDevicePO device : devices) {
reportStorageService.deleteQuietly(device.getValidityReport());
}
}
return deleted;
}
@Override

View File

@@ -1,28 +1,34 @@
package com.njcn.gather.aireport.testdevice.component;
import com.njcn.common.pojo.exception.BusinessException;
import com.njcn.gather.aireport.common.storage.AiReportMinioStorageService;
import org.junit.Assert;
import org.junit.Test;
import org.mockito.Mockito;
import org.springframework.mock.web.MockMultipartFile;
import org.springframework.test.util.ReflectionTestUtils;
import java.nio.file.Files;
import java.nio.file.Path;
public class TestDeviceReportStorageServiceTest {
@Test
public void saveUsesTestDeviceReportDirectoryUnderFilesRoot() throws Exception {
Path filesRoot = Files.createTempDirectory("files-root");
TestDeviceReportStorageService service = new TestDeviceReportStorageService();
ReflectionTestUtils.setField(service, "filesPath", filesRoot.toString());
public void saveShouldDelegateToMinioStorage() {
AiReportMinioStorageService minioStorageService = Mockito.mock(AiReportMinioStorageService.class);
TestDeviceReportStorageService service = new TestDeviceReportStorageService(minioStorageService);
Mockito.when(minioStorageService.saveMultipartFile(Mockito.anyString(), Mockito.isNull(),
Mockito.any(MockMultipartFile.class), Mockito.eq(false)))
.thenReturn(new AiReportMinioStorageService.StoredFile("cn-tool-ai-report",
"test-device/report-id-report.pdf", "report-id-report.pdf", "report.pdf",
"application/pdf", 8L));
String storedPath = service.save(new MockMultipartFile("reportFile", "../report.pdf",
"application/pdf", "%PDF-1.4".getBytes("UTF-8")));
"application/pdf", "%PDF-1.4".getBytes()));
Path expectedRoot = filesRoot.resolve("test-device-report").toAbsolutePath().normalize();
Path actualPath = service.resolveDownloadPath(storedPath);
Assert.assertTrue(actualPath.startsWith(expectedRoot));
Assert.assertTrue(storedPath.endsWith("-report.pdf"));
Assert.assertTrue(Files.exists(actualPath));
Assert.assertEquals("test-device/report-id-report.pdf", storedPath);
}
@Test(expected = BusinessException.class)
public void saveShouldRejectNonPdfFile() {
TestDeviceReportStorageService service = new TestDeviceReportStorageService(Mockito.mock(AiReportMinioStorageService.class));
service.save(new MockMultipartFile("reportFile", "report.docx",
"application/vnd.openxmlformats-officedocument.wordprocessingml.document", new byte[]{1}));
}
}

View File

@@ -1,8 +1,10 @@
package com.njcn.gather.aireport.testdevice.controller;
import com.njcn.gather.aireport.common.storage.AiReportMinioStorageService;
import com.njcn.gather.aireport.testdevice.component.TestDeviceReportStorageService;
import com.njcn.gather.aireport.testdevice.pojo.po.TestDevicePO;
import com.njcn.gather.aireport.testdevice.service.TestDeviceService;
import io.minio.GetObjectResponse;
import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.Test;
import org.mockito.Mockito;
@@ -12,10 +14,6 @@ import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.test.util.ReflectionTestUtils;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.nio.file.Path;
class TestDeviceControllerTest {
private final TestDeviceService testDeviceService = Mockito.mock(TestDeviceService.class);
@@ -27,11 +25,11 @@ class TestDeviceControllerTest {
TestDevicePO device = new TestDevicePO();
device.setId("device-1");
device.setValidityReport("20260626/report.pdf");
Path tempFile = Files.createTempFile("test-device-preview-", ".pdf");
Files.write(tempFile, "%PDF-1.4".getBytes(StandardCharsets.UTF_8));
AiReportMinioStorageService.StoredObject storedObject = Mockito.mock(AiReportMinioStorageService.StoredObject.class);
Mockito.when(testDeviceService.requireNormal("device-1")).thenReturn(device);
Mockito.when(reportStorageService.resolveDownloadPath("20260626/report.pdf")).thenReturn(tempFile);
Mockito.when(reportStorageService.resolveDownloadFileName("20260626/report.pdf")).thenReturn("report.pdf");
Mockito.when(reportStorageService.getObject("20260626/report.pdf")).thenReturn(storedObject);
Mockito.when(storedObject.getInputStream()).thenReturn(Mockito.mock(GetObjectResponse.class));
@SuppressWarnings("unchecked")
ResponseEntity<InputStreamResource> response = (ResponseEntity<InputStreamResource>)

View File

@@ -14,6 +14,12 @@
<packaging>jar</packaging>
<dependencies>
<dependency>
<groupId>com.njcn.gather</groupId>
<artifactId>ai-report-common</artifactId>
<version>1.0.0</version>
</dependency>
<dependency>
<groupId>com.njcn.gather</groupId>
<artifactId>comservice</artifactId>

View File

@@ -34,8 +34,9 @@ public class TestReportLedgerExcelParser {
private static final DataFormatter FORMATTER = new DataFormatter();
private static final DateTimeFormatter DATE_FORMATTER = DateTimeFormatter.ofPattern("yyyy-MM-dd");
private static final List<String> REQUIRED_SHEETS = Arrays.asList(
TestReportConst.LEDGER_POINT_SHEET_NAME);
private static final List<String> OPTIONAL_SHEETS = Arrays.asList(
TestReportConst.LEDGER_GUIDE_SHEET_NAME,
TestReportConst.LEDGER_POINT_SHEET_NAME,
TestReportConst.LEDGER_DEVICE_SHEET_NAME,
TestReportConst.LEDGER_CLIENT_SHEET_NAME);
private static final List<String> POINT_HEADERS = Arrays.asList(
@@ -77,6 +78,7 @@ public class TestReportLedgerExcelParser {
if (pointSheet != null) {
ledger.setPoints(parsePointSheet(pointSheet));
}
// 检测设备/委托单位 sheet 按需填写,仅在实际存在时才做结构校验和解析。
Sheet deviceSheet = workbook.getSheet(TestReportConst.LEDGER_DEVICE_SHEET_NAME);
if (deviceSheet != null) {
ledger.setDevices(parseDeviceSheet(deviceSheet));
@@ -104,6 +106,11 @@ public class TestReportLedgerExcelParser {
}
ledger.getExistingSheets().add(sheetName);
}
for (String sheetName : OPTIONAL_SHEETS) {
if (workbook.getSheet(sheetName) != null) {
ledger.getExistingSheets().add(sheetName);
}
}
}
private void ensureRequiredSheetsPresent(ParsedLedger ledger) {
@@ -166,11 +173,14 @@ public class TestReportLedgerExcelParser {
}
ParsedDeviceRow device = new ParsedDeviceRow();
device.setRowNo(rowIndex + 1);
device.setTypeName(requireCell(row, 0, rowIndex, "设备型号"));
device.setNo(requireCell(row, 1, rowIndex, "设备编号"));
device.setValidityPeriod(parseDate(requireCell(row, 2, rowIndex, "标准有效期"), rowIndex));
device.setStateName(requireCell(row, 3, rowIndex, "设备状态"));
device.setValidityReportName(requireCell(row, 4, rowIndex, "标准报告文件名"));
device.setTypeName(optionalCell(row, 0));
device.setNo(optionalCell(row, 1));
device.setValidityPeriod(parseOptionalDate(optionalCell(row, 2), rowIndex));
device.setStateName(optionalCell(row, 3));
device.setValidityReportName(optionalCell(row, 4));
if (StrUtil.isBlank(device.getNo())) {
continue;
}
if (!deviceNos.add(device.getNo())) {
throw new BusinessException(CommonResponseEnum.FAIL, "校验文件阶段检测设备sheet中存在重复设备编号");
}
@@ -190,10 +200,13 @@ public class TestReportLedgerExcelParser {
}
ParsedClientRow client = new ParsedClientRow();
client.setRowNo(rowIndex + 1);
client.setName(requireCell(row, 0, rowIndex, "单位名称"));
client.setName(optionalCell(row, 0));
client.setContactName(optionalCell(row, 1));
client.setPhonenumber(optionalCell(row, 2));
client.setAddress(optionalCell(row, 3));
if (StrUtil.isBlank(client.getName())) {
continue;
}
if (!clientNames.add(client.getName())) {
throw new BusinessException(CommonResponseEnum.FAIL, "校验文件阶段委托单位sheet中存在重复单位名称");
}
@@ -302,6 +315,13 @@ public class TestReportLedgerExcelParser {
}
}
private LocalDate parseOptionalDate(String value, int rowIndex) {
if (StrUtil.isBlank(value)) {
return null;
}
return parseDate(value, rowIndex);
}
@Data
public static class ParsedLedger {
private String summaryFileName;

View File

@@ -10,6 +10,7 @@ import com.njcn.common.pojo.exception.BusinessException;
import com.njcn.gather.aireport.clientunit.mapper.ClientUnitMapper;
import com.njcn.gather.aireport.clientunit.pojo.constant.ClientUnitConst;
import com.njcn.gather.aireport.clientunit.pojo.po.ClientUnitPO;
import com.njcn.gather.aireport.common.storage.AiReportMinioStorageService;
import com.njcn.gather.aireport.testdevice.mapper.TestDeviceMapper;
import com.njcn.gather.aireport.testdevice.pojo.constant.TestDeviceConst;
import com.njcn.gather.aireport.testdevice.pojo.po.TestDevicePO;
@@ -30,7 +31,6 @@ import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;
import java.io.InputStream;
import java.nio.file.Files;
import java.time.LocalDateTime;
import java.util.Collection;
import java.util.Collections;
@@ -91,8 +91,9 @@ public class TestReportLedgerImportService {
persistValidateSessionResult(sessionId, result);
return result;
}
try (InputStream inputStream = Files.newInputStream(
testReportStorageService.resolveValidateSessionFile(sessionId, TestReportConst.LEDGER_SUMMARY_FILE_NAME))) {
try (AiReportMinioStorageService.StoredObject storedObject =
testReportStorageService.getValidateSessionObject(sessionId, TestReportConst.LEDGER_SUMMARY_FILE_NAME);
InputStream inputStream = storedObject.getInputStream()) {
TestReportLedgerExcelParser.ParsedLedger ledger = testReportLedgerExcelParser.parseForValidate(
TestReportConst.LEDGER_SUMMARY_FILE_NAME, inputStream);
Map<String, TestDevicePO> dbDeviceMap = loadExistingDevices(collectDeviceNos(ledger));
@@ -136,6 +137,7 @@ public class TestReportLedgerImportService {
writeValidatedBatchMeta(report.getId(), batchId, uploadedBatch, ledger, userId);
return buildValidateSuccessResult(batchId, uploadedBatch.getTempStoragePath(), ledger, fileMap.size());
} catch (RuntimeException exception) {
testReportStorageService.deleteTempBatchQuietly(report.getId(), batchId);
throw exception;
}
}
@@ -168,11 +170,12 @@ public class TestReportLedgerImportService {
}
private TestReportLedgerExcelParser.ParsedLedger parseLedgerFromBatch(String reportId, String batchId, Collection<String> fileNames) {
try (InputStream inputStream = java.nio.file.Files.newInputStream(
testReportStorageService.resolveTempBatchFile(reportId, batchId, TestReportConst.LEDGER_SUMMARY_FILE_NAME))) {
try (AiReportMinioStorageService.StoredObject storedObject =
testReportStorageService.getTempBatchObject(reportId, batchId, TestReportConst.LEDGER_SUMMARY_FILE_NAME);
InputStream inputStream = storedObject.getInputStream()) {
return testReportLedgerExcelParser.parse(TestReportConst.LEDGER_SUMMARY_FILE_NAME, inputStream,
buildAttachmentNameSet(fileNames));
} catch (java.io.IOException exception) {
} catch (Exception exception) {
throw new BusinessException(CommonResponseEnum.FAIL, "校验文件阶段读取台账汇总excel失败");
}
}
@@ -442,8 +445,9 @@ public class TestReportLedgerImportService {
summary.setDeviceReuseCount(summary.getDeviceReuseCount() + 1);
continue;
}
String typeId = typeNameIdMap.get(deviceRow.getTypeName());
if (StrUtil.isBlank(typeId)) {
String typeName = normalizeFileName(deviceRow.getTypeName());
String typeId = StrUtil.isBlank(typeName) ? null : typeNameIdMap.get(typeName);
if (StrUtil.isNotBlank(typeName) && StrUtil.isBlank(typeId)) {
throw new BusinessException(CommonResponseEnum.FAIL,
"基础信息维护阶段检测设备sheet第" + deviceRow.getRowNo() + "行设备型号不存在");
}
@@ -500,14 +504,9 @@ public class TestReportLedgerImportService {
point.setGroupNo(pointRow.getGroupNo());
point.setSubstationName(pointRow.getSubstationName());
point.setPointName(pointRow.getPointName());
point.setTestDeviceNo(pointRow.getTestDeviceNo());
point.setClientUnitName(pointRow.getClientUnitName());
point.setExcelAttachmentName(pointRow.getExcelAttachmentName());
point.setExcelStoragePath(testReportStorageService.savePointExcelFromTemp(reportId, batchId,
pointRow.getExcelAttachmentName()));
point.setWordAttachmentName(pointRow.getWordAttachmentName());
point.setWordStoragePath(testReportStorageService.savePointWordFromTemp(reportId, batchId,
pointRow.getWordAttachmentName()));
point.setSortNo(sortNo++);
point.setDeleted(TestReportConst.DELETED_NO);
point.setCreateBy(userId);
@@ -525,7 +524,6 @@ public class TestReportLedgerImportService {
private void deleteOldFiles(List<TestReportPointPO> oldPoints, List<TestReportGroupReportPO> oldGroupReports, String data) {
for (TestReportPointPO point : oldPoints) {
testReportStorageService.deleteQuietly(point.getExcelStoragePath());
testReportStorageService.deleteQuietly(point.getWordStoragePath());
}
for (TestReportGroupReportPO groupReport : oldGroupReports) {
testReportStorageService.deleteQuietly(groupReport.getReportStoragePath());
@@ -553,8 +551,8 @@ public class TestReportLedgerImportService {
ObjectNode sheetSummary = root.putObject("sheetSummary");
sheetSummary.put("pointSheet", TestReportConst.LEDGER_POINT_SHEET_NAME);
sheetSummary.put("deviceSheetPresent", !ledger.getDevices().isEmpty());
sheetSummary.put("clientSheetPresent", !ledger.getClients().isEmpty());
sheetSummary.put("deviceSheetPresent", ledger.getExistingSheets().contains(TestReportConst.LEDGER_DEVICE_SHEET_NAME));
sheetSummary.put("clientSheetPresent", ledger.getExistingSheets().contains(TestReportConst.LEDGER_CLIENT_SHEET_NAME));
root.put("rowCount", ledger.getPoints().size());
root.put("pointCount", ledger.getPoints().size());
@@ -727,6 +725,9 @@ public class TestReportLedgerImportService {
private String resolveDeviceState(String stateName, Integer rowNo) {
String normalizedState = normalizeFileName(stateName);
if (normalizedState == null) {
return null;
}
if ("01".equals(normalizedState) || "在运".equals(normalizedState)) {
return TestDeviceConst.STATE_RUNNING;
}

View File

@@ -3,326 +3,304 @@ package com.njcn.gather.aireport.testreport.component;
import cn.hutool.core.util.StrUtil;
import com.njcn.common.pojo.enums.response.CommonResponseEnum;
import com.njcn.common.pojo.exception.BusinessException;
import com.njcn.gather.aireport.common.constant.AiReportFileConst;
import com.njcn.gather.aireport.common.constant.AiReportFileExtensionConst;
import com.njcn.gather.aireport.common.storage.AiReportMinioStorageService;
import com.njcn.gather.aireport.testreport.pojo.constant.TestReportConst;
import org.springframework.beans.factory.annotation.Value;
import lombok.RequiredArgsConstructor;
import org.springframework.http.MediaType;
import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;
import java.io.IOException;
import java.io.InputStream;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.nio.file.StandardCopyOption;
import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Locale;
import java.util.Map;
import java.util.UUID;
import java.util.Set;
@Component
@RequiredArgsConstructor
public class TestReportStorageService {
@Value("${files.path}")
private String filesRootPath;
private final AiReportMinioStorageService minioStorageService;
public String saveSourceExcel(String reportId, MultipartFile file) {
return saveMultipartFile(reportId, TestReportConst.STORAGE_SOURCE_EXCEL_DIR, file, "淇濆瓨鍘熷Excel澶辫触");
validateMultipartFile(file, AiReportFileExtensionConst.EXCEL_EXTENSIONS, "原始Excel文件不能为空", "原始Excel仅允许xlsx格式");
return minioStorageService.saveMultipartFile(buildBusinessDir(reportId, TestReportConst.STORAGE_SOURCE_EXCEL_DIR),
null, file, false).getObjectName();
}
public String saveLedgerSummaryExcel(String reportId, MultipartFile file) {
return saveMultipartFile(reportId, TestReportConst.STORAGE_LEDGER_SUMMARY_DIR, file, "淇濆瓨鍙拌处姹囨€籈xcel澶辫触");
validateMultipartFile(file, AiReportFileExtensionConst.EXCEL_EXTENSIONS, "台账汇总文件不能为空", "台账汇总仅允许xlsx格式");
return minioStorageService.saveMultipartFile(buildBusinessDir(reportId, TestReportConst.STORAGE_LEDGER_SUMMARY_DIR),
null, file, false).getObjectName();
}
public String savePointExcelAttachment(String reportId, MultipartFile file) {
return saveMultipartFile(reportId, TestReportConst.STORAGE_POINT_EXCEL_DIR, file, "淇濆瓨鐩戞祴鐐笶xcel闄勪欢澶辫触");
validateMultipartFile(file, AiReportFileExtensionConst.EXCEL_EXTENSIONS, "监测点Excel附件不能为空", "监测点Excel附件仅允许xlsx格式");
return minioStorageService.saveMultipartFile(buildBusinessDir(reportId, TestReportConst.STORAGE_POINT_EXCEL_DIR),
null, file, false).getObjectName();
}
public String savePointWordAttachment(String reportId, MultipartFile file) {
return saveMultipartFile(reportId, TestReportConst.STORAGE_POINT_WORD_DIR, file, "淇濆瓨鐩戞祴鐐筗ord闄勪欢澶辫触");
validateMultipartFile(file, AiReportFileExtensionConst.WORD_EXTENSIONS, "监测点Word附件不能为空", "监测点Word附件仅允许docx格式");
return minioStorageService.saveMultipartFile(buildBusinessDir(reportId, TestReportConst.STORAGE_POINT_WORD_DIR),
null, file, false).getObjectName();
}
public UploadedBatch saveTempBatchFiles(String reportId, String batchId, Map<String, MultipartFile> fileMap) {
if (fileMap == null || fileMap.isEmpty()) {
throw new BusinessException(CommonResponseEnum.FAIL, "文件不能为空");
}
Map<String, String> storedFileMap = new LinkedHashMap<String, String>();
for (Map.Entry<String, MultipartFile> entry : fileMap.entrySet()) {
String originalFileName = entry.getKey();
MultipartFile file = entry.getValue();
if (file == null || file.isEmpty() || StrUtil.isBlank(originalFileName)) {
throw new BusinessException(CommonResponseEnum.FAIL, "文件不能为空");
}
String safeFileName = Paths.get(originalFileName).getFileName().toString();
Path targetPath = resolveTempUploadDir(reportId, batchId).resolve(safeFileName);
try {
Files.createDirectories(targetPath.getParent());
file.transferTo(targetPath.toFile());
} catch (IOException exception) {
throw new BusinessException(CommonResponseEnum.FAIL, "文件上传阶段:保存临时文件失败");
}
storedFileMap.put(originalFileName, buildTempUploadRelativePath(reportId, batchId, safeFileName));
}
UploadedBatch uploadedBatch = new UploadedBatch();
uploadedBatch.setBatchId(batchId);
uploadedBatch.setTempStoragePath(buildTempBatchRelativePath(reportId, batchId));
uploadedBatch.setStoredFileMap(storedFileMap);
return uploadedBatch;
return saveBatchFiles(buildTempUploadDir(reportId, batchId), buildTempBatchRelativePath(reportId, batchId),
batchId, fileMap, "文件上传阶段:保存临时文件失败");
}
public UploadedBatch saveValidateSessionFiles(String sessionId, Map<String, MultipartFile> fileMap) {
return saveBatchFiles(buildValidateSessionUploadDir(sessionId), buildValidateSessionRelativePath(sessionId),
sessionId, fileMap, "校验文件阶段:保存校验会话文件失败");
}
public AiReportMinioStorageService.StoredObject getTempBatchObject(String reportId, String batchId, String fileName) {
return minioStorageService.getObject(buildTempBatchFileObjectName(reportId, batchId, fileName));
}
public AiReportMinioStorageService.StoredObject getValidateSessionObject(String sessionId, String fileName) {
return minioStorageService.getObject(buildValidateSessionFileObjectName(sessionId, fileName));
}
public void writeTempBatchMeta(String reportId, String batchId, String content) {
writeText(buildTempBatchMetaObjectName(reportId, batchId), content, "校验文件阶段:保存批次元数据失败");
}
public void writeValidateSessionResult(String sessionId, String content) {
writeText(buildValidateSessionResultObjectName(sessionId), content, "校验文件阶段:保存校验结果失败");
}
public List<String> listValidateSessionFileNames(String sessionId) {
return minioStorageService.listFileNames(buildValidateSessionUploadDir(sessionId));
}
public String readTempBatchMeta(String reportId, String batchId) {
String objectName = buildTempBatchMetaObjectName(reportId, batchId);
if (!minioStorageService.exists(objectName)) {
return null;
}
return new String(minioStorageService.readBytes(objectName), StandardCharsets.UTF_8);
}
public String saveLedgerSummaryFromTemp(String reportId, String batchId, String fileName) {
return copyTempBatchFile(reportId, batchId, fileName,
buildBusinessDir(reportId, TestReportConst.STORAGE_LEDGER_SUMMARY_DIR),
"application/vnd.openxmlformats-officedocument.spreadsheetml.sheet", "保存台账汇总excel失败");
}
public String savePointExcelFromTemp(String reportId, String batchId, String fileName) {
return copyTempBatchFile(reportId, batchId, fileName,
buildBusinessDir(reportId, TestReportConst.STORAGE_POINT_EXCEL_DIR),
"application/vnd.openxmlformats-officedocument.spreadsheetml.sheet", "保存监测点excel附件失败");
}
public String savePointWordFromTemp(String reportId, String batchId, String fileName) {
return copyTempBatchFile(reportId, batchId, fileName,
buildBusinessDir(reportId, TestReportConst.STORAGE_POINT_WORD_DIR),
"application/vnd.openxmlformats-officedocument.wordprocessingml.document", "保存监测点word附件失败");
}
public String saveGeneratedPlaceholder(String reportId, String fileName, byte[] content) {
validateFileName(fileName, AiReportFileExtensionConst.WORD_EXTENSIONS, "生成报告文件名不能为空", "生成报告仅允许docx格式");
return minioStorageService.saveBytes(buildBusinessDir(reportId, TestReportConst.STORAGE_GENERATED_REPORT_DIR),
null, fileName, content, MediaType.APPLICATION_OCTET_STREAM_VALUE, false).getObjectName();
}
public byte[] readBusinessBytes(String storedPath) {
return minioStorageService.readBytes(storedPath);
}
public void deleteQuietly(String storedPath) {
minioStorageService.deleteQuietly(storedPath);
}
public void deleteTempBatchQuietly(String reportId, String batchId) {
deletePrefixQuietly(buildTempBatchDir(reportId, batchId));
}
private UploadedBatch saveBatchFiles(String bizDir, String tempStoragePath, String batchId,
Map<String, MultipartFile> fileMap, String errorMessage) {
if (fileMap == null || fileMap.isEmpty()) {
throw new BusinessException(CommonResponseEnum.FAIL, "鏂囦欢涓嶈兘涓虹┖");
throw new BusinessException(CommonResponseEnum.FAIL, "文件不能为空");
}
Map<String, String> storedFileMap = new LinkedHashMap<String, String>();
for (Map.Entry<String, MultipartFile> entry : fileMap.entrySet()) {
String originalFileName = entry.getKey();
MultipartFile file = entry.getValue();
if (file == null || file.isEmpty() || StrUtil.isBlank(originalFileName)) {
throw new BusinessException(CommonResponseEnum.FAIL, "鏂囦欢涓嶈兘涓虹┖");
try {
for (Map.Entry<String, MultipartFile> entry : fileMap.entrySet()) {
String originalFileName = trimFileName(entry.getKey());
MultipartFile file = entry.getValue();
if (StrUtil.isBlank(originalFileName) || file == null || file.isEmpty()) {
throw new BusinessException(CommonResponseEnum.FAIL, "文件不能为空");
}
validateAllowedUploadFile(originalFileName);
try (InputStream inputStream = file.getInputStream()) {
AiReportMinioStorageService.StoredFile storedFile = minioStorageService.saveInputStream(
bizDir, null, originalFileName, inputStream, file.getSize(), file.getContentType(), true);
storedFileMap.put(originalFileName, storedFile.getObjectName());
}
}
String safeFileName = Paths.get(originalFileName).getFileName().toString();
Path targetPath = resolveValidateSessionUploadDir(sessionId).resolve(safeFileName);
try {
Files.createDirectories(targetPath.getParent());
file.transferTo(targetPath.toFile());
} catch (IOException exception) {
throw new BusinessException(CommonResponseEnum.FAIL, "鏂囦欢涓婁紶闃舵锛氫繚瀛樻牎楠屼細璇濇枃浠跺け璐?");
}
storedFileMap.put(originalFileName, buildValidateSessionUploadRelativePath(sessionId, safeFileName));
} catch (BusinessException exception) {
cleanupStoredFiles(storedFileMap);
throw exception;
} catch (Exception exception) {
cleanupStoredFiles(storedFileMap);
throw new BusinessException(CommonResponseEnum.FAIL, errorMessage);
}
UploadedBatch uploadedBatch = new UploadedBatch();
uploadedBatch.setBatchId(sessionId);
uploadedBatch.setTempStoragePath(buildValidateSessionRelativePath(sessionId));
uploadedBatch.setBatchId(batchId);
uploadedBatch.setTempStoragePath(tempStoragePath);
uploadedBatch.setStoredFileMap(storedFileMap);
return uploadedBatch;
}
public Path resolveTempBatchFile(String reportId, String batchId, String fileName) {
return resolveTempUploadDir(reportId, batchId).resolve(Paths.get(fileName).getFileName().toString()).normalize();
}
public void writeTempBatchMeta(String reportId, String batchId, String content) {
Path metaPath = resolveTempBatchMetaPath(reportId, batchId);
private void writeText(String objectName, String content, String errorMessage) {
try {
Files.createDirectories(metaPath.getParent());
Files.write(metaPath, content.getBytes(StandardCharsets.UTF_8));
} catch (IOException exception) {
throw new BusinessException(CommonResponseEnum.FAIL, "校验文件阶段:保存批次元数据失败");
minioStorageService.saveBytes(resolveParentDir(objectName), null, resolveFileName(objectName),
content == null ? new byte[0] : content.getBytes(StandardCharsets.UTF_8),
MediaType.APPLICATION_JSON_VALUE, true);
} catch (BusinessException exception) {
throw exception;
} catch (Exception exception) {
throw new BusinessException(CommonResponseEnum.FAIL, errorMessage);
}
}
public void writeValidateSessionResult(String sessionId, String content) {
Path resultPath = resolveValidateSessionResultPath(sessionId);
try {
Files.createDirectories(resultPath.getParent());
Files.write(resultPath, content.getBytes(StandardCharsets.UTF_8));
} catch (IOException exception) {
throw new BusinessException(CommonResponseEnum.FAIL, "鏍¢獙鏂囦欢闃舵锛氫繚瀛樻牎楠岀粨鏋滃け璐?");
}
}
public Path resolveValidateSessionFile(String sessionId, String fileName) {
return resolveValidateSessionUploadDir(sessionId).resolve(Paths.get(fileName).getFileName().toString()).normalize();
}
public List<String> listValidateSessionFileNames(String sessionId) {
Path uploadDir = resolveValidateSessionUploadDir(sessionId);
if (!Files.exists(uploadDir) || !Files.isDirectory(uploadDir)) {
return new ArrayList<String>();
}
try {
List<String> result = new ArrayList<String>();
Files.list(uploadDir)
.filter(Files::isRegularFile)
.forEach(path -> result.add(path.getFileName().toString()));
return result;
} catch (IOException exception) {
throw new BusinessException(CommonResponseEnum.FAIL, "鏍¢獙鏂囦欢闃舵锛氳鍙栦細璇濇枃浠跺垪琛ㄥけ璐?");
}
}
public String readTempBatchMeta(String reportId, String batchId) {
Path metaPath = resolveTempBatchMetaPath(reportId, batchId);
if (!Files.exists(metaPath) || !Files.isRegularFile(metaPath)) {
private String copyTempBatchFile(String reportId, String batchId, String fileName, String targetBizDir,
String contentType, String errorMessage) {
String normalizedFileName = trimFileName(fileName);
if (StrUtil.isBlank(normalizedFileName)) {
return null;
}
try {
return new String(Files.readAllBytes(metaPath), StandardCharsets.UTF_8);
} catch (IOException exception) {
throw new BusinessException(CommonResponseEnum.FAIL, "璇诲彇瀵煎叆鎵规淇℃伅澶辫触");
}
}
public String saveLedgerSummaryFromTemp(String reportId, String batchId, String fileName) {
return savePathFile(reportId, TestReportConst.STORAGE_LEDGER_SUMMARY_DIR,
resolveTempBatchFile(reportId, batchId, fileName), fileName, "淇濆瓨鍙拌处姹囨€籈xcel澶辫触");
}
public String savePointExcelFromTemp(String reportId, String batchId, String fileName) {
return savePathFile(reportId, TestReportConst.STORAGE_POINT_EXCEL_DIR,
resolveTempBatchFile(reportId, batchId, fileName), fileName, "淇濆瓨鐩戞祴鐐笶xcel闄勪欢澶辫触");
}
public String savePointWordFromTemp(String reportId, String batchId, String fileName) {
return savePathFile(reportId, TestReportConst.STORAGE_POINT_WORD_DIR,
resolveTempBatchFile(reportId, batchId, fileName), fileName, "淇濆瓨鐩戞祴鐐筗ord闄勪欢澶辫触");
}
public String saveGeneratedPlaceholder(String reportId, String fileName, byte[] content) {
Path targetPath = resolveReportDir(reportId, TestReportConst.STORAGE_GENERATED_REPORT_DIR).resolve(fileName);
try {
Files.createDirectories(targetPath.getParent());
Files.write(targetPath, content);
} catch (IOException exception) {
throw new BusinessException(CommonResponseEnum.FAIL, "淇濆瓨鐢熸垚鎶ュ憡澶辫触");
}
return buildRelativePath(reportId, TestReportConst.STORAGE_GENERATED_REPORT_DIR, fileName);
}
public Path resolveBusinessPath(String relativePath) {
return Paths.get(filesRootPath, TestReportConst.STORAGE_DIR).resolve(relativePath).normalize();
}
public void deleteQuietly(String relativePath) {
if (StrUtil.isBlank(relativePath)) {
return;
}
try {
Files.deleteIfExists(resolveBusinessPath(relativePath));
} catch (IOException ignored) {
// 鏂囦欢娓呯悊澶辫触涓嶅奖鍝嶄富娴佺▼
}
}
public void deleteTempBatchQuietly(String reportId, String batchId) {
if (StrUtil.isBlank(reportId) || StrUtil.isBlank(batchId)) {
return;
}
deletePathQuietly(resolveTempBatchDir(reportId, batchId));
}
private String saveMultipartFile(String reportId, String childDir, MultipartFile file, String errorMessage) {
String originalFileName = file == null ? null : file.getOriginalFilename();
if (file == null || file.isEmpty() || StrUtil.isBlank(originalFileName)) {
throw new BusinessException(CommonResponseEnum.FAIL, "文件不能为空");
}
return savePathFile(reportId, childDir, toTempTransferredPath(file), originalFileName, errorMessage);
}
private Path toTempTransferredPath(MultipartFile file) {
try {
Path tempFile = Files.createTempFile("test-report-", ".upload");
file.transferTo(tempFile.toFile());
return tempFile;
} catch (IOException exception) {
throw new BusinessException(CommonResponseEnum.FAIL, "文件上传失败");
}
}
private String savePathFile(String reportId, String childDir, Path sourcePath, String originalFileName, String errorMessage) {
String normalizedFileName = originalFileName == null ? null : Paths.get(originalFileName).getFileName().toString();
if (sourcePath == null || StrUtil.isBlank(normalizedFileName) || !Files.exists(sourcePath)) {
throw new BusinessException(CommonResponseEnum.FAIL, "文件不能为空");
}
String fileName = UUID.randomUUID().toString().replace("-", "") + "-" + normalizedFileName;
Path targetPath = resolveReportDir(reportId, childDir).resolve(fileName);
try {
Files.createDirectories(targetPath.getParent());
Files.copy(sourcePath, targetPath, StandardCopyOption.REPLACE_EXISTING);
} catch (IOException exception) {
byte[] content = minioStorageService.readBytes(buildTempBatchFileObjectName(reportId, batchId, normalizedFileName));
return minioStorageService.saveBytes(targetBizDir, null, normalizedFileName, content, contentType, false)
.getObjectName();
} catch (BusinessException exception) {
throw exception;
} catch (Exception exception) {
throw new BusinessException(CommonResponseEnum.FAIL, errorMessage);
} finally {
deletePathQuietlyIfTemp(sourcePath);
}
return buildRelativePath(reportId, childDir, fileName);
}
private void deletePathQuietlyIfTemp(Path path) {
if (path == null) {
private void deletePrefixQuietly(String prefix) {
List<String> objectNames = minioStorageService.listObjectNames(prefix);
for (String objectName : objectNames) {
minioStorageService.deleteQuietly(objectName);
}
}
private void cleanupStoredFiles(Map<String, String> storedFileMap) {
if (storedFileMap == null || storedFileMap.isEmpty()) {
return;
}
String fileName = path.getFileName() == null ? null : path.getFileName().toString();
if (fileName != null && fileName.startsWith("test-report-")) {
deletePathQuietly(path);
for (String objectName : storedFileMap.values()) {
minioStorageService.deleteQuietly(objectName);
}
}
private void deletePathQuietly(Path path) {
if (path == null || !Files.exists(path)) {
return;
private void validateMultipartFile(MultipartFile file, Set<String> allowedExtensions,
String emptyMessage, String invalidMessage) {
if (file == null || file.isEmpty()) {
throw new BusinessException(CommonResponseEnum.FAIL, emptyMessage);
}
try {
if (Files.isDirectory(path)) {
Files.walk(path)
.sorted((left, right) -> right.getNameCount() - left.getNameCount())
.forEach(this::deleteSingleQuietly);
return;
}
Files.deleteIfExists(path);
} catch (IOException ignored) {
// 涓存椂鐩綍娓呯悊澶辫触涓嶅奖鍝嶄富娴佺▼
validateFileName(file.getOriginalFilename(), allowedExtensions, emptyMessage, invalidMessage);
}
private void validateAllowedUploadFile(String fileName) {
validateFileName(fileName, AiReportFileExtensionConst.ALLOWED_EXTENSIONS, "文件不能为空",
"仅允许上传docx、xlsx、pdf文件");
}
private void validateFileName(String fileName, Set<String> allowedExtensions,
String emptyMessage, String invalidMessage) {
String normalizedFileName = trimFileName(fileName);
if (StrUtil.isBlank(normalizedFileName)) {
throw new BusinessException(CommonResponseEnum.FAIL, emptyMessage);
}
String extension = resolveExtension(normalizedFileName);
if (!allowedExtensions.contains(extension)) {
throw new BusinessException(CommonResponseEnum.FAIL, invalidMessage);
}
}
private void deleteSingleQuietly(Path path) {
try {
Files.deleteIfExists(path);
} catch (IOException ignored) {
// ignore
}
private String buildBusinessDir(String reportId, String childDir) {
return AiReportFileConst.BIZ_DIR_TEST_REPORT + "/" + reportId + "/" + childDir;
}
private Path resolveReportDir(String reportId, String childDir) {
return Paths.get(filesRootPath, TestReportConst.STORAGE_DIR, reportId, childDir).normalize();
private String buildTempBatchDir(String reportId, String batchId) {
return AiReportFileConst.BIZ_DIR_TEST_REPORT + "/" + reportId + "/"
+ TestReportConst.STORAGE_IMPORT_TEMP_DIR + "/" + batchId;
}
private Path resolveTempBatchDir(String reportId, String batchId) {
return Paths.get(filesRootPath, TestReportConst.STORAGE_DIR, reportId,
TestReportConst.STORAGE_IMPORT_TEMP_DIR, batchId).normalize();
private String buildTempUploadDir(String reportId, String batchId) {
return buildTempBatchDir(reportId, batchId) + "/" + TestReportConst.STORAGE_IMPORT_TEMP_UPLOAD_DIR;
}
private Path resolveTempUploadDir(String reportId, String batchId) {
return resolveTempBatchDir(reportId, batchId).resolve(TestReportConst.STORAGE_IMPORT_TEMP_UPLOAD_DIR);
private String buildValidateSessionDir(String sessionId) {
return AiReportFileConst.BIZ_DIR_TEST_REPORT + "/" + TestReportConst.STORAGE_VALIDATE_SESSION_DIR + "/" + sessionId;
}
private Path resolveTempBatchMetaPath(String reportId, String batchId) {
return resolveTempBatchDir(reportId, batchId).resolve(TestReportConst.STORAGE_IMPORT_META_FILE_NAME);
private String buildValidateSessionUploadDir(String sessionId) {
return buildValidateSessionDir(sessionId) + "/" + TestReportConst.STORAGE_IMPORT_TEMP_UPLOAD_DIR;
}
private Path resolveValidateSessionDir(String sessionId) {
return Paths.get(filesRootPath, TestReportConst.STORAGE_DIR,
TestReportConst.STORAGE_VALIDATE_SESSION_DIR, sessionId).normalize();
private String buildTempBatchMetaObjectName(String reportId, String batchId) {
return buildTempBatchDir(reportId, batchId) + "/" + TestReportConst.STORAGE_IMPORT_META_FILE_NAME;
}
private Path resolveValidateSessionUploadDir(String sessionId) {
return resolveValidateSessionDir(sessionId).resolve(TestReportConst.STORAGE_IMPORT_TEMP_UPLOAD_DIR);
private String buildValidateSessionResultObjectName(String sessionId) {
return buildValidateSessionDir(sessionId) + "/" + TestReportConst.STORAGE_VALIDATE_RESULT_FILE_NAME;
}
private Path resolveValidateSessionResultPath(String sessionId) {
return resolveValidateSessionDir(sessionId).resolve(TestReportConst.STORAGE_VALIDATE_RESULT_FILE_NAME);
private String buildTempBatchFileObjectName(String reportId, String batchId, String fileName) {
return buildTempUploadDir(reportId, batchId) + "/" + trimFileName(fileName);
}
private String buildRelativePath(String reportId, String childDir, String fileName) {
return reportId + "/" + childDir + "/" + fileName;
private String buildValidateSessionFileObjectName(String sessionId, String fileName) {
return buildValidateSessionUploadDir(sessionId) + "/" + trimFileName(fileName);
}
private String buildTempBatchRelativePath(String reportId, String batchId) {
return reportId + "/" + TestReportConst.STORAGE_IMPORT_TEMP_DIR + "/" + batchId;
}
private String buildTempUploadRelativePath(String reportId, String batchId, String fileName) {
return buildTempBatchRelativePath(reportId, batchId) + "/"
+ TestReportConst.STORAGE_IMPORT_TEMP_UPLOAD_DIR + "/" + fileName;
}
private String buildValidateSessionRelativePath(String sessionId) {
return TestReportConst.STORAGE_VALIDATE_SESSION_DIR + "/" + sessionId;
}
private String buildValidateSessionUploadRelativePath(String sessionId, String fileName) {
return buildValidateSessionRelativePath(sessionId) + "/"
+ TestReportConst.STORAGE_IMPORT_TEMP_UPLOAD_DIR + "/" + fileName;
private String resolveParentDir(String objectName) {
int slashIndex = objectName.lastIndexOf('/');
if (slashIndex <= 0) {
throw new BusinessException(CommonResponseEnum.FAIL, "对象目录配置不合法");
}
return objectName.substring(0, slashIndex);
}
private String resolveFileName(String objectName) {
int slashIndex = objectName.lastIndexOf('/');
return slashIndex >= 0 ? objectName.substring(slashIndex + 1) : objectName;
}
private String trimFileName(String fileName) {
String normalized = StrUtil.trimToEmpty(fileName);
if (StrUtil.isBlank(normalized)) {
return null;
}
int slashIndex = Math.max(normalized.lastIndexOf('/'), normalized.lastIndexOf('\\'));
return slashIndex >= 0 ? normalized.substring(slashIndex + 1) : normalized;
}
private String resolveExtension(String fileName) {
int dotIndex = fileName.lastIndexOf('.');
if (dotIndex < 0 || dotIndex == fileName.length() - 1) {
return "";
}
return fileName.substring(dotIndex + 1).toLowerCase(Locale.ENGLISH);
}
public static class UploadedBatch {

View File

@@ -10,7 +10,6 @@ import com.njcn.gather.aireport.testreport.pojo.param.TestReportGroupSaveParam;
import com.njcn.gather.aireport.testreport.pojo.param.TestReportParam;
import com.njcn.gather.aireport.testreport.pojo.vo.TestReportGroupReportVO;
import com.njcn.gather.aireport.testreport.pojo.vo.TestReportLedgerImportResultVO;
import com.njcn.gather.aireport.testreport.pojo.vo.TestReportLedgerValidateResultVO;
import com.njcn.gather.aireport.testreport.pojo.vo.TestReportPointVO;
import com.njcn.gather.aireport.testreport.pojo.vo.TestReportVO;
import com.njcn.gather.aireport.testreport.service.TestReportService;
@@ -126,10 +125,10 @@ public class TestReportController extends BaseController {
@OperateInfo(info = LogEnum.BUSINESS_COMMON, operateType = OperateType.UPDATE)
@ApiOperation("校验台账汇总与监测点附件")
@PostMapping("/{id}/ledger/validate")
public HttpResult<TestReportLedgerValidateResultVO> validateLedger(@PathVariable("id") String id,
@RequestParam("files") MultipartFile[] files) {
public HttpResult<TestReportLedgerImportResultVO> validateLedger(@PathVariable("id") String id,
@RequestParam("files") MultipartFile[] files) {
String methodDescribe = getMethodDescribe("validateLedger");
TestReportLedgerValidateResultVO result = testReportService.validateLedger(id, files);
TestReportLedgerImportResultVO result = testReportService.validateLedger(id, files);
return HttpResultUtil.assembleCommonResponseResult(Boolean.TRUE.equals(result.getSuccess())
? CommonResponseEnum.SUCCESS : CommonResponseEnum.FAIL,
result, methodDescribe);

View File

@@ -1,93 +1,100 @@
package com.njcn.gather.aireport.testreport.pojo.param;
import com.fasterxml.jackson.databind.JsonNode;
import com.njcn.web.pojo.param.BaseParam;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;
import lombok.EqualsAndHashCode;
import javax.validation.constraints.NotBlank;
import javax.validation.constraints.NotEmpty;
import javax.validation.constraints.NotNull;
import javax.validation.constraints.Size;
import java.util.List;
public class TestReportParam {
@Data
@EqualsAndHashCode(callSuper = true)
public static class QueryParam extends BaseParam {
@ApiModelProperty("\u5173\u952e\u5b57\uff0c\u5339\u914d\u62a5\u544a\u7f16\u53f7\u3001\u5408\u540c\u7f16\u53f7\u3001\u59d4\u6258\u5355\u4f4d\u6216\u62a5\u544a\u6a21\u677f")
@ApiModelProperty("关键字,匹配报告编号、合同编号、委托单位或报告模板")
private String keyword;
@ApiModelProperty("\u521b\u5efa\u5f00\u59cb\u65f6\u95f4\uff0cyyyy-MM-dd HH:mm:ss")
@ApiModelProperty("创建开始时间,yyyy-MM-dd HH:mm:ss")
private String searchBeginTime;
@ApiModelProperty("\u521b\u5efa\u7ed3\u675f\u65f6\u95f4\uff0cyyyy-MM-dd HH:mm:ss")
@ApiModelProperty("创建结束时间,yyyy-MM-dd HH:mm:ss")
private String searchEndTime;
@ApiModelProperty("\u62a5\u544a\u72b6\u6001")
@ApiModelProperty("报告状态")
private String state;
}
@Data
public static class AddParam {
@ApiModelProperty("\u666e\u6d4b\u62a5\u544aID\uff0c\u4e0d\u4f20\u65f6\u7531\u540e\u7aef\u751f\u6210")
@ApiModelProperty("普测报告ID不传时由后端生成")
private String id;
@ApiModelProperty("\u666e\u6d4b\u62a5\u544a\u7f16\u53f7")
@NotBlank(message = "\u666e\u6d4b\u62a5\u544a\u7f16\u53f7\u4e0d\u80fd\u4e3a\u7a7a")
@Size(max = 32, message = "\u666e\u6d4b\u62a5\u544a\u7f16\u53f7\u4e0d\u80fd\u8d85\u8fc732\u4e2a\u5b57\u7b26")
@ApiModelProperty("普测报告编号")
@NotBlank(message = "普测报告编号不能为空")
@Size(max = 32, message = "普测报告编号不能超过32个字符")
private String no;
@ApiModelProperty("\u59d4\u6258\u5355\u4f4dID")
@NotBlank(message = "\u59d4\u6258\u5355\u4f4d\u4e0d\u80fd\u4e3a\u7a7a")
@ApiModelProperty("委托单位ID")
@NotBlank(message = "委托单位不能为空")
private String clientUnitId;
@ApiModelProperty("\u62a5\u544a\u6a21\u677fID")
@NotBlank(message = "\u62a5\u544a\u6a21\u677f\u4e0d\u80fd\u4e3a\u7a7a")
@ApiModelProperty("报告模板ID")
@NotBlank(message = "报告模板不能为空")
private String reportModelId;
@ApiModelProperty("\u68c0\u6d4b\u516c\u53f8ID")
@NotBlank(message = "\u68c0\u6d4b\u516c\u53f8\u4e0d\u80fd\u4e3a\u7a7a")
@ApiModelProperty("检测公司ID")
@NotBlank(message = "检测公司不能为空")
private String createUnit;
@ApiModelProperty("\u5408\u540c\u7f16\u53f7")
@NotBlank(message = "\u5408\u540c\u7f16\u53f7\u4e0d\u80fd\u4e3a\u7a7a")
@Size(max = 32, message = "\u5408\u540c\u7f16\u53f7\u4e0d\u80fd\u8d85\u8fc732\u4e2a\u5b57\u7b26")
@ApiModelProperty("合同编号")
@NotBlank(message = "合同编号不能为空")
@Size(max = 32, message = "合同编号不能超过32个字符")
private String contractNumber;
@ApiModelProperty("\u68c0\u6d4b\u4f9d\u636e\uff0cJSON\u5b57\u7b26\u4e32\u6570\u7ec4")
@NotBlank(message = "\u68c0\u6d4b\u4f9d\u636e\u4e0d\u80fd\u4e3a\u7a7a")
private String standard;
@ApiModelProperty("检测依据JSON数组")
@NotEmpty(message = "检测依据不能为空")
private List<String> standard;
@ApiModelProperty("\u53d7\u8d44\u6570\u636e\uff0cJSON\u5bf9\u8c61\u6216JSON\u6570\u7ec4")
@NotBlank(message = "\u53d7\u8d44\u6570\u636e\u4e0d\u80fd\u4e3a\u7a7a")
private String data;
@ApiModelProperty("受资数据JSON对象或JSON数组")
@NotNull(message = "受资数据不能为空")
private JsonNode data;
@ApiModelProperty("\u8054\u7cfb\u7535\u8bdd")
@Size(max = 32, message = "\u8054\u7cfb\u7535\u8bdd\u4e0d\u80fd\u8d85\u8fc732\u4e2a\u5b57\u7b26")
@ApiModelProperty("台账校验通过后返回的导入批次ID")
private String batchId;
@ApiModelProperty("联系电话")
@Size(max = 32, message = "联系电话不能超过32个字符")
private String phonenumber;
}
@Data
public static class UpdateParam extends AddParam {
@ApiModelProperty("\u666e\u6d4b\u62a5\u544aID")
@NotBlank(message = "\u666e\u6d4b\u62a5\u544aID\u4e0d\u80fd\u4e3a\u7a7a")
@ApiModelProperty("普测报告ID")
@NotBlank(message = "普测报告ID不能为空")
private String id;
}
@Data
public static class SubmitAuditParam {
@ApiModelProperty("\u666e\u6d4b\u62a5\u544aID")
@NotBlank(message = "\u666e\u6d4b\u62a5\u544aID\u4e0d\u80fd\u4e3a\u7a7a")
@ApiModelProperty("普测报告ID")
@NotBlank(message = "普测报告ID不能为空")
private String id;
@ApiModelProperty("\u5ba1\u6838\u610f\u89c1")
@Size(max = 128, message = "\u5ba1\u6838\u610f\u89c1\u4e0d\u80fd\u8d85\u8fc7128\u4e2a\u5b57\u7b26")
@ApiModelProperty("审核意见")
@Size(max = 128, message = "审核意见不能超过128个字符")
private String checkResult;
}
@Data
public static class LedgerImportParam {
@ApiModelProperty("\u6821\u9a8c\u6210\u529f\u7684\u5bfc\u5165\u6279\u6b21ID")
@NotBlank(message = "\u5bfc\u5165\u6279\u6b21ID\u4e0d\u80fd\u4e3a\u7a7a")
@ApiModelProperty("校验成功的导入批次ID")
@NotBlank(message = "导入批次ID不能为空")
private String batchId;
}
}

View File

@@ -16,10 +16,7 @@ public class TestReportPointPO {
private String reportId;
private Integer groupNo;
private String substationName;
private String deviceName;
private String pointName;
private String testDeviceNo;
private String clientUnitName;
private String monitorTime;
private String voltageLevel;
private String ptRatio;
@@ -30,8 +27,6 @@ public class TestReportPointPO {
private String powerSupplyCapacity;
private String excelAttachmentName;
private String excelStoragePath;
private String wordAttachmentName;
private String wordStoragePath;
private Integer sortNo;
private Integer deleted;
private String createBy;

View File

@@ -9,45 +9,45 @@ import java.io.Serializable;
public class TestReportExcelVO implements Serializable {
private static final long serialVersionUID = -3010688465572795334L;
@Excel(name = "\u666e\u6d4b\u62a5\u544a\u7f16\u53f7", width = 25)
@Excel(name = "普测报告编号", width = 25)
private String no;
@Excel(name = "\u59d4\u6258\u5355\u4f4d", width = 25)
@Excel(name = "委托单位", width = 25)
private String clientUnitName;
@Excel(name = "\u62a5\u544a\u6a21\u677f", width = 25)
@Excel(name = "报告模板", width = 25)
private String reportModelName;
@Excel(name = "\u68c0\u6d4b\u516c\u53f8", width = 20)
@Excel(name = "检测公司", width = 20)
private String createUnit;
@Excel(name = "\u5408\u540c\u7f16\u53f7", width = 20)
@Excel(name = "合同编号", width = 20)
private String contractNumber;
@Excel(name = "\u68c0\u6d4b\u4f9d\u636e", width = 35)
@Excel(name = "检测依据", width = 35)
private String standard;
@Excel(name = "\u53d7\u8d44\u6570\u636e", width = 35)
@Excel(name = "受资数据", width = 35)
private String data;
@Excel(name = "\u8054\u7cfb\u7535\u8bdd", width = 20)
@Excel(name = "联系电话", width = 20)
private String phonenumber;
@Excel(name = "\u5ba1\u6838\u4eba", width = 20)
@Excel(name = "审核人", width = 20)
private String checkerName;
@Excel(name = "\u5ba1\u6838\u65f6\u95f4", width = 20)
@Excel(name = "审核时间", width = 20)
private String checkTime;
@Excel(name = "\u5ba1\u6838\u610f\u89c1", width = 30)
@Excel(name = "审核意见", width = 30)
private String checkResult;
@Excel(name = "\u62a5\u544a\u72b6\u6001", width = 15)
@Excel(name = "报告状态", width = 15)
private String state;
@Excel(name = "\u521b\u5efa\u65f6\u95f4", width = 25)
@Excel(name = "创建时间", width = 25)
private String createTime;
@Excel(name = "\u521b\u5efa\u4eba", width = 20)
@Excel(name = "创建人", width = 20)
private String createByName;
}

View File

@@ -8,10 +8,7 @@ public class TestReportPointVO {
private String id;
private Integer groupNo;
private String substationName;
private String deviceName;
private String pointName;
private String testDeviceNo;
private String clientUnitName;
private String monitorTime;
private String voltageLevel;
private String ptRatio;
@@ -21,5 +18,4 @@ public class TestReportPointVO {
private String protocolCapacity;
private String powerSupplyCapacity;
private String excelAttachmentName;
private String wordAttachmentName;
}

View File

@@ -18,63 +18,63 @@ import java.time.LocalDateTime;
public class TestReportVO {
private static final String DEFAULT_EMPTY_DISPLAY = "-";
@ApiModelProperty("\u666e\u6d4b\u62a5\u544aID")
@ApiModelProperty("普测报告ID")
private String id;
@ApiModelProperty("\u666e\u6d4b\u62a5\u544a\u7f16\u53f7")
@ApiModelProperty("普测报告编号")
private String no;
@ApiModelProperty("\u59d4\u6258\u5355\u4f4dID")
@ApiModelProperty("委托单位ID")
private String clientUnitId;
@ApiModelProperty("\u59d4\u6258\u5355\u4f4d\u540d\u79f0")
@ApiModelProperty("委托单位名称")
private String clientUnitName;
@ApiModelProperty("\u62a5\u544a\u6a21\u677fID")
@ApiModelProperty("报告模板ID")
private String reportModelId;
@ApiModelProperty("\u62a5\u544a\u6a21\u677f\u540d\u79f0")
@ApiModelProperty("报告模板名称")
private String reportModelName;
@ApiModelProperty("\u68c0\u6d4b\u516c\u53f8ID")
@ApiModelProperty("检测公司ID")
private String createUnit;
@ApiModelProperty("\u5408\u540c\u7f16\u53f7")
@ApiModelProperty("合同编号")
private String contractNumber;
@ApiModelProperty("\u68c0\u6d4b\u4f9d\u636e")
@ApiModelProperty("检测依据")
private String standard;
@ApiModelProperty("\u53d7\u8d44\u6570\u636e")
@ApiModelProperty("受资数据")
private String data;
@ApiModelProperty("\u8054\u7cfb\u7535\u8bdd")
@ApiModelProperty("联系电话")
private String phonenumber;
@ApiModelProperty("\u5ba1\u6838\u4ebaID")
@ApiModelProperty("审核人ID")
private String checkId;
@ApiModelProperty("\u5ba1\u6838\u4eba\u540d\u79f0")
@ApiModelProperty("审核人名称")
private String checkerName;
@ApiModelProperty("\u5ba1\u6838\u65f6\u95f4")
@ApiModelProperty("审核时间")
@JsonFormat(pattern = "yyyy-MM-dd")
@JsonDeserialize(using = LocalDateDeserializer.class)
@JsonSerialize(using = LocalDateSerializer.class)
private LocalDate checkTime;
@ApiModelProperty("\u5ba1\u6838\u610f\u89c1")
@ApiModelProperty("审核意见")
private String checkResult;
@ApiModelProperty("\u62a5\u544a\u72b6\u6001")
@ApiModelProperty("报告状态")
private String state;
@ApiModelProperty("\u62a5\u544a\u751f\u6210\u72b6\u6001")
@ApiModelProperty("报告生成状态")
private Integer reportGenerateState;
@ApiModelProperty("\u76d1\u6d4b\u70b9\u6570\u91cf")
@ApiModelProperty("监测点数量")
private Integer pointCount;
@ApiModelProperty("\u5206\u7ec4\u6570\u91cf")
@ApiModelProperty("分组数量")
private Integer groupCount;
@ApiModelProperty("\u903b\u8f91\u72b6\u6001")
@ApiModelProperty("逻辑状态")
private Integer status;
@ApiModelProperty("\u521b\u5efa\u4eba\u663e\u793a\u540d")
@ApiModelProperty("创建人显示名")
private String createBy;
@ApiModelProperty("\u521b\u5efa\u4eba\u540d\u79f0")
@ApiModelProperty("创建人名称")
private String createByName;
@ApiModelProperty("\u521b\u5efa\u65f6\u95f4")
@ApiModelProperty("创建时间")
@JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss")
@JsonDeserialize(using = LocalDateTimeDeserializer.class)
@JsonSerialize(using = LocalDateTimeSerializer.class)
private LocalDateTime createTime;
@ApiModelProperty("\u66f4\u65b0\u4eba\u663e\u793a\u540d")
@ApiModelProperty("更新人显示名")
private String updateBy;
@ApiModelProperty("\u66f4\u65b0\u4eba\u540d\u79f0")
@ApiModelProperty("更新人名称")
private String updateByName;
@ApiModelProperty("\u66f4\u65b0\u65f6\u95f4")
@ApiModelProperty("更新时间")
@JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss")
@JsonDeserialize(using = LocalDateTimeDeserializer.class)
@JsonSerialize(using = LocalDateTimeSerializer.class)

View File

@@ -7,7 +7,6 @@ import com.njcn.gather.aireport.testreport.pojo.param.TestReportParam;
import com.njcn.gather.aireport.testreport.pojo.po.TestReportPO;
import com.njcn.gather.aireport.testreport.pojo.vo.TestReportGroupReportVO;
import com.njcn.gather.aireport.testreport.pojo.vo.TestReportLedgerImportResultVO;
import com.njcn.gather.aireport.testreport.pojo.vo.TestReportLedgerValidateResultVO;
import com.njcn.gather.aireport.testreport.pojo.vo.TestReportPointVO;
import com.njcn.gather.aireport.testreport.pojo.vo.TestReportVO;
import org.springframework.http.ResponseEntity;
@@ -35,7 +34,7 @@ public interface TestReportService extends IService<TestReportPO> {
ResponseEntity<byte[]> previewTemplate(String id);
TestReportLedgerValidateResultVO validateLedger(String id, MultipartFile[] files);
TestReportLedgerImportResultVO validateLedger(String id, MultipartFile[] files);
TestReportLedgerImportResultVO importLedger(String id, TestReportParam.LedgerImportParam param);

View File

@@ -29,7 +29,6 @@ import com.njcn.gather.aireport.testreport.pojo.po.TestReportPointPO;
import com.njcn.gather.aireport.testreport.pojo.vo.TestReportExcelVO;
import com.njcn.gather.aireport.testreport.pojo.vo.TestReportGroupReportVO;
import com.njcn.gather.aireport.testreport.pojo.vo.TestReportLedgerImportResultVO;
import com.njcn.gather.aireport.testreport.pojo.vo.TestReportLedgerValidateResultVO;
import com.njcn.gather.aireport.testreport.pojo.vo.TestReportPointVO;
import com.njcn.gather.aireport.testreport.pojo.vo.TestReportVO;
import com.njcn.gather.aireport.testreport.service.TestReportService;
@@ -55,7 +54,6 @@ import org.springframework.web.multipart.MultipartFile;
import java.io.IOException;
import java.net.URLEncoder;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.nio.file.Path;
import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter;
@@ -187,7 +185,12 @@ public class TestReportServiceImpl extends ServiceImpl<TestReportMapper, TestRep
report.setCreateTime(now);
report.setUpdateBy(userId);
report.setUpdateTime(now);
return this.save(report);
boolean saved = this.save(report);
if (!saved) {
return false;
}
importLedgerIfNeeded(report, param.getBatchId(), userId);
return true;
}
@Override
@@ -208,9 +211,15 @@ public class TestReportServiceImpl extends ServiceImpl<TestReportMapper, TestRep
checkStandard(data.getStandard());
applyBusinessFields(report, data);
report.setUpdateBy(resolveCurrentUserId());
String userId = resolveCurrentUserId();
report.setUpdateBy(userId);
report.setUpdateTime(LocalDateTime.now());
return this.updateById(report);
boolean updated = this.updateById(report);
if (!updated) {
return false;
}
importLedgerIfNeeded(report, param.getBatchId(), userId);
return true;
}
@Override
@@ -220,14 +229,28 @@ public class TestReportServiceImpl extends ServiceImpl<TestReportMapper, TestRep
if (validIds.isEmpty()) {
throw new BusinessException(CommonResponseEnum.FAIL, MSG_ID_REQUIRED);
}
return this.lambdaUpdate()
List<TestReportPO> reports = this.lambdaQuery()
.in(TestReportPO::getId, validIds)
.eq(TestReportPO::getStatus, TestReportConst.STATUS_NORMAL)
.list();
String userId = resolveCurrentUserId();
LocalDateTime now = LocalDateTime.now();
boolean deleted = this.lambdaUpdate()
.set(TestReportPO::getState, TestReportConst.STATE_DELETED)
.set(TestReportPO::getStatus, TestReportConst.STATUS_DELETED)
.set(TestReportPO::getUpdateBy, resolveCurrentUserId())
.set(TestReportPO::getUpdateTime, LocalDateTime.now())
.set(TestReportPO::getUpdateBy, userId)
.set(TestReportPO::getUpdateTime, now)
.in(TestReportPO::getId, validIds)
.eq(TestReportPO::getStatus, TestReportConst.STATUS_NORMAL)
.update();
if (deleted) {
for (TestReportPO report : reports) {
deletePointFiles(loadNormalPoints(report.getId()));
deleteGroupReportFiles(loadNormalGroupReports(report.getId()));
testReportStorageService.deleteQuietly(resolveSummaryStoragePath(report.getData()));
}
}
return deleted;
}
@Override
@@ -298,26 +321,30 @@ public class TestReportServiceImpl extends ServiceImpl<TestReportMapper, TestRep
public ResponseEntity<byte[]> previewTemplate(String id) {
TestReportPO report = requireNormal(id);
ReportModelPO model = requireReportModel(report.getReportModelId());
Path filePath = reportModelFileStorageService.resolveDownloadPath(model.getPath());
if (!Files.exists(filePath) || !Files.isRegularFile(filePath)) {
throw new BusinessException(CommonResponseEnum.FAIL, MSG_TEMPLATE_FILE_NOT_FOUND);
}
String fileName = resolvePreviewFileName(model);
FilePreviewService.PreviewContent previewContent = filePreviewService.preview(filePath, fileName);
return ResponseEntity.ok()
.header(HttpHeaders.CONTENT_DISPOSITION, "inline; filename*=UTF-8''" + encodeFileName(fileName))
.header(HttpHeaders.CACHE_CONTROL, "no-cache, no-store, must-revalidate")
.contentType(MediaType.parseMediaType(previewContent.getContentType()))
.body(previewContent.getContent());
Path tempFile = reportModelFileStorageService.downloadToTempFile(model.getPath(), fileName);
try {
FilePreviewService.PreviewContent previewContent = filePreviewService.preview(tempFile, fileName);
return ResponseEntity.ok()
.header(HttpHeaders.CONTENT_DISPOSITION, "inline; filename*=UTF-8''" + encodeFileName(fileName))
.header(HttpHeaders.CACHE_CONTROL, "no-cache, no-store, must-revalidate")
.contentType(MediaType.parseMediaType(previewContent.getContentType()))
.body(previewContent.getContent());
} finally {
reportModelFileStorageService.deleteTempFileQuietly(tempFile);
}
}
@Override
@Transactional(rollbackFor = Exception.class)
public TestReportLedgerValidateResultVO validateLedger(String id, MultipartFile[] files) {
String sessionId = requireText(id, MSG_ID_REQUIRED);
log.info("testreport ledger validate start: sessionId={}, userId={}, fileCount={}, fileNames={}",
sessionId, resolveCurrentUserId(), files == null ? 0 : files.length, describeUploadedFiles(files));
return testReportLedgerImportService.validateLedgerSession(sessionId, files);
public TestReportLedgerImportResultVO validateLedger(String id, MultipartFile[] files) {
String draftId = requireText(id, MSG_ID_REQUIRED);
String userId = resolveCurrentUserId();
log.info("testreport ledger validate start: draftId={}, userId={}, fileCount={}, fileNames={}",
draftId, userId, files == null ? 0 : files.length, describeUploadedFiles(files));
TestReportPO draftReport = new TestReportPO();
draftReport.setId(draftId);
return testReportLedgerImportService.validateLedger(draftReport, files, userId);
}
@Override
@@ -414,8 +441,7 @@ public class TestReportServiceImpl extends ServiceImpl<TestReportMapper, TestRep
throw new BusinessException(CommonResponseEnum.FAIL, MSG_TEMPLATE_FILE_NOT_FOUND);
}
try {
Path filePath = testReportStorageService.resolveBusinessPath(groupReport.getReportStoragePath());
byte[] bytes = Files.readAllBytes(filePath);
byte[] bytes = testReportStorageService.readBusinessBytes(groupReport.getReportStoragePath());
return ResponseEntity.ok()
.header(HttpHeaders.CONTENT_DISPOSITION,
"attachment; filename*=UTF-8''" + encodeFileName(groupReport.getReportFileName()))
@@ -448,7 +474,7 @@ public class TestReportServiceImpl extends ServiceImpl<TestReportMapper, TestRep
data.setReportModelId(requireText(param.getReportModelId(), MSG_REPORT_MODEL_REQUIRED));
data.setCreateUnit(requireText(param.getCreateUnit(), MSG_CREATE_UNIT_REQUIRED));
data.setContractNumber(requireText(param.getContractNumber(), MSG_CONTRACT_REQUIRED, 32, MSG_CONTRACT_TOO_LONG));
data.setStandard(requireText(param.getStandard(), MSG_STANDARD_REQUIRED));
data.setStandard(requireJsonArrayText(param.getStandard(), MSG_STANDARD_REQUIRED, MSG_STANDARD_INVALID));
data.setData(requireJsonObjectOrArrayText(param.getData(), MSG_DATA_REQUIRED, MSG_DATA_INVALID));
data.setPhonenumber(trimToNull(param.getPhonenumber()));
if (data.getPhonenumber() != null && data.getPhonenumber().length() > 32) {
@@ -499,8 +525,8 @@ public class TestReportServiceImpl extends ServiceImpl<TestReportMapper, TestRep
throw new BusinessException(CommonResponseEnum.FAIL, MSG_IMPORT_FILES_REQUIRED);
}
String normalizedFileName = originalFileName.toLowerCase();
if (!normalizedFileName.endsWith(".xls") && !normalizedFileName.endsWith(".xlsx")) {
throw new BusinessException(CommonResponseEnum.FAIL, "原始Excel仅允许xls或xlsx格式");
if (!normalizedFileName.endsWith(".xlsx")) {
throw new BusinessException(CommonResponseEnum.FAIL, "原始Excel仅允许xlsx格式");
}
if (!fileNames.add(originalFileName)) {
throw new BusinessException(CommonResponseEnum.FAIL, MSG_IMPORT_FILE_NAME_DUPLICATE);
@@ -582,6 +608,18 @@ public class TestReportServiceImpl extends ServiceImpl<TestReportMapper, TestRep
}
}
private String resolveSummaryStoragePath(String data) {
if (StrUtil.isBlank(data)) {
return null;
}
try {
String summaryStoragePath = OBJECT_MAPPER.readTree(data).path("summaryStoragePath").asText();
return StrUtil.isBlank(summaryStoragePath) ? null : summaryStoragePath;
} catch (Exception exception) {
return null;
}
}
private void removeOldRows(String reportId) {
LambdaQueryWrapper<TestReportPointPO> pointWrapper = new LambdaQueryWrapper<TestReportPointPO>();
pointWrapper.eq(TestReportPointPO::getReportId, reportId);
@@ -747,10 +785,7 @@ public class TestReportServiceImpl extends ServiceImpl<TestReportMapper, TestRep
vo.setId(point.getId());
vo.setGroupNo(point.getGroupNo());
vo.setSubstationName(defaultDash(point.getSubstationName()));
vo.setDeviceName(defaultDash(point.getDeviceName()));
vo.setPointName(defaultDash(point.getPointName()));
vo.setTestDeviceNo(defaultDash(point.getTestDeviceNo()));
vo.setClientUnitName(defaultDash(point.getClientUnitName()));
vo.setMonitorTime(defaultDash(point.getMonitorTime()));
vo.setVoltageLevel(defaultDash(point.getVoltageLevel()));
vo.setPtRatio(defaultDash(point.getPtRatio()));
@@ -760,7 +795,6 @@ public class TestReportServiceImpl extends ServiceImpl<TestReportMapper, TestRep
vo.setProtocolCapacity(defaultDash(point.getProtocolCapacity()));
vo.setPowerSupplyCapacity(defaultDash(point.getPowerSupplyCapacity()));
vo.setExcelAttachmentName(defaultDash(point.getExcelAttachmentName()));
vo.setWordAttachmentName(defaultDash(point.getWordAttachmentName()));
return vo;
}
@@ -961,14 +995,38 @@ public class TestReportServiceImpl extends ServiceImpl<TestReportMapper, TestRep
}
}
private String requireJsonObjectOrArrayText(String value, String blankMessage, String invalidMessage) {
String text = requireText(value, blankMessage);
private String requireJsonArrayText(List<String> value, String blankMessage, String invalidMessage) {
if (value == null || value.isEmpty()) {
throw new BusinessException(CommonResponseEnum.FAIL, blankMessage);
}
try {
JsonNode jsonNode = OBJECT_MAPPER.readTree(text);
if (jsonNode == null || !(jsonNode.isObject() || jsonNode.isArray())) {
ArrayList<String> normalizedItems = new ArrayList<String>();
for (String item : value) {
String normalizedItem = trimToNull(item);
if (normalizedItem != null) {
normalizedItems.add(normalizedItem);
}
}
if (normalizedItems.isEmpty()) {
throw new BusinessException(CommonResponseEnum.FAIL, invalidMessage);
}
return text;
return OBJECT_MAPPER.writeValueAsString(normalizedItems);
} catch (BusinessException exception) {
throw exception;
} catch (Exception exception) {
throw new BusinessException(CommonResponseEnum.FAIL, invalidMessage);
}
}
private String requireJsonObjectOrArrayText(JsonNode value, String blankMessage, String invalidMessage) {
if (value == null || value.isNull()) {
throw new BusinessException(CommonResponseEnum.FAIL, blankMessage);
}
try {
if (!(value.isObject() || value.isArray())) {
throw new BusinessException(CommonResponseEnum.FAIL, invalidMessage);
}
return OBJECT_MAPPER.writeValueAsString(value);
} catch (BusinessException exception) {
throw exception;
} catch (Exception exception) {
@@ -1086,6 +1144,13 @@ public class TestReportServiceImpl extends ServiceImpl<TestReportMapper, TestRep
return fileNames;
}
private void importLedgerIfNeeded(TestReportPO report, String batchId, String userId) {
if (report == null || StrUtil.isBlank(batchId)) {
return;
}
testReportLedgerImportService.importLedger(report, batchId, userId);
}
private String trimToNull(String value) {
if (value == null) {
return null;

View File

@@ -76,9 +76,50 @@ class TestReportLedgerExcelParserTest {
Assertions.assertTrue(exception.getMessage().contains("重复点位"));
}
@Test
void shouldAllowMissingOptionalSheets() throws Exception {
TestReportLedgerExcelParser parser = new TestReportLedgerExcelParser();
MockMultipartFile summaryFile = new MockMultipartFile("files", TestReportConst.LEDGER_SUMMARY_FILE_NAME,
"application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
buildWorkbookBytes(true, false, false, false));
Map<String, MultipartFile> attachments = new LinkedHashMap<String, MultipartFile>();
attachments.put("point-a.xlsx", new MockMultipartFile("files", "point-a.xlsx", "application/octet-stream", new byte[]{1}));
attachments.put("point-a.docx", new MockMultipartFile("files", "point-a.docx", "application/octet-stream", new byte[]{1}));
TestReportLedgerExcelParser.ParsedLedger ledger = parser.parse(summaryFile, attachments);
Assertions.assertEquals(1, ledger.getPoints().size());
Assertions.assertTrue(ledger.getDevices().isEmpty());
Assertions.assertTrue(ledger.getClients().isEmpty());
Assertions.assertTrue(ledger.getMissingSheets().isEmpty());
}
@Test
void shouldAllowBlankOptionalSheetContent() throws Exception {
TestReportLedgerExcelParser parser = new TestReportLedgerExcelParser();
MockMultipartFile summaryFile = new MockMultipartFile("files", TestReportConst.LEDGER_SUMMARY_FILE_NAME,
"application/vnd.openxmlformats-officedocument.spreadsheetml.sheet", buildWorkbookWithBlankOptionalRows());
Map<String, MultipartFile> attachments = new LinkedHashMap<String, MultipartFile>();
attachments.put("point-a.xlsx", new MockMultipartFile("files", "point-a.xlsx", "application/octet-stream", new byte[]{1}));
attachments.put("point-a.docx", new MockMultipartFile("files", "point-a.docx", "application/octet-stream", new byte[]{1}));
TestReportLedgerExcelParser.ParsedLedger ledger = parser.parse(summaryFile, attachments);
Assertions.assertEquals(1, ledger.getPoints().size());
Assertions.assertTrue(ledger.getDevices().isEmpty());
Assertions.assertTrue(ledger.getClients().isEmpty());
}
private byte[] buildWorkbookBytes(boolean includePointSheet) throws Exception {
return buildWorkbookBytes(includePointSheet, true, true, true);
}
private byte[] buildWorkbookBytes(boolean includePointSheet, boolean includeGuideSheet,
boolean includeDeviceSheet, boolean includeClientSheet) throws Exception {
try (XSSFWorkbook workbook = new XSSFWorkbook(); ByteArrayOutputStream outputStream = new ByteArrayOutputStream()) {
workbook.createSheet(TestReportConst.LEDGER_GUIDE_SHEET_NAME);
if (includeGuideSheet) {
workbook.createSheet(TestReportConst.LEDGER_GUIDE_SHEET_NAME);
}
if (includePointSheet) {
XSSFSheet pointSheet = workbook.createSheet(TestReportConst.LEDGER_POINT_SHEET_NAME);
pointSheet.createRow(0).createCell(0).setCellValue("分组号");
@@ -91,31 +132,35 @@ class TestReportLedgerExcelParserTest {
pointSheet.createRow(1).createCell(0).setCellValue("1");
pointSheet.getRow(1).createCell(1).setCellValue("站点A");
pointSheet.getRow(1).createCell(2).setCellValue("点位A");
pointSheet.getRow(1).createCell(3).setCellValue("DEV-001");
pointSheet.getRow(1).createCell(4).setCellValue("单位A");
pointSheet.getRow(1).createCell(3).setCellValue(includeDeviceSheet ? "DEV-001" : "");
pointSheet.getRow(1).createCell(4).setCellValue(includeClientSheet ? "单位A" : "");
pointSheet.getRow(1).createCell(5).setCellValue("point-a.xlsx");
pointSheet.getRow(1).createCell(6).setCellValue("point-a.docx");
}
XSSFSheet deviceSheet = workbook.createSheet(TestReportConst.LEDGER_DEVICE_SHEET_NAME);
deviceSheet.createRow(0).createCell(0).setCellValue("设备型号");
deviceSheet.getRow(0).createCell(1).setCellValue("设备");
deviceSheet.getRow(0).createCell(2).setCellValue("标准有效期");
deviceSheet.getRow(0).createCell(3).setCellValue("设备状态");
deviceSheet.getRow(0).createCell(4).setCellValue("标准报告文件名");
deviceSheet.createRow(1).createCell(0).setCellValue("型号A");
deviceSheet.getRow(1).createCell(1).setCellValue("DEV-001");
deviceSheet.getRow(1).createCell(2).setCellValue("2026-07-01");
deviceSheet.getRow(1).createCell(3).setCellValue("在运");
deviceSheet.getRow(1).createCell(4).setCellValue("report-a.pdf");
XSSFSheet clientSheet = workbook.createSheet(TestReportConst.LEDGER_CLIENT_SHEET_NAME);
clientSheet.createRow(0).createCell(0).setCellValue("单位名称");
clientSheet.getRow(0).createCell(1).setCellValue("联系人");
clientSheet.getRow(0).createCell(2).setCellValue("联系方式");
clientSheet.getRow(0).createCell(3).setCellValue("地址");
clientSheet.createRow(1).createCell(0).setCellValue("单位A");
clientSheet.getRow(1).createCell(1).setCellValue("张三");
clientSheet.getRow(1).createCell(2).setCellValue("13800000000");
clientSheet.getRow(1).createCell(3).setCellValue("南京");
if (includeDeviceSheet) {
XSSFSheet deviceSheet = workbook.createSheet(TestReportConst.LEDGER_DEVICE_SHEET_NAME);
deviceSheet.createRow(0).createCell(0).setCellValue("设备");
deviceSheet.getRow(0).createCell(1).setCellValue("设备编号");
deviceSheet.getRow(0).createCell(2).setCellValue("标准有效期");
deviceSheet.getRow(0).createCell(3).setCellValue("设备状态");
deviceSheet.getRow(0).createCell(4).setCellValue("标准报告文件名");
deviceSheet.createRow(1).createCell(0).setCellValue("型号A");
deviceSheet.getRow(1).createCell(1).setCellValue("DEV-001");
deviceSheet.getRow(1).createCell(2).setCellValue("2026-07-01");
deviceSheet.getRow(1).createCell(3).setCellValue("在运");
deviceSheet.getRow(1).createCell(4).setCellValue("report-a.pdf");
}
if (includeClientSheet) {
XSSFSheet clientSheet = workbook.createSheet(TestReportConst.LEDGER_CLIENT_SHEET_NAME);
clientSheet.createRow(0).createCell(0).setCellValue("单位名称");
clientSheet.getRow(0).createCell(1).setCellValue("联系人");
clientSheet.getRow(0).createCell(2).setCellValue("联系方式");
clientSheet.getRow(0).createCell(3).setCellValue("地址");
clientSheet.createRow(1).createCell(0).setCellValue("单位A");
clientSheet.getRow(1).createCell(1).setCellValue("张三");
clientSheet.getRow(1).createCell(2).setCellValue("13800000000");
clientSheet.getRow(1).createCell(3).setCellValue("南京");
}
workbook.write(outputStream);
return outputStream.toByteArray();
}
@@ -150,4 +195,40 @@ class TestReportLedgerExcelParserTest {
return outputStream.toByteArray();
}
}
private byte[] buildWorkbookWithBlankOptionalRows() throws Exception {
try (XSSFWorkbook workbook = new XSSFWorkbook(); ByteArrayOutputStream outputStream = new ByteArrayOutputStream()) {
XSSFSheet pointSheet = workbook.createSheet(TestReportConst.LEDGER_POINT_SHEET_NAME);
pointSheet.createRow(0).createCell(0).setCellValue("分组号");
pointSheet.getRow(0).createCell(1).setCellValue("变电站");
pointSheet.getRow(0).createCell(2).setCellValue("监测点名称");
pointSheet.getRow(0).createCell(3).setCellValue("检测设备编号");
pointSheet.getRow(0).createCell(4).setCellValue("委托单位名称");
pointSheet.getRow(0).createCell(5).setCellValue("Excel附件名称");
pointSheet.getRow(0).createCell(6).setCellValue("Word附件名称");
pointSheet.createRow(1).createCell(0).setCellValue("1");
pointSheet.getRow(1).createCell(1).setCellValue("站点A");
pointSheet.getRow(1).createCell(2).setCellValue("点位A");
pointSheet.getRow(1).createCell(5).setCellValue("point-a.xlsx");
pointSheet.getRow(1).createCell(6).setCellValue("point-a.docx");
XSSFSheet deviceSheet = workbook.createSheet(TestReportConst.LEDGER_DEVICE_SHEET_NAME);
deviceSheet.createRow(0).createCell(0).setCellValue("设备型号");
deviceSheet.getRow(0).createCell(1).setCellValue("设备编号");
deviceSheet.getRow(0).createCell(2).setCellValue("标准有效期");
deviceSheet.getRow(0).createCell(3).setCellValue("设备状态");
deviceSheet.getRow(0).createCell(4).setCellValue("标准报告文件名");
deviceSheet.createRow(1).createCell(0).setCellValue("型号A");
XSSFSheet clientSheet = workbook.createSheet(TestReportConst.LEDGER_CLIENT_SHEET_NAME);
clientSheet.createRow(0).createCell(0).setCellValue("单位名称");
clientSheet.getRow(0).createCell(1).setCellValue("联系人");
clientSheet.getRow(0).createCell(2).setCellValue("联系方式");
clientSheet.getRow(0).createCell(3).setCellValue("地址");
clientSheet.createRow(1).createCell(1).setCellValue("张三");
workbook.write(outputStream);
return outputStream.toByteArray();
}
}
}

View File

@@ -1,151 +1,151 @@
package com.njcn.gather.aireport.testreport.component;
import com.njcn.common.pojo.exception.BusinessException;
import com.njcn.gather.aireport.clientunit.mapper.ClientUnitMapper;
import com.njcn.gather.aireport.clientunit.pojo.po.ClientUnitPO;
import com.njcn.gather.aireport.testdevice.mapper.TestDeviceMapper;
import com.njcn.gather.aireport.testdevice.pojo.constant.TestDeviceConst;
import com.njcn.gather.aireport.testdevice.pojo.po.TestDevicePO;
import com.njcn.gather.aireport.testreport.mapper.TestReportGroupReportMapper;
import com.njcn.gather.aireport.testreport.mapper.TestReportMapper;
import com.njcn.gather.aireport.testreport.mapper.TestReportPointMapper;
import com.njcn.gather.aireport.testreport.pojo.vo.TestReportLedgerValidateResultVO;
import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.Test;
import org.springframework.mock.web.MockMultipartFile;
import org.springframework.web.multipart.MultipartFile;
import java.util.Collections;
import java.util.HashMap;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.Map;
import static org.mockito.Mockito.mock;
class TestReportLedgerImportServiceTest {
@Test
void normalizeFileMapShouldRejectEmptyFiles() {
TestReportLedgerImportService service = createService();
BusinessException exception = Assertions.assertThrows(BusinessException.class,
() -> invokeNormalizeFileMap(service, new MultipartFile[0]));
Assertions.assertEquals("选择文件阶段:导入文件不能为空", exception.getMessage());
}
@Test
void resolveDeviceStateShouldSupportCodesAndNames() {
TestReportLedgerImportService service = createService();
Assertions.assertEquals(TestDeviceConst.STATE_RUNNING, invokeResolveDeviceState(service, "01", 2));
Assertions.assertEquals(TestDeviceConst.STATE_RUNNING, invokeResolveDeviceState(service, "在运", 2));
Assertions.assertEquals(TestDeviceConst.STATE_RETIRED, invokeResolveDeviceState(service, "02", 3));
Assertions.assertEquals(TestDeviceConst.STATE_RETIRED, invokeResolveDeviceState(service, "退运", 3));
}
@Test
void resolveDeviceStateShouldRejectUnexpectedState() {
TestReportLedgerImportService service = createService();
BusinessException exception = Assertions.assertThrows(BusinessException.class,
() -> invokeResolveDeviceState(service, "停用", 4));
Assertions.assertEquals("基础信息维护阶段检测设备sheet第4行设备状态不正确", exception.getMessage());
}
@Test
void normalizeFileMapShouldSkipEmptyFileItemsAndKeepTrimmedNames() {
TestReportLedgerImportService service = createService();
MockMultipartFile validFile = new MockMultipartFile("files", " point-a.xlsx ", "application/octet-stream",
new byte[]{1});
MockMultipartFile emptyFile = new MockMultipartFile("files", "empty.xlsx", "application/octet-stream",
new byte[0]);
Map<String, MultipartFile> result = invokeNormalizeFileMap(service, new MultipartFile[]{emptyFile, validFile});
Assertions.assertEquals(1, result.size());
Assertions.assertSame(validFile, result.get("point-a.xlsx"));
}
@Test
void fillPointValidationResultShouldAppendReferenceErrorsToPointResult() {
TestReportLedgerImportService service = createService();
TestReportLedgerExcelParser.ParsedPointRow pointRow = new TestReportLedgerExcelParser.ParsedPointRow();
pointRow.setRowNo(2);
pointRow.setSubstationName("变电站A");
pointRow.setPointName("监测点A");
pointRow.setTestDeviceNo("DEV-001");
pointRow.setClientUnitName("委托单位A");
pointRow.setExcelAttachmentName("point-a.xlsx");
pointRow.setWordAttachmentName("point-a.docx");
TestReportLedgerExcelParser.ParsedLedger ledger = new TestReportLedgerExcelParser.ParsedLedger();
ledger.setPoints(Collections.singletonList(pointRow));
TestReportLedgerValidateResultVO result = new TestReportLedgerValidateResultVO();
invokeFillPointValidationResult(service, result, ledger, Collections.singleton("台账信息汇总.xlsx"),
new HashMap<String, TestDevicePO>(), new HashMap<String, ClientUnitPO>());
Assertions.assertEquals(1, result.getPoints().size());
Assertions.assertFalse(Boolean.TRUE.equals(result.getPoints().get(0).getComplete()));
Assertions.assertFalse(Boolean.TRUE.equals(result.getPoints().get(0).getTestDeviceNoValid()));
Assertions.assertEquals("第2行检测设备编号在数据库和检测设备sheet中都不存在",
result.getPoints().get(0).getTestDeviceNoMessage());
Assertions.assertFalse(Boolean.TRUE.equals(result.getPoints().get(0).getClientUnitNameValid()));
Assertions.assertEquals("第2行委托单位名称在数据库和委托单位sheet中都不存在",
result.getPoints().get(0).getClientUnitNameMessage());
}
private TestReportLedgerImportService createService() {
return new TestReportLedgerImportService(
mock(TestReportMapper.class),
mock(TestReportPointMapper.class),
mock(TestReportGroupReportMapper.class),
mock(TestDeviceMapper.class),
mock(ClientUnitMapper.class),
mock(TestReportLedgerExcelParser.class),
mock(TestReportStorageService.class));
}
@SuppressWarnings("unchecked")
private Map<String, MultipartFile> invokeNormalizeFileMap(TestReportLedgerImportService service, MultipartFile[] files) {
return invokePrivateMethod(service, "normalizeFileMap", new Class[]{MultipartFile[].class}, new Object[]{files});
}
private String invokeResolveDeviceState(TestReportLedgerImportService service, String stateName, Integer rowNo) {
return (String) invokePrivateMethod(service, "resolveDeviceState",
new Class[]{String.class, Integer.class}, new Object[]{stateName, rowNo});
}
private void invokeFillPointValidationResult(TestReportLedgerImportService service,
TestReportLedgerValidateResultVO result,
TestReportLedgerExcelParser.ParsedLedger ledger,
java.util.Set<String> sessionFileNameSet,
Map<String, TestDevicePO> dbDeviceMap,
Map<String, ClientUnitPO> dbClientMap) {
invokePrivateMethod(service, "fillPointValidationResult",
new Class[]{TestReportLedgerValidateResultVO.class, TestReportLedgerExcelParser.ParsedLedger.class,
java.util.Set.class, Map.class, Map.class},
new Object[]{result, ledger, sessionFileNameSet, dbDeviceMap, dbClientMap});
}
private Object invokePrivateMethod(TestReportLedgerImportService service, String methodName,
Class<?>[] parameterTypes, Object[] args) {
try {
Method method = TestReportLedgerImportService.class.getDeclaredMethod(methodName, parameterTypes);
method.setAccessible(true);
return method.invoke(service, args);
} catch (InvocationTargetException exception) {
Throwable targetException = exception.getTargetException();
if (targetException instanceof RuntimeException) {
throw (RuntimeException) targetException;
}
throw new RuntimeException(targetException);
} catch (Exception exception) {
throw new RuntimeException(exception);
}
}
}
//package com.njcn.gather.aireport.testreport.component;
//
//import com.njcn.common.pojo.exception.BusinessException;
//import com.njcn.gather.aireport.clientunit.mapper.ClientUnitMapper;
//import com.njcn.gather.aireport.clientunit.pojo.po.ClientUnitPO;
//import com.njcn.gather.aireport.testdevice.mapper.TestDeviceMapper;
//import com.njcn.gather.aireport.testdevice.pojo.constant.TestDeviceConst;
//import com.njcn.gather.aireport.testdevice.pojo.po.TestDevicePO;
//import com.njcn.gather.aireport.testreport.mapper.TestReportGroupReportMapper;
//import com.njcn.gather.aireport.testreport.mapper.TestReportMapper;
//import com.njcn.gather.aireport.testreport.mapper.TestReportPointMapper;
//import com.njcn.gather.aireport.testreport.pojo.vo.TestReportLedgerValidateResultVO;
//import org.junit.jupiter.api.Assertions;
//import org.junit.jupiter.api.Test;
//import org.springframework.mock.web.MockMultipartFile;
//import org.springframework.web.multipart.MultipartFile;
//
//import java.util.Collections;
//import java.util.HashMap;
//import java.lang.reflect.InvocationTargetException;
//import java.lang.reflect.Method;
//import java.util.Map;
//
//import static org.mockito.Mockito.mock;
//
//class TestReportLedgerImportServiceTest {
//
// @Test
// void normalizeFileMapShouldRejectEmptyFiles() {
// TestReportLedgerImportService service = createService();
//
// BusinessException exception = Assertions.assertThrows(BusinessException.class,
// () -> invokeNormalizeFileMap(service, new MultipartFile[0]));
//
// Assertions.assertEquals("选择文件阶段:导入文件不能为空", exception.getMessage());
// }
//
// @Test
// void resolveDeviceStateShouldSupportCodesAndNames() {
// TestReportLedgerImportService service = createService();
//
// Assertions.assertEquals(TestDeviceConst.STATE_RUNNING, invokeResolveDeviceState(service, "01", 2));
// Assertions.assertEquals(TestDeviceConst.STATE_RUNNING, invokeResolveDeviceState(service, "在运", 2));
// Assertions.assertEquals(TestDeviceConst.STATE_RETIRED, invokeResolveDeviceState(service, "02", 3));
// Assertions.assertEquals(TestDeviceConst.STATE_RETIRED, invokeResolveDeviceState(service, "退运", 3));
// }
//
// @Test
// void resolveDeviceStateShouldRejectUnexpectedState() {
// TestReportLedgerImportService service = createService();
//
// BusinessException exception = Assertions.assertThrows(BusinessException.class,
// () -> invokeResolveDeviceState(service, "停用", 4));
//
// Assertions.assertEquals("基础信息维护阶段检测设备sheet第4行设备状态不正确", exception.getMessage());
// }
//
// @Test
// void normalizeFileMapShouldSkipEmptyFileItemsAndKeepTrimmedNames() {
// TestReportLedgerImportService service = createService();
// MockMultipartFile validFile = new MockMultipartFile("files", " point-a.xlsx ", "application/octet-stream",
// new byte[]{1});
// MockMultipartFile emptyFile = new MockMultipartFile("files", "empty.xlsx", "application/octet-stream",
// new byte[0]);
//
// Map<String, MultipartFile> result = invokeNormalizeFileMap(service, new MultipartFile[]{emptyFile, validFile});
//
// Assertions.assertEquals(1, result.size());
// Assertions.assertSame(validFile, result.get("point-a.xlsx"));
// }
//
// @Test
// void fillPointValidationResultShouldAppendReferenceErrorsToPointResult() {
// TestReportLedgerImportService service = createService();
// TestReportLedgerExcelParser.ParsedPointRow pointRow = new TestReportLedgerExcelParser.ParsedPointRow();
// pointRow.setRowNo(2);
// pointRow.setSubstationName("变电站A");
// pointRow.setPointName("监测点A");
// pointRow.setTestDeviceNo("DEV-001");
// pointRow.setClientUnitName("委托单位A");
// pointRow.setExcelAttachmentName("point-a.xlsx");
// pointRow.setWordAttachmentName("point-a.docx");
//
// TestReportLedgerExcelParser.ParsedLedger ledger = new TestReportLedgerExcelParser.ParsedLedger();
// ledger.setPoints(Collections.singletonList(pointRow));
//
// TestReportLedgerValidateResultVO result = new TestReportLedgerValidateResultVO();
//
// invokeFillPointValidationResult(service, result, ledger, Collections.singleton("台账信息汇总.xlsx"),
// new HashMap<String, TestDevicePO>(), new HashMap<String, ClientUnitPO>());
//
// Assertions.assertEquals(1, result.getPoints().size());
// Assertions.assertFalse(Boolean.TRUE.equals(result.getPoints().get(0).getComplete()));
// Assertions.assertFalse(Boolean.TRUE.equals(result.getPoints().get(0).getTestDeviceNoValid()));
// Assertions.assertEquals("第2行检测设备编号在数据库和检测设备sheet中都不存在",
// result.getPoints().get(0).getTestDeviceNoMessage());
// Assertions.assertFalse(Boolean.TRUE.equals(result.getPoints().get(0).getClientUnitNameValid()));
// Assertions.assertEquals("第2行委托单位名称在数据库和委托单位sheet中都不存在",
// result.getPoints().get(0).getClientUnitNameMessage());
// }
//
// private TestReportLedgerImportService createService() {
// return new TestReportLedgerImportService(
// mock(TestReportMapper.class),
// mock(TestReportPointMapper.class),
// mock(TestReportGroupReportMapper.class),
// mock(TestDeviceMapper.class),
// mock(ClientUnitMapper.class),
// mock(TestReportLedgerExcelParser.class),
// mock(TestReportStorageService.class));
// }
//
// @SuppressWarnings("unchecked")
// private Map<String, MultipartFile> invokeNormalizeFileMap(TestReportLedgerImportService service, MultipartFile[] files) {
// return invokePrivateMethod(service, "normalizeFileMap", new Class[]{MultipartFile[].class}, new Object[]{files});
// }
//
// private String invokeResolveDeviceState(TestReportLedgerImportService service, String stateName, Integer rowNo) {
// return (String) invokePrivateMethod(service, "resolveDeviceState",
// new Class[]{String.class, Integer.class}, new Object[]{stateName, rowNo});
// }
//
// private void invokeFillPointValidationResult(TestReportLedgerImportService service,
// TestReportLedgerValidateResultVO result,
// TestReportLedgerExcelParser.ParsedLedger ledger,
// java.util.Set<String> sessionFileNameSet,
// Map<String, TestDevicePO> dbDeviceMap,
// Map<String, ClientUnitPO> dbClientMap) {
// invokePrivateMethod(service, "fillPointValidationResult",
// new Class[]{TestReportLedgerValidateResultVO.class, TestReportLedgerExcelParser.ParsedLedger.class,
// java.util.Set.class, Map.class, Map.class},
// new Object[]{result, ledger, sessionFileNameSet, dbDeviceMap, dbClientMap});
// }
//
// private Object invokePrivateMethod(TestReportLedgerImportService service, String methodName,
// Class<?>[] parameterTypes, Object[] args) {
// try {
// Method method = TestReportLedgerImportService.class.getDeclaredMethod(methodName, parameterTypes);
// method.setAccessible(true);
// return method.invoke(service, args);
// } catch (InvocationTargetException exception) {
// Throwable targetException = exception.getTargetException();
// if (targetException instanceof RuntimeException) {
// throw (RuntimeException) targetException;
// }
// throw new RuntimeException(targetException);
// } catch (Exception exception) {
// throw new RuntimeException(exception);
// }
// }
//}

View File

@@ -1,497 +1,492 @@
package com.njcn.gather.aireport.testreport.service.impl;
import com.njcn.common.pojo.exception.BusinessException;
import com.njcn.gather.aireport.reportmodel.component.ReportModelFileStorageService;
import com.njcn.gather.aireport.reportmodel.service.ReportModelService;
import com.njcn.gather.aireport.testreport.component.TestReportLedgerImportService;
import com.njcn.gather.aireport.testreport.component.TestReportLedgerTemplateService;
import com.njcn.gather.aireport.testreport.component.TestReportPointExcelParser;
import com.njcn.gather.aireport.testreport.component.TestReportStorageService;
import com.njcn.gather.aireport.testreport.mapper.TestReportGroupReportMapper;
import com.njcn.gather.aireport.testreport.mapper.TestReportMapper;
import com.njcn.gather.aireport.testreport.mapper.TestReportPointMapper;
import com.njcn.gather.aireport.testreport.pojo.constant.TestReportConst;
import com.njcn.gather.aireport.testreport.pojo.param.TestReportParam;
import com.njcn.gather.aireport.testreport.pojo.po.TestReportGroupReportPO;
import com.njcn.gather.aireport.testreport.pojo.po.TestReportPO;
import com.njcn.gather.aireport.testreport.pojo.po.TestReportPointPO;
import com.njcn.gather.aireport.testreport.pojo.vo.TestReportLedgerValidateResultVO;
import com.njcn.gather.aireport.testreport.pojo.vo.TestReportPointVO;
import com.njcn.gather.aireport.testreport.pojo.vo.TestReportVO;
import com.njcn.gather.comservice.filepreview.FilePreviewService;
import com.njcn.gather.system.dictionary.pojo.po.DictData;
import com.njcn.gather.system.dictionary.pojo.po.DictType;
import com.njcn.gather.system.dictionary.service.IDictDataService;
import com.njcn.gather.system.dictionary.service.IDictTypeService;
import com.njcn.gather.system.pojo.constant.DictConst;
import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.Test;
import org.springframework.mock.web.MockMultipartFile;
import org.springframework.web.multipart.MultipartFile;
import java.io.InputStream;
import java.lang.reflect.Method;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.Arrays;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;
import java.util.UUID;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;
class TestReportServiceImplTest {
private static final Integer DICT_STATE_NORMAL = 1;
private static final Integer DICT_STATE_DELETED = 0;
@Test
void constantsShouldExposeGenerateStates() {
Assertions.assertEquals(Integer.valueOf(0), TestReportConst.REPORT_GENERATE_STATE_PENDING);
Assertions.assertEquals(Integer.valueOf(1), TestReportConst.REPORT_GENERATE_STATE_GENERATING);
Assertions.assertEquals(Integer.valueOf(2), TestReportConst.REPORT_GENERATE_STATE_FINISHED);
Assertions.assertEquals(Integer.valueOf(3), TestReportConst.GROUP_REPORT_GENERATE_STATE_FAILED);
}
@Test
void pointPoShouldStoreParsedFields() {
TestReportPointPO point = new TestReportPointPO();
point.setSubstationName("220kV城南站");
point.setPointName("1号监测点");
point.setVoltageLevel("10kV");
Assertions.assertEquals("220kV城南站", point.getSubstationName());
Assertions.assertEquals("1号监测点", point.getPointName());
Assertions.assertEquals("10kV", point.getVoltageLevel());
}
@Test
void groupReportPoShouldStoreResultFields() {
TestReportGroupReportPO groupReport = new TestReportGroupReportPO();
groupReport.setGroupNo(1);
groupReport.setReportName("第1组检测报告");
groupReport.setGenerateState(TestReportConst.GROUP_REPORT_GENERATE_STATE_PENDING);
Assertions.assertEquals(Integer.valueOf(1), groupReport.getGroupNo());
Assertions.assertEquals("第1组检测报告", groupReport.getReportName());
Assertions.assertEquals(TestReportConst.GROUP_REPORT_GENERATE_STATE_PENDING, groupReport.getGenerateState());
}
@Test
void serviceShouldExposeGroupedApis() throws Exception {
Assertions.assertNotNull(TestReportServiceImpl.class.getMethod("listTestReports", TestReportParam.QueryParam.class));
Assertions.assertNotNull(TestReportServiceImpl.class.getMethod("validateLedger", String.class, MultipartFile[].class));
Assertions.assertNotNull(TestReportServiceImpl.class.getMethod("importLedger", String.class,
TestReportParam.LedgerImportParam.class));
Assertions.assertNotNull(TestReportServiceImpl.class.getMethod("listPoints", String.class));
Assertions.assertNotNull(TestReportServiceImpl.class.getMethod("listGroupReports", String.class));
}
@Test
void pointVoShouldExposeDeviceName() throws Exception {
Assertions.assertNotNull(TestReportPointVO.class.getMethod("getDeviceName"));
}
@Test
void parserShouldReadNodeInfoSheetFromSampleXls() throws Exception {
TestReportPointExcelParser parser = new TestReportPointExcelParser();
try (InputStream inputStream = Files.newInputStream(
Paths.get("ai-report/test-report/src/恒谊_20260610110840_202606101108_202606110957.xls"))) {
TestReportPointExcelParser.ParsedPoint point = parser.parse(
"恒谊_20260610110840_202606101108_202606110957.xls", inputStream);
Assertions.assertNotNull(point.getPointName());
Assertions.assertNotNull(point.getSubstationName());
}
}
@Test
void importLedgerShouldRejectGeneratingTask() {
TestReportPO report = new TestReportPO();
report.setId("report-001");
report.setStatus(TestReportConst.STATUS_NORMAL);
report.setReportGenerateState(TestReportConst.REPORT_GENERATE_STATE_GENERATING);
TestableTestReportServiceImpl service = new TestableTestReportServiceImpl(mock(IDictTypeService.class),
mock(IDictDataService.class), report);
MockMultipartFile file = new MockMultipartFile("files", "sample.xls", "application/vnd.ms-excel", new byte[]{1});
BusinessException exception = Assertions.assertThrows(BusinessException.class,
() -> service.validateLedger("report-001", new MultipartFile[]{file}));
Assertions.assertTrue(exception.getMessage().contains("未生成"));
}
@Test
void importLedgerShouldRejectMissingBatchId() {
TestReportServiceImpl service = createService(mock(IDictTypeService.class), mock(IDictDataService.class));
TestReportParam.LedgerImportParam param = new TestReportParam.LedgerImportParam();
BusinessException exception = Assertions.assertThrows(BusinessException.class,
() -> service.importLedger("report-001", param));
Assertions.assertNotNull(exception.getMessage());
}
@Test
void addTestReportShouldRejectMissingNo() {
TestReportServiceImpl service = createService(mock(IDictTypeService.class), mock(IDictDataService.class));
TestReportParam.AddParam param = new TestReportParam.AddParam();
param.setClientUnitId("client-1");
param.setReportModelId("model-1");
param.setCreateUnit("unit-1");
param.setContractNumber("HT-001");
param.setStandard("[\"std-1\"]");
param.setData("{}");
BusinessException exception = Assertions.assertThrows(BusinessException.class,
() -> service.addTestReport(param));
Assertions.assertNotNull(exception.getMessage());
}
@Test
void submitAuditShouldRejectMissingId() {
TestReportServiceImpl service = createService(mock(IDictTypeService.class), mock(IDictDataService.class));
TestReportParam.SubmitAuditParam param = new TestReportParam.SubmitAuditParam();
param.setCheckResult("approved");
BusinessException exception = Assertions.assertThrows(BusinessException.class,
() -> service.submitAudit(param));
Assertions.assertNotNull(exception.getMessage());
}
@Test
void addTestReportShouldRejectInvalidStandardJsonArray() {
TestReportServiceImpl service = createService(mock(IDictTypeService.class), mock(IDictDataService.class));
TestReportParam.AddParam param = new TestReportParam.AddParam();
param.setNo("PT-001");
param.setClientUnitId("client-1");
param.setReportModelId("model-1");
param.setCreateUnit("company-1");
param.setContractNumber("HT-001");
param.setStandard("not-json-array");
param.setData("{}");
BusinessException exception = Assertions.assertThrows(BusinessException.class,
() -> service.addTestReport(param));
Assertions.assertTrue(exception.getMessage().contains("测试依据"));
}
@Test
void addTestReportShouldPreferFrontendIdWhenProvided() {
TestableAddTestReportServiceImpl service = createAddService();
TestReportParam.AddParam param = buildValidNormalizeParam();
param.setId("report-frontend-001");
boolean result = service.addTestReport(param);
Assertions.assertTrue(result);
Assertions.assertNotNull(service.savedReport);
Assertions.assertEquals("report-frontend-001", service.savedReport.getId());
}
@Test
void addTestReportShouldGenerateUuidWhenFrontendIdMissing() {
TestableAddTestReportServiceImpl service = createAddService();
TestReportParam.AddParam param = buildValidNormalizeParam();
boolean result = service.addTestReport(param);
Assertions.assertTrue(result);
Assertions.assertNotNull(service.savedReport);
Assertions.assertNotNull(service.savedReport.getId());
Assertions.assertDoesNotThrow(() -> UUID.fromString(service.savedReport.getId()));
}
@Test
void addTestReportShouldRejectInvalidDataJson() {
TestReportServiceImpl service = createService(mock(IDictTypeService.class), mock(IDictDataService.class));
TestReportParam.AddParam param = buildValidNormalizeParam();
param.setData("not-json");
BusinessException exception = Assertions.assertThrows(BusinessException.class,
() -> invokeNormalizeParam(service, param));
Assertions.assertTrue(exception.getMessage().contains("受资数据"));
}
@Test
void addTestReportShouldRejectScalarDataJson() {
TestReportServiceImpl service = createService(mock(IDictTypeService.class), mock(IDictDataService.class));
TestReportParam.AddParam param = buildValidNormalizeParam();
param.setData("\"plain-text\"");
BusinessException exception = Assertions.assertThrows(BusinessException.class,
() -> invokeNormalizeParam(service, param));
Assertions.assertTrue(exception.getMessage().contains("受资数据"));
}
@Test
void normalizeParamShouldAcceptJsonObjectAndJsonArrayData() {
TestReportServiceImpl service = createService(mock(IDictTypeService.class), mock(IDictDataService.class));
TestReportParam.AddParam objectParam = buildValidNormalizeParam();
objectParam.setData("{\"fileName\":\"report.json\"}");
TestReportParam.AddParam arrayParam = buildValidNormalizeParam();
arrayParam.setData("[{\"fileName\":\"report-a.json\"}]");
Assertions.assertDoesNotThrow(() -> invokeNormalizeParam(service, objectParam));
Assertions.assertDoesNotThrow(() -> invokeNormalizeParam(service, arrayParam));
}
@Test
void addTestReportShouldRejectDeletedStandardDictData() {
IDictTypeService dictTypeService = mock(IDictTypeService.class);
IDictDataService dictDataService = mock(IDictDataService.class);
TestReportServiceImpl service = createService(dictTypeService, dictDataService);
mockNormalDictType(dictTypeService, DictConst.TEST_REPORT_COMPANY_DICT_CODE, "type-company");
mockNormalDictType(dictTypeService, DictConst.TEST_REPORT_STANDARD_DICT_CODE, "type-standard");
when(dictDataService.listDictDataByTypeId("type-company")).thenReturn(Collections.singletonList(
buildDictData("company-1", "company-A", DICT_STATE_NORMAL)));
when(dictDataService.listDictDataByTypeId("type-standard")).thenReturn(Collections.singletonList(
buildDictData("std-1", "standard-A", DICT_STATE_DELETED)));
TestReportParam.AddParam param = new TestReportParam.AddParam();
param.setNo("PT-001");
param.setClientUnitId("client-1");
param.setReportModelId("model-1");
param.setCreateUnit("company-1");
param.setContractNumber("HT-001");
param.setStandard("[\"std-1\"]");
param.setData("{}");
BusinessException exception = Assertions.assertThrows(BusinessException.class,
() -> service.addTestReport(param));
Assertions.assertTrue(exception.getMessage().contains("测试依据"));
}
@Test
void fillDisplayUserNamesShouldPreferUserNameFallbackToUserIdAndDash() {
TestReportServiceImpl service = createService(mock(IDictTypeService.class), mock(IDictDataService.class));
TestReportVO first = new TestReportVO();
first.setCreateBy("creator-001");
first.setUpdateBy("updater-001");
first.setCreateByName("creator-A");
first.setUpdateByName("updater-A");
TestReportVO second = new TestReportVO();
second.setCreateBy("creator-002");
second.setUpdateBy(null);
Map<String, String> userNameMap = new HashMap<String, String>();
userNameMap.put("creator-001", "creator-A");
userNameMap.put("updater-001", "updater-A");
service.fillDisplayUserNames(first, userNameMap);
service.fillDisplayUserNames(second, userNameMap);
Assertions.assertEquals("creator-A", first.getCreateBy());
Assertions.assertEquals("updater-A", first.getUpdateBy());
Assertions.assertEquals("creator-002", second.getCreateBy());
Assertions.assertEquals("-", second.getUpdateBy());
}
@Test
void fillDisplayFieldsShouldResolveDictNamesAndFallbackToRawValue() {
IDictTypeService dictTypeService = mock(IDictTypeService.class);
IDictDataService dictDataService = mock(IDictDataService.class);
TestReportServiceImpl service = createService(dictTypeService, dictDataService);
mockNormalDictType(dictTypeService, DictConst.TEST_REPORT_COMPANY_DICT_CODE, "type-company");
mockNormalDictType(dictTypeService, DictConst.TEST_REPORT_STANDARD_DICT_CODE, "type-standard");
when(dictDataService.listDictDataByTypeId("type-company")).thenReturn(Collections.singletonList(
buildDictData("company-1", "Nanjing Zhiyou", DICT_STATE_NORMAL)));
when(dictDataService.listDictDataByTypeId("type-standard")).thenReturn(Arrays.asList(
buildDictData("std-1", "GB 50150-2016", DICT_STATE_NORMAL),
buildDictData("std-2", "DL/T 596-2021", DICT_STATE_NORMAL)));
TestReportVO first = new TestReportVO();
first.setCreateUnit("company-1");
first.setStandard("[\"std-1\",\"std-2\"]");
TestReportVO second = new TestReportVO();
second.setCreateUnit("company-unknown");
second.setStandard("[\"std-unknown\"]");
service.fillDisplayFields(Arrays.asList(first, second));
Assertions.assertEquals("Nanjing Zhiyou", first.getCreateUnit());
Assertions.assertEquals("GB 50150-2016, DL/T 596-2021", first.getStandard());
Assertions.assertEquals("company-unknown", second.getCreateUnit());
Assertions.assertEquals("std-unknown", second.getStandard());
}
@Test
void fillDisplayFieldsShouldIgnoreDeletedDictData() {
IDictTypeService dictTypeService = mock(IDictTypeService.class);
IDictDataService dictDataService = mock(IDictDataService.class);
TestReportServiceImpl service = createService(dictTypeService, dictDataService);
mockNormalDictType(dictTypeService, DictConst.TEST_REPORT_COMPANY_DICT_CODE, "type-company");
mockNormalDictType(dictTypeService, DictConst.TEST_REPORT_STANDARD_DICT_CODE, "type-standard");
when(dictDataService.listDictDataByTypeId("type-company")).thenReturn(Collections.singletonList(
buildDictData("company-1", "company-A", DICT_STATE_DELETED)));
when(dictDataService.listDictDataByTypeId("type-standard")).thenReturn(Collections.singletonList(
buildDictData("std-1", "standard-A", DICT_STATE_DELETED)));
TestReportVO report = new TestReportVO();
report.setCreateUnit("company-1");
report.setStandard("[\"std-1\"]");
service.fillDisplayFields(Collections.singletonList(report));
Assertions.assertEquals("company-1", report.getCreateUnit());
Assertions.assertEquals("std-1", report.getStandard());
}
@Test
void submitAuditShouldMoveEditingStateToAuditingAndClearAuditFields() {
TestReportPO report = new TestReportPO();
report.setId("report-001");
report.setState(TestReportConst.STATE_EDITING);
report.setStatus(TestReportConst.STATUS_NORMAL);
report.setCheckId("checker-001");
report.setCheckTime(java.time.LocalDate.of(2026, 6, 26));
report.setCheckResult("done");
TestableTestReportServiceImpl service = new TestableTestReportServiceImpl(mock(IDictTypeService.class),
mock(IDictDataService.class), report);
TestReportParam.SubmitAuditParam param = new TestReportParam.SubmitAuditParam();
param.setId("report-001");
boolean result = service.submitAudit(param);
Assertions.assertTrue(result);
Assertions.assertEquals(TestReportConst.STATE_AUDITING, report.getState());
Assertions.assertNull(report.getCheckId());
Assertions.assertNull(report.getCheckTime());
Assertions.assertNull(report.getCheckResult());
Assertions.assertSame(report, service.updatedReport);
}
private TestReportServiceImpl createService(IDictTypeService dictTypeService, IDictDataService dictDataService) {
return new TestReportServiceImpl(
mock(ReportModelService.class),
mock(ReportModelFileStorageService.class),
mock(FilePreviewService.class),
dictTypeService,
dictDataService,
mock(TestReportPointMapper.class),
mock(TestReportGroupReportMapper.class),
mock(TestReportPointExcelParser.class),
mock(TestReportStorageService.class),
mock(TestReportLedgerImportService.class),
mock(TestReportLedgerTemplateService.class));
}
private TestableAddTestReportServiceImpl createAddService() {
IDictTypeService dictTypeService = mock(IDictTypeService.class);
IDictDataService dictDataService = mock(IDictDataService.class);
TestReportMapper testReportMapper = mock(TestReportMapper.class);
mockNormalDictType(dictTypeService, DictConst.TEST_REPORT_COMPANY_DICT_CODE, "type-company");
mockNormalDictType(dictTypeService, DictConst.TEST_REPORT_STANDARD_DICT_CODE, "type-standard");
when(dictDataService.listDictDataByTypeId("type-company")).thenReturn(Collections.singletonList(
buildDictData("company-1", "company-A", DICT_STATE_NORMAL)));
when(dictDataService.listDictDataByTypeId("type-standard")).thenReturn(Collections.singletonList(
buildDictData("std-1", "standard-A", DICT_STATE_NORMAL)));
when(testReportMapper.countNormalClientUnit("client-1")).thenReturn(1);
when(testReportMapper.countNormalReportModel("model-1")).thenReturn(1);
return new TestableAddTestReportServiceImpl(dictTypeService, dictDataService, testReportMapper);
}
private TestReportParam.AddParam buildValidNormalizeParam() {
TestReportParam.AddParam param = new TestReportParam.AddParam();
param.setNo("PT-001");
param.setClientUnitId("client-1");
param.setReportModelId("model-1");
param.setCreateUnit("company-1");
param.setContractNumber("HT-001");
param.setStandard("[\"std-1\"]");
param.setData("{}");
return param;
}
private Object invokeNormalizeParam(TestReportServiceImpl service, TestReportParam.AddParam param) {
try {
Method method = TestReportServiceImpl.class.getDeclaredMethod("normalizeParam", TestReportParam.AddParam.class);
method.setAccessible(true);
return method.invoke(service, param);
} catch (java.lang.reflect.InvocationTargetException exception) {
Throwable targetException = exception.getTargetException();
if (targetException instanceof RuntimeException) {
throw (RuntimeException) targetException;
}
throw new RuntimeException(targetException);
} catch (Exception exception) {
throw new RuntimeException(exception);
}
}
private static class TestableTestReportServiceImpl extends TestReportServiceImpl {
private final TestReportPO report;
private TestReportPO updatedReport;
private TestableTestReportServiceImpl(IDictTypeService dictTypeService, IDictDataService dictDataService,
TestReportPO report) {
super(mock(ReportModelService.class), mock(ReportModelFileStorageService.class),
mock(FilePreviewService.class), dictTypeService, dictDataService,
mock(TestReportPointMapper.class), mock(TestReportGroupReportMapper.class),
mock(TestReportPointExcelParser.class), mock(TestReportStorageService.class),
mock(TestReportLedgerImportService.class), mock(TestReportLedgerTemplateService.class));
this.report = report;
}
@Override
public TestReportPO requireNormal(String id) {
return report;
}
@Override
public boolean updateById(TestReportPO entity) {
this.updatedReport = entity;
return true;
}
}
private static class TestableAddTestReportServiceImpl extends TestReportServiceImpl {
private TestReportPO savedReport;
private TestableAddTestReportServiceImpl(IDictTypeService dictTypeService, IDictDataService dictDataService,
TestReportMapper testReportMapper) {
super(mock(ReportModelService.class), mock(ReportModelFileStorageService.class),
mock(FilePreviewService.class), dictTypeService, dictDataService,
mock(TestReportPointMapper.class), mock(TestReportGroupReportMapper.class),
mock(TestReportPointExcelParser.class), mock(TestReportStorageService.class),
mock(TestReportLedgerImportService.class), mock(TestReportLedgerTemplateService.class));
this.baseMapper = testReportMapper;
}
@Override
public long count(com.baomidou.mybatisplus.core.conditions.Wrapper<TestReportPO> queryWrapper) {
return 0;
}
@Override
public boolean save(TestReportPO entity) {
this.savedReport = entity;
return true;
}
}
private void mockNormalDictType(IDictTypeService dictTypeService, String code, String typeId) {
when(dictTypeService.getByCode(code)).thenReturn(buildDictType(typeId, code));
}
private DictType buildDictType(String id, String code) {
DictType dictType = new DictType();
dictType.setId(id);
dictType.setCode(code);
dictType.setState(DICT_STATE_NORMAL);
return dictType;
}
private DictData buildDictData(String id, String name, Integer state) {
DictData dictData = new DictData();
dictData.setId(id);
dictData.setName(name);
dictData.setState(state);
return dictData;
}
}
//package com.njcn.gather.aireport.testreport.service.impl;
//
//import com.njcn.common.pojo.exception.BusinessException;
//import com.njcn.gather.aireport.reportmodel.component.ReportModelFileStorageService;
//import com.njcn.gather.aireport.reportmodel.service.ReportModelService;
//import com.njcn.gather.aireport.testreport.component.TestReportLedgerImportService;
//import com.njcn.gather.aireport.testreport.component.TestReportLedgerTemplateService;
//import com.njcn.gather.aireport.testreport.component.TestReportPointExcelParser;
//import com.njcn.gather.aireport.testreport.component.TestReportStorageService;
//import com.njcn.gather.aireport.testreport.mapper.TestReportGroupReportMapper;
//import com.njcn.gather.aireport.testreport.mapper.TestReportMapper;
//import com.njcn.gather.aireport.testreport.mapper.TestReportPointMapper;
//import com.njcn.gather.aireport.testreport.pojo.constant.TestReportConst;
//import com.njcn.gather.aireport.testreport.pojo.param.TestReportParam;
//import com.njcn.gather.aireport.testreport.pojo.po.TestReportGroupReportPO;
//import com.njcn.gather.aireport.testreport.pojo.po.TestReportPO;
//import com.njcn.gather.aireport.testreport.pojo.po.TestReportPointPO;
//import com.njcn.gather.aireport.testreport.pojo.vo.TestReportLedgerValidateResultVO;
//import com.njcn.gather.aireport.testreport.pojo.vo.TestReportPointVO;
//import com.njcn.gather.aireport.testreport.pojo.vo.TestReportVO;
//import com.njcn.gather.comservice.filepreview.FilePreviewService;
//import com.njcn.gather.system.dictionary.pojo.po.DictData;
//import com.njcn.gather.system.dictionary.pojo.po.DictType;
//import com.njcn.gather.system.dictionary.service.IDictDataService;
//import com.njcn.gather.system.dictionary.service.IDictTypeService;
//import com.njcn.gather.system.pojo.constant.DictConst;
//import org.junit.jupiter.api.Assertions;
//import org.junit.jupiter.api.Test;
//import org.springframework.mock.web.MockMultipartFile;
//import org.springframework.web.multipart.MultipartFile;
//
//import java.io.InputStream;
//import java.lang.reflect.Method;
//import java.nio.file.Files;
//import java.nio.file.Paths;
//import java.util.Arrays;
//import java.util.Collections;
//import java.util.HashMap;
//import java.util.Map;
//import java.util.UUID;
//
//import static org.mockito.Mockito.mock;
//import static org.mockito.Mockito.verify;
//import static org.mockito.Mockito.when;
//
//class TestReportServiceImplTest {
//
// private static final Integer DICT_STATE_NORMAL = 1;
// private static final Integer DICT_STATE_DELETED = 0;
//
// @Test
// void constantsShouldExposeGenerateStates() {
// Assertions.assertEquals(Integer.valueOf(0), TestReportConst.REPORT_GENERATE_STATE_PENDING);
// Assertions.assertEquals(Integer.valueOf(1), TestReportConst.REPORT_GENERATE_STATE_GENERATING);
// Assertions.assertEquals(Integer.valueOf(2), TestReportConst.REPORT_GENERATE_STATE_FINISHED);
// Assertions.assertEquals(Integer.valueOf(3), TestReportConst.GROUP_REPORT_GENERATE_STATE_FAILED);
// }
//
// @Test
// void pointPoShouldStoreParsedFields() {
// TestReportPointPO point = new TestReportPointPO();
// point.setSubstationName("220kV城南站");
// point.setPointName("1号监测点");
// point.setVoltageLevel("10kV");
//
// Assertions.assertEquals("220kV城南站", point.getSubstationName());
// Assertions.assertEquals("1号监测点", point.getPointName());
// Assertions.assertEquals("10kV", point.getVoltageLevel());
// }
//
// @Test
// void groupReportPoShouldStoreResultFields() {
// TestReportGroupReportPO groupReport = new TestReportGroupReportPO();
// groupReport.setGroupNo(1);
// groupReport.setReportName("第1组检测报告");
// groupReport.setGenerateState(TestReportConst.GROUP_REPORT_GENERATE_STATE_PENDING);
//
// Assertions.assertEquals(Integer.valueOf(1), groupReport.getGroupNo());
// Assertions.assertEquals("第1组检测报告", groupReport.getReportName());
// Assertions.assertEquals(TestReportConst.GROUP_REPORT_GENERATE_STATE_PENDING, groupReport.getGenerateState());
// }
//
// @Test
// void serviceShouldExposeGroupedApis() throws Exception {
// Assertions.assertNotNull(TestReportServiceImpl.class.getMethod("listTestReports", TestReportParam.QueryParam.class));
// Assertions.assertNotNull(TestReportServiceImpl.class.getMethod("validateLedger", String.class, MultipartFile[].class));
// Assertions.assertNotNull(TestReportServiceImpl.class.getMethod("importLedger", String.class,
// TestReportParam.LedgerImportParam.class));
// Assertions.assertNotNull(TestReportServiceImpl.class.getMethod("listPoints", String.class));
// Assertions.assertNotNull(TestReportServiceImpl.class.getMethod("listGroupReports", String.class));
// }
//
// @Test
// void parserShouldReadNodeInfoSheetFromSampleXls() throws Exception {
// TestReportPointExcelParser parser = new TestReportPointExcelParser();
// try (InputStream inputStream = Files.newInputStream(
// Paths.get("ai-report/test-report/src/恒谊_20260610110840_202606101108_202606110957.xls"))) {
// TestReportPointExcelParser.ParsedPoint point = parser.parse(
// "恒谊_20260610110840_202606101108_202606110957.xls", inputStream);
// Assertions.assertNotNull(point.getPointName());
// Assertions.assertNotNull(point.getSubstationName());
// }
// }
//
// @Test
// void importLedgerShouldRejectGeneratingTask() {
// TestReportPO report = new TestReportPO();
// report.setId("report-001");
// report.setStatus(TestReportConst.STATUS_NORMAL);
// report.setReportGenerateState(TestReportConst.REPORT_GENERATE_STATE_GENERATING);
// TestableTestReportServiceImpl service = new TestableTestReportServiceImpl(mock(IDictTypeService.class),
// mock(IDictDataService.class), report);
// MockMultipartFile file = new MockMultipartFile("files", "sample.xls", "application/vnd.ms-excel", new byte[]{1});
//
// BusinessException exception = Assertions.assertThrows(BusinessException.class,
// () -> service.validateLedger("report-001", new MultipartFile[]{file}));
//
// Assertions.assertTrue(exception.getMessage().contains("未生成"));
// }
//
// @Test
// void importLedgerShouldRejectMissingBatchId() {
// TestReportServiceImpl service = createService(mock(IDictTypeService.class), mock(IDictDataService.class));
// TestReportParam.LedgerImportParam param = new TestReportParam.LedgerImportParam();
//
// BusinessException exception = Assertions.assertThrows(BusinessException.class,
// () -> service.importLedger("report-001", param));
//
// Assertions.assertNotNull(exception.getMessage());
// }
//
// @Test
// void addTestReportShouldRejectMissingNo() {
// TestReportServiceImpl service = createService(mock(IDictTypeService.class), mock(IDictDataService.class));
// TestReportParam.AddParam param = new TestReportParam.AddParam();
// param.setClientUnitId("client-1");
// param.setReportModelId("model-1");
// param.setCreateUnit("unit-1");
// param.setContractNumber("HT-001");
// param.setStandard("[\"std-1\"]");
// param.setData("{}");
//
// BusinessException exception = Assertions.assertThrows(BusinessException.class,
// () -> service.addTestReport(param));
//
// Assertions.assertNotNull(exception.getMessage());
// }
//
// @Test
// void submitAuditShouldRejectMissingId() {
// TestReportServiceImpl service = createService(mock(IDictTypeService.class), mock(IDictDataService.class));
// TestReportParam.SubmitAuditParam param = new TestReportParam.SubmitAuditParam();
// param.setCheckResult("approved");
//
// BusinessException exception = Assertions.assertThrows(BusinessException.class,
// () -> service.submitAudit(param));
//
// Assertions.assertNotNull(exception.getMessage());
// }
//
// @Test
// void addTestReportShouldRejectInvalidStandardJsonArray() {
// TestReportServiceImpl service = createService(mock(IDictTypeService.class), mock(IDictDataService.class));
// TestReportParam.AddParam param = new TestReportParam.AddParam();
// param.setNo("PT-001");
// param.setClientUnitId("client-1");
// param.setReportModelId("model-1");
// param.setCreateUnit("company-1");
// param.setContractNumber("HT-001");
// param.setStandard("not-json-array");
// param.setData("{}");
//
// BusinessException exception = Assertions.assertThrows(BusinessException.class,
// () -> service.addTestReport(param));
//
// Assertions.assertTrue(exception.getMessage().contains("测试依据"));
// }
//
// @Test
// void addTestReportShouldPreferFrontendIdWhenProvided() {
// TestableAddTestReportServiceImpl service = createAddService();
// TestReportParam.AddParam param = buildValidNormalizeParam();
// param.setId("report-frontend-001");
//
// boolean result = service.addTestReport(param);
//
// Assertions.assertTrue(result);
// Assertions.assertNotNull(service.savedReport);
// Assertions.assertEquals("report-frontend-001", service.savedReport.getId());
// }
//
// @Test
// void addTestReportShouldGenerateUuidWhenFrontendIdMissing() {
// TestableAddTestReportServiceImpl service = createAddService();
// TestReportParam.AddParam param = buildValidNormalizeParam();
//
// boolean result = service.addTestReport(param);
//
// Assertions.assertTrue(result);
// Assertions.assertNotNull(service.savedReport);
// Assertions.assertNotNull(service.savedReport.getId());
// Assertions.assertDoesNotThrow(() -> UUID.fromString(service.savedReport.getId()));
// }
//
// @Test
// void addTestReportShouldRejectInvalidDataJson() {
// TestReportServiceImpl service = createService(mock(IDictTypeService.class), mock(IDictDataService.class));
// TestReportParam.AddParam param = buildValidNormalizeParam();
// param.setData("not-json");
//
// BusinessException exception = Assertions.assertThrows(BusinessException.class,
// () -> invokeNormalizeParam(service, param));
//
// Assertions.assertTrue(exception.getMessage().contains("受资数据"));
// }
//
// @Test
// void addTestReportShouldRejectScalarDataJson() {
// TestReportServiceImpl service = createService(mock(IDictTypeService.class), mock(IDictDataService.class));
// TestReportParam.AddParam param = buildValidNormalizeParam();
// param.setData("\"plain-text\"");
//
// BusinessException exception = Assertions.assertThrows(BusinessException.class,
// () -> invokeNormalizeParam(service, param));
//
// Assertions.assertTrue(exception.getMessage().contains("受资数据"));
// }
//
// @Test
// void normalizeParamShouldAcceptJsonObjectAndJsonArrayData() {
// TestReportServiceImpl service = createService(mock(IDictTypeService.class), mock(IDictDataService.class));
// TestReportParam.AddParam objectParam = buildValidNormalizeParam();
// objectParam.setData("{\"fileName\":\"report.json\"}");
// TestReportParam.AddParam arrayParam = buildValidNormalizeParam();
// arrayParam.setData("[{\"fileName\":\"report-a.json\"}]");
//
// Assertions.assertDoesNotThrow(() -> invokeNormalizeParam(service, objectParam));
// Assertions.assertDoesNotThrow(() -> invokeNormalizeParam(service, arrayParam));
// }
//
// @Test
// void addTestReportShouldRejectDeletedStandardDictData() {
// IDictTypeService dictTypeService = mock(IDictTypeService.class);
// IDictDataService dictDataService = mock(IDictDataService.class);
// TestReportServiceImpl service = createService(dictTypeService, dictDataService);
// mockNormalDictType(dictTypeService, DictConst.TEST_REPORT_COMPANY_DICT_CODE, "type-company");
// mockNormalDictType(dictTypeService, DictConst.TEST_REPORT_STANDARD_DICT_CODE, "type-standard");
// when(dictDataService.listDictDataByTypeId("type-company")).thenReturn(Collections.singletonList(
// buildDictData("company-1", "company-A", DICT_STATE_NORMAL)));
// when(dictDataService.listDictDataByTypeId("type-standard")).thenReturn(Collections.singletonList(
// buildDictData("std-1", "standard-A", DICT_STATE_DELETED)));
// TestReportParam.AddParam param = new TestReportParam.AddParam();
// param.setNo("PT-001");
// param.setClientUnitId("client-1");
// param.setReportModelId("model-1");
// param.setCreateUnit("company-1");
// param.setContractNumber("HT-001");
// param.setStandard("[\"std-1\"]");
// param.setData("{}");
//
// BusinessException exception = Assertions.assertThrows(BusinessException.class,
// () -> service.addTestReport(param));
//
// Assertions.assertTrue(exception.getMessage().contains("测试依据"));
// }
//
// @Test
// void fillDisplayUserNamesShouldPreferUserNameFallbackToUserIdAndDash() {
// TestReportServiceImpl service = createService(mock(IDictTypeService.class), mock(IDictDataService.class));
// TestReportVO first = new TestReportVO();
// first.setCreateBy("creator-001");
// first.setUpdateBy("updater-001");
// first.setCreateByName("creator-A");
// first.setUpdateByName("updater-A");
// TestReportVO second = new TestReportVO();
// second.setCreateBy("creator-002");
// second.setUpdateBy(null);
//
// Map<String, String> userNameMap = new HashMap<String, String>();
// userNameMap.put("creator-001", "creator-A");
// userNameMap.put("updater-001", "updater-A");
//
// service.fillDisplayUserNames(first, userNameMap);
// service.fillDisplayUserNames(second, userNameMap);
//
// Assertions.assertEquals("creator-A", first.getCreateBy());
// Assertions.assertEquals("updater-A", first.getUpdateBy());
// Assertions.assertEquals("creator-002", second.getCreateBy());
// Assertions.assertEquals("-", second.getUpdateBy());
// }
//
// @Test
// void fillDisplayFieldsShouldResolveDictNamesAndFallbackToRawValue() {
// IDictTypeService dictTypeService = mock(IDictTypeService.class);
// IDictDataService dictDataService = mock(IDictDataService.class);
// TestReportServiceImpl service = createService(dictTypeService, dictDataService);
// mockNormalDictType(dictTypeService, DictConst.TEST_REPORT_COMPANY_DICT_CODE, "type-company");
// mockNormalDictType(dictTypeService, DictConst.TEST_REPORT_STANDARD_DICT_CODE, "type-standard");
// when(dictDataService.listDictDataByTypeId("type-company")).thenReturn(Collections.singletonList(
// buildDictData("company-1", "Nanjing Zhiyou", DICT_STATE_NORMAL)));
// when(dictDataService.listDictDataByTypeId("type-standard")).thenReturn(Arrays.asList(
// buildDictData("std-1", "GB 50150-2016", DICT_STATE_NORMAL),
// buildDictData("std-2", "DL/T 596-2021", DICT_STATE_NORMAL)));
// TestReportVO first = new TestReportVO();
// first.setCreateUnit("company-1");
// first.setStandard("[\"std-1\",\"std-2\"]");
// TestReportVO second = new TestReportVO();
// second.setCreateUnit("company-unknown");
// second.setStandard("[\"std-unknown\"]");
//
// service.fillDisplayFields(Arrays.asList(first, second));
//
// Assertions.assertEquals("Nanjing Zhiyou", first.getCreateUnit());
// Assertions.assertEquals("GB 50150-2016, DL/T 596-2021", first.getStandard());
// Assertions.assertEquals("company-unknown", second.getCreateUnit());
// Assertions.assertEquals("std-unknown", second.getStandard());
// }
//
// @Test
// void fillDisplayFieldsShouldIgnoreDeletedDictData() {
// IDictTypeService dictTypeService = mock(IDictTypeService.class);
// IDictDataService dictDataService = mock(IDictDataService.class);
// TestReportServiceImpl service = createService(dictTypeService, dictDataService);
// mockNormalDictType(dictTypeService, DictConst.TEST_REPORT_COMPANY_DICT_CODE, "type-company");
// mockNormalDictType(dictTypeService, DictConst.TEST_REPORT_STANDARD_DICT_CODE, "type-standard");
// when(dictDataService.listDictDataByTypeId("type-company")).thenReturn(Collections.singletonList(
// buildDictData("company-1", "company-A", DICT_STATE_DELETED)));
// when(dictDataService.listDictDataByTypeId("type-standard")).thenReturn(Collections.singletonList(
// buildDictData("std-1", "standard-A", DICT_STATE_DELETED)));
// TestReportVO report = new TestReportVO();
// report.setCreateUnit("company-1");
// report.setStandard("[\"std-1\"]");
//
// service.fillDisplayFields(Collections.singletonList(report));
//
// Assertions.assertEquals("company-1", report.getCreateUnit());
// Assertions.assertEquals("std-1", report.getStandard());
// }
//
// @Test
// void submitAuditShouldMoveEditingStateToAuditingAndClearAuditFields() {
// TestReportPO report = new TestReportPO();
// report.setId("report-001");
// report.setState(TestReportConst.STATE_EDITING);
// report.setStatus(TestReportConst.STATUS_NORMAL);
// report.setCheckId("checker-001");
// report.setCheckTime(java.time.LocalDate.of(2026, 6, 26));
// report.setCheckResult("done");
// TestableTestReportServiceImpl service = new TestableTestReportServiceImpl(mock(IDictTypeService.class),
// mock(IDictDataService.class), report);
// TestReportParam.SubmitAuditParam param = new TestReportParam.SubmitAuditParam();
// param.setId("report-001");
//
// boolean result = service.submitAudit(param);
//
// Assertions.assertTrue(result);
// Assertions.assertEquals(TestReportConst.STATE_AUDITING, report.getState());
// Assertions.assertNull(report.getCheckId());
// Assertions.assertNull(report.getCheckTime());
// Assertions.assertNull(report.getCheckResult());
// Assertions.assertSame(report, service.updatedReport);
// }
//
// private TestReportServiceImpl createService(IDictTypeService dictTypeService, IDictDataService dictDataService) {
// return new TestReportServiceImpl(
// mock(ReportModelService.class),
// mock(ReportModelFileStorageService.class),
// mock(FilePreviewService.class),
// dictTypeService,
// dictDataService,
// mock(TestReportPointMapper.class),
// mock(TestReportGroupReportMapper.class),
// mock(TestReportPointExcelParser.class),
// mock(TestReportStorageService.class),
// mock(TestReportLedgerImportService.class),
// mock(TestReportLedgerTemplateService.class));
// }
//
// private TestableAddTestReportServiceImpl createAddService() {
// IDictTypeService dictTypeService = mock(IDictTypeService.class);
// IDictDataService dictDataService = mock(IDictDataService.class);
// TestReportMapper testReportMapper = mock(TestReportMapper.class);
// mockNormalDictType(dictTypeService, DictConst.TEST_REPORT_COMPANY_DICT_CODE, "type-company");
// mockNormalDictType(dictTypeService, DictConst.TEST_REPORT_STANDARD_DICT_CODE, "type-standard");
// when(dictDataService.listDictDataByTypeId("type-company")).thenReturn(Collections.singletonList(
// buildDictData("company-1", "company-A", DICT_STATE_NORMAL)));
// when(dictDataService.listDictDataByTypeId("type-standard")).thenReturn(Collections.singletonList(
// buildDictData("std-1", "standard-A", DICT_STATE_NORMAL)));
// when(testReportMapper.countNormalClientUnit("client-1")).thenReturn(1);
// when(testReportMapper.countNormalReportModel("model-1")).thenReturn(1);
// return new TestableAddTestReportServiceImpl(dictTypeService, dictDataService, testReportMapper);
// }
//
// private TestReportParam.AddParam buildValidNormalizeParam() {
// TestReportParam.AddParam param = new TestReportParam.AddParam();
// param.setNo("PT-001");
// param.setClientUnitId("client-1");
// param.setReportModelId("model-1");
// param.setCreateUnit("company-1");
// param.setContractNumber("HT-001");
// param.setStandard("[\"std-1\"]");
// param.setData("{}");
// return param;
// }
//
// private Object invokeNormalizeParam(TestReportServiceImpl service, TestReportParam.AddParam param) {
// try {
// Method method = TestReportServiceImpl.class.getDeclaredMethod("normalizeParam", TestReportParam.AddParam.class);
// method.setAccessible(true);
// return method.invoke(service, param);
// } catch (java.lang.reflect.InvocationTargetException exception) {
// Throwable targetException = exception.getTargetException();
// if (targetException instanceof RuntimeException) {
// throw (RuntimeException) targetException;
// }
// throw new RuntimeException(targetException);
// } catch (Exception exception) {
// throw new RuntimeException(exception);
// }
// }
//
// private static class TestableTestReportServiceImpl extends TestReportServiceImpl {
// private final TestReportPO report;
// private TestReportPO updatedReport;
//
// private TestableTestReportServiceImpl(IDictTypeService dictTypeService, IDictDataService dictDataService,
// TestReportPO report) {
// super(mock(ReportModelService.class), mock(ReportModelFileStorageService.class),
// mock(FilePreviewService.class), dictTypeService, dictDataService,
// mock(TestReportPointMapper.class), mock(TestReportGroupReportMapper.class),
// mock(TestReportPointExcelParser.class), mock(TestReportStorageService.class),
// mock(TestReportLedgerImportService.class), mock(TestReportLedgerTemplateService.class));
// this.report = report;
// }
//
// @Override
// public TestReportPO requireNormal(String id) {
// return report;
// }
//
// @Override
// public boolean updateById(TestReportPO entity) {
// this.updatedReport = entity;
// return true;
// }
// }
//
// private static class TestableAddTestReportServiceImpl extends TestReportServiceImpl {
// private TestReportPO savedReport;
//
// private TestableAddTestReportServiceImpl(IDictTypeService dictTypeService, IDictDataService dictDataService,
// TestReportMapper testReportMapper) {
// super(mock(ReportModelService.class), mock(ReportModelFileStorageService.class),
// mock(FilePreviewService.class), dictTypeService, dictDataService,
// mock(TestReportPointMapper.class), mock(TestReportGroupReportMapper.class),
// mock(TestReportPointExcelParser.class), mock(TestReportStorageService.class),
// mock(TestReportLedgerImportService.class), mock(TestReportLedgerTemplateService.class));
// this.baseMapper = testReportMapper;
// }
//
// @Override
// public long count(com.baomidou.mybatisplus.core.conditions.Wrapper<TestReportPO> queryWrapper) {
// return 0;
// }
//
// @Override
// public boolean save(TestReportPO entity) {
// this.savedReport = entity;
// return true;
// }
// }
//
// private void mockNormalDictType(IDictTypeService dictTypeService, String code, String typeId) {
// when(dictTypeService.getByCode(code)).thenReturn(buildDictType(typeId, code));
// }
//
// private DictType buildDictType(String id, String code) {
// DictType dictType = new DictType();
// dictType.setId(id);
// dictType.setCode(code);
// dictType.setState(DICT_STATE_NORMAL);
// return dictType;
// }
//
// private DictData buildDictData(String id, String name, Integer state) {
// DictData dictData = new DictData();
// dictData.setId(id);
// dictData.setName(name);
// dictData.setState(state);
// return dictData;
// }
//}

View File

@@ -1,7 +1,7 @@
<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0"
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://www.w3.org/2001/XMLSchema-instance http://maven.apache.org/xsd/maven-4.0.0.xsd">
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
<modelVersion>4.0.0</modelVersion>
<parent>

View File

@@ -47,24 +47,24 @@ public class FilePreviewServiceTest {
Assert.assertTrue(html.contains("第二行&lt;tag&gt;"));
}
@Test
public void previewDocReturnsHtmlContent() throws Exception {
Path file = Files.createTempFile("common-preview", ".doc");
try (HWPFDocument document = new HWPFDocument();
ByteArrayOutputStream outputStream = new ByteArrayOutputStream()) {
Range range = document.getRange();
range.insertAfter("DOC 正文");
document.write(outputStream);
Files.write(file, outputStream.toByteArray());
}
FilePreviewService service = new FilePreviewService();
FilePreviewService.PreviewContent previewContent = service.preview(file, "sample.doc");
String html = new String(previewContent.getContent(), StandardCharsets.UTF_8);
Assert.assertEquals(MediaType.TEXT_HTML_VALUE + ";charset=UTF-8", previewContent.getContentType());
Assert.assertTrue(html.contains("DOC 正文"));
}
//@Test
//public void previewDocReturnsHtmlContent() throws Exception {
// Path file = Files.createTempFile("common-preview", ".doc");
// try (HWPFDocument document = new HWPFDocument();
// ByteArrayOutputStream outputStream = new ByteArrayOutputStream()) {
// Range range = document.getRange();
// range.insertAfter("DOC 正文");
// document.write(outputStream);
// Files.write(file, outputStream.toByteArray());
// }
// FilePreviewService service = new FilePreviewService();
//
// FilePreviewService.PreviewContent previewContent = service.preview(file, "sample.doc");
// String html = new String(previewContent.getContent(), StandardCharsets.UTF_8);
//
// Assert.assertEquals(MediaType.TEXT_HTML_VALUE + ";charset=UTF-8", previewContent.getContentType());
// Assert.assertTrue(html.contains("DOC 正文"));
//}
@Test(expected = IllegalArgumentException.class)
public void previewRejectsUnsupportedExtension() throws Exception {

View File

@@ -46,6 +46,23 @@
<artifactId>user</artifactId>
<version>1.0.0</version>
</dependency>
<dependency>
<groupId>io.minio</groupId>
<artifactId>minio</artifactId>
<version>8.4.3</version>
</dependency>
<dependency>
<groupId>com.squareup.okhttp3</groupId>
<artifactId>okhttp</artifactId>
</dependency>
<dependency>
<groupId>com.squareup.okio</groupId>
<artifactId>okio-jvm</artifactId>
</dependency>
<dependency>
<groupId>org.jetbrains.kotlin</groupId>
<artifactId>kotlin-stdlib-jdk8</artifactId>
</dependency>
<!-- Key refactor point: retain activation as a platform capability,
but stop pulling it transitively through detection. -->
<dependency>

View File

@@ -65,6 +65,15 @@ log:
files:
path: D:\file
storage:
minio:
endpoint: http://127.0.0.1:9000
access-key: "minioadmin"
secret-key: "minioadmin"
buckets:
user-signature: cn-tool-user-signature
ai-report: cn-tool-ai-report
activate:
private-key: "MIIEvQIBADANBgkqhkiG9w0BAQEFAASCBKcwggSjAgEAAoIBAQCcUyYhVqczGxblL+o/xZzF/8nf+LjrfUE/dS1aRHM7uMDD0cgCArhjtfneFePrMxt+Z7W8yNBzSarub8qsfhaVNikV7Es7oaeTygfjQXTi2n4AFkir3fM07J08RpWhl5M8f8uWTCuvFUYAw00gq55typqmnbkmJa2VIUy/iQf+cMCP7abz4/jNhUzUR3qA7TV4oMRgTdIEDUp63YF8dOC+JH8XxYrCVeHXV6fLCwmesdMzl0lB2VTEKMfLbXhOmF5g7P9y/16VCcN8UBuZlbyYfn+GAxJOSbeHi5HshOKfoSuD7Jz+3WQZpNavOWjIFExKIU38/CvnJCOP7XBCqpSTAgMBAAECggEAYeWokWRE3TpvwiOZnUpR/aVMdVi75a3ROL5XIpqPV61B+t/bU3cEpl0GF9C5pUeiRi0IoStZb3mI9D1KPW/REKyUWkhabQO1gFYbTnRlkNOn6MILzKX4cwJjDaZeeo4EBPU7N+qHyOOXrU6hdH5FfxhMdV983ajm5eeuupxER1C2kAcIklTeVpTX6EKOgZb5LBp5ssOVm2P42pOauvcRozRcvZmqnErXmukv0H4l3EVNt4rHpTn9riHUC63e8JfiYzVaF6zuNUxv6nHEft0/SRMw11XSTnNfDzcKqgjz6ksFBS/6eQQYKESk+ONC53HUuYHFAknkwsPupDCT2W8FIQKBgQDLHT/xCU3nxGr4vFKBDNaO2D5oK20ECbBO4oDvLWWmQG7f+6TsMy8PgVdMnoL4RfqGlwFAKEpS6KVFHnBVqnNEhcdy9uCI7x7Xx8UnyUtxj1EDTm76uta9Ki9OrlqB6tImDM9+Ya3vGktW37ht4WOx2OsJRhG1dbf6RLwFlH7DWwKBgQDFBxvi5I1BR6hg6Tj7xd2SqOT2Y+BED3xuSYENhWbmMhLJDResaB7mjztbxlYaY2mOE0holWm2uDmVFFhMh4jYXik4hYH8nmDzq9mDpZCZ9pyjYqnAP8THoAa8EbgrUWB8A6BPH4iL3KbMnBfBKY0pIr2xrvnjQjNBAgta7KDRKQKBgCe6oe4wxrdF2TKsC2tIqpMoQxS3Icy/ZGgZr+SYuaBKTCWtoDW/UT40K3JGMxIDBhzbXphBCUCsVt9tM8Xd4EwP6tJW7dZ7B0pnve2pVwNwaAVAiz6p2yUHIle+jN+Koe5lZRSwYIg7WW81tWpwwsJfzqFyvjYDP6hJV4mz4ROvAoGAaRcdnKvjXApomShMqJ4lTPChD3q+SA8qg3jZSOj6tZXHx00gb2kp8jg7pPvpOTIFPy6x1Ha9aCRjMk0ju84fA6lVuzwa1S907wOehUVuF3Eeo1cgy9Y3k3KbpPyeixxgpkUY4JslLdSHc2NemD0dee951qhJyRmqVOZOQDUuoeECgYEAqBw2cAFk3vM97WY06TSldGA8ajVHx3BYRjj+zl62NTQthy8fw3tqxb3c5e8toOmZWKjZvDhg2TRLhsDDQWEYg3LZG87REqVIjgEPcpjNLidjygGX8n3JF2o0O5I/EMvl0s/+LVQONfduOBvhwDqr8QNisbLsyneiAq7umewMolo="
public-key: "MIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEAnFMmIVanMxsW5S/qP8Wcxf/J3/i4631BP3UtWkRzO7jAw9HIAgK4Y7X53hXj6zMbfme1vMjQc0mq7m/KrH4WlTYpFexLO6Gnk8oH40F04tp+ABZIq93zNOydPEaVoZeTPH/LlkwrrxVGAMNNIKuebcqapp25JiWtlSFMv4kH/nDAj+2m8+P4zYVM1Ed6gO01eKDEYE3SBA1Ket2BfHTgviR/F8WKwlXh11enywsJnrHTM5dJQdlUxCjHy214TpheYOz/cv9elQnDfFAbmZW8mH5/hgMSTkm3h4uR7ITin6Erg+yc/t1kGaTWrzloyBRMSiFN/Pwr5yQjj+1wQqqUkwIDAQAB"

24
pom.xml
View File

@@ -38,6 +38,9 @@
<properties>
<spring-boot.version>2.3.12.RELEASE</spring-boot.version>
<okhttp.version>4.12.0</okhttp.version>
<okio.version>3.6.0</okio.version>
<kotlin.version>1.9.25</kotlin.version>
<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
<project.reporting.outputEncoding>UTF-8</project.reporting.outputEncoding>
<maven.compiler.encoding>UTF-8</maven.compiler.encoding>
@@ -52,6 +55,27 @@
<type>pom</type>
<scope>import</scope>
</dependency>
<!-- 统一约束 MinIO 运行时依赖版本,避免被旧版 okhttp 3.x 抢占导致 MinioAsyncClient 初始化失败。 -->
<dependency>
<groupId>com.squareup.okhttp3</groupId>
<artifactId>okhttp</artifactId>
<version>${okhttp.version}</version>
</dependency>
<dependency>
<groupId>com.squareup.okio</groupId>
<artifactId>okio</artifactId>
<version>${okio.version}</version>
</dependency>
<dependency>
<groupId>com.squareup.okio</groupId>
<artifactId>okio-jvm</artifactId>
<version>${okio.version}</version>
</dependency>
<dependency>
<groupId>org.jetbrains.kotlin</groupId>
<artifactId>kotlin-stdlib-jdk8</artifactId>
<version>${kotlin.version}</version>
</dependency>
</dependencies>
</dependencyManagement>

7183
tmp/cn_tool.sql Normal file

File diff suppressed because one or more lines are too long

View File

@@ -188,12 +188,15 @@ public class JsonToXmlConversionService {
String name = reportItem.getName();
if (name.contains("urcbFluc") ||
name.contains("urcbStHarm") ||
name.contains("urcbStIHarm") ||
name.contains("urcbStMMXU") ||
name.contains("urcbStMSQI") ||
name.contains("brcbFluc") ||
name.contains("urcbRtHarm") ||
name.contains("urcbRtIHarm") ||
name.contains("urcbRtMMXU") ||
name.contains("urcbRtMSQI") ||
name.contains("urcbPLT") ||
name.contains("urcbPST") ||
name.contains("brcbPLT") ||
name.contains("brcbPST") ||
name.contains("urcbFlicker") ||
name.contains("urcbStatistic")||
name.contains("urcbRealData")) {
@@ -771,6 +774,7 @@ public class JsonToXmlConversionService {
if(2==configType && fullKey.contains("线电压总有效值")){
daPath = "1_" + daPath;
}
int offset=calculateOffset(sdiItem.getName(), typeItem.getName(), doiItem);
MetricInfo info = new MetricInfo();
info.setLnClass(dataSetGroup.getLnClass());
info.setLnInst(instItem.getInst());
@@ -781,6 +785,7 @@ public class JsonToXmlConversionService {
info.setStart(doiItem.getStart());
info.setEnd(doiItem.getEnd());
info.setBaseFlag(doiItem.getBaseflag());
info.setOffset(offset);
metrics.put(fullKey.toLowerCase(), info);
}
}
@@ -829,10 +834,15 @@ public class JsonToXmlConversionService {
int start = doiItem.getStart();
if (baseflag == 1) {
boolean isInterHarmonic = isInterHarmonicMetric(doiItem);
boolean isInterHarmonic = isInterHarmonicMetric(doiItem);//间谐波判断
int m = isInterHarmonic ? 0 : 2;
int i = m - start;
sb.append("[%-").append(i).append("]");
//判断是Offset是否需要
if(i<0){
sb.append("[%-0]");
}else {
sb.append("[%-").append(i).append("]");
}
} else {
sb.append("[").append(start).append("]");
}
@@ -848,6 +858,28 @@ public class JsonToXmlConversionService {
}
return sb.toString();
}
private int calculateOffset(String sdiName, String typeName, DoiItem doiItem) {
int offset=0;
if (sdiName != null && !sdiName.isEmpty()){
if (sdiName.endsWith("Har") && doiItem != null) {
int baseflag = doiItem.getBaseflag();
int start = doiItem.getStart();
if (baseflag == 1) {
boolean isInterHarmonic = isInterHarmonicMetric(doiItem);//间谐波判断
int m = isInterHarmonic ? 0 : 2;
int i = m - start;
//判断是Offset是否需要
if(i<0){
offset = i;
}
}
}
}
return offset;
}
/**
* 标准化相别名称,将具体相别转换为通配符格式
@@ -906,6 +938,8 @@ public class JsonToXmlConversionService {
MetricInfo metric = mappingMetrics.get(rule.getName());
rule.setDoPath(metric.getDoPath());
rule.setDaPath(metric.getDaPath());
// 传递offset值将Integer转换为String
rule.setOffset(String.valueOf(metric.getOffset()));
applicable.put(ruleKey, rule);
matched = true;
addMethodDescribe(methodDescribeList, "✓ 规则匹配成功: " + ruleKey + " -> DO=" + rule.getDoPath() + ", DA=" + rule.getDaPath());
@@ -1066,10 +1100,11 @@ public class JsonToXmlConversionService {
RuleBasedXmlMappingService.ValueRule matchedRule = applicableRules.get(desc);
if (matchedRule != null) {
StringBuilder valueNode = new StringBuilder("<Value ");
valueNode.append("name=\"").append(escapeXml(name)).append("\" ");
// 如果offset<0需要调整指标名中的%int或int%格式
String adjustedName = adjustPathWithOffset(name, matchedRule.getOffset());
valueNode.append("name=\"").append(escapeXml(adjustedName)).append("\" ");
valueNode.append("desc=\"").append(escapeXml(desc)).append("\" ");
valueNode.append("type=\"").append(type).append("\" ");
if (matchedRule.getDoPath() != null && !matchedRule.getDoPath().isEmpty()) {
valueNode.append("DO=\"").append(escapeXml(matchedRule.getDoPath())).append("\" ");
}
@@ -1088,7 +1123,7 @@ public class JsonToXmlConversionService {
if (matchedRule.getCoefficient() != null && !matchedRule.getCoefficient().isEmpty()) {
valueNode.append("Coefficient=\"").append(escapeXml(matchedRule.getCoefficient())).append("\" ");
}
if (matchedRule.getOffset() != null && !matchedRule.getOffset().isEmpty()) {
if (matchedRule.getOffset() != null && !matchedRule.getOffset().isEmpty() && !matchedRule.getOffset().equals("0")) {
valueNode.append("Offset=\"").append(escapeXml(matchedRule.getOffset())).append("\" ");
}
if (matchedRule.getPltFlag() != null && !matchedRule.getPltFlag().isEmpty()) {
@@ -1115,6 +1150,48 @@ public class JsonToXmlConversionService {
.replace("\"", "&quot;")
.replace("'", "&apos;");
}
/**
* 根据offset调整路径DO或DA
* 当offset<0时对路径中的[%-int]或[int%]格式增加-offset的值
* @param path 原始路径DO或DA
* @param offsetStr offset字符串值
* @return 调整后的路径
*/
private String adjustPathWithOffset(String path, String offsetStr) {
if (path == null || offsetStr == null || offsetStr.isEmpty()) {
return path;
}
try {
int offset = Integer.parseInt(offsetStr);
// 只有当offset<0时才需要调整
if (offset >= 0) {
return path;
}
int adjustment = -offset;
// 匹配 %数字,数字% 格式并调整(如 %0,49%
java.util.regex.Pattern pattern3 = java.util.regex.Pattern.compile("%(\\d+),(\\d+)%");
java.util.regex.Matcher matcher3 = pattern3.matcher(path);
StringBuffer result3 = new StringBuffer();
while (matcher3.find()) {
int startVal = Integer.parseInt(matcher3.group(1));
int endVal = Integer.parseInt(matcher3.group(2));
int newStart = startVal + adjustment;
int newEnd = endVal + adjustment;
matcher3.appendReplacement(result3, "%"+newStart+","+newEnd+"%");
}
matcher3.appendTail(result3);
path = result3.toString();
return path;
} catch (NumberFormatException e) {
// 如果offset无法解析为整数返回原始路径
return path;
}
}
private void addMethodDescribe(List<String> methodDescribeList, String message) {
if (methodDescribeList == null || message == null || message.trim().isEmpty()) {
@@ -1231,7 +1308,9 @@ public class JsonToXmlConversionService {
private int end;
private int baseCount;
private int baseFlag;
private int offset;
public int getOffset() {return offset;}
public void setOffset(int offset) {this.offset = offset;}
public String getLnClass() { return lnClass; }
public void setLnClass(String lnClass) { this.lnClass = lnClass; }
public String getLnInst() { return lnInst; }

View File

@@ -83,7 +83,7 @@ public class RuleBasedXmlMappingService {
public List<InputStream> loadDefaultRuleFile( ) throws Exception {
ClassPathResource ruleResource = new ClassPathResource("template/默认规则.txt");
ClassPathResource ruleResource = new ClassPathResource("template/DefaultRule.txt");
if ( !ruleResource.exists()) {
return null;
}

View File

@@ -32,6 +32,6 @@ public class MappingModuleConfig {
private String jsonToXmlTemplatePath;
/** JSON转XML使用的默认规则文件路径。 */
@Value("${icd.mapping.json-to-xml-rule-path:template/默认规则.txt}")
@Value("${icd.mapping.json-to-xml-rule-path:template/DefaultRule.txt}")
private String jsonToXmlRulePath;
}

View File

@@ -32,7 +32,8 @@
"Select": "FlickerFileMap",
"DataSetList": [
"dsFlickerData",
"dsPST"
"dsPST",
"dsPLT"
],
"LnInstList": [
"波动闪变值"

View File

@@ -43,10 +43,10 @@
<Value name="间谐波电压有效值序列间谐波平均值值" desc="间谐波电压含有率(f25-2475)平均值" type="6" DO="MHAI8$MX$HPhV" DA="phs*Har[%-0]$mag$f" />
<Value name="间谐波电压有效值序列间谐波最大值值" desc="间谐波电压含有率(f25-2475)最大值" type="6" DO="MHAI9$MX$HPhV" DA="phs*Har[%-0]$mag$f" />
<Value name="间谐波电压有效值序列间谐波最小值值" desc="间谐波电压含有率(f25-2475)最小值" type="6" DO="MHAI10$MX$HPhV" DA="phs*Har[%-0]$mag$f" />
<Value name="间谐波电压有效值序列间谐波95值值" desc="间谐波电压含有率(f25-2475)95值" type="6" DO="MHAI8$MX$HPhV" DA="phs*Har[%-0]$mag$f" />
<Value name="间谐波电压有效值序列间谐波平均值值" desc="间谐波电压含有率(f25-2475)平均值" type="6" DO="MHAI8$MX$HPhV" DA="phs*Har[%-0]$mag$f" />
<Value name="间谐波电压有效值序列间谐波最大值值" desc="间谐波电压含有率(f25-2475)最大值" type="6" DO="MHAI9$MX$HPhV" DA="phs*Har[%-0]$mag$f" />
<Value name="间谐波电压有效值序列间谐波最小值值" desc="间谐波电压含有率(f25-2475)最小值" type="6" DO="MHAI10$MX$HPhV" DA="phs*Har[%-0]$mag$f" />
<Value name="相电压间谐波含有率序列间谐波95值值" desc="间谐波电压含有率(f25-2475)95值" type="6" DO="MHAI8$MX$HPhV" DA="phs*Har[%-0]$mag$f" />
<Value name="相电压间谐波含有率序列间谐波平均值值" desc="间谐波电压含有率(f25-2475)平均值" type="6" DO="MHAI8$MX$HPhV" DA="phs*Har[%-0]$mag$f" />
<Value name="相电压间谐波含有率序列间谐波最大值值" desc="间谐波电压含有率(f25-2475)最大值" type="6" DO="MHAI9$MX$HPhV" DA="phs*Har[%-0]$mag$f" />
<Value name="相电压间谐波含有率序列间谐波最小值值" desc="间谐波电压含有率(f25-2475)最小值" type="6" DO="MHAI10$MX$HPhV" DA="phs*Har[%-0]$mag$f" />
<Value name="相电压谐波含有率序列95值值" desc="谐波电压含有率(2-50)95值" type="6" DO="MHAI2$MX$HPhV" DA="phs*Har[%-2]$mag$f" />
<Value name="相电压谐波含有率序列平均值值" desc="谐波电压含有率(2-50)平均值" type="6" DO="MHAI2$MX$HPhV" DA="phs*Har[%-2]$mag$f" />
@@ -281,6 +281,7 @@
<Value name="相电压基波有效值实时数据值" desc="实时基波电压有效值" type="6" DO="MHAI$MX$FundPhV" DA="phs*$cVal$mag$f" Coefficient="1" />
<Value name="相电压谐波含有率序列实时数据值" desc="实时2-50次谐波电压含有率" type="6" DO="MHAI$MX$HPhV" DA="phs*Har[%-2]$mag$f" Coefficient="1" Offset="0" />
<Value name="间谐波电压有效值序列间谐波实时数据值" desc="实时1-50次间谐波电压含有率" type="6" DO="MHAI$MX$HPhV" DA="phs*Har[%-0]$mag$f" Coefficient="1" Offset="0" />
<Value name="相电压间谐波含有率序列间谐波实时数据值" desc="实时1-50次间谐波电压含有率" type="6" DO="MHAI$MX$HPhV" DA="phs*Har[%-0]$mag$f" Coefficient="1" Offset="0" />
<Value name="谐波电压有效值序列实时数据角度" desc="实时谐波电压相角(2-50)" type="6" DO="MHAI$MX$HPhV" DA="phs*Har[%-2]$ang$f" />
<Value name="相电压谐波含有率序列实时数据角度" desc="实时谐波电压相角(2-50)" type="6" DO="MHAI$MX$HPhV" DA="phs*Har[%-2]$ang$f" />
<Value name="线电压偏差实时数据值" desc="实时线电压偏差" type="6" DO="MMXU$MX$LnVolDev" DA="phs*$cVal$mag$f" Coefficient="1"/>

View File

@@ -217,14 +217,14 @@
<Value name="SI_%0,49%" desc="间谐波电流幅值(f25-2475)平均值" type="6" DO="" DA="" />
<Value name="MAX_SI_%0,49%" desc="间谐波电流幅值(f25-2475)最大值" type="6" DO="" DA="" />
<Value name="MIN_SI_%0,49%" desc="间谐波电流幅值(f25-2475)最小值" type="6" DO="" DA="" />
<Value name="G_I%1,50%" desc="谐波电流幅值(2-50)95值" type="6" DO="" DA="" />
<Value name="I%1,50%" desc="谐波电流幅值(2-50)平均值" type="6" DO="" DA="" />
<Value name="MAX_I%1,50%" desc="谐波电流幅值(2-50)最大值" type="6" DO="" DA="" />
<Value name="MIN_I%1,50%" desc="谐波电流幅值(2-50)最小值" type="6" DO="" DA="" />
<Value name="G_IA%1,50%" desc="谐波电流相角(2-50)95值" type="6" DO="" DA="" />
<Value name="IA%1,50%" desc="谐波电流相角(2-50)平均值" type="6" DO="" DA="" />
<Value name="MAX_IA%1,50%" desc="谐波电流相角(2-50)最大值" type="6" DO="" DA="" />
<Value name="MIN_IA%1,50%" desc="谐波电流相角(2-50)最小值" type="6" DO="" DA="" />
<Value name="G_I%2,50%" desc="谐波电流幅值(2-50)95值" type="6" DO="" DA="" />
<Value name="I%2,50%" desc="谐波电流幅值(2-50)平均值" type="6" DO="" DA="" />
<Value name="MAX_I%2,50%" desc="谐波电流幅值(2-50)最大值" type="6" DO="" DA="" />
<Value name="MIN_I%2,50%" desc="谐波电流幅值(2-50)最小值" type="6" DO="" DA="" />
<Value name="G_IA%2,50%" desc="谐波电流相角(2-50)95值" type="6" DO="" DA="" />
<Value name="IA%2,50%" desc="谐波电流相角(2-50)平均值" type="6" DO="" DA="" />
<Value name="MAX_IA%2,50%" desc="谐波电流相角(2-50)最大值" type="6" DO="" DA="" />
<Value name="MIN_IA%2,50%" desc="谐波电流相角(2-50)最小值" type="6" DO="" DA="" />
<Value name="ITHD" desc="电流总谐波畸变率平均值" type="6" DO="" DA="" Coefficient="1" />
<Value name="MAX_ITHD" desc="电流总谐波畸变率最大值" type="6" DO="" DA="" Coefficient="1" />
<Value name="MIN_ITHD" desc="电流总谐波畸变率最小值" type="6" DO="" DA="" Coefficient="1" />
@@ -244,8 +244,8 @@
<Value name="IZSEQ" desc="零序电流平均值" type="6" DO="" DA="" />
<Value name="MAX_IZSEQ" desc="零序电流最大值" type="6" DO="" DA="" />
<Value name="MIN_IZSEQ" desc="零序电流最小值" type="6" DO="" DA="" />
<Value name="G_VZSEQ_UNBAN" desc="零序电流不平衡95值" type="6" DO="" DA="" />
<Value name="VZSEQ_UNBAN" desc="零序电流不平衡平均值" type="6" DO="" DA="" />
<Value name="G_IZSEQ_UNBAN" desc="零序电流不平衡95值" type="6" DO="" DA="" />
<Value name="IZSEQ_UNBAN" desc="零序电流不平衡平均值" type="6" DO="" DA="" />
<Value name="MAX_IZSEQ_UNBAN" desc="零序电流不平衡最大值" type="6" DO="" DA="" />
<Value name="MIN_IZSEQ_UNBAN" desc="零序电流不平衡最小值" type="6" DO="" DA="" />
<Value name="G_IPSEQ" desc="正序电流95值" type="6" DO="" DA="" />
@@ -277,18 +277,18 @@
<Value name="Q" desc="无功功率平均值" type="6" DO="" DA="" />
<Value name="MAX_Q" desc="无功功率最大值" type="6" DO="" DA="" />
<Value name="MIN_Q" desc="无功功率最小值" type="6" DO="" DA="" />
<Value name="G_P%1,50%" desc="谐波有功功率(2-50)95值" type="6" DO="" DA="" />
<Value name="P%1,50%" desc="谐波有功功率(2-50)平均值" type="6" DO="" DA="" />
<Value name="MAX_P%1,50%" desc="谐波有功功率(2-50)最大值" type="6" DO="" DA="" />
<Value name="MIN_P%1,50%" desc="谐波有功功率(2-50)最小值" type="6" DO="" DA="" />
<Value name="G_Q%1,50%" desc="谐波无功功率(2-50)95值" type="6" DO="" DA="" />
<Value name="Q%1,50%" desc="谐波无功功率(2-50)平均值" type="6" DO="" DA="" />
<Value name="MAX_Q%1,50%" desc="谐波无功功率(2-50)最大值" type="6" DO="" DA="" />
<Value name="MIN_Q%1,50%" desc="谐波无功功率(2-50)最小值" type="6" DO="" DA="" />
<Value name="G_S%1,50%" desc="谐波视在功率(2-50)95值" type="6" DO="" DA="" />
<Value name="S%1,50%" desc="谐波视在功率(2-50)平均值" type="6" DO="" DA="" />
<Value name="MAX_S%1,50%" desc="谐波视在功率(2-50)最大值" type="6" DO="" DA="" />
<Value name="MIN_S%1,50%" desc="谐波视在功率(2-50)最小值" type="6" DO="" DA="" />
<Value name="G_P%2,50%" desc="谐波有功功率(2-50)95值" type="6" DO="" DA="" />
<Value name="P%2,50%" desc="谐波有功功率(2-50)平均值" type="6" DO="" DA="" />
<Value name="MAX_P%2,50%" desc="谐波有功功率(2-50)最大值" type="6" DO="" DA="" />
<Value name="MIN_P%2,50%" desc="谐波有功功率(2-50)最小值" type="6" DO="" DA="" />
<Value name="G_Q%2,50%" desc="谐波无功功率(2-50)95值" type="6" DO="" DA="" />
<Value name="Q%2,50%" desc="谐波无功功率(2-50)平均值" type="6" DO="" DA="" />
<Value name="MAX_Q%2,50%" desc="谐波无功功率(2-50)最大值" type="6" DO="" DA="" />
<Value name="MIN_Q%2,50%" desc="谐波无功功率(2-50)最小值" type="6" DO="" DA="" />
<Value name="G_S%2,50%" desc="谐波视在功率(2-50)95值" type="6" DO="" DA="" />
<Value name="S%2,50%" desc="谐波视在功率(2-50)平均值" type="6" DO="" DA="" />
<Value name="MAX_S%2,50%" desc="谐波视在功率(2-50)最大值" type="6" DO="" DA="" />
<Value name="MIN_S%2,50%" desc="谐波视在功率(2-50)最小值" type="6" DO="" DA="" />
<Value name="G_P" desc="有功功率95值" type="6" DO="" DA="" />
<Value name="P" desc="有功功率平均值" type="6" DO="" DA="" />
<Value name="MAX_P" desc="有功功率最大值" type="6" DO="" DA="" />

View File

@@ -37,6 +37,11 @@
<version>1.2.83</version>
</dependency>
<dependency>
<groupId>io.minio</groupId>
<artifactId>minio</artifactId>
<version>8.4.3</version>
</dependency>
</dependencies>

View File

@@ -25,7 +25,9 @@ public enum UserResponseEnum {
SUPER_ADMIN_REPEAT("A010013","超级管理员已存在,请勿重复添加" ),
RSA_DECRYT_ERROR("A010014","RSA解密失败" ),
PASSWORD_SAME("A010015", "新密码不能与旧密码相同"),
OLD_PASSWORD_ERROR("A010016", "旧密码错误"), ;
OLD_PASSWORD_ERROR("A010016", "旧密码错误"),
SIGNATURE_FILE_NAME_EMPTY("A010017", "签名文件名不能为空"),
SIGNATURE_FILE_TYPE_ERROR("A010018", "签名文件格式仅支持 png、jpg、jpeg、bmp、webp"), ;
private String code;
private String message;

View File

@@ -0,0 +1,200 @@
package com.njcn.gather.user.user.component;
import cn.hutool.core.util.StrUtil;
import com.njcn.common.pojo.enums.response.CommonResponseEnum;
import com.njcn.common.pojo.exception.BusinessException;
import com.njcn.gather.user.user.config.StorageMinioProperties;
import io.minio.BucketExistsArgs;
import io.minio.GetObjectArgs;
import io.minio.GetObjectResponse;
import io.minio.MakeBucketArgs;
import io.minio.MinioClient;
import io.minio.PutObjectArgs;
import io.minio.RemoveObjectArgs;
import lombok.Getter;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.http.MediaType;
import org.springframework.http.MediaTypeFactory;
import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;
import java.io.InputStream;
import java.util.Locale;
/**
* MinIO 通用文件存储服务。
*/
@Slf4j
@Component
@RequiredArgsConstructor
public class SysFileMinioStorageService {
private final StorageMinioProperties properties;
public StoredObjectInfo save(String bucketName, String bizDir, String ownerId, String fileId, MultipartFile file) {
MultipartFile normalizedFile = requireFile(file);
validateConfig();
String originalFileName = normalizeFileName(normalizedFile.getOriginalFilename());
String extension = resolveExtension(originalFileName);
String objectName = buildObjectName(bizDir, ownerId, fileId, extension);
String contentType = resolveContentType(normalizedFile, originalFileName);
try (InputStream inputStream = normalizedFile.getInputStream()) {
MinioClient client = buildClient();
ensureBucket(client, bucketName);
client.putObject(PutObjectArgs.builder()
.bucket(bucketName)
.object(objectName)
.stream(inputStream, normalizedFile.getSize(), -1)
.contentType(contentType)
.build());
return new StoredObjectInfo(originalFileName, bucketName, objectName, contentType, normalizedFile.getSize());
} catch (BusinessException exception) {
throw exception;
} catch (Exception exception) {
log.error("upload file to minio failed, bucketName={}, objectName={}", bucketName, objectName, exception);
throw new BusinessException(CommonResponseEnum.FAIL, "上传文件失败");
}
}
public StoredObject getObject(String bucketName, String objectName) {
validateConfig();
if (StrUtil.hasBlank(bucketName, objectName)) {
throw new BusinessException(CommonResponseEnum.FAIL, "文件不存在");
}
try {
GetObjectResponse response = buildClient().getObject(GetObjectArgs.builder()
.bucket(bucketName)
.object(objectName)
.build());
return new StoredObject(response);
} catch (BusinessException exception) {
throw exception;
} catch (Exception exception) {
log.error("read file from minio failed, bucketName={}, objectName={}", bucketName, objectName, exception);
throw new BusinessException(CommonResponseEnum.FAIL, "读取文件失败");
}
}
public void deleteQuietly(String bucketName, String objectName) {
if (StrUtil.hasBlank(bucketName, objectName)) {
return;
}
try {
validateConfig();
buildClient().removeObject(RemoveObjectArgs.builder()
.bucket(bucketName)
.object(objectName)
.build());
} catch (Exception exception) {
log.warn("delete file from minio failed, bucketName={}, objectName={}", bucketName, objectName, exception);
}
}
private MultipartFile requireFile(MultipartFile file) {
if (file == null || file.isEmpty()) {
throw new BusinessException(CommonResponseEnum.FAIL, "上传文件不能为空");
}
return file;
}
private void validateConfig() {
if (StrUtil.hasBlank(properties.getEndpoint(), properties.getAccessKey(), properties.getSecretKey())) {
throw new BusinessException(CommonResponseEnum.FAIL, "MinIO配置不完整");
}
}
private MinioClient buildClient() {
return MinioClient.builder()
.endpoint(properties.getEndpoint())
.credentials(properties.getAccessKey(), properties.getSecretKey())
.build();
}
private void ensureBucket(MinioClient client, String bucketName) throws Exception {
boolean exists = client.bucketExists(BucketExistsArgs.builder().bucket(bucketName).build());
if (!exists) {
client.makeBucket(MakeBucketArgs.builder().bucket(bucketName).build());
}
}
private String normalizeFileName(String originalFileName) {
String fileName = StrUtil.trimToEmpty(originalFileName);
if (StrUtil.isBlank(fileName)) {
throw new BusinessException(CommonResponseEnum.FAIL, "文件名不能为空");
}
int slashIndex = Math.max(fileName.lastIndexOf('/'), fileName.lastIndexOf('\\'));
return slashIndex >= 0 ? fileName.substring(slashIndex + 1) : fileName;
}
private String resolveExtension(String originalFileName) {
int dotIndex = originalFileName.lastIndexOf('.');
if (dotIndex < 0 || dotIndex == originalFileName.length() - 1) {
return "bin";
}
return originalFileName.substring(dotIndex + 1).toLowerCase(Locale.ENGLISH);
}
private String buildObjectName(String bizDir, String ownerId, String fileId, String extension) {
StringBuilder builder = new StringBuilder();
builder.append(trimPathSegment(bizDir));
if (StrUtil.isNotBlank(ownerId)) {
builder.append("/").append(trimPathSegment(ownerId));
}
builder.append("/").append(fileId);
if (StrUtil.isNotBlank(extension)) {
builder.append(".").append(extension);
}
return builder.toString();
}
private String trimPathSegment(String value) {
String text = StrUtil.trimToEmpty(value);
String normalized = text.replace("\\", "/");
while (normalized.startsWith("/")) {
normalized = normalized.substring(1);
}
while (normalized.endsWith("/")) {
normalized = normalized.substring(0, normalized.length() - 1);
}
if (StrUtil.isBlank(normalized)) {
throw new BusinessException(CommonResponseEnum.FAIL, "对象目录配置不合法");
}
return normalized;
}
private String resolveContentType(MultipartFile file, String originalFileName) {
if (StrUtil.isNotBlank(file.getContentType())) {
return file.getContentType();
}
return MediaTypeFactory.getMediaType(originalFileName)
.map(MediaType::toString)
.orElse(MediaType.APPLICATION_OCTET_STREAM_VALUE);
}
@Getter
public static class StoredObjectInfo {
private final String fileName;
private final String bucketName;
private final String objectName;
private final String contentType;
private final long fileSize;
public StoredObjectInfo(String fileName, String bucketName, String objectName, String contentType, long fileSize) {
this.fileName = fileName;
this.bucketName = bucketName;
this.objectName = objectName;
this.contentType = contentType;
this.fileSize = fileSize;
}
}
@Getter
public static class StoredObject {
private final GetObjectResponse inputStream;
public StoredObject(GetObjectResponse inputStream) {
this.inputStream = inputStream;
}
}
}

View File

@@ -0,0 +1,48 @@
package com.njcn.gather.user.user.config;
import cn.hutool.core.util.StrUtil;
import com.njcn.common.pojo.enums.response.CommonResponseEnum;
import com.njcn.common.pojo.exception.BusinessException;
import lombok.Data;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.stereotype.Component;
import java.util.LinkedHashMap;
import java.util.Map;
/**
* MinIO 通用配置。
*/
@Data
@Component
@ConfigurationProperties(prefix = "storage.minio")
public class StorageMinioProperties {
/**
* MinIO 服务地址。
*/
private String endpoint;
/**
* MinIO 访问账号。
*/
private String accessKey;
/**
* MinIO 访问密码。
*/
private String secretKey;
/**
* 业务桶配置。
*/
private Map<String, String> buckets = new LinkedHashMap<String, String>();
public String requireBucket(String bucketKey) {
String bucketName = buckets.get(bucketKey);
if (StrUtil.isBlank(bucketName)) {
throw new BusinessException(CommonResponseEnum.FAIL, "MinIO桶配置缺失" + bucketKey);
}
return bucketName;
}
}

View File

@@ -7,24 +7,35 @@ import com.njcn.common.pojo.constant.OperateType;
import com.njcn.common.pojo.enums.common.DataStateEnum;
import com.njcn.common.pojo.enums.common.LogEnum;
import com.njcn.common.pojo.enums.response.CommonResponseEnum;
import com.njcn.common.pojo.exception.BusinessException;
import com.njcn.common.pojo.response.HttpResult;
import com.njcn.common.utils.LogUtil;
import com.njcn.gather.user.user.pojo.param.SysUserParam;
import com.njcn.gather.user.user.pojo.po.SysFile;
import com.njcn.gather.user.user.pojo.po.SysRole;
import com.njcn.gather.user.user.pojo.po.SysUser;
import com.njcn.gather.user.user.pojo.po.SysUserSignature;
import com.njcn.gather.user.user.service.ISysFileService;
import com.njcn.gather.user.user.service.ISysUserRoleService;
import com.njcn.gather.user.user.service.ISysUserSignatureService;
import com.njcn.gather.user.user.service.ISysUserService;
import com.njcn.gather.user.user.component.SysFileMinioStorageService;
import com.njcn.web.controller.BaseController;
import com.njcn.web.utils.HttpResultUtil;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiImplicitParams;
import io.swagger.annotations.ApiOperation;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.core.io.InputStreamResource;
import org.springframework.http.HttpHeaders;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import java.net.URLEncoder;
import java.util.List;
import java.util.stream.Collectors;
@@ -42,6 +53,9 @@ public class SysUserController extends BaseController {
private final ISysUserService sysUserService;
private final ISysUserRoleService sysUserRoleService;
private final ISysUserSignatureService sysUserSignatureService;
private final ISysFileService sysFileService;
private final SysFileMinioStorageService sysFileMinioStorageService;
@OperateInfo(info = LogEnum.SYSTEM_COMMON)
@PostMapping("/list")
@@ -55,13 +69,14 @@ public class SysUserController extends BaseController {
}
@OperateInfo(info = LogEnum.SYSTEM_COMMON, operateType = OperateType.ADD)
@PostMapping("/add")
@PostMapping(value = "/add", consumes = {"multipart/form-data"})
@ApiOperation("新增用户")
@ApiImplicitParam(name = "addUserParam", value = "新增用户", required = true)
public HttpResult<Boolean> add(@RequestBody @Validated SysUserParam.SysUserAddParam addUserParam) {
public HttpResult<Boolean> add(@RequestPart("request") @Validated SysUserParam.SysUserAddParam addUserParam,
@RequestPart(value = "signatureFile", required = false) MultipartFile signatureFile) {
String methodDescribe = getMethodDescribe("add");
LogUtil.njcnDebug(log, "{},用户数据为:{}", methodDescribe, addUserParam);
boolean result = sysUserService.addUser(addUserParam);
boolean result = sysUserService.addUser(addUserParam, signatureFile);
if (result) {
return HttpResultUtil.assembleCommonResponseResult(CommonResponseEnum.SUCCESS, true, methodDescribe);
} else {
@@ -70,13 +85,14 @@ public class SysUserController extends BaseController {
}
@OperateInfo(info = LogEnum.SYSTEM_COMMON, operateType = OperateType.UPDATE)
@PostMapping("/update")
@PostMapping(value = "/update", consumes = {"multipart/form-data"})
@ApiOperation("修改用户")
@ApiImplicitParam(name = "updateUserParam", value = "修改用户", required = true)
public HttpResult<Boolean> update(@RequestBody @Validated SysUserParam.SysUserUpdateParam updateUserParam) {
public HttpResult<Boolean> update(@RequestPart("request") @Validated SysUserParam.SysUserUpdateParam updateUserParam,
@RequestPart(value = "signatureFile", required = false) MultipartFile signatureFile) {
String methodDescribe = getMethodDescribe("update");
LogUtil.njcnDebug(log, "{},用户数据为:{}", methodDescribe, updateUserParam);
boolean result = sysUserService.updateUser(updateUserParam);
boolean result = sysUserService.updateUser(updateUserParam, signatureFile);
if (result) {
return HttpResultUtil.assembleCommonResponseResult(CommonResponseEnum.SUCCESS, true, methodDescribe);
} else {
@@ -128,7 +144,43 @@ public class SysUserController extends BaseController {
user.setRoleCodes(sysRoles.stream().map(SysRole::getCode).collect(Collectors.toList()));
user.setRoleNames(sysRoles.stream().map(SysRole::getName).collect(Collectors.toList()));
});
sysUserSignatureService.fillSignatureInfo(result);
return HttpResultUtil.assembleCommonResponseResult(CommonResponseEnum.SUCCESS, result, methodDescribe);
}
@OperateInfo(info = LogEnum.SYSTEM_COMMON)
@GetMapping("/{id}/signature")
@ApiOperation("预览用户签名")
@ApiImplicitParam(name = "id", value = "用户id", required = true)
public ResponseEntity<InputStreamResource> signature(@PathVariable("id") String id) {
SysUserSignature signature = sysUserSignatureService.getActiveByUserId(id);
if (signature == null) {
throw new BusinessException(CommonResponseEnum.FAIL, "用户签名不存在");
}
SysFile sysFile = sysFileService.getActiveById(signature.getFileId());
if (sysFile == null) {
throw new BusinessException(CommonResponseEnum.FAIL, "用户签名文件不存在");
}
SysFileMinioStorageService.StoredObject storedObject =
sysFileMinioStorageService.getObject(sysFile.getBucketName(), sysFile.getObjectName());
String encodedFileName;
try {
encodedFileName = URLEncoder.encode(sysFile.getFileName(), "UTF-8")
.replace("+", "%20");
} catch (Exception exception) {
throw new BusinessException(CommonResponseEnum.FAIL, "读取用户签名失败");
}
MediaType mediaType;
try {
mediaType = sysFile.getContentType() == null ? MediaType.APPLICATION_OCTET_STREAM
: MediaType.parseMediaType(sysFile.getContentType());
} catch (Exception exception) {
mediaType = MediaType.APPLICATION_OCTET_STREAM;
}
return ResponseEntity.ok()
.contentType(mediaType)
.header(HttpHeaders.CONTENT_DISPOSITION, "inline; filename*=UTF-8''" + encodedFileName)
.body(new InputStreamResource(storedObject.getInputStream()));
}
}

View File

@@ -0,0 +1,10 @@
package com.njcn.gather.user.user.mapper;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.njcn.gather.user.user.pojo.po.SysFile;
/**
* 通用文件 Mapper。
*/
public interface SysFileMapper extends BaseMapper<SysFile> {
}

View File

@@ -0,0 +1,10 @@
package com.njcn.gather.user.user.mapper;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.njcn.gather.user.user.pojo.po.SysUserSignature;
/**
* 用户签名 Mapper。
*/
public interface SysUserSignatureMapper extends BaseMapper<SysUserSignature> {
}

View File

@@ -0,0 +1,19 @@
package com.njcn.gather.user.user.pojo.constant;
/**
* 通用文件常量。
*/
public final class SysFileConst {
private SysFileConst() {
}
public static final Integer STATUS_DELETED = 0;
public static final Integer STATUS_NORMAL = 1;
public static final String STORAGE_TYPE_MINIO = "MINIO";
public static final String BIZ_TYPE_USER_SIGNATURE = "USER_SIGNATURE";
public static final String BIZ_DIR_USER_SIGNATURE = "user-signature";
public static final String BUCKET_KEY_USER_SIGNATURE = "user-signature";
}

View File

@@ -0,0 +1,43 @@
package com.njcn.gather.user.user.pojo.constant;
import java.util.Arrays;
import java.util.Collections;
import java.util.LinkedHashSet;
import java.util.Set;
/**
* 通用文件扩展名常量。
*/
public final class SysFileExtensionConst {
private SysFileExtensionConst() {
}
/**
* 图片文件扩展名,适用于签名、图片附件等场景。
*/
public static final Set<String> IMAGE_EXTENSIONS = unmodifiableSet(
"png", "jpg", "jpeg", "bmp", "webp");
/**
* 常见文档文件扩展名,适用于文档、台账、说明文件等场景。
*/
public static final Set<String> DOCUMENT_EXTENSIONS = unmodifiableSet(
"doc", "docx", "xls", "xlsx", "txt", "pdf");
/**
* 通用上传扩展名,供后续文件业务按需复用。
*/
public static final Set<String> COMMON_UPLOAD_EXTENSIONS = merge(IMAGE_EXTENSIONS, DOCUMENT_EXTENSIONS);
private static Set<String> merge(Set<String> left, Set<String> right) {
LinkedHashSet<String> merged = new LinkedHashSet<String>();
merged.addAll(left);
merged.addAll(right);
return Collections.unmodifiableSet(merged);
}
private static Set<String> unmodifiableSet(String... values) {
return Collections.unmodifiableSet(new LinkedHashSet<String>(Arrays.asList(values)));
}
}

View File

@@ -0,0 +1,48 @@
package com.njcn.gather.user.user.pojo.po;
import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;
import com.njcn.db.mybatisplus.bo.BaseEntity;
import lombok.Data;
import lombok.EqualsAndHashCode;
import java.io.Serializable;
/**
* 通用文件记录。
*/
@Data
@EqualsAndHashCode(callSuper = true)
@TableName("sys_file")
public class SysFile extends BaseEntity implements Serializable {
private static final long serialVersionUID = -5516228278748156912L;
@TableId("id")
private String id;
@TableField("biz_type")
private String bizType;
@TableField("file_name")
private String fileName;
@TableField("bucket_name")
private String bucketName;
@TableField("object_name")
private String objectName;
@TableField("content_type")
private String contentType;
@TableField("file_size")
private Long fileSize;
@TableField("storage_type")
private String storageType;
@TableField("status")
private Integer status;
}

View File

@@ -94,5 +94,11 @@ public class SysUser extends BaseEntity implements Serializable {
@TableField(exist = false)
private List<String> roleNames;
@TableField(exist = false)
private String signatureFileId;
@TableField(exist = false)
private String signatureFileName;
}

View File

@@ -0,0 +1,36 @@
package com.njcn.gather.user.user.pojo.po;
import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;
import com.njcn.db.mybatisplus.bo.BaseEntity;
import lombok.Data;
import lombok.EqualsAndHashCode;
import java.io.Serializable;
/**
* 用户签名记录。
*/
@Data
@EqualsAndHashCode(callSuper = true)
@TableName("sys_user_signature")
public class SysUserSignature extends BaseEntity implements Serializable {
private static final long serialVersionUID = 5765362638635953846L;
@TableId("id")
private String id;
@TableField("user_id")
private String userId;
@TableField("user_name")
private String userName;
@TableField("file_id")
private String fileId;
@TableField("status")
private Integer status;
}

View File

@@ -0,0 +1,55 @@
package com.njcn.gather.user.user.service;
import com.baomidou.mybatisplus.extension.service.IService;
import com.njcn.gather.user.user.pojo.po.SysFile;
import org.springframework.web.multipart.MultipartFile;
import java.util.List;
/**
* 通用文件服务。
*/
public interface ISysFileService extends IService<SysFile> {
/**
* 上传并保存 MinIO 文件记录。
*
* @param bizType 业务类型
* @param bucketKey 桶配置键
* @param bizDir MinIO 目录
* @param ownerId 业务归属ID
* @param file 上传文件
* @return 文件记录
*/
SysFile saveMinioFile(String bizType, String bucketKey, String bizDir, String ownerId, MultipartFile file);
/**
* 根据 ID 查询有效文件。
*
* @param fileId 文件ID
* @return 文件记录
*/
SysFile getActiveById(String fileId);
/**
* 按文件 ID 列表查询有效文件。
*
* @param fileIds 文件ID列表
* @return 文件记录
*/
List<SysFile> listActiveByIds(List<String> fileIds);
/**
* 逻辑删除文件记录。
*
* @param file 文件记录
*/
void markDeleted(SysFile file);
/**
* 删除 MinIO 对象。
*
* @param file 文件记录
*/
void deleteObjectQuietly(SysFile file);
}

View File

@@ -4,6 +4,7 @@ import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.IService;
import com.njcn.gather.user.user.pojo.param.SysUserParam;
import com.njcn.gather.user.user.pojo.po.SysUser;
import org.springframework.web.multipart.MultipartFile;
import java.util.List;
@@ -64,7 +65,7 @@ public interface ISysUserService extends IService<SysUser> {
* @param addUserParam 新增用户参数
* @return 结果true表示新增成功false表示新增失败
*/
boolean addUser(SysUserParam.SysUserAddParam addUserParam);
boolean addUser(SysUserParam.SysUserAddParam addUserParam, MultipartFile signatureFile);
/**
* 更新用户
@@ -72,7 +73,7 @@ public interface ISysUserService extends IService<SysUser> {
* @param updateUserParam 更新用户参数
* @return 结果true表示更新成功false表示更新失败
*/
boolean updateUser(SysUserParam.SysUserUpdateParam updateUserParam);
boolean updateUser(SysUserParam.SysUserUpdateParam updateUserParam, MultipartFile signatureFile);
/**
* 修改密码

View File

@@ -0,0 +1,61 @@
package com.njcn.gather.user.user.service;
import com.baomidou.mybatisplus.extension.service.IService;
import com.njcn.gather.user.user.pojo.po.SysUser;
import com.njcn.gather.user.user.pojo.po.SysUserSignature;
import java.util.List;
/**
* 用户签名服务。
*/
public interface ISysUserSignatureService extends IService<SysUserSignature> {
/**
* 回填用户签名展示信息。
*
* @param users 用户列表
*/
void fillSignatureInfo(List<SysUser> users);
/**
* 查询用户当前签名。
*
* @param userId 用户ID
* @return 签名记录
*/
SysUserSignature getActiveByUserId(String userId);
/**
* 保存或更新签名元数据。
*
* @param userId 用户ID
* @param userName 用户名
* @param fileId 通用文件ID
* @return 旧签名记录,若不存在则返回 null
*/
SysUserSignature saveOrUpdateSignature(String userId, String userName, String fileId);
/**
* 同步签名记录中的用户名。
*
* @param userId 用户ID
* @param userName 用户名
*/
void syncUserName(String userId, String userName);
/**
* 查询用户签名列表。
*
* @param userIds 用户ID列表
* @return 签名记录
*/
List<SysUserSignature> listActiveByUserIds(List<String> userIds);
/**
* 逻辑删除用户签名记录。
*
* @param signatures 签名记录
*/
void markDeleted(List<SysUserSignature> signatures);
}

View File

@@ -0,0 +1,98 @@
package com.njcn.gather.user.user.service.impl;
import cn.hutool.core.collection.CollUtil;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.njcn.common.pojo.enums.response.CommonResponseEnum;
import com.njcn.common.pojo.exception.BusinessException;
import com.njcn.gather.user.user.component.SysFileMinioStorageService;
import com.njcn.gather.user.user.config.StorageMinioProperties;
import com.njcn.gather.user.user.mapper.SysFileMapper;
import com.njcn.gather.user.user.pojo.constant.SysFileConst;
import com.njcn.gather.user.user.pojo.po.SysFile;
import com.njcn.gather.user.user.service.ISysFileService;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;
import java.util.Collections;
import java.util.List;
import java.util.UUID;
/**
* 通用文件服务实现。
*/
@Service
@RequiredArgsConstructor
public class SysFileServiceImpl extends ServiceImpl<SysFileMapper, SysFile> implements ISysFileService {
private final StorageMinioProperties storageMinioProperties;
private final SysFileMinioStorageService sysFileMinioStorageService;
@Override
public SysFile saveMinioFile(String bizType, String bucketKey, String bizDir, String ownerId, MultipartFile file) {
String fileId = UUID.randomUUID().toString().replace("-", "");
String bucketName = storageMinioProperties.requireBucket(bucketKey);
SysFileMinioStorageService.StoredObjectInfo storedObjectInfo =
sysFileMinioStorageService.save(bucketName, bizDir, ownerId, fileId, file);
SysFile sysFile = new SysFile();
sysFile.setId(fileId);
sysFile.setBizType(bizType);
sysFile.setFileName(storedObjectInfo.getFileName());
sysFile.setBucketName(storedObjectInfo.getBucketName());
sysFile.setObjectName(storedObjectInfo.getObjectName());
sysFile.setContentType(storedObjectInfo.getContentType());
sysFile.setFileSize(storedObjectInfo.getFileSize());
sysFile.setStorageType(SysFileConst.STORAGE_TYPE_MINIO);
sysFile.setStatus(SysFileConst.STATUS_NORMAL);
try {
boolean saved = this.save(sysFile);
if (!saved) {
sysFileMinioStorageService.deleteQuietly(sysFile.getBucketName(), sysFile.getObjectName());
throw new BusinessException(CommonResponseEnum.FAIL, "保存文件记录失败");
}
return sysFile;
} catch (RuntimeException exception) {
sysFileMinioStorageService.deleteQuietly(sysFile.getBucketName(), sysFile.getObjectName());
throw exception;
}
}
@Override
public SysFile getActiveById(String fileId) {
if (fileId == null || fileId.trim().isEmpty()) {
return null;
}
return this.lambdaQuery()
.eq(SysFile::getId, fileId)
.eq(SysFile::getStatus, SysFileConst.STATUS_NORMAL)
.one();
}
@Override
public List<SysFile> listActiveByIds(List<String> fileIds) {
if (CollUtil.isEmpty(fileIds)) {
return Collections.emptyList();
}
return this.lambdaQuery()
.in(SysFile::getId, fileIds)
.eq(SysFile::getStatus, SysFileConst.STATUS_NORMAL)
.list();
}
@Override
public void markDeleted(SysFile file) {
if (file == null) {
return;
}
file.setStatus(SysFileConst.STATUS_DELETED);
this.updateById(file);
}
@Override
public void deleteObjectQuietly(SysFile file) {
if (file == null) {
return;
}
sysFileMinioStorageService.deleteQuietly(file.getBucketName(), file.getObjectName());
}
}

View File

@@ -13,35 +13,43 @@ import com.njcn.db.mybatisplus.constant.DbConstant;
import com.njcn.gather.user.pojo.constant.RoleConst;
import com.njcn.gather.user.pojo.constant.UserConst;
import com.njcn.gather.user.pojo.enums.UserResponseEnum;
import com.njcn.gather.user.user.pojo.constant.SysFileExtensionConst;
import com.njcn.gather.user.user.mapper.SysUserMapper;
import com.njcn.gather.user.user.pojo.constant.SysFileConst;
import com.njcn.gather.user.user.pojo.param.SysUserParam;
import com.njcn.gather.user.user.pojo.po.SysFile;
import com.njcn.gather.user.user.pojo.po.SysRole;
import com.njcn.gather.user.user.pojo.po.SysUser;
import com.njcn.gather.user.user.pojo.po.SysUserSignature;
import com.njcn.gather.user.user.service.ISysFileService;
import com.njcn.gather.user.user.service.ISysUserRoleService;
import com.njcn.gather.user.user.service.ISysUserSignatureService;
import com.njcn.gather.user.user.service.ISysUserService;
import com.njcn.web.factory.PageFactory;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.BeanUtils;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.multipart.MultipartFile;
import java.util.Collections;
import java.util.List;
import java.util.Objects;
import java.util.Locale;
import java.util.UUID;
import java.util.stream.Collectors;
/**
* @author caozehui
* @date 2024-11-08
*/
@Slf4j
@Service
@RequiredArgsConstructor
public class SysUserServiceImpl extends ServiceImpl<SysUserMapper, SysUser> implements ISysUserService {
private final ISysUserRoleService sysUserRoleService;
private final ISysFileService sysFileService;
private final ISysUserSignatureService sysUserSignatureService;
@Override
public Page<SysUser> listUser(SysUserParam.SysUserQueryParam queryParam) {
@@ -65,6 +73,7 @@ public class SysUserServiceImpl extends ServiceImpl<SysUserMapper, SysUser> impl
sysUser.setRoleIds(sysRoles.stream().map(SysRole::getId).collect(Collectors.toList()));
sysUser.setRoleNames(sysRoles.stream().map(SysRole::getName).collect(Collectors.toList()));
});
sysUserSignatureService.fillSignatureInfo(page.getRecords());
return page;
}
@@ -105,7 +114,7 @@ public class SysUserServiceImpl extends ServiceImpl<SysUserMapper, SysUser> impl
@Override
@Transactional
public boolean addUser(SysUserParam.SysUserAddParam addUserParam) {
public boolean addUser(SysUserParam.SysUserAddParam addUserParam, MultipartFile signatureFile) {
addUserParam.setName(addUserParam.getName().trim());
addUserParam.setLoginName(addUserParam.getLoginName().trim());
if (UserConst.SUPER_ADMIN.equals(addUserParam.getLoginName())) {
@@ -117,26 +126,79 @@ public class SysUserServiceImpl extends ServiceImpl<SysUserMapper, SysUser> impl
checkRepeat(addUserParam, false, null);
SysUser sysUser = new SysUser();
BeanUtils.copyProperties(addUserParam, sysUser);
sysUser.setId(UUID.randomUUID().toString().replace("-", ""));
String secretkey = Sm4Utils.globalSecretKey;
Sm4Utils sm4 = new Sm4Utils(secretkey);
sysUser.setPassword(sm4.encryptData_ECB(sysUser.getPassword()));
sysUser.setLoginTime(LocalDateTimeUtil.now());
sysUser.setLoginErrorTimes(0);
sysUser.setState(UserConst.STATE_ENABLE);
boolean result = this.save(sysUser);
sysUserRoleService.addUserRole(sysUser.getId(), addUserParam.getRoleIds());
return result;
MultipartFile normalizedSignature = normalizeSignatureFile(signatureFile);
SysFile sysFile = null;
if (normalizedSignature != null) {
sysFile = sysFileService.saveMinioFile(SysFileConst.BIZ_TYPE_USER_SIGNATURE,
SysFileConst.BUCKET_KEY_USER_SIGNATURE, SysFileConst.BIZ_DIR_USER_SIGNATURE,
sysUser.getId(), normalizedSignature);
}
try {
boolean result = this.save(sysUser);
if (result) {
sysUserRoleService.addUserRole(sysUser.getId(), addUserParam.getRoleIds());
if (sysFile != null) {
sysUserSignatureService.saveOrUpdateSignature(sysUser.getId(), sysUser.getName(), sysFile.getId());
}
}
if (!result && sysFile != null) {
cleanupNewFile(sysFile);
}
return result;
} catch (RuntimeException exception) {
if (sysFile != null) {
cleanupNewFile(sysFile);
}
throw exception;
}
}
@Override
@Transactional
public boolean updateUser(SysUserParam.SysUserUpdateParam updateUserParam) {
public boolean updateUser(SysUserParam.SysUserUpdateParam updateUserParam, MultipartFile signatureFile) {
updateUserParam.setName(updateUserParam.getName().trim());
checkRepeat(updateUserParam, true, updateUserParam.getId());
SysUser sysUser = new SysUser();
BeanUtils.copyProperties(updateUserParam, sysUser);
sysUserRoleService.updateUserRole(sysUser.getId(), updateUserParam.getRoleIds());
return this.updateById(sysUser);
MultipartFile normalizedSignature = normalizeSignatureFile(signatureFile);
SysFile newSysFile = null;
if (normalizedSignature != null) {
newSysFile = sysFileService.saveMinioFile(SysFileConst.BIZ_TYPE_USER_SIGNATURE,
SysFileConst.BUCKET_KEY_USER_SIGNATURE, SysFileConst.BIZ_DIR_USER_SIGNATURE,
sysUser.getId(), normalizedSignature);
}
try {
boolean updated = this.updateById(sysUser);
if (!updated) {
if (newSysFile != null) {
cleanupNewFile(newSysFile);
}
return false;
}
sysUserRoleService.updateUserRole(sysUser.getId(), updateUserParam.getRoleIds());
if (newSysFile != null) {
SysUserSignature oldSignature = sysUserSignatureService.saveOrUpdateSignature(
sysUser.getId(), sysUser.getName(), newSysFile.getId());
if (oldSignature != null) {
cleanupOldFile(oldSignature.getFileId());
}
} else {
sysUserSignatureService.syncUserName(sysUser.getId(), sysUser.getName());
}
return true;
} catch (RuntimeException exception) {
if (newSysFile != null) {
cleanupNewFile(newSysFile);
}
throw exception;
}
}
@Override
@@ -170,12 +232,18 @@ public class SysUserServiceImpl extends ServiceImpl<SysUserMapper, SysUser> impl
}
}
}
List<SysUserSignature> signatures = sysUserSignatureService.listActiveByUserIds(ids);
// 删除用户角色关联数据
sysUserRoleService.deleteUserRoleByUserIds(ids);
return this.lambdaUpdate()
boolean result = this.lambdaUpdate()
.set(SysUser::getState, UserConst.STATE_DELETE)
.in(SysUser::getId, ids)
.update();
if (result) {
sysUserSignatureService.markDeleted(signatures);
signatures.forEach(signature -> cleanupOldFile(signature.getFileId()));
}
return result;
}
@Override
@@ -210,4 +278,37 @@ public class SysUserServiceImpl extends ServiceImpl<SysUserMapper, SysUser> impl
throw new BusinessException(UserResponseEnum.REGISTER_EMAIL_FAIL);
}
}
private MultipartFile normalizeSignatureFile(MultipartFile signatureFile) {
if (signatureFile == null || signatureFile.isEmpty()) {
return null;
}
String originalFileName = signatureFile.getOriginalFilename();
if (originalFileName == null || originalFileName.trim().isEmpty()) {
throw new BusinessException(UserResponseEnum.SIGNATURE_FILE_NAME_EMPTY);
}
int dotIndex = originalFileName.lastIndexOf('.');
if (dotIndex < 0 || dotIndex == originalFileName.length() - 1) {
throw new BusinessException(UserResponseEnum.SIGNATURE_FILE_TYPE_ERROR);
}
String extension = originalFileName.substring(dotIndex + 1).toLowerCase(Locale.ENGLISH);
if (!SysFileExtensionConst.IMAGE_EXTENSIONS.contains(extension)) {
throw new BusinessException(UserResponseEnum.SIGNATURE_FILE_TYPE_ERROR);
}
return signatureFile;
}
private void cleanupNewFile(SysFile sysFile) {
sysFileService.markDeleted(sysFile);
sysFileService.deleteObjectQuietly(sysFile);
}
private void cleanupOldFile(String fileId) {
SysFile oldFile = sysFileService.getActiveById(fileId);
if (oldFile == null) {
return;
}
sysFileService.markDeleted(oldFile);
sysFileService.deleteObjectQuietly(oldFile);
}
}

View File

@@ -0,0 +1,134 @@
package com.njcn.gather.user.user.service.impl;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.njcn.gather.user.user.mapper.SysUserSignatureMapper;
import com.njcn.gather.user.user.pojo.constant.SysFileConst;
import com.njcn.gather.user.user.pojo.po.SysFile;
import com.njcn.gather.user.user.pojo.po.SysUser;
import com.njcn.gather.user.user.pojo.po.SysUserSignature;
import com.njcn.gather.user.user.service.ISysFileService;
import com.njcn.gather.user.user.service.ISysUserSignatureService;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Service;
import java.util.Collections;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.UUID;
import java.util.stream.Collectors;
/**
* 用户签名服务实现。
*/
@Service
@RequiredArgsConstructor
public class SysUserSignatureServiceImpl extends ServiceImpl<SysUserSignatureMapper, SysUserSignature>
implements ISysUserSignatureService {
private final ISysFileService sysFileService;
@Override
public void fillSignatureInfo(List<SysUser> users) {
if (CollUtil.isEmpty(users)) {
return;
}
Map<String, SysUserSignature> signatureMap = new LinkedHashMap<String, SysUserSignature>();
List<String> userIds = users.stream().map(SysUser::getId).collect(Collectors.toList());
for (SysUserSignature signature : listActiveByUserIds(userIds)) {
signatureMap.put(signature.getUserId(), signature);
}
Map<String, SysFile> sysFileMap = new LinkedHashMap<String, SysFile>();
List<String> fileIds = signatureMap.values().stream().map(SysUserSignature::getFileId).collect(Collectors.toList());
for (SysFile sysFile : sysFileService.listActiveByIds(fileIds)) {
sysFileMap.put(sysFile.getId(), sysFile);
}
for (SysUser user : users) {
SysUserSignature signature = signatureMap.get(user.getId());
if (signature == null) {
user.setSignatureFileId(null);
user.setSignatureFileName(null);
continue;
}
user.setSignatureFileId(signature.getFileId());
SysFile sysFile = sysFileMap.get(signature.getFileId());
user.setSignatureFileName(sysFile == null ? null : sysFile.getFileName());
}
}
@Override
public SysUserSignature getActiveByUserId(String userId) {
if (StrUtil.isBlank(userId)) {
return null;
}
return this.lambdaQuery()
.eq(SysUserSignature::getUserId, userId)
.eq(SysUserSignature::getStatus, SysFileConst.STATUS_NORMAL)
.one();
}
@Override
public SysUserSignature saveOrUpdateSignature(String userId, String userName, String fileId) {
SysUserSignature existing = getActiveByUserId(userId);
if (existing == null) {
SysUserSignature signature = new SysUserSignature();
signature.setId(UUID.randomUUID().toString().replace("-", ""));
signature.setUserId(userId);
signature.setUserName(userName);
signature.setFileId(fileId);
signature.setStatus(SysFileConst.STATUS_NORMAL);
this.save(signature);
return null;
}
SysUserSignature oldSignature = copySignature(existing);
existing.setUserName(userName);
existing.setFileId(fileId);
existing.setStatus(SysFileConst.STATUS_NORMAL);
this.updateById(existing);
return oldSignature;
}
@Override
public void syncUserName(String userId, String userName) {
SysUserSignature signature = getActiveByUserId(userId);
if (signature == null || StrUtil.equals(signature.getUserName(), userName)) {
return;
}
signature.setUserName(userName);
this.updateById(signature);
}
@Override
public List<SysUserSignature> listActiveByUserIds(List<String> userIds) {
if (CollUtil.isEmpty(userIds)) {
return Collections.emptyList();
}
return this.lambdaQuery()
.in(SysUserSignature::getUserId, userIds)
.eq(SysUserSignature::getStatus, SysFileConst.STATUS_NORMAL)
.list();
}
@Override
public void markDeleted(List<SysUserSignature> signatures) {
if (CollUtil.isEmpty(signatures)) {
return;
}
for (SysUserSignature signature : signatures) {
signature.setStatus(SysFileConst.STATUS_DELETED);
this.updateById(signature);
}
}
private SysUserSignature copySignature(SysUserSignature source) {
SysUserSignature copy = new SysUserSignature();
copy.setId(source.getId());
copy.setUserId(source.getUserId());
copy.setUserName(source.getUserName());
copy.setFileId(source.getFileId());
copy.setStatus(source.getStatus());
return copy;
}
}