27 changed files with 1830 additions and 5 deletions
@ -0,0 +1,47 @@ |
|||||||
|
package com.mh.web.controller.ai; |
||||||
|
|
||||||
|
import com.mh.common.core.domain.dto.AsyncTaskDTO; |
||||||
|
import com.mh.system.service.ai.impl.AsyncReportServiceImpl; |
||||||
|
import com.mh.system.service.ai.impl.TaskStoreService; |
||||||
|
import org.springframework.http.MediaType; |
||||||
|
import org.springframework.http.ResponseEntity; |
||||||
|
import org.springframework.web.bind.annotation.*; |
||||||
|
import org.springframework.web.servlet.mvc.method.annotation.SseEmitter; |
||||||
|
|
||||||
|
import java.util.Map; |
||||||
|
|
||||||
|
/** |
||||||
|
* @author LJF |
||||||
|
* @version 1.0 |
||||||
|
* @project EEMCS |
||||||
|
* @description 一步报表controller |
||||||
|
* @date 2026-02-27 11:36:57 |
||||||
|
*/ |
||||||
|
@RestController |
||||||
|
@RequestMapping("/api/reports/async") |
||||||
|
public class AsyncReportController { |
||||||
|
private final AsyncReportServiceImpl asyncReportService; |
||||||
|
private final TaskStoreService taskStore; |
||||||
|
|
||||||
|
public AsyncReportController(AsyncReportServiceImpl asyncReportService, TaskStoreService taskStore) { |
||||||
|
this.asyncReportService = asyncReportService; |
||||||
|
this.taskStore = taskStore; |
||||||
|
} |
||||||
|
|
||||||
|
@PostMapping |
||||||
|
public ResponseEntity<Map<String, String>> submitTask(@RequestBody String userMessage) { |
||||||
|
String taskId = asyncReportService.createAsyncTask(userMessage); |
||||||
|
return ResponseEntity.accepted().body(Map.of("taskId", taskId)); |
||||||
|
} |
||||||
|
|
||||||
|
@GetMapping(value = "/{taskId}/subscribe", produces = MediaType.TEXT_EVENT_STREAM_VALUE) |
||||||
|
public SseEmitter subscribe(@PathVariable String taskId) { |
||||||
|
return asyncReportService.registerEmitter(taskId); |
||||||
|
} |
||||||
|
|
||||||
|
@GetMapping("/{taskId}") |
||||||
|
public ResponseEntity<AsyncTaskDTO> getTask(@PathVariable String taskId) { |
||||||
|
AsyncTaskDTO task = taskStore.get(taskId); |
||||||
|
return task != null ? ResponseEntity.ok(task) : ResponseEntity.notFound().build(); |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,39 @@ |
|||||||
|
package com.mh.web.controller.ai; |
||||||
|
|
||||||
|
import org.springframework.beans.factory.annotation.Value; |
||||||
|
import org.springframework.core.io.Resource; |
||||||
|
import org.springframework.core.io.UrlResource; |
||||||
|
import org.springframework.http.HttpHeaders; |
||||||
|
import org.springframework.http.ResponseEntity; |
||||||
|
import org.springframework.web.bind.annotation.GetMapping; |
||||||
|
import org.springframework.web.bind.annotation.PathVariable; |
||||||
|
import org.springframework.web.bind.annotation.RequestMapping; |
||||||
|
|
||||||
|
import java.nio.file.Path; |
||||||
|
import java.nio.file.Paths; |
||||||
|
|
||||||
|
/** |
||||||
|
* @author LJF |
||||||
|
* @version 1.0 |
||||||
|
* @project EEMCS |
||||||
|
* @description 文件下载链接 |
||||||
|
* @date 2026-02-27 11:40:07 |
||||||
|
*/ |
||||||
|
@RequestMapping("/api/files") |
||||||
|
public class FileDownloadController { |
||||||
|
@Value("${mh.profile:/tmp/energy-reports}") |
||||||
|
private String storageDir; |
||||||
|
|
||||||
|
@GetMapping("/{filename}") |
||||||
|
public ResponseEntity<Resource> downloadFile(@PathVariable String filename) throws Exception { |
||||||
|
Path filePath = Paths.get(storageDir).resolve(filename); |
||||||
|
Resource resource = new UrlResource(filePath.toUri()); |
||||||
|
if (resource.exists()) { |
||||||
|
return ResponseEntity.ok() |
||||||
|
.header(HttpHeaders.CONTENT_DISPOSITION, "attachment; filename=\"" + filename + "\"") |
||||||
|
.body(resource); |
||||||
|
} |
||||||
|
return ResponseEntity.notFound().build(); |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
@ -0,0 +1,35 @@ |
|||||||
|
package com.mh.common.config; |
||||||
|
|
||||||
|
import org.springframework.beans.factory.annotation.Value; |
||||||
|
import org.springframework.stereotype.Component; |
||||||
|
import org.springframework.web.servlet.support.ServletUriComponentsBuilder; |
||||||
|
|
||||||
|
import java.io.IOException; |
||||||
|
import java.nio.file.Files; |
||||||
|
import java.nio.file.Path; |
||||||
|
import java.nio.file.Paths; |
||||||
|
|
||||||
|
/** |
||||||
|
* @author LJF |
||||||
|
* @version 1.0 |
||||||
|
* @project EEMCS |
||||||
|
* @description 文件服务类 |
||||||
|
* @date 2026-02-27 11:32:04 |
||||||
|
*/ |
||||||
|
@Component |
||||||
|
public class FileStorageUtil { |
||||||
|
@Value("${mh.profile:/tmp/energy-reports}") |
||||||
|
private String storageDir; |
||||||
|
|
||||||
|
public String saveFile(String filename, byte[] content) throws IOException { |
||||||
|
Path dir = Paths.get(storageDir); |
||||||
|
if (!Files.exists(dir)) Files.createDirectories(dir); |
||||||
|
Path filePath = dir.resolve(filename); |
||||||
|
Files.write(filePath, content); |
||||||
|
// 生成可下载的URL（假设当前应用提供静态资源访问或下载接口）
|
||||||
|
return ServletUriComponentsBuilder.fromCurrentContextPath() |
||||||
|
.path("/api/files/") |
||||||
|
.path(filename) |
||||||
|
.toUriString(); |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,24 @@ |
|||||||
|
package com.mh.common.core.domain.dto; |
||||||
|
|
||||||
|
import lombok.Data; |
||||||
|
|
||||||
|
import java.time.LocalDateTime; |
||||||
|
|
||||||
|
/** |
||||||
|
* @author LJF |
||||||
|
* @version 1.0 |
||||||
|
* @project EEMCS |
||||||
|
* @description 任务状态dto |
||||||
|
* @date 2026-02-27 10:22:40 |
||||||
|
*/ |
||||||
|
@Data |
||||||
|
public class AsyncTaskDTO { |
||||||
|
|
||||||
|
private String taskId; |
||||||
|
private String status; // PROCESSING, COMPLETED, FAILED
|
||||||
|
private String message; // 成功时的消息或失败原因
|
||||||
|
private String downloadUrl; // 生成文件的下载URL（成功后填充）
|
||||||
|
private LocalDateTime createTime; |
||||||
|
private LocalDateTime completeTime; |
||||||
|
|
||||||
|
} |
||||||
@ -0,0 +1,99 @@ |
|||||||
|
package com.mh.common.core.domain.dto; |
||||||
|
|
||||||
|
import com.fasterxml.jackson.annotation.JsonPropertyDescription; |
||||||
|
import lombok.Builder; |
||||||
|
import lombok.Data; |
||||||
|
|
||||||
|
import java.util.List; |
||||||
|
|
||||||
|
/** |
||||||
|
* @author LJF |
||||||
|
* @version 1.0 |
||||||
|
* @project EEMCS |
||||||
|
* @description ai机房能效报表 |
||||||
|
* @date 2026-02-27 10:11:57 |
||||||
|
*/ |
||||||
|
@Data |
||||||
|
@Builder |
||||||
|
public class EnergyReportDTO { |
||||||
|
|
||||||
|
@JsonPropertyDescription("报表生成日期") |
||||||
|
private String reportDate; |
||||||
|
|
||||||
|
@JsonPropertyDescription("整体能效评分 (0-100)") |
||||||
|
private Integer overallScore; |
||||||
|
|
||||||
|
@JsonPropertyDescription("系统综合能效比") |
||||||
|
private Double systemEER; |
||||||
|
|
||||||
|
@JsonPropertyDescription("总能耗 (kWh)") |
||||||
|
private Double totalEnergyConsumption; |
||||||
|
|
||||||
|
@JsonPropertyDescription("各设备能耗明细") |
||||||
|
private List<DeviceEnergy> deviceDetails; |
||||||
|
|
||||||
|
@JsonPropertyDescription("异常情况列表") |
||||||
|
private List<Anomaly> anomalies; |
||||||
|
|
||||||
|
@JsonPropertyDescription("优化建议") |
||||||
|
private List<Suggestion> suggestions; |
||||||
|
|
||||||
|
@Data |
||||||
|
@Builder |
||||||
|
public static class DeviceEnergy { |
||||||
|
@JsonPropertyDescription("设备名称") |
||||||
|
private String deviceName; |
||||||
|
|
||||||
|
@JsonPropertyDescription("设备类型 (冷机/冷冻泵/冷却泵/冷却塔)") |
||||||
|
private String deviceType; |
||||||
|
|
||||||
|
@JsonPropertyDescription("能耗 (kWh)") |
||||||
|
private Double energyConsumption; |
||||||
|
|
||||||
|
@JsonPropertyDescription("运行效率 (%)") |
||||||
|
private Double efficiency; |
||||||
|
|
||||||
|
@JsonPropertyDescription("对比基准的偏差率") |
||||||
|
private Double deviationRate; |
||||||
|
} |
||||||
|
|
||||||
|
@Data |
||||||
|
@Builder |
||||||
|
public static class Anomaly { |
||||||
|
@JsonPropertyDescription("异常设备") |
||||||
|
private String deviceName; |
||||||
|
|
||||||
|
@JsonPropertyDescription("异常类型") |
||||||
|
private String anomalyType; |
||||||
|
|
||||||
|
@JsonPropertyDescription("异常描述") |
||||||
|
private String description; |
||||||
|
|
||||||
|
@JsonPropertyDescription("严重程度 (高/中/低)") |
||||||
|
private String severity; |
||||||
|
|
||||||
|
@JsonPropertyDescription("预估能耗损失 (kWh)") |
||||||
|
private Double estimatedLoss; |
||||||
|
} |
||||||
|
|
||||||
|
@Data |
||||||
|
@Builder |
||||||
|
public static class Suggestion { |
||||||
|
@JsonPropertyDescription("建议标题") |
||||||
|
private String title; |
||||||
|
|
||||||
|
@JsonPropertyDescription("建议详细内容") |
||||||
|
private String description; |
||||||
|
|
||||||
|
@JsonPropertyDescription("预期节能效果 (kWh)") |
||||||
|
private Double expectedSaving; |
||||||
|
|
||||||
|
@JsonPropertyDescription("优先级 (高/中/低)") |
||||||
|
private String priority; |
||||||
|
|
||||||
|
@JsonPropertyDescription("建议执行的操作 (如:检查/维修/调整参数)") |
||||||
|
private String action; |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
} |
||||||
@ -0,0 +1,21 @@ |
|||||||
|
package com.mh.common.core.domain.dto; |
||||||
|
|
||||||
|
import lombok.Data; |
||||||
|
|
||||||
|
/** |
||||||
|
* @author LJF |
||||||
|
* @version 1.0 |
||||||
|
* @project EEMCS |
||||||
|
* @description 接收前端经过deepseek结构化后的请求 |
||||||
|
* @date 2026-02-27 10:18:04 |
||||||
|
*/ |
||||||
|
@Data |
||||||
|
public class ReportRequestDTO { |
||||||
|
|
||||||
|
private String reportType; // 报表类型，如 "能效报表"
|
||||||
|
private String timeRange; // 时间范围，如 "昨天"、"2026-02-01至2026-02-07"
|
||||||
|
private String format; // 输出格式: word, excel, pdf
|
||||||
|
private Boolean watermark; // 是否需要水印 (默认true)
|
||||||
|
private String watermarkText; // 自定义水印文字，若为空则使用默认
|
||||||
|
|
||||||
|
} |
||||||
@ -0,0 +1,29 @@ |
|||||||
|
package com.mh.framework.config; |
||||||
|
|
||||||
|
import org.springframework.context.annotation.Configuration; |
||||||
|
import org.springframework.scheduling.annotation.EnableAsync; |
||||||
|
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor; |
||||||
|
|
||||||
|
import java.util.concurrent.Executor; |
||||||
|
|
||||||
|
/** |
||||||
|
* @author LJF |
||||||
|
* @version 1.0 |
||||||
|
* @project EEMCS |
||||||
|
* @description 自定义线程池 |
||||||
|
* @date 2026-02-27 11:41:52 |
||||||
|
*/ |
||||||
|
@Configuration |
||||||
|
@EnableAsync |
||||||
|
public class AsyncConfig { |
||||||
|
// 可选：自定义线程池
|
||||||
|
public Executor taskExecutor() { |
||||||
|
ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor(); |
||||||
|
executor.setCorePoolSize(5); |
||||||
|
executor.setMaxPoolSize(10); |
||||||
|
executor.setQueueCapacity(100); |
||||||
|
executor.setThreadNamePrefix("async-"); |
||||||
|
executor.initialize(); |
||||||
|
return executor; |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,17 @@ |
|||||||
|
package com.mh.system.service.ai; |
||||||
|
|
||||||
|
import com.mh.common.core.domain.dto.EnergyReportDTO; |
||||||
|
|
||||||
|
import java.util.function.Consumer; |
||||||
|
|
||||||
|
/** |
||||||
|
* @author LJF |
||||||
|
* @version 1.0 |
||||||
|
* @project EEMCS |
||||||
|
* @description 流式推理版 |
||||||
|
* @date 2026-02-27 10:36:03 |
||||||
|
*/ |
||||||
|
public interface IEnergyReportService { |
||||||
|
|
||||||
|
public EnergyReportDTO generateReportWithReasoning(String rawData, Consumer<String> reasoningConsumer); |
||||||
|
} |
||||||
@ -0,0 +1,20 @@ |
|||||||
|
package com.mh.system.service.ai; |
||||||
|
|
||||||
|
import com.mh.common.core.domain.dto.EnergyReportDTO; |
||||||
|
import com.mh.common.core.domain.dto.ReportRequestDTO; |
||||||
|
|
||||||
|
import java.io.IOException; |
||||||
|
|
||||||
|
/** |
||||||
|
* @author LJF |
||||||
|
* @version 1.0 |
||||||
|
* @project EEMCS |
||||||
|
* @description 报表生成服务类 |
||||||
|
* @date 2026-02-27 10:50:42 |
||||||
|
*/ |
||||||
|
public interface IFileGenerationService { |
||||||
|
|
||||||
|
byte[] generateReport(EnergyReportDTO report, ReportRequestDTO request) throws IOException; |
||||||
|
String getFileExtension(); |
||||||
|
|
||||||
|
} |
||||||
@ -0,0 +1,15 @@ |
|||||||
|
package com.mh.system.service.ai; |
||||||
|
|
||||||
|
import com.mh.common.core.domain.dto.ReportRequestDTO; |
||||||
|
|
||||||
|
/** |
||||||
|
* @author LJF |
||||||
|
* @version 1.0 |
||||||
|
* @project EEMCS |
||||||
|
* @description 自然语言解析服务类 |
||||||
|
* @date 2026-02-27 10:25:40 |
||||||
|
*/ |
||||||
|
public interface INLPService { |
||||||
|
|
||||||
|
public ReportRequestDTO parseUserIntent(String userMessage); |
||||||
|
} |
||||||
@ -0,0 +1,114 @@ |
|||||||
|
package com.mh.system.service.ai.impl; |
||||||
|
|
||||||
|
import com.mh.common.config.FileStorageUtil; |
||||||
|
import com.mh.common.core.domain.dto.AsyncTaskDTO; |
||||||
|
import com.mh.common.core.domain.dto.EnergyReportDTO; |
||||||
|
import com.mh.common.core.domain.dto.ReportRequestDTO; |
||||||
|
import com.mh.common.utils.uuid.UUID; |
||||||
|
import com.mh.system.service.ai.IEnergyReportService; |
||||||
|
import com.mh.system.service.ai.INLPService; |
||||||
|
import lombok.extern.slf4j.Slf4j; |
||||||
|
import org.springframework.scheduling.annotation.Async; |
||||||
|
import org.springframework.stereotype.Service; |
||||||
|
import org.springframework.web.servlet.mvc.method.annotation.SseEmitter; |
||||||
|
|
||||||
|
import java.io.IOException; |
||||||
|
import java.time.LocalDateTime; |
||||||
|
import java.util.Map; |
||||||
|
import java.util.concurrent.ConcurrentHashMap; |
||||||
|
|
||||||
|
/** |
||||||
|
* @author LJF |
||||||
|
* @version 1.0 |
||||||
|
* @project EEMCS |
||||||
|
* @description 异步任务服务 |
||||||
|
* @date 2026-02-27 11:27:21 |
||||||
|
*/ |
||||||
|
@Service |
||||||
|
@Slf4j |
||||||
|
public class AsyncReportServiceImpl { |
||||||
|
|
||||||
|
private final INLPService nlpService; |
||||||
|
private final IEnergyReportService energyReportService; |
||||||
|
private final ReportOrchestrationService orchestrationService; |
||||||
|
private final TaskStoreService taskStore; |
||||||
|
private final FileStorageUtil fileStorageUtil; |
||||||
|
|
||||||
|
private final Map<String, SseEmitter> emitters = new ConcurrentHashMap<>(); |
||||||
|
|
||||||
|
public AsyncReportServiceImpl(INLPService nlpService, |
||||||
|
IEnergyReportService energyReportService, |
||||||
|
ReportOrchestrationService orchestrationService, |
||||||
|
TaskStoreService taskStore, |
||||||
|
FileStorageUtil fileStorageUtil) { |
||||||
|
this.nlpService = nlpService; |
||||||
|
this.energyReportService = energyReportService; |
||||||
|
this.orchestrationService = orchestrationService; |
||||||
|
this.taskStore = taskStore; |
||||||
|
this.fileStorageUtil = fileStorageUtil; |
||||||
|
} |
||||||
|
|
||||||
|
public SseEmitter registerEmitter(String taskId) { |
||||||
|
SseEmitter emitter = new SseEmitter(300_000L); |
||||||
|
emitters.put(taskId, emitter); |
||||||
|
emitter.onCompletion(() -> emitters.remove(taskId)); |
||||||
|
emitter.onTimeout(() -> emitters.remove(taskId)); |
||||||
|
return emitter; |
||||||
|
} |
||||||
|
|
||||||
|
public String createAsyncTask(String userMessage) { |
||||||
|
String taskId = UUID.randomUUID().toString(); |
||||||
|
AsyncTaskDTO task = new AsyncTaskDTO(); |
||||||
|
task.setTaskId(taskId); |
||||||
|
task.setStatus("PROCESSING"); |
||||||
|
task.setCreateTime(LocalDateTime.now()); |
||||||
|
taskStore.save(task); |
||||||
|
processTask(taskId, userMessage); |
||||||
|
return taskId; |
||||||
|
} |
||||||
|
|
||||||
|
@Async |
||||||
|
public void processTask(String taskId, String userMessage) { |
||||||
|
try { |
||||||
|
// 解析自然语言
|
||||||
|
ReportRequestDTO request = nlpService.parseUserIntent(userMessage); |
||||||
|
// 获取原始数据（模拟）
|
||||||
|
String rawData = fetchRawData(request.getTimeRange()); |
||||||
|
// 生成报表并实时推送推理过程
|
||||||
|
EnergyReportDTO report = energyReportService.generateReportWithReasoning(rawData, |
||||||
|
reasoningChunk -> sendEvent(taskId, "reasoning", reasoningChunk)); |
||||||
|
// 生成文件
|
||||||
|
ReportOrchestrationService.ReportFile reportFile = orchestrationService.generateFile(report, request); |
||||||
|
// 保存文件并获取下载URL
|
||||||
|
String downloadUrl = fileStorageUtil.saveFile(reportFile.filename(), reportFile.content()); |
||||||
|
// 更新任务状态
|
||||||
|
taskStore.updateStatus(taskId, "COMPLETED", "报表生成成功", downloadUrl); |
||||||
|
sendEvent(taskId, "COMPLETED", downloadUrl); |
||||||
|
} catch (Exception e) { |
||||||
|
log.error("任务处理失败", e); |
||||||
|
taskStore.updateStatus(taskId, "FAILED", e.getMessage(), null); |
||||||
|
sendEvent(taskId, "FAILED", e.getMessage()); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
private void sendEvent(String taskId, String event, Object data) { |
||||||
|
SseEmitter emitter = emitters.get(taskId); |
||||||
|
if (emitter != null) { |
||||||
|
try { |
||||||
|
emitter.send(SseEmitter.event().name(event).data(data)); |
||||||
|
if ("COMPLETED".equals(event) || "FAILED".equals(event)) { |
||||||
|
emitter.complete(); |
||||||
|
} |
||||||
|
} catch (IOException e) { |
||||||
|
log.error("推送事件失败", e); |
||||||
|
emitter.completeWithError(e); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
private String fetchRawData(String timeRange) { |
||||||
|
// 模拟数据，实际应从数据库或IoT平台获取
|
||||||
|
return "{\"date\":\"2026-02-26\",\"outdoorTemp\":32.5,\"coolingLoad\":850}"; |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
@ -0,0 +1,189 @@ |
|||||||
|
package com.mh.system.service.ai.impl; |
||||||
|
|
||||||
|
import com.fasterxml.jackson.databind.ObjectMapper; |
||||||
|
import com.mh.common.core.domain.dto.EnergyReportDTO; |
||||||
|
import com.mh.system.service.ai.IEnergyReportService; |
||||||
|
import lombok.extern.slf4j.Slf4j; |
||||||
|
import org.springframework.ai.chat.client.ChatClient; |
||||||
|
import org.springframework.stereotype.Service; |
||||||
|
import reactor.core.publisher.Flux; |
||||||
|
|
||||||
|
import java.time.LocalDate; |
||||||
|
import java.time.format.DateTimeFormatter; |
||||||
|
import java.util.Map; |
||||||
|
import java.util.function.Consumer; |
||||||
|
import java.util.regex.Matcher; |
||||||
|
import java.util.regex.Pattern; |
||||||
|
|
||||||
|
/** |
||||||
|
* @author LJF |
||||||
|
* @version 1.0 |
||||||
|
* @project EEMCS |
||||||
|
* @description 流式推理版实现类 |
||||||
|
* @date 2026-02-27 10:37:03 |
||||||
|
*/ |
||||||
|
@Slf4j |
||||||
|
@Service |
||||||
|
public class EnergyReportServiceImpl implements IEnergyReportService { |
||||||
|
|
||||||
|
private final ChatClient chatClient; |
||||||
|
private final ObjectMapper objectMapper; |
||||||
|
|
||||||
|
public EnergyReportServiceImpl(ChatClient.Builder builder, ObjectMapper objectMapper) { |
||||||
|
this.chatClient = builder.build(); |
||||||
|
this.objectMapper = objectMapper; |
||||||
|
} |
||||||
|
|
||||||
|
public EnergyReportDTO generateReportWithReasoning(String rawData, Consumer<String> reasoningConsumer) { |
||||||
|
String prompt = """ |
||||||
|
你是一位专业的中央空调系统能效分析专家。 |
||||||
|
原始数据：{rawData} |
||||||
|
分析要求： |
||||||
|
1. 计算系统综合能效比 (EER = 总制冷量 / 总电耗) |
||||||
|
2. 识别能效异常的设备 (效率低于基准值80%视为异常) |
||||||
|
3. 评估各设备能耗占比和运行效率 |
||||||
|
4. 提供具体的优化建议 |
||||||
|
|
||||||
|
请先逐步推理你的分析过程，然后输出最终的 JSON 报表。输出格式如下： |
||||||
|
|
||||||
|
推理过程： |
||||||
|
（分析步骤） |
||||||
|
|
||||||
|
请根据数据生成一份能效分析报表，输出必须为严格的 JSON 格式，包含以下字段： |
||||||
|
- reportDate: 报表生成日期，格式 yyyy-MM-dd |
||||||
|
- overallScore: 整体能效评分 (0-100 的整数) |
||||||
|
- systemEER: 系统综合能效比（数值，保留一位小数） |
||||||
|
- totalEnergyConsumption: 总能耗，单位 kWh |
||||||
|
- deviceDetails: 数组，每个元素包含： |
||||||
|
deviceName: 设备名称 |
||||||
|
deviceType: 设备类型（冷机/冷冻泵/冷却泵/冷却塔） |
||||||
|
energyConsumption: 能耗 (kWh) |
||||||
|
efficiency: 运行效率 (%) |
||||||
|
deviationRate: 对比基准的偏差率 (%) |
||||||
|
- anomalies: 数组，每个元素包含： |
||||||
|
deviceName: 异常设备 |
||||||
|
anomalyType: 异常类型 |
||||||
|
description: 异常描述 |
||||||
|
severity: 严重程度 (高/中/低) |
||||||
|
estimatedLoss: 预估能耗损失 (kWh) |
||||||
|
- suggestions: 数组，每个元素包含： |
||||||
|
title: 建议标题 |
||||||
|
description: 建议详细内容 |
||||||
|
expectedSaving: 预期节能效果 (kWh) |
||||||
|
priority: 优先级 (高/中/低) |
||||||
|
action: 建议执行的操作 |
||||||
|
请先逐步推理你的分析过程，然后输出最终的 JSON 报表。 |
||||||
|
最终报表： |
||||||
|
```json |
||||||
|
<严格按照 EnergyReportDTO JSON 格式输出> |
||||||
|
``` |
||||||
|
今天的日期：{currentDate} |
||||||
|
"""; |
||||||
|
|
||||||
|
Map<String, Object> params = Map.of( |
||||||
|
"rawData", rawData, |
||||||
|
"currentDate", LocalDate.now().format(DateTimeFormatter.ISO_LOCAL_DATE) |
||||||
|
); |
||||||
|
|
||||||
|
StringBuilder fullResponse = new StringBuilder(); |
||||||
|
|
||||||
|
// Flux<String> flux = chatClient.prompt()
|
||||||
|
// .system("你是一个JSON API，但请先详细推理再输出最终JSON。")
|
||||||
|
// .user(userSpec -> userSpec.text(prompt).params(params))
|
||||||
|
// .stream()
|
||||||
|
// .content();
|
||||||
|
//
|
||||||
|
// flux.doOnNext(chunk -> {
|
||||||
|
// fullResponse.append(chunk);
|
||||||
|
// if (!fullResponse.toString().contains("最终报表")) {
|
||||||
|
// if (reasoningConsumer != null) {
|
||||||
|
// reasoningConsumer.accept(chunk);
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
// }).blockLast();
|
||||||
|
|
||||||
|
log.info("AI原始格式输出：{}", fullResponse); |
||||||
|
// String jsonStr = extractJson(fullResponse.toString());
|
||||||
|
String jsonStr = "{\n" + |
||||||
|
" \"reportDate\": \"2026-02-27\",\n" + |
||||||
|
" \"overallScore\": 70,\n" + |
||||||
|
" \"systemEER\": 4.0,\n" + |
||||||
|
" \"totalEnergyConsumption\": 212.5,\n" + |
||||||
|
" \"deviceDetails\": [\n" + |
||||||
|
" {\n" + |
||||||
|
" \"deviceName\": \"冷机-1\",\n" + |
||||||
|
" \"deviceType\": \"冷机\",\n" + |
||||||
|
" \"energyConsumption\": 138.1,\n" + |
||||||
|
" \"efficiency\": 75.0,\n" + |
||||||
|
" \"deviationRate\": -25.0\n" + |
||||||
|
" },\n" + |
||||||
|
" {\n" + |
||||||
|
" \"deviceName\": \"冷冻泵-1\",\n" + |
||||||
|
" \"deviceType\": \"冷冻泵\",\n" + |
||||||
|
" \"energyConsumption\": 31.9,\n" + |
||||||
|
" \"efficiency\": 95.0,\n" + |
||||||
|
" \"deviationRate\": -5.0\n" + |
||||||
|
" },\n" + |
||||||
|
" {\n" + |
||||||
|
" \"deviceName\": \"冷却泵-1\",\n" + |
||||||
|
" \"deviceType\": \"冷却泵\",\n" + |
||||||
|
" \"energyConsumption\": 21.3,\n" + |
||||||
|
" \"efficiency\": 92.0,\n" + |
||||||
|
" \"deviationRate\": -8.0\n" + |
||||||
|
" },\n" + |
||||||
|
" {\n" + |
||||||
|
" \"deviceName\": \"冷却塔-1\",\n" + |
||||||
|
" \"deviceType\": \"冷却塔\",\n" + |
||||||
|
" \"energyConsumption\": 21.3,\n" + |
||||||
|
" \"efficiency\": 88.0,\n" + |
||||||
|
" \"deviationRate\": -12.0\n" + |
||||||
|
" }\n" + |
||||||
|
" ],\n" + |
||||||
|
" \"anomalies\": [\n" + |
||||||
|
" {\n" + |
||||||
|
" \"deviceName\": \"冷机-1\",\n" + |
||||||
|
" \"anomalyType\": \"能效低下\",\n" + |
||||||
|
" \"description\": \"冷机运行效率为75%，低于基准值80%，可能由于换热器污垢或制冷剂不足导致能耗增加。\",\n" + |
||||||
|
" \"severity\": \"高\",\n" + |
||||||
|
" \"estimatedLoss\": 34.5\n" + |
||||||
|
" }\n" + |
||||||
|
" ],\n" + |
||||||
|
" \"suggestions\": [\n" + |
||||||
|
" {\n" + |
||||||
|
" \"title\": \"清洗冷机换热器\",\n" + |
||||||
|
" \"description\": \"检查并清洗冷凝器和蒸发器，清除污垢和沉积物，以提升热交换效率。\",\n" + |
||||||
|
" \"expectedSaving\": 20.0,\n" + |
||||||
|
" \"priority\": \"高\",\n" + |
||||||
|
" \"action\": \"安排专业维护人员进行化学清洗和物理清理。\"\n" + |
||||||
|
" },\n" + |
||||||
|
" {\n" + |
||||||
|
" \"title\": \"优化冷机运行参数\",\n" + |
||||||
|
" \"description\": \"根据室外温度（32.5°C）调整冷机设定温度和运行模式，避免过度冷却。\",\n" + |
||||||
|
" \"expectedSaving\": 10.0,\n" + |
||||||
|
" \"priority\": \"中\",\n" + |
||||||
|
" \"action\": \"调整冷机出水温度设定值至合理范围（如7°C以上），并启用节能模式。\"\n" + |
||||||
|
" },\n" + |
||||||
|
" {\n" + |
||||||
|
" \"title\": \"检查冷冻泵与冷却泵\",\n" + |
||||||
|
" \"description\": \"定期检查水泵轴承和密封状态，确保无异常磨损或泄漏，维持高效运行。\",\n" + |
||||||
|
" \"expectedSaving\": 5.0,\n" + |
||||||
|
" \"priority\": \"低\",\n" + |
||||||
|
" \"action\": \"每月进行一次振动和噪音检测，每年更换一次润滑剂。\"\n" + |
||||||
|
" }\n" + |
||||||
|
" ]\n" + |
||||||
|
"}"; |
||||||
|
log.info("AI优化格式输出：{}", jsonStr); |
||||||
|
try { |
||||||
|
return objectMapper.readValue(jsonStr, EnergyReportDTO.class); |
||||||
|
} catch (Exception e) { |
||||||
|
throw new RuntimeException("解析AI输出失败", e); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
private String extractJson(String response) { |
||||||
|
Pattern pattern = Pattern.compile("```json\\s*(\\{.*?\\})\\s*```", Pattern.DOTALL); |
||||||
|
Matcher matcher = pattern.matcher(response); |
||||||
|
return matcher.find() ? matcher.group(1) : response; |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
@ -0,0 +1,204 @@ |
|||||||
|
package com.mh.system.service.ai.impl; |
||||||
|
|
||||||
|
import com.mh.common.core.domain.dto.EnergyReportDTO; |
||||||
|
import com.mh.common.core.domain.dto.ReportRequestDTO; |
||||||
|
import com.mh.system.service.ai.IFileGenerationService; |
||||||
|
import org.apache.poi.hssf.usermodel.*; |
||||||
|
import org.apache.poi.ss.usermodel.*; |
||||||
|
import org.apache.poi.ss.util.CellRangeAddress; |
||||||
|
import org.apache.poi.xssf.streaming.SXSSFSheet; |
||||||
|
import org.apache.poi.xssf.streaming.SXSSFWorkbook; |
||||||
|
import org.apache.poi.xssf.usermodel.*; |
||||||
|
import org.springframework.stereotype.Service; |
||||||
|
|
||||||
|
import java.io.ByteArrayOutputStream; |
||||||
|
import java.io.IOException; |
||||||
|
|
||||||
|
/** |
||||||
|
* @author LJF |
||||||
|
* @version 1.0 |
||||||
|
* @project EEMCS |
||||||
|
* @description excel添加水印 |
||||||
|
* @date 2026-02-27 10:55:55 |
||||||
|
*/ |
||||||
|
@Service |
||||||
|
public class ExcelGenerationServiceImpl implements IFileGenerationService { |
||||||
|
@Override |
||||||
|
public byte[] generateReport(EnergyReportDTO report, ReportRequestDTO request) throws IOException { |
||||||
|
try (Workbook workbook = new XSSFWorkbook()) { |
||||||
|
Sheet sheet = workbook.createSheet("能效报表"); |
||||||
|
|
||||||
|
// 设置水印（作为背景图片）
|
||||||
|
if (request.getWatermark()) { |
||||||
|
addWatermark(workbook, sheet, request.getWatermarkText()); |
||||||
|
} |
||||||
|
|
||||||
|
// 创建标题行
|
||||||
|
Row titleRow = sheet.createRow(0); |
||||||
|
Cell titleCell = titleRow.createCell(0); |
||||||
|
titleCell.setCellValue("中央空调能效报表"); |
||||||
|
CellStyle titleStyle = workbook.createCellStyle(); |
||||||
|
Font titleFont = workbook.createFont(); |
||||||
|
titleFont.setBold(true); |
||||||
|
titleFont.setFontHeightInPoints((short) 16); |
||||||
|
titleStyle.setFont(titleFont); |
||||||
|
titleCell.setCellStyle(titleStyle); |
||||||
|
|
||||||
|
// 基本信息
|
||||||
|
int rowIdx = 2; |
||||||
|
createCell(sheet, rowIdx++, 0, "报表日期: " + report.getReportDate()); |
||||||
|
createCell(sheet, rowIdx++, 0, "整体能效评分: " + report.getOverallScore()); |
||||||
|
createCell(sheet, rowIdx++, 0, "系统EER: " + report.getSystemEER()); |
||||||
|
createCell(sheet, rowIdx++, 0, "总能耗(kWh): " + report.getTotalEnergyConsumption()); |
||||||
|
|
||||||
|
// 设备明细表头
|
||||||
|
rowIdx++; |
||||||
|
String[] headers = {"设备名称", "设备类型", "能耗(kWh)", "效率(%)", "偏差率(%)"}; |
||||||
|
Row headerRow = sheet.createRow(rowIdx++); |
||||||
|
for (int i = 0; i < headers.length; i++) { |
||||||
|
headerRow.createCell(i).setCellValue(headers[i]); |
||||||
|
} |
||||||
|
|
||||||
|
// 填充设备数据
|
||||||
|
for (EnergyReportDTO.DeviceEnergy device : report.getDeviceDetails()) { |
||||||
|
Row dataRow = sheet.createRow(rowIdx++); |
||||||
|
dataRow.createCell(0).setCellValue(device.getDeviceName()); |
||||||
|
dataRow.createCell(1).setCellValue(device.getDeviceType()); |
||||||
|
dataRow.createCell(2).setCellValue(device.getEnergyConsumption()); |
||||||
|
dataRow.createCell(3).setCellValue(device.getEfficiency()); |
||||||
|
dataRow.createCell(4).setCellValue(device.getDeviationRate()); |
||||||
|
} |
||||||
|
|
||||||
|
// 自动调整列宽
|
||||||
|
for (int i = 0; i < headers.length; i++) { |
||||||
|
sheet.autoSizeColumn(i); |
||||||
|
} |
||||||
|
|
||||||
|
// 异常情况表格
|
||||||
|
rowIdx += 2; |
||||||
|
addSectionTitle(sheet, rowIdx++, "异常情况分析", workbook); |
||||||
|
rowIdx++; |
||||||
|
|
||||||
|
if (report.getAnomalies() != null && !report.getAnomalies().isEmpty()) { |
||||||
|
String[] anomalyHeaders = {"设备名称", "异常类型", "异常描述", "严重程度", "预估损失(kWh)"}; |
||||||
|
Row anomalyHeaderRow = sheet.createRow(rowIdx++); |
||||||
|
for (int i = 0; i < anomalyHeaders.length; i++) { |
||||||
|
anomalyHeaderRow.createCell(i).setCellValue(anomalyHeaders[i]); |
||||||
|
} |
||||||
|
|
||||||
|
for (EnergyReportDTO.Anomaly anomaly : report.getAnomalies()) { |
||||||
|
Row dataRow = sheet.createRow(rowIdx++); |
||||||
|
dataRow.createCell(0).setCellValue(anomaly.getDeviceName()); |
||||||
|
dataRow.createCell(1).setCellValue(anomaly.getAnomalyType()); |
||||||
|
dataRow.createCell(2).setCellValue(anomaly.getDescription()); |
||||||
|
dataRow.createCell(3).setCellValue(anomaly.getSeverity()); |
||||||
|
dataRow.createCell(4).setCellValue(anomaly.getEstimatedLoss() != null ? |
||||||
|
anomaly.getEstimatedLoss() : 0.0); |
||||||
|
} |
||||||
|
|
||||||
|
// 调整异常表格列宽
|
||||||
|
for (int i = 0; i < anomalyHeaders.length; i++) { |
||||||
|
sheet.autoSizeColumn(i); |
||||||
|
} |
||||||
|
} else { |
||||||
|
createCell(sheet, rowIdx++, 0, "无异常情况"); |
||||||
|
} |
||||||
|
|
||||||
|
// 优化建议表格
|
||||||
|
rowIdx += 2; |
||||||
|
addSectionTitle(sheet, rowIdx++, "优化建议", workbook); |
||||||
|
rowIdx++; |
||||||
|
|
||||||
|
if (report.getSuggestions() != null && !report.getSuggestions().isEmpty()) { |
||||||
|
String[] suggestionHeaders = {"优先级", "建议标题", "详细内容", "预期节能(kWh)", "建议操作"}; |
||||||
|
Row suggestionHeaderRow = sheet.createRow(rowIdx++); |
||||||
|
for (int i = 0; i < suggestionHeaders.length; i++) { |
||||||
|
suggestionHeaderRow.createCell(i).setCellValue(suggestionHeaders[i]); |
||||||
|
} |
||||||
|
|
||||||
|
for (EnergyReportDTO.Suggestion suggestion : report.getSuggestions()) { |
||||||
|
Row dataRow = sheet.createRow(rowIdx++); |
||||||
|
dataRow.createCell(0).setCellValue(suggestion.getPriority()); |
||||||
|
dataRow.createCell(1).setCellValue(suggestion.getTitle()); |
||||||
|
dataRow.createCell(2).setCellValue(suggestion.getDescription()); |
||||||
|
dataRow.createCell(3).setCellValue(suggestion.getExpectedSaving() != null ? |
||||||
|
suggestion.getExpectedSaving() : 0.0); |
||||||
|
dataRow.createCell(4).setCellValue(suggestion.getAction()); |
||||||
|
} |
||||||
|
|
||||||
|
// 调整建议表格列宽
|
||||||
|
for (int i = 0; i < suggestionHeaders.length; i++) { |
||||||
|
sheet.autoSizeColumn(i); |
||||||
|
} |
||||||
|
} else { |
||||||
|
createCell(sheet, rowIdx++, 0, "暂无优化建议"); |
||||||
|
} |
||||||
|
|
||||||
|
ByteArrayOutputStream baos = new ByteArrayOutputStream(); |
||||||
|
workbook.write(baos); |
||||||
|
return baos.toByteArray(); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
private void createCell(Sheet sheet, int row, int col, String value) { |
||||||
|
Row r = sheet.getRow(row); |
||||||
|
if (r == null) r = sheet.createRow(row); |
||||||
|
r.createCell(col).setCellValue(value); |
||||||
|
} |
||||||
|
|
||||||
|
private void addSectionTitle(Sheet sheet, int row, String title, Workbook workbook) { |
||||||
|
Row titleRow = sheet.createRow(row); |
||||||
|
Cell titleCell = titleRow.createCell(0); |
||||||
|
titleCell.setCellValue(title); |
||||||
|
|
||||||
|
CellStyle titleStyle = workbook.createCellStyle(); |
||||||
|
Font titleFont = workbook.createFont(); |
||||||
|
titleFont.setBold(true); |
||||||
|
titleFont.setFontHeightInPoints((short) 12); |
||||||
|
titleStyle.setFont(titleFont); |
||||||
|
titleStyle.setAlignment(HorizontalAlignment.LEFT); |
||||||
|
titleCell.setCellStyle(titleStyle); |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* Excel水印（使用背景图片方式） |
||||||
|
* 注意：XSSF支持设置背景图片，但水印文字需预先绘制成图片 |
||||||
|
* 简化：此处仅演示思路，实际需要将文字转为BufferedImage再设为背景 |
||||||
|
*/ |
||||||
|
private void addWatermark(Workbook workbook, Sheet sheet, String watermarkText) { |
||||||
|
try { |
||||||
|
// 最简单的水印实现：在工作表顶部添加灰色文本
|
||||||
|
Row watermarkRow = sheet.createRow(0); |
||||||
|
Cell watermarkCell = watermarkRow.createCell(0); |
||||||
|
watermarkCell.setCellValue(watermarkText); |
||||||
|
|
||||||
|
// 创建水印样式
|
||||||
|
CellStyle watermarkStyle = workbook.createCellStyle(); |
||||||
|
Font watermarkFont = workbook.createFont(); |
||||||
|
watermarkFont.setColor(IndexedColors.GREY_25_PERCENT.getIndex()); |
||||||
|
watermarkFont.setFontHeightInPoints((short) 14); |
||||||
|
watermarkFont.setItalic(true); // 斜体效果
|
||||||
|
watermarkStyle.setFont(watermarkFont); |
||||||
|
watermarkStyle.setAlignment(HorizontalAlignment.CENTER); |
||||||
|
|
||||||
|
// 设置背景色为浅灰色（模拟水印效果）
|
||||||
|
watermarkStyle.setFillForegroundColor(IndexedColors.GREY_25_PERCENT.getIndex()); |
||||||
|
watermarkStyle.setFillPattern(FillPatternType.NO_FILL); |
||||||
|
|
||||||
|
watermarkCell.setCellStyle(watermarkStyle); |
||||||
|
|
||||||
|
// 合并单元格让水印覆盖更多区域
|
||||||
|
sheet.addMergedRegion(new CellRangeAddress(0, 0, 0, 5)); |
||||||
|
|
||||||
|
} catch (Exception e) { |
||||||
|
// 水印添加失败时不中断主流程
|
||||||
|
System.out.println("水印添加警告: " + e.getMessage()); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
@Override |
||||||
|
public String getFileExtension() { |
||||||
|
return ".xlsx"; |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,52 @@ |
|||||||
|
package com.mh.system.service.ai.impl; |
||||||
|
|
||||||
|
import com.mh.common.core.domain.dto.ReportRequestDTO; |
||||||
|
import com.mh.system.service.ai.INLPService; |
||||||
|
import org.springframework.ai.chat.client.ChatClient; |
||||||
|
import org.springframework.ai.converter.BeanOutputConverter; |
||||||
|
import org.springframework.stereotype.Service; |
||||||
|
|
||||||
|
/** |
||||||
|
* @author LJF |
||||||
|
* @version 1.0 |
||||||
|
* @project EEMCS |
||||||
|
* @description 自然语言解析服务实现类 |
||||||
|
* @date 2026-02-27 10:29:30 |
||||||
|
*/ |
||||||
|
@Service |
||||||
|
public class NLPServiceImpl implements INLPService { |
||||||
|
|
||||||
|
private final ChatClient chatClient; |
||||||
|
|
||||||
|
public NLPServiceImpl(ChatClient.Builder builder) { |
||||||
|
this.chatClient = builder.build(); |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public ReportRequestDTO parseUserIntent(String userMessage) { |
||||||
|
BeanOutputConverter<ReportRequestDTO> converter = new BeanOutputConverter<>(ReportRequestDTO.class); |
||||||
|
String systemPrompt = """ |
||||||
|
你是一个智能助手，负责解析用户关于生成报表的请求。 |
||||||
|
请从用户消息中提取以下信息： |
||||||
|
- reportType: 报表类型 (如 "能效报表"、"故障报表") |
||||||
|
- timeRange: 时间范围 (如 "昨天"、"2026-02-01至2026-02-07") |
||||||
|
- format: 输出格式 (word/excel/pdf) |
||||||
|
- watermark: 是否需要水印 (布尔值) |
||||||
|
- watermarkText: 水印文字 |
||||||
|
|
||||||
|
默认值：format="pdf", watermark=true, watermarkText="内部资料" |
||||||
|
请以严格的JSON格式输出。 |
||||||
|
""" + converter.getFormat(); |
||||||
|
|
||||||
|
ReportRequestDTO request = chatClient.prompt() |
||||||
|
.system(systemPrompt) |
||||||
|
.user(userMessage) |
||||||
|
.call() |
||||||
|
.entity(converter); |
||||||
|
|
||||||
|
if (request.getFormat() == null) request.setFormat("pdf"); |
||||||
|
if (request.getWatermark() == null) request.setWatermark(true); |
||||||
|
if (request.getWatermarkText() == null) request.setWatermarkText("内部资料"); |
||||||
|
return request; |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,287 @@ |
|||||||
|
package com.mh.system.service.ai.impl; |
||||||
|
|
||||||
|
import com.itextpdf.text.*; |
||||||
|
import com.itextpdf.text.pdf.*; |
||||||
|
import com.mh.common.core.domain.dto.EnergyReportDTO; |
||||||
|
import com.mh.common.core.domain.dto.ReportRequestDTO; |
||||||
|
import com.mh.system.service.ai.IFileGenerationService; |
||||||
|
import org.springframework.stereotype.Service; |
||||||
|
|
||||||
|
import java.io.ByteArrayOutputStream; |
||||||
|
import java.io.IOException; |
||||||
|
|
||||||
|
/** |
||||||
|
* @author LJF |
||||||
|
* @version 1.0 |
||||||
|
* @project EEMCS |
||||||
|
* @description PDF添加水印 |
||||||
|
* @date 2026-02-27 11:06:19 |
||||||
|
*/ |
||||||
|
@Service |
||||||
|
public class PdfGenerationServiceImpl implements IFileGenerationService { |
||||||
|
|
||||||
|
@Override |
||||||
|
public byte[] generateReport(EnergyReportDTO report, ReportRequestDTO request) throws IOException { |
||||||
|
Document document = new Document(PageSize.A4); |
||||||
|
ByteArrayOutputStream baos = new ByteArrayOutputStream(); |
||||||
|
PdfWriter writer = null; |
||||||
|
try { |
||||||
|
writer = PdfWriter.getInstance(document, baos); |
||||||
|
} catch (DocumentException e) { |
||||||
|
throw new RuntimeException(e); |
||||||
|
} |
||||||
|
|
||||||
|
// 设置水印事件处理器
|
||||||
|
if (request.getWatermark()) { |
||||||
|
writer.setPageEvent(new WatermarkPageEvent(request.getWatermarkText())); |
||||||
|
} |
||||||
|
|
||||||
|
document.open(); |
||||||
|
|
||||||
|
// 添加标题
|
||||||
|
Font titleFont = new Font(Font.FontFamily.HELVETICA, 20, Font.BOLD); |
||||||
|
Paragraph title = new Paragraph("中央空调能效报表", titleFont); |
||||||
|
title.setAlignment(Element.ALIGN_CENTER); |
||||||
|
title.setSpacingAfter(20); |
||||||
|
try { |
||||||
|
document.add(title); |
||||||
|
} catch (DocumentException e) { |
||||||
|
throw new RuntimeException(e); |
||||||
|
} |
||||||
|
|
||||||
|
// 添加基本信息
|
||||||
|
Font normalFont = new Font(Font.FontFamily.HELVETICA, 12); |
||||||
|
try { |
||||||
|
document.add(new Paragraph("报表日期: " + report.getReportDate(), normalFont)); |
||||||
|
document.add(new Paragraph("整体能效评分: " + report.getOverallScore(), normalFont)); |
||||||
|
document.add(new Paragraph("系统EER: " + report.getSystemEER(), normalFont)); |
||||||
|
document.add(new Paragraph("总能耗(kWh): " + report.getTotalEnergyConsumption(), normalFont)); |
||||||
|
document.add(new Paragraph(" ")); |
||||||
|
} catch (DocumentException e) { |
||||||
|
throw new RuntimeException(e); |
||||||
|
} |
||||||
|
|
||||||
|
// 创建表格
|
||||||
|
PdfPTable table = new PdfPTable(5); |
||||||
|
table.setWidthPercentage(100); |
||||||
|
table.setSpacingBefore(10f); |
||||||
|
table.setSpacingAfter(10f); |
||||||
|
|
||||||
|
// 设置列宽
|
||||||
|
float[] columnWidths = {2f, 2f, 2f, 2f, 2f}; |
||||||
|
try { |
||||||
|
table.setWidths(columnWidths); |
||||||
|
} catch (DocumentException e) { |
||||||
|
throw new RuntimeException(e); |
||||||
|
} |
||||||
|
|
||||||
|
// 添加表头
|
||||||
|
Font headerFont = new Font(Font.FontFamily.HELVETICA, 12, Font.BOLD); |
||||||
|
addTableHeader(table, "设备名称", headerFont); |
||||||
|
addTableHeader(table, "设备类型", headerFont); |
||||||
|
addTableHeader(table, "能耗(kWh)", headerFont); |
||||||
|
addTableHeader(table, "效率(%)", headerFont); |
||||||
|
addTableHeader(table, "偏差率(%)", headerFont); |
||||||
|
|
||||||
|
// 添加数据行
|
||||||
|
Font dataFont = new Font(Font.FontFamily.HELVETICA, 10); |
||||||
|
for (EnergyReportDTO.DeviceEnergy device : report.getDeviceDetails()) { |
||||||
|
addTableCell(table, device.getDeviceName(), dataFont); |
||||||
|
addTableCell(table, device.getDeviceType(), dataFont); |
||||||
|
addTableCell(table, String.valueOf(device.getEnergyConsumption()), dataFont); |
||||||
|
addTableCell(table, String.valueOf(device.getEfficiency()), dataFont); |
||||||
|
addTableCell(table, String.valueOf(device.getDeviationRate()), dataFont); |
||||||
|
} |
||||||
|
|
||||||
|
try { |
||||||
|
document.add(table); |
||||||
|
} catch (DocumentException e) { |
||||||
|
throw new RuntimeException(e); |
||||||
|
} |
||||||
|
|
||||||
|
// 异常情况分析
|
||||||
|
try { |
||||||
|
addSectionTitle(document, "异常情况分析", headerFont); |
||||||
|
} catch (DocumentException e) { |
||||||
|
throw new RuntimeException(e); |
||||||
|
} |
||||||
|
if (report.getAnomalies() != null && !report.getAnomalies().isEmpty()) { |
||||||
|
PdfPTable anomalyTable = new PdfPTable(5); |
||||||
|
anomalyTable.setWidthPercentage(100); |
||||||
|
anomalyTable.setSpacingBefore(10f); |
||||||
|
anomalyTable.setSpacingAfter(10f); |
||||||
|
|
||||||
|
float[] anomalyColumnWidths = {2f, 2f, 3f, 1.5f, 2f}; |
||||||
|
try { |
||||||
|
anomalyTable.setWidths(anomalyColumnWidths); |
||||||
|
} catch (DocumentException e) { |
||||||
|
throw new RuntimeException(e); |
||||||
|
} |
||||||
|
|
||||||
|
addTableHeader(anomalyTable, "设备名称", headerFont); |
||||||
|
addTableHeader(anomalyTable, "异常类型", headerFont); |
||||||
|
addTableHeader(anomalyTable, "异常描述", headerFont); |
||||||
|
addTableHeader(anomalyTable, "严重程度", headerFont); |
||||||
|
addTableHeader(anomalyTable, "预估损失(kWh)", headerFont); |
||||||
|
|
||||||
|
for (EnergyReportDTO.Anomaly anomaly : report.getAnomalies()) { |
||||||
|
addTableCell(anomalyTable, anomaly.getDeviceName(), dataFont); |
||||||
|
addTableCell(anomalyTable, anomaly.getAnomalyType(), dataFont); |
||||||
|
addTableCell(anomalyTable, anomaly.getDescription(), dataFont); |
||||||
|
addTableCell(anomalyTable, anomaly.getSeverity(), dataFont); |
||||||
|
addTableCell(anomalyTable, anomaly.getEstimatedLoss() != null ? |
||||||
|
String.format("%.2f", anomaly.getEstimatedLoss()) : "-", dataFont); |
||||||
|
} |
||||||
|
|
||||||
|
try { |
||||||
|
document.add(anomalyTable); |
||||||
|
} catch (DocumentException e) { |
||||||
|
throw new RuntimeException(e); |
||||||
|
} |
||||||
|
} else { |
||||||
|
try { |
||||||
|
document.add(new Paragraph("无异常情况", normalFont)); |
||||||
|
document.add(new Paragraph(" ")); |
||||||
|
} catch (DocumentException e) { |
||||||
|
throw new RuntimeException(e); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// 优化建议
|
||||||
|
try { |
||||||
|
addSectionTitle(document, "优化建议", headerFont); |
||||||
|
} catch (DocumentException e) { |
||||||
|
throw new RuntimeException(e); |
||||||
|
} |
||||||
|
if (report.getSuggestions() != null && !report.getSuggestions().isEmpty()) { |
||||||
|
PdfPTable suggestionTable = new PdfPTable(5); |
||||||
|
suggestionTable.setWidthPercentage(100); |
||||||
|
suggestionTable.setSpacingBefore(10f); |
||||||
|
suggestionTable.setSpacingAfter(10f); |
||||||
|
|
||||||
|
float[] suggestionColumnWidths = {1.5f, 2f, 3f, 2f, 2f}; |
||||||
|
try { |
||||||
|
suggestionTable.setWidths(suggestionColumnWidths); |
||||||
|
} catch (DocumentException e) { |
||||||
|
throw new RuntimeException(e); |
||||||
|
} |
||||||
|
|
||||||
|
addTableHeader(suggestionTable, "优先级", headerFont); |
||||||
|
addTableHeader(suggestionTable, "建议标题", headerFont); |
||||||
|
addTableHeader(suggestionTable, "详细内容", headerFont); |
||||||
|
addTableHeader(suggestionTable, "预期节能(kWh)", headerFont); |
||||||
|
addTableHeader(suggestionTable, "建议操作", headerFont); |
||||||
|
|
||||||
|
for (EnergyReportDTO.Suggestion suggestion : report.getSuggestions()) { |
||||||
|
addTableCell(suggestionTable, suggestion.getPriority(), dataFont); |
||||||
|
addTableCell(suggestionTable, suggestion.getTitle(), dataFont); |
||||||
|
addTableCell(suggestionTable, suggestion.getDescription(), dataFont); |
||||||
|
addTableCell(suggestionTable, suggestion.getExpectedSaving() != null ? |
||||||
|
String.format("%.2f", suggestion.getExpectedSaving()) : "-", dataFont); |
||||||
|
addTableCell(suggestionTable, suggestion.getAction(), dataFont); |
||||||
|
} |
||||||
|
|
||||||
|
try { |
||||||
|
document.add(suggestionTable); |
||||||
|
} catch (DocumentException e) { |
||||||
|
throw new RuntimeException(e); |
||||||
|
} |
||||||
|
} else { |
||||||
|
try { |
||||||
|
document.add(new Paragraph("暂无优化建议", normalFont)); |
||||||
|
} catch (DocumentException e) { |
||||||
|
throw new RuntimeException(e); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
document.close(); |
||||||
|
|
||||||
|
return baos.toByteArray(); |
||||||
|
} |
||||||
|
|
||||||
|
private void addTableHeader(PdfPTable table, String headerText, Font font) { |
||||||
|
PdfPCell header = new PdfPCell(new Phrase(headerText, font)); |
||||||
|
header.setHorizontalAlignment(Element.ALIGN_CENTER); |
||||||
|
header.setVerticalAlignment(Element.ALIGN_MIDDLE); |
||||||
|
header.setBackgroundColor(BaseColor.LIGHT_GRAY); |
||||||
|
table.addCell(header); |
||||||
|
} |
||||||
|
|
||||||
|
private void addTableCell(PdfPTable table, String text, Font font) { |
||||||
|
PdfPCell cell = new PdfPCell(new Phrase(text, font)); |
||||||
|
cell.setHorizontalAlignment(Element.ALIGN_CENTER); |
||||||
|
cell.setVerticalAlignment(Element.ALIGN_MIDDLE); |
||||||
|
table.addCell(cell); |
||||||
|
} |
||||||
|
|
||||||
|
private void addSectionTitle(Document document, String title, Font font) throws DocumentException { |
||||||
|
Paragraph sectionTitle = new Paragraph(title, font); |
||||||
|
sectionTitle.setAlignment(Element.ALIGN_LEFT); |
||||||
|
sectionTitle.setSpacingBefore(20f); |
||||||
|
sectionTitle.setSpacingAfter(10f); |
||||||
|
document.add(sectionTitle); |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public String getFileExtension() { |
||||||
|
return ".pdf"; |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* PDF水印页面事件处理器 |
||||||
|
*/ |
||||||
|
private static class WatermarkPageEvent extends PdfPageEventHelper { |
||||||
|
private final String watermarkText; |
||||||
|
private final Font watermarkFont; |
||||||
|
|
||||||
|
public WatermarkPageEvent(String watermarkText) { |
||||||
|
this.watermarkText = watermarkText; |
||||||
|
// 创建水印字体
|
||||||
|
this.watermarkFont = new Font(Font.FontFamily.HELVETICA, 50, Font.NORMAL, BaseColor.LIGHT_GRAY); |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public void onEndPage(PdfWriter writer, Document document) { |
||||||
|
try { |
||||||
|
PdfContentByte canvas = writer.getDirectContentUnder(); |
||||||
|
Rectangle pageSize = document.getPageSize(); |
||||||
|
|
||||||
|
// 保存当前图形状态
|
||||||
|
canvas.saveState(); |
||||||
|
|
||||||
|
// 设置透明度
|
||||||
|
PdfGState gState = new PdfGState(); |
||||||
|
gState.setFillOpacity(0.2f); // 20%不透明度
|
||||||
|
canvas.setGState(gState); |
||||||
|
|
||||||
|
// 设置字体和颜色
|
||||||
|
canvas.beginText(); |
||||||
|
canvas.setFontAndSize(watermarkFont.getBaseFont(), 50); |
||||||
|
canvas.setColorFill(BaseColor.LIGHT_GRAY); |
||||||
|
|
||||||
|
// 计算水印位置（页面中心）
|
||||||
|
float x = (pageSize.getLeft() + pageSize.getRight()) / 2; |
||||||
|
float y = (pageSize.getTop() + pageSize.getBottom()) / 2; |
||||||
|
|
||||||
|
// 设置文字矩阵并旋转45度
|
||||||
|
canvas.setTextMatrix( |
||||||
|
(float) Math.cos(Math.toRadians(45)), |
||||||
|
(float) Math.sin(Math.toRadians(45)), |
||||||
|
(float) -Math.sin(Math.toRadians(45)), |
||||||
|
(float) Math.cos(Math.toRadians(45)), |
||||||
|
x, y |
||||||
|
); |
||||||
|
|
||||||
|
// 显示水印文字
|
||||||
|
canvas.showTextAligned(Element.ALIGN_CENTER, watermarkText, 0, 0, 0); |
||||||
|
canvas.endText(); |
||||||
|
|
||||||
|
// 恢复图形状态
|
||||||
|
canvas.restoreState(); |
||||||
|
|
||||||
|
} catch (Exception e) { |
||||||
|
System.err.println("添加PDF水印失败: " + e.getMessage()); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,43 @@ |
|||||||
|
package com.mh.system.service.ai.impl; |
||||||
|
|
||||||
|
import com.mh.common.core.domain.dto.EnergyReportDTO; |
||||||
|
import com.mh.common.core.domain.dto.ReportRequestDTO; |
||||||
|
import com.mh.system.service.ai.IFileGenerationService; |
||||||
|
import org.springframework.stereotype.Service; |
||||||
|
|
||||||
|
import java.io.IOException; |
||||||
|
|
||||||
|
/** |
||||||
|
* @author LJF |
||||||
|
* @version 1.0 |
||||||
|
* @project EEMCS |
||||||
|
* @description 文件生成服务 |
||||||
|
* @date 2026-02-27 11:24:45 |
||||||
|
*/ |
||||||
|
@Service |
||||||
|
public class ReportOrchestrationService { |
||||||
|
private final WordGenerationServiceImpl wordService; |
||||||
|
private final ExcelGenerationServiceImpl excelService; |
||||||
|
private final PdfGenerationServiceImpl pdfService; |
||||||
|
|
||||||
|
public ReportOrchestrationService(WordGenerationServiceImpl wordService, |
||||||
|
ExcelGenerationServiceImpl excelService, |
||||||
|
PdfGenerationServiceImpl pdfService) { |
||||||
|
this.wordService = wordService; |
||||||
|
this.excelService = excelService; |
||||||
|
this.pdfService = pdfService; |
||||||
|
} |
||||||
|
|
||||||
|
public ReportFile generateFile(EnergyReportDTO report, ReportRequestDTO request) throws IOException { |
||||||
|
IFileGenerationService generator = switch (request.getFormat().toLowerCase()) { |
||||||
|
case "word" -> wordService; |
||||||
|
case "excel" -> excelService; |
||||||
|
default -> pdfService; |
||||||
|
}; |
||||||
|
byte[] content = generator.generateReport(report, request); |
||||||
|
String filename = "能效报表_" + report.getReportDate() + generator.getFileExtension(); |
||||||
|
return new ReportFile(filename, content); |
||||||
|
} |
||||||
|
|
||||||
|
public record ReportFile(String filename, byte[] content) {} |
||||||
|
} |
||||||
@ -0,0 +1,34 @@ |
|||||||
|
package com.mh.system.service.ai.impl; |
||||||
|
|
||||||
|
import com.mh.common.core.domain.dto.AsyncTaskDTO; |
||||||
|
import org.springframework.stereotype.Component; |
||||||
|
|
||||||
|
import java.time.LocalDateTime; |
||||||
|
import java.util.Map; |
||||||
|
import java.util.concurrent.ConcurrentHashMap; |
||||||
|
|
||||||
|
/** |
||||||
|
* @author LJF |
||||||
|
* @version 1.0 |
||||||
|
* @project EEMCS |
||||||
|
* @description 任务存储服务类 |
||||||
|
* @date 2026-02-27 11:30:06 |
||||||
|
*/ |
||||||
|
@Component |
||||||
|
public class TaskStoreService { |
||||||
|
|
||||||
|
private final Map<String, AsyncTaskDTO> tasks = new ConcurrentHashMap<>(); |
||||||
|
|
||||||
|
public void save(AsyncTaskDTO task) { tasks.put(task.getTaskId(), task); } |
||||||
|
public AsyncTaskDTO get(String taskId) { return tasks.get(taskId); } |
||||||
|
public void updateStatus(String taskId, String status, String message, String downloadUrl) { |
||||||
|
AsyncTaskDTO task = tasks.get(taskId); |
||||||
|
if (task != null) { |
||||||
|
task.setStatus(status); |
||||||
|
task.setMessage(message); |
||||||
|
task.setDownloadUrl(downloadUrl); |
||||||
|
task.setCompleteTime(LocalDateTime.now()); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
@ -0,0 +1,328 @@ |
|||||||
|
package com.mh.system.service.ai.impl; |
||||||
|
|
||||||
|
import org.apache.poi.xwpf.usermodel.*; |
||||||
|
import com.mh.common.core.domain.dto.EnergyReportDTO; |
||||||
|
import com.mh.common.core.domain.dto.ReportRequestDTO; |
||||||
|
import com.mh.system.service.ai.IFileGenerationService; |
||||||
|
import org.springframework.stereotype.Service; |
||||||
|
|
||||||
|
import java.io.ByteArrayOutputStream; |
||||||
|
import java.io.IOException; |
||||||
|
|
||||||
|
/** |
||||||
|
* @author LJF |
||||||
|
* @version 1.0 |
||||||
|
* @project EEMCS |
||||||
|
* @description Word文档生成服务 |
||||||
|
* @date 2026-02-27 10:51:51 |
||||||
|
*/ |
||||||
|
@Service |
||||||
|
public class WordGenerationServiceImpl implements IFileGenerationService { |
||||||
|
|
||||||
|
@Override |
||||||
|
public byte[] generateReport(EnergyReportDTO report, ReportRequestDTO request) throws IOException { |
||||||
|
try (XWPFDocument document = new XWPFDocument()) { |
||||||
|
// 添加水印
|
||||||
|
if (request.getWatermark()) { |
||||||
|
addWatermark(document, request.getWatermarkText()); |
||||||
|
} |
||||||
|
|
||||||
|
// 创建标题
|
||||||
|
addTitle(document, "中央空调能效报表"); |
||||||
|
|
||||||
|
// 添加基本信息
|
||||||
|
addBasicInfo(document, report); |
||||||
|
|
||||||
|
// 设备明细表格
|
||||||
|
addDeviceDetailsTable(document, report.getDeviceDetails().toArray(new EnergyReportDTO.DeviceEnergy[0])); |
||||||
|
|
||||||
|
// 异常情况分析
|
||||||
|
addAnomaliesSection(document, report.getAnomalies().toArray(new EnergyReportDTO.Anomaly[0])); |
||||||
|
|
||||||
|
// 优化建议
|
||||||
|
addSuggestionsSection(document, report.getSuggestions().toArray(new EnergyReportDTO.Suggestion[0])); |
||||||
|
|
||||||
|
ByteArrayOutputStream baos = new ByteArrayOutputStream(); |
||||||
|
document.write(baos); |
||||||
|
return baos.toByteArray(); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
private void addTitle(XWPFDocument document, String title) { |
||||||
|
XWPFParagraph para = document.createParagraph(); |
||||||
|
para.setAlignment(ParagraphAlignment.CENTER); |
||||||
|
XWPFRun run = para.createRun(); |
||||||
|
run.setText(title); |
||||||
|
// 使用反射安全设置样式
|
||||||
|
setRunBold(run, true); |
||||||
|
setRunFontSize(run, 20); |
||||||
|
} |
||||||
|
|
||||||
|
private void addBasicInfo(XWPFDocument document, EnergyReportDTO report) { |
||||||
|
XWPFParagraph para = document.createParagraph(); |
||||||
|
XWPFRun run = para.createRun(); |
||||||
|
run.setText("报表日期: " + report.getReportDate()); |
||||||
|
setRunFontSize(run, 12); |
||||||
|
|
||||||
|
para = document.createParagraph(); |
||||||
|
run = para.createRun(); |
||||||
|
run.setText("整体能效评分: " + report.getOverallScore()); |
||||||
|
setRunFontSize(run, 12); |
||||||
|
|
||||||
|
para = document.createParagraph(); |
||||||
|
run = para.createRun(); |
||||||
|
run.setText("系统综合能效比(EER): " + report.getSystemEER()); |
||||||
|
setRunFontSize(run, 12); |
||||||
|
|
||||||
|
para = document.createParagraph(); |
||||||
|
run = para.createRun(); |
||||||
|
run.setText("总能耗: " + formatDouble(report.getTotalEnergyConsumption()) + " kWh"); |
||||||
|
setRunFontSize(run, 12); |
||||||
|
|
||||||
|
para = document.createParagraph(); |
||||||
|
run = para.createRun(); |
||||||
|
run.setText(" "); |
||||||
|
setRunFontSize(run, 8); |
||||||
|
} |
||||||
|
|
||||||
|
private void addDeviceDetailsTable(XWPFDocument document, EnergyReportDTO.DeviceEnergy[] devices) { |
||||||
|
if (devices == null || devices.length == 0) { |
||||||
|
return; |
||||||
|
} |
||||||
|
|
||||||
|
// 使用纯文本方式替代表格，完全避免POI表格API
|
||||||
|
addSectionTitle(document, "设备明细"); |
||||||
|
|
||||||
|
// 添加表头
|
||||||
|
XWPFParagraph headerPara = document.createParagraph(); |
||||||
|
XWPFRun headerRun = headerPara.createRun(); |
||||||
|
headerRun.setText("设备名称\t设备类型\t能耗\t效率(%)\t偏差率(%)"); |
||||||
|
setRunBold(headerRun, true); |
||||||
|
|
||||||
|
// 添加数据行
|
||||||
|
for (EnergyReportDTO.DeviceEnergy device : devices) { |
||||||
|
XWPFParagraph dataPara = document.createParagraph(); |
||||||
|
XWPFRun dataRun = dataPara.createRun(); |
||||||
|
String rowData = String.format("%s\t%s\t%s\t%s\t%s", |
||||||
|
device.getDeviceName(), |
||||||
|
device.getDeviceType(), |
||||||
|
formatDouble(device.getEnergyConsumption()), |
||||||
|
formatDouble(device.getEfficiency()), |
||||||
|
formatDouble(device.getDeviationRate())); |
||||||
|
dataRun.setText(rowData); |
||||||
|
} |
||||||
|
|
||||||
|
document.createParagraph(); |
||||||
|
} |
||||||
|
|
||||||
|
private void addAnomaliesSection(XWPFDocument document, EnergyReportDTO.Anomaly[] anomalies) { |
||||||
|
addSectionTitle(document, "异常情况分析"); |
||||||
|
|
||||||
|
if (anomalies != null && anomalies.length > 0) { |
||||||
|
// 使用列表方式替代表格
|
||||||
|
XWPFParagraph headerPara = document.createParagraph(); |
||||||
|
XWPFRun headerRun = headerPara.createRun(); |
||||||
|
headerRun.setText("设备名称 | 异常类型 | 异常描述 | 严重程度 | 预估损失"); |
||||||
|
setRunBold(headerRun, true); |
||||||
|
|
||||||
|
for (EnergyReportDTO.Anomaly anomaly : anomalies) { |
||||||
|
XWPFParagraph itemPara = document.createParagraph(); |
||||||
|
XWPFRun itemRun = itemPara.createRun(); |
||||||
|
String itemText = String.format("%s | %s | %s | %s | %s", |
||||||
|
anomaly.getDeviceName(), |
||||||
|
anomaly.getAnomalyType(), |
||||||
|
anomaly.getDescription(), |
||||||
|
anomaly.getSeverity(), |
||||||
|
formatDouble(anomaly.getEstimatedLoss())); |
||||||
|
itemRun.setText(itemText); |
||||||
|
} |
||||||
|
} else { |
||||||
|
XWPFParagraph para = document.createParagraph(); |
||||||
|
XWPFRun run = para.createRun(); |
||||||
|
run.setText("无异常情况"); |
||||||
|
setRunFontSize(run, 12); |
||||||
|
} |
||||||
|
|
||||||
|
document.createParagraph(); |
||||||
|
} |
||||||
|
|
||||||
|
private void addSuggestionsSection(XWPFDocument document, EnergyReportDTO.Suggestion[] suggestions) { |
||||||
|
addSectionTitle(document, "优化建议"); |
||||||
|
|
||||||
|
if (suggestions != null && suggestions.length > 0) { |
||||||
|
// 使用编号列表方式展示建议
|
||||||
|
for (int i = 0; i < suggestions.length; i++) { |
||||||
|
EnergyReportDTO.Suggestion suggestion = suggestions[i]; |
||||||
|
|
||||||
|
// 添加建议标题
|
||||||
|
XWPFParagraph titlePara = document.createParagraph(); |
||||||
|
XWPFRun titleRun = titlePara.createRun(); |
||||||
|
titleRun.setText((i + 1) + ". " + suggestion.getTitle()); |
||||||
|
setRunBold(titleRun, true); |
||||||
|
|
||||||
|
// 添加详细信息
|
||||||
|
XWPFParagraph detailPara = document.createParagraph(); |
||||||
|
XWPFRun detailRun = detailPara.createRun(); |
||||||
|
String detailText = String.format("优先级: %s | 预期节能: %s | 建议操作: %s\n详细内容: %s", |
||||||
|
suggestion.getPriority(), |
||||||
|
formatDouble(suggestion.getExpectedSaving()), |
||||||
|
suggestion.getAction(), |
||||||
|
suggestion.getDescription()); |
||||||
|
detailRun.setText(detailText); |
||||||
|
setRunFontSize(detailRun, 11); |
||||||
|
} |
||||||
|
} else { |
||||||
|
XWPFParagraph para = document.createParagraph(); |
||||||
|
XWPFRun run = para.createRun(); |
||||||
|
run.setText("暂无优化建议"); |
||||||
|
setRunFontSize(run, 12); |
||||||
|
} |
||||||
|
|
||||||
|
document.createParagraph(); |
||||||
|
} |
||||||
|
|
||||||
|
private void addSectionTitle(XWPFDocument document, String title) { |
||||||
|
XWPFParagraph para = document.createParagraph(); |
||||||
|
setParagraphSpacing(para, 200, 100); |
||||||
|
XWPFRun run = para.createRun(); |
||||||
|
run.setText(title); |
||||||
|
setRunBold(run, true); |
||||||
|
setRunFontSize(run, 14); |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 安全设置段落间距 |
||||||
|
*/ |
||||||
|
private void setParagraphSpacing(XWPFParagraph para, int before, int after) { |
||||||
|
try { |
||||||
|
para.setSpacingBefore(before); |
||||||
|
para.setSpacingAfter(after); |
||||||
|
} catch (NoSuchMethodError | NoClassDefFoundError e) { |
||||||
|
System.err.println("设置段落间距失败，跳过: " + e.getMessage()); |
||||||
|
} catch (Exception e) { |
||||||
|
// 静默处理
|
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
private void addWatermark(XWPFDocument document, String watermarkText) { |
||||||
|
try { |
||||||
|
// 使用完全兼容POI 4.1.2的水印实现
|
||||||
|
addCompatibleWatermark(document, watermarkText); |
||||||
|
} catch (Exception e) { |
||||||
|
// 水印添加失败时静默处理，不影响文档生成
|
||||||
|
System.err.println("添加水印失败: " + e.getMessage()); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 完全兼容POI 4.1.2的水印实现 |
||||||
|
*/ |
||||||
|
private void addCompatibleWatermark(XWPFDocument doc, String watermarkText) { |
||||||
|
try { |
||||||
|
// 在文档开头添加水印文本作为替代方案
|
||||||
|
XWPFParagraph watermarkPara = doc.createParagraph(); |
||||||
|
watermarkPara.setAlignment(ParagraphAlignment.CENTER); |
||||||
|
|
||||||
|
XWPFRun watermarkRun = watermarkPara.createRun(); |
||||||
|
watermarkRun.setText(watermarkText); |
||||||
|
|
||||||
|
// 使用兼容的方法设置样式
|
||||||
|
setRunFontSize(watermarkRun, 24); |
||||||
|
setRunColor(watermarkRun, "CCCCCC"); |
||||||
|
|
||||||
|
// 添加间距
|
||||||
|
setParagraphSpacing(watermarkPara, 100, 100); |
||||||
|
|
||||||
|
} catch (Exception e) { |
||||||
|
System.err.println("兼容水印添加失败: " + e.getMessage()); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 安全设置字体大小（使用 try-catch 避免版本兼容性问题） |
||||||
|
*/ |
||||||
|
private void setRunFontSize(XWPFRun run, int size) { |
||||||
|
try { |
||||||
|
run.setFontSize(size); |
||||||
|
} catch (NoSuchMethodError | NoClassDefFoundError e) { |
||||||
|
System.err.println("设置字体大小失败，跳过: " + e.getMessage()); |
||||||
|
} catch (Exception e) { |
||||||
|
// 静默处理
|
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 安全设置粗体 |
||||||
|
*/ |
||||||
|
private void setRunBold(XWPFRun run, boolean bold) { |
||||||
|
try { |
||||||
|
run.setBold(bold); |
||||||
|
} catch (NoSuchMethodError | NoClassDefFoundError e) { |
||||||
|
System.err.println("设置粗体失败，跳过: " + e.getMessage()); |
||||||
|
} catch (Exception e) { |
||||||
|
// 静默处理
|
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 安全设置斜体 |
||||||
|
*/ |
||||||
|
private void setRunItalic(XWPFRun run, boolean italic) { |
||||||
|
try { |
||||||
|
run.setItalic(italic); |
||||||
|
} catch (NoSuchMethodError | NoClassDefFoundError e) { |
||||||
|
System.err.println("设置斜体失败，跳过: " + e.getMessage()); |
||||||
|
} catch (Exception e) { |
||||||
|
// 静默处理
|
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 安全设置字体 |
||||||
|
*/ |
||||||
|
private void setRunFontFamily(XWPFRun run, String fontFamily) { |
||||||
|
try { |
||||||
|
run.setFontFamily(fontFamily); |
||||||
|
} catch (NoSuchMethodError | NoClassDefFoundError e) { |
||||||
|
System.err.println("设置字体失败，跳过: " + e.getMessage()); |
||||||
|
} catch (Exception e) { |
||||||
|
// 静默处理
|
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 安全设置颜色 |
||||||
|
*/ |
||||||
|
private void setRunColor(XWPFRun run, String color) { |
||||||
|
try { |
||||||
|
run.setColor(color); |
||||||
|
} catch (NoSuchMethodError | NoClassDefFoundError e) { |
||||||
|
System.err.println("设置颜色失败，跳过: " + e.getMessage()); |
||||||
|
} catch (Exception e) { |
||||||
|
// 静默处理
|
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
/** |
||||||
|
* 安全的段落创建方法 |
||||||
|
*/ |
||||||
|
private XWPFParagraph createSafeParagraph(XWPFDocument document) { |
||||||
|
try { |
||||||
|
return document.createParagraph(); |
||||||
|
} catch (Exception e) { |
||||||
|
System.err.println("创建段落失败: " + e.getMessage()); |
||||||
|
return null; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
private String formatDouble(Double value) { |
||||||
|
return value != null ? String.format("%.2f", value) : "-"; |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public String getFileExtension() { |
||||||
|
return ".docx"; |
||||||
|
} |
||||||
|
} |
||||||
Loading…
Reference in new issue