新增工具模块，并添加了子模块-报告工具生成模块
This commit is contained in:
@@ -81,6 +81,9 @@ public class AdPlan extends BaseEntity implements Serializable {
|
||||
|
||||
/**
|
||||
* 是否关联报告，0-不关联 1-关联
|
||||
* 目前这个阶段
|
||||
* 0-不关联的是采用替换占位符的方式生成测试报告
|
||||
* 1-关联的是采用模板生成测试报告
|
||||
*/
|
||||
private Integer associateReport;
|
||||
|
||||
|
||||
1
pom.xml
1
pom.xml
@@ -14,6 +14,7 @@
|
||||
<module>detection</module>
|
||||
<module>storage</module>
|
||||
<module>event_smart</module>
|
||||
<module>tools</module>
|
||||
</modules>
|
||||
<packaging>pom</packaging>
|
||||
<name>融合各工具的项目</name>
|
||||
|
||||
52
tools/README.md
Normal file
52
tools/README.md
Normal file
@@ -0,0 +1,52 @@
|
||||
# Tools 工具模块
|
||||
|
||||
这是CN_Gather项目的工具集合模块，用于管理各种通用工具。每个工具作为独立的子模块，拥有完整的MVC架构。
|
||||
|
||||
## 架构设计
|
||||
|
||||
```
|
||||
tools/
|
||||
├── report-generator/ # 报告生成工具
|
||||
├── data-generator/ # 数据生成工具（未来扩展）
|
||||
└── file-processor/ # 文件处理工具（未来扩展）
|
||||
```
|
||||
|
||||
## 子模块说明
|
||||
|
||||
### 1. report-generator（报告生成工具）
|
||||
- **功能**: 通用的文档模板处理和报告生成
|
||||
- **技术栈**: Apache POI, docx4j
|
||||
- **特性**:
|
||||
- 占位符替换
|
||||
- 书签定位插入
|
||||
- 动态表格生成
|
||||
- 多文档合并
|
||||
- 样式管理
|
||||
|
||||
### 2. data-generator（数据生成工具）
|
||||
- **功能**: MySQL数据生成、测试数据构造
|
||||
- **状态**: 待开发
|
||||
|
||||
### 3. file-processor（文件处理工具）
|
||||
- **功能**: 文件转换、批量处理等
|
||||
- **状态**: 待开发
|
||||
|
||||
## 设计原则
|
||||
|
||||
1. **独立性**: 每个工具子模块独立部署和使用
|
||||
2. **通用性**: 脱离具体业务逻辑，提供纯技术能力
|
||||
3. **可扩展**: 支持插件化扩展和自定义处理器
|
||||
4. **易集成**: 提供简洁的API接口
|
||||
|
||||
## 使用方式
|
||||
|
||||
每个工具模块都提供HTTP接口和Java API两种使用方式：
|
||||
- HTTP接口：适合微服务架构，跨语言调用
|
||||
- Java API：适合同项目内直接依赖调用
|
||||
|
||||
## 模块间依赖
|
||||
|
||||
tools模块作为独立的工具集合，尽量减少对其他业务模块的依赖，主要依赖：
|
||||
- 基础的Spring Boot框架
|
||||
- 通用工具库（hutool等）
|
||||
- 各工具特定的技术栈依赖
|
||||
25
tools/pom.xml
Normal file
25
tools/pom.xml
Normal file
@@ -0,0 +1,25 @@
|
||||
<?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>CN_Gather</artifactId>
|
||||
<version>1.0.0</version>
|
||||
</parent>
|
||||
|
||||
<artifactId>tools</artifactId>
|
||||
<packaging>pom</packaging>
|
||||
<name>工具集合模块</name>
|
||||
<description>各种通用工具的集合，每个工具作为独立子模块</description>
|
||||
|
||||
<modules>
|
||||
<module>report-generator</module>
|
||||
<!-- 未来可以添加更多工具子模块 -->
|
||||
<!-- <module>data-generator</module> -->
|
||||
<!-- <module>file-processor</module> -->
|
||||
</modules>
|
||||
|
||||
</project>
|
||||
184
tools/report-generator/PROJECT_SUMMARY.md
Normal file
184
tools/report-generator/PROJECT_SUMMARY.md
Normal file
@@ -0,0 +1,184 @@
|
||||
# Report Generator 项目总结
|
||||
|
||||
## 项目概述
|
||||
|
||||
成功从CN_Gather项目的detection模块中抽取出了报告生成功能，创建了独立的通用工具模块。该模块完全脱离了业务逻辑，提供纯技术能力，可以被其他项目复用。
|
||||
|
||||
## 架构设计
|
||||
|
||||
### 模块结构
|
||||
```
|
||||
CN_Gather/
|
||||
├── tools/ # 工具集合模块（新增）
|
||||
│ ├── report-generator/ # 报告生成工具子模块（新增）
|
||||
│ │ ├── src/main/java/com/njcn/gather/tools/report/
|
||||
│ │ │ ├── controller/ # HTTP接口层
|
||||
│ │ │ ├── service/ # 业务服务层
|
||||
│ │ │ ├── engine/ # 模板处理引擎
|
||||
│ │ │ ├── model/ # 数据模型
|
||||
│ │ │ ├── util/ # 工具类
|
||||
│ │ └── pom.xml
|
||||
│ ├── pom.xml
|
||||
│ └── README.md
|
||||
├── detection/ # 原有检测模块（保持不变）
|
||||
├── entrance/ # 应用入口
|
||||
└── pom.xml # 已更新，包含tools模块
|
||||
```
|
||||
|
||||
### 核心组件
|
||||
|
||||
#### 1. 数据模型层 (model/)
|
||||
- **TemplateSource**: 模板来源定义，支持文件、流、字节数组输入
|
||||
- **TemplateRequest**: 模板处理请求封装
|
||||
- **ProcessResult**: 处理结果封装，包含详细统计和错误信息
|
||||
- **ProcessOptions**: 处理选项配置，支持各种功能开关
|
||||
- **TemplateType**: 模板类型枚举，支持扩展
|
||||
|
||||
#### 2. 引擎层 (engine/)
|
||||
- **DocumentProcessor**: 文档处理器接口
|
||||
- **WordDocumentProcessor**: Word文档处理实现
|
||||
|
||||
#### 3. 工具类层 (util/)
|
||||
- **WordDocumentUtil**: 基于Apache POI的通用Word操作
|
||||
- **Docx4jAdvancedUtil**: 基于docx4j的高级Word操作
|
||||
|
||||
#### 4. 服务层 (service/)
|
||||
- **ReportGeneratorService**: 报告生成服务接口
|
||||
- **ReportGeneratorServiceImpl**: 服务实现，支持同步/异步/批量处理
|
||||
|
||||
#### 5. 控制器层 (controller/)
|
||||
- **ReportGeneratorController**: REST API接口，提供多种调用方式
|
||||
|
||||
## 技术特性
|
||||
|
||||
### 已实现功能
|
||||
1. ✅ **占位符替换**: `${key}`、`#{key}`、`{{key}}` 格式支持
|
||||
2. ✅ **多输入方式**: 文件路径、输入流、字节数组
|
||||
3. ✅ **多输出方式**: 文件、流、字节数组
|
||||
4. ✅ **同步处理**: 立即返回结果
|
||||
5. ✅ **异步处理**: 支持长时间处理任务
|
||||
6. ✅ **批量处理**: 一次处理多个模板
|
||||
7. ✅ **错误处理**: 完整的异常处理和错误信息
|
||||
8. ✅ **性能监控**: 处理时间和统计信息
|
||||
9. ✅ **参数验证**: 请求参数完整性验证
|
||||
10. ✅ **模板验证**: 模板文件有效性检查
|
||||
|
||||
### 从原有代码抽取的功能
|
||||
- **WordUtil → WordDocumentUtil**: 基础Word文档操作
|
||||
- **Docx4jUtil → Docx4jAdvancedUtil**: 高级文档处理功能
|
||||
- **BookmarkUtil 功能**: 整合到高级工具类中
|
||||
- **占位符替换逻辑**: 通用化并支持多种格式
|
||||
- **文档合并功能**: 保留并优化
|
||||
- **样式管理功能**: 字体、颜色、对齐等
|
||||
|
||||
## API 设计
|
||||
|
||||
### HTTP接口
|
||||
```
|
||||
POST /api/tools/report/process/simple # 简单处理
|
||||
POST /api/tools/report/process/advanced # 高级处理
|
||||
POST /api/tools/report/process/async # 异步处理
|
||||
GET /api/tools/report/process/async/{id} # 查询异步结果
|
||||
POST /api/tools/report/process/batch # 批量处理
|
||||
POST /api/tools/report/validate # 模板验证
|
||||
```
|
||||
|
||||
### Java API
|
||||
```java
|
||||
// 核心服务接口
|
||||
ProcessResult processTemplate(TemplateRequest request)
|
||||
String processTemplateAsync(TemplateRequest request)
|
||||
ProcessResult getAsyncResult(String requestId)
|
||||
List<ProcessResult> batchProcessTemplates(List<TemplateRequest> requests)
|
||||
ProcessResult validateTemplate(TemplateRequest request)
|
||||
```
|
||||
|
||||
## 设计优势
|
||||
|
||||
### 1. 高内聚低耦合
|
||||
- 完全独立的模块，不依赖具体业务
|
||||
- 清晰的分层架构
|
||||
- 接口与实现分离
|
||||
|
||||
### 2. 可扩展性强
|
||||
- 支持插件化的文档处理器
|
||||
- 可方便扩展新的模板类型（PDF、Excel等）
|
||||
- 处理选项可灵活配置
|
||||
|
||||
### 3. 易于使用
|
||||
- 提供多种使用方式（HTTP、Java API）
|
||||
- Builder模式简化对象构建
|
||||
- 完整的使用文档和示例
|
||||
|
||||
### 4. 健壮性好
|
||||
- 完整的错误处理机制
|
||||
- 参数验证和模板验证
|
||||
- 资源自动管理和清理
|
||||
|
||||
### 5. 性能优异
|
||||
- 支持异步处理避免阻塞
|
||||
- 批量处理提高效率
|
||||
- 详细的性能统计信息
|
||||
|
||||
## 使用场景
|
||||
|
||||
### 1. 报告自动化生成
|
||||
- 检测报告、试验报告
|
||||
- 财务报表、统计报表
|
||||
- 证书、合格证等
|
||||
|
||||
### 2. 文档批量处理
|
||||
- 合同批量生成
|
||||
- 通知书批量制作
|
||||
- 标签批量打印
|
||||
|
||||
### 3. 模板管理系统
|
||||
- 模板上传和验证
|
||||
- 模板版本管理
|
||||
- 模板效果预览
|
||||
|
||||
## 部署和集成
|
||||
|
||||
### 1. 独立部署
|
||||
tools模块可以作为独立的微服务部署，对外提供HTTP接口。
|
||||
|
||||
### 2. 嵌入式集成
|
||||
其他项目可以直接依赖report-generator模块，使用Java API调用。
|
||||
|
||||
### 3. 与现有系统集成
|
||||
detection模块可以逐步迁移到使用新的report-generator工具。
|
||||
|
||||
## 后续扩展建议
|
||||
|
||||
### 1. 功能扩展
|
||||
- 支持PDF模板处理
|
||||
- 支持Excel模板处理
|
||||
- 支持图片插入和处理
|
||||
- 支持复杂表达式计算
|
||||
|
||||
### 2. 性能优化
|
||||
- 添加模板缓存机制
|
||||
- 支持流式处理大文件
|
||||
- 增加并发控制和限流
|
||||
|
||||
### 3. 管理功能
|
||||
- 模板管理界面
|
||||
- 处理任务监控面板
|
||||
- 统计分析报表
|
||||
|
||||
### 4. 安全增强
|
||||
- 用户权限控制
|
||||
- 模板安全检查
|
||||
- 操作审计日志
|
||||
|
||||
## 总结
|
||||
|
||||
成功创建了一个完全独立、通用的报告生成工具模块，实现了以下目标：
|
||||
|
||||
1. **脱离业务**: 完全剥离了电能质量检测的业务逻辑
|
||||
2. **通用化**: 可以处理任何Word模板和数据
|
||||
3. **易扩展**: 支持新的文档类型和处理方式
|
||||
4. **高可用**: 提供多种调用方式和完善的错误处理
|
||||
5. **高性能**: 支持异步和批量处理
|
||||
|
||||
该工具不仅可以满足当前CN_Gather项目的需求，也为未来的数据生成工具、文件处理工具等提供了良好的架构基础。
|
||||
126
tools/report-generator/README.md
Normal file
126
tools/report-generator/README.md
Normal file
@@ -0,0 +1,126 @@
|
||||
# Report Generator 报告生成工具
|
||||
|
||||
通用的文档模板处理和报告生成工具，支持Word文档的动态内容生成。
|
||||
|
||||
## 功能特性
|
||||
|
||||
### 1. 基础功能
|
||||
- ✅ 占位符替换 (`${key}` 格式)
|
||||
- ✅ Word文档处理 (基于Apache POI)
|
||||
- ✅ 模板文件管理
|
||||
- ✅ 多种输入输出方式 (文件路径、流、字节数组)
|
||||
|
||||
### 2. 高级功能
|
||||
- ✅ 书签定位插入 (基于docx4j)
|
||||
- ✅ 动态表格生成
|
||||
- ✅ 文档合并
|
||||
- ✅ 样式管理 (字体、颜色、对齐)
|
||||
- ✅ 自动分页控制
|
||||
- ✅ 深拷贝支持
|
||||
|
||||
### 3. 扩展功能
|
||||
- 🔄 自定义处理器插件化
|
||||
- 🔄 复杂表达式支持
|
||||
- 🔄 模板验证
|
||||
- 🔄 缓存优化
|
||||
|
||||
## 架构设计
|
||||
|
||||
```
|
||||
report-generator/
|
||||
├── controller/ # HTTP接口层
|
||||
├── service/ # 业务逻辑层
|
||||
├── engine/ # 模板引擎核心
|
||||
│ ├── DocumentProcessor # 文档处理器接口
|
||||
│ ├── WordEngine # Word处理引擎
|
||||
│ ├── PlaceholderEngine # 占位符处理引擎
|
||||
│ └── BookmarkEngine # 书签处理引擎
|
||||
├── model/ # 数据模型
|
||||
│ ├── TemplateRequest # 处理请求模型
|
||||
│ ├── TemplateSource # 模板源定义
|
||||
│ └── ProcessResult # 处理结果
|
||||
├── util/ # 工具类
|
||||
└── exception/ # 异常处理
|
||||
```
|
||||
|
||||
## 使用示例
|
||||
|
||||
### 1. HTTP接口方式
|
||||
|
||||
```bash
|
||||
# 简单占位符替换
|
||||
POST /api/report/process
|
||||
Content-Type: multipart/form-data
|
||||
|
||||
{
|
||||
"templateFile": <template.docx>,
|
||||
"data": {
|
||||
"name": "张三",
|
||||
"date": "2024-01-01",
|
||||
"amount": "1000.00"
|
||||
},
|
||||
"outputFileName": "report.docx"
|
||||
}
|
||||
```
|
||||
|
||||
### 2. Java API方式
|
||||
|
||||
```java
|
||||
// 注入服务
|
||||
@Autowired
|
||||
private ReportGeneratorService reportService;
|
||||
|
||||
// 创建请求
|
||||
TemplateRequest request = TemplateRequest.builder()
|
||||
.templatePath("template.docx")
|
||||
.data(Map.of("name", "张三", "date", "2024-01-01"))
|
||||
.outputPath("result.docx")
|
||||
.options(ProcessOptions.builder()
|
||||
.enablePlaceholder(true)
|
||||
.enableBookmark(true)
|
||||
.build())
|
||||
.build();
|
||||
|
||||
// 处理模板
|
||||
ProcessResult result = reportService.processTemplate(request);
|
||||
```
|
||||
|
||||
## 配置说明
|
||||
|
||||
### 处理选项 (ProcessOptions)
|
||||
```java
|
||||
ProcessOptions options = ProcessOptions.builder()
|
||||
.enablePlaceholder(true) // 启用占位符替换
|
||||
.enableBookmark(true) // 启用书签处理
|
||||
.enableDynamicTable(true) // 启用动态表格
|
||||
.enableAutoPage(true) // 启用自动分页
|
||||
.placeholderPattern("${}") // 占位符格式
|
||||
.build();
|
||||
```
|
||||
|
||||
### 模板源配置 (TemplateSource)
|
||||
```java
|
||||
// 文件路径方式
|
||||
TemplateSource.fromFile("path/to/template.docx")
|
||||
|
||||
// 输入流方式
|
||||
TemplateSource.fromStream(inputStream)
|
||||
|
||||
// 字节数组方式
|
||||
TemplateSource.fromBytes(byteArray)
|
||||
```
|
||||
|
||||
## 技术栈
|
||||
|
||||
- **Apache POI**: Word文档基础操作
|
||||
- **docx4j**: 高级Word文档处理
|
||||
- **Spring Boot**: 框架支持
|
||||
- **Hutool**: 工具库
|
||||
- **Jackson**: JSON处理
|
||||
|
||||
## 性能优化
|
||||
|
||||
1. **模板缓存**: 常用模板缓存减少IO
|
||||
2. **流式处理**: 大文件流式处理避免内存溢出
|
||||
3. **异步处理**: 支持异步批量处理
|
||||
4. **资源管理**: 自动资源清理和释放
|
||||
144
tools/report-generator/pom.xml
Normal file
144
tools/report-generator/pom.xml
Normal file
@@ -0,0 +1,144 @@
|
||||
<?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>tools</artifactId>
|
||||
<version>1.0.0</version>
|
||||
</parent>
|
||||
|
||||
<artifactId>report-generator</artifactId>
|
||||
<packaging>jar</packaging>
|
||||
<name>报告生成工具</name>
|
||||
<description>通用的文档模板处理和报告生成工具，支持占位符替换、书签插入、动态表格等功能</description>
|
||||
|
||||
<properties>
|
||||
<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
|
||||
<project.reporting.outputEncoding>UTF-8</project.reporting.outputEncoding>
|
||||
<maven.compiler.encoding>UTF-8</maven.compiler.encoding>
|
||||
<java.version>1.8</java.version>
|
||||
<maven.compiler.source>1.8</maven.compiler.source>
|
||||
<maven.compiler.target>1.8</maven.compiler.target>
|
||||
</properties>
|
||||
|
||||
<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>
|
||||
|
||||
<!-- Apache POI for Word document processing - 与detection模块版本保持一致 -->
|
||||
<dependency>
|
||||
<groupId>org.apache.poi</groupId>
|
||||
<artifactId>poi</artifactId>
|
||||
<version>4.1.2</version>
|
||||
</dependency>
|
||||
|
||||
<dependency>
|
||||
<groupId>org.apache.poi</groupId>
|
||||
<artifactId>poi-ooxml</artifactId>
|
||||
<version>4.1.2</version>
|
||||
</dependency>
|
||||
|
||||
<dependency>
|
||||
<groupId>org.apache.poi</groupId>
|
||||
<artifactId>poi-ooxml-schemas</artifactId>
|
||||
<version>4.1.2</version>
|
||||
</dependency>
|
||||
|
||||
<dependency>
|
||||
<groupId>org.apache.poi</groupId>
|
||||
<artifactId>poi-scratchpad</artifactId>
|
||||
<version>4.1.2</version>
|
||||
</dependency>
|
||||
|
||||
<!-- docx4j for advanced Word processing - 与detection模块版本保持一致 -->
|
||||
<dependency>
|
||||
<groupId>jakarta.xml.bind</groupId>
|
||||
<artifactId>jakarta.xml.bind-api</artifactId>
|
||||
<version>2.3.3</version>
|
||||
</dependency>
|
||||
|
||||
<dependency>
|
||||
<groupId>org.glassfish.jaxb</groupId>
|
||||
<artifactId>jaxb-runtime</artifactId>
|
||||
<version>2.3.3</version>
|
||||
</dependency>
|
||||
|
||||
<dependency>
|
||||
<groupId>org.docx4j</groupId>
|
||||
<artifactId>docx4j</artifactId>
|
||||
<version>6.1.0</version>
|
||||
</dependency>
|
||||
|
||||
<!-- Validation API -->
|
||||
<dependency>
|
||||
<groupId>org.springframework.boot</groupId>
|
||||
<artifactId>spring-boot-starter-validation</artifactId>
|
||||
</dependency>
|
||||
|
||||
<!-- JSON processing - 与detection模块版本保持一致 -->
|
||||
<dependency>
|
||||
<groupId>com.fasterxml.jackson.core</groupId>
|
||||
<artifactId>jackson-databind</artifactId>
|
||||
<version>2.12.0</version>
|
||||
</dependency>
|
||||
|
||||
<dependency>
|
||||
<groupId>com.fasterxml.jackson.core</groupId>
|
||||
<artifactId>jackson-core</artifactId>
|
||||
<version>2.12.0</version>
|
||||
</dependency>
|
||||
|
||||
<dependency>
|
||||
<groupId>com.fasterxml.jackson.core</groupId>
|
||||
<artifactId>jackson-annotations</artifactId>
|
||||
<version>2.12.0</version>
|
||||
</dependency>
|
||||
|
||||
<!-- FastJSON - 与detection模块保持一致 -->
|
||||
<dependency>
|
||||
<groupId>com.alibaba</groupId>
|
||||
<artifactId>fastjson</artifactId>
|
||||
<version>1.2.83</version>
|
||||
</dependency>
|
||||
|
||||
<!-- File upload support -->
|
||||
<dependency>
|
||||
<groupId>commons-fileupload</groupId>
|
||||
<artifactId>commons-fileupload</artifactId>
|
||||
<version>1.4</version>
|
||||
</dependency>
|
||||
|
||||
|
||||
</dependencies>
|
||||
|
||||
<build>
|
||||
<plugins>
|
||||
<plugin>
|
||||
<groupId>org.springframework.boot</groupId>
|
||||
<artifactId>spring-boot-maven-plugin</artifactId>
|
||||
<configuration>
|
||||
<excludes>
|
||||
<exclude>
|
||||
<groupId>org.projectlombok</groupId>
|
||||
<artifactId>lombok</artifactId>
|
||||
</exclude>
|
||||
</excludes>
|
||||
</configuration>
|
||||
</plugin>
|
||||
</plugins>
|
||||
</build>
|
||||
|
||||
</project>
|
||||
@@ -0,0 +1,392 @@
|
||||
package com.njcn.gather.tools.report.controller;
|
||||
|
||||
import com.njcn.gather.tools.report.model.*;
|
||||
import com.njcn.gather.tools.report.service.ReportGeneratorService;
|
||||
import lombok.RequiredArgsConstructor;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.http.HttpHeaders;
|
||||
import org.springframework.http.HttpStatus;
|
||||
import org.springframework.http.MediaType;
|
||||
import org.springframework.http.ResponseEntity;
|
||||
import org.springframework.web.bind.annotation.*;
|
||||
import org.springframework.web.multipart.MultipartFile;
|
||||
|
||||
import javax.servlet.http.HttpServletResponse;
|
||||
import java.io.IOException;
|
||||
import java.io.OutputStream;
|
||||
import java.util.HashMap;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
|
||||
/**
|
||||
* 报告生成工具HTTP接口
|
||||
* 提供REST风格的报告生成API
|
||||
*
|
||||
* @author hongawen
|
||||
*/
|
||||
@Slf4j
|
||||
@RestController
|
||||
@RequestMapping("/api/tools/report")
|
||||
@RequiredArgsConstructor
|
||||
public class ReportGeneratorController {
|
||||
|
||||
private final ReportGeneratorService reportGeneratorService;
|
||||
|
||||
/**
|
||||
* 简单模板处理 - 占位符替换
|
||||
*
|
||||
* @param templateFile 模板文件
|
||||
* @param dataJson 数据JSON字符串
|
||||
* @param outputFileName 输出文件名（可选）
|
||||
* @param response HTTP响应
|
||||
*/
|
||||
@PostMapping("/process/simple")
|
||||
public void processSimpleTemplate(
|
||||
@RequestParam("templateFile") MultipartFile templateFile,
|
||||
@RequestParam("data") String dataJson,
|
||||
@RequestParam(value = "outputFileName", required = false) String outputFileName,
|
||||
HttpServletResponse response) throws IOException {
|
||||
|
||||
log.info("接收到简单模板处理请求: {}", templateFile.getOriginalFilename());
|
||||
|
||||
try {
|
||||
// 解析数据
|
||||
@SuppressWarnings("unchecked")
|
||||
Map<String, Object> data = com.fasterxml.jackson.databind.ObjectMapper.class
|
||||
.getDeclaredConstructor().newInstance().readValue(dataJson, Map.class);
|
||||
|
||||
// 构建请求
|
||||
TemplateRequest request = TemplateRequest.builder()
|
||||
.templateSource(TemplateSource.fromStream(templateFile.getInputStream(),
|
||||
templateFile.getOriginalFilename()))
|
||||
.data(data)
|
||||
.options(ProcessOptions.simplePlaceholder())
|
||||
.outputTarget(TemplateRequest.OutputTarget.toBytes())
|
||||
.build();
|
||||
|
||||
// 处理模板
|
||||
ProcessResult result = reportGeneratorService.processTemplate(request);
|
||||
|
||||
if (result.isSuccess() && result.getOutputBytes() != null) {
|
||||
// 设置响应头
|
||||
String fileName = outputFileName != null ? outputFileName :
|
||||
"report_" + System.currentTimeMillis() + ".docx";
|
||||
response.setContentType("application/vnd.openxmlformats-officedocument.wordprocessingml.document");
|
||||
response.setHeader("Content-Disposition", "attachment; filename=\"" + fileName + "\"");
|
||||
response.setContentLength(result.getOutputBytes().length);
|
||||
|
||||
// 输出文件
|
||||
try (OutputStream os = response.getOutputStream()) {
|
||||
os.write(result.getOutputBytes());
|
||||
os.flush();
|
||||
}
|
||||
|
||||
log.info("简单模板处理成功: {}", fileName);
|
||||
} else {
|
||||
response.setStatus(HttpStatus.INTERNAL_SERVER_ERROR.value());
|
||||
response.getWriter().write("模板处理失败: " + result.getMessage());
|
||||
}
|
||||
|
||||
} catch (Exception e) {
|
||||
log.error("简单模板处理异常: {}", e.getMessage(), e);
|
||||
response.setStatus(HttpStatus.INTERNAL_SERVER_ERROR.value());
|
||||
response.getWriter().write("处理异常: " + e.getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 高级模板处理 - 支持更多功能
|
||||
*/
|
||||
@PostMapping("/process/advanced")
|
||||
public ResponseEntity<Map<String, Object>> processAdvancedTemplate(@RequestBody AdvancedProcessRequest request) {
|
||||
log.info("接收到高级模板处理请求");
|
||||
|
||||
try {
|
||||
// 构建模板请求
|
||||
TemplateRequest templateRequest = buildTemplateRequest(request);
|
||||
|
||||
// 处理模板
|
||||
ProcessResult result = reportGeneratorService.processTemplate(templateRequest);
|
||||
|
||||
// 构建响应
|
||||
Map<String, Object> response = new HashMap<>();
|
||||
response.put("success", result.isSuccess());
|
||||
response.put("message", result.getMessage());
|
||||
response.put("requestId", result.getRequestId());
|
||||
|
||||
if (result.isSuccess()) {
|
||||
if (result.getOutputBytes() != null) {
|
||||
// 返回base64编码的文件内容
|
||||
String base64Content = java.util.Base64.getEncoder().encodeToString(result.getOutputBytes());
|
||||
response.put("fileContent", base64Content);
|
||||
}
|
||||
if (result.getStats() != null) {
|
||||
response.put("stats", result.getStats());
|
||||
}
|
||||
response.put("processingTimeMs", result.getProcessingTimeMs());
|
||||
} else {
|
||||
response.put("errorCode", result.getErrorCode());
|
||||
response.put("errorDetails", result.getErrorDetails());
|
||||
}
|
||||
|
||||
return ResponseEntity.ok(response);
|
||||
|
||||
} catch (Exception e) {
|
||||
log.error("高级模板处理异常: {}", e.getMessage(), e);
|
||||
Map<String, Object> errorResponse = new HashMap<>();
|
||||
errorResponse.put("success", false);
|
||||
errorResponse.put("message", "处理异常");
|
||||
errorResponse.put("errorDetails", e.getMessage());
|
||||
return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body(errorResponse);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 异步处理模板
|
||||
*/
|
||||
@PostMapping("/process/async")
|
||||
public ResponseEntity<Map<String, Object>> processTemplateAsync(@RequestBody AdvancedProcessRequest request) {
|
||||
log.info("接收到异步模板处理请求");
|
||||
|
||||
try {
|
||||
// 构建模板请求
|
||||
TemplateRequest templateRequest = buildTemplateRequest(request);
|
||||
|
||||
// 启动异步处理
|
||||
String requestId = reportGeneratorService.processTemplateAsync(templateRequest);
|
||||
|
||||
Map<String, Object> response = new HashMap<>();
|
||||
response.put("success", true);
|
||||
response.put("message", "异步处理已启动");
|
||||
response.put("requestId", requestId);
|
||||
|
||||
return ResponseEntity.ok(response);
|
||||
|
||||
} catch (Exception e) {
|
||||
log.error("异步模板处理启动失败: {}", e.getMessage(), e);
|
||||
Map<String, Object> errorResponse = new HashMap<>();
|
||||
errorResponse.put("success", false);
|
||||
errorResponse.put("message", "启动异步处理失败");
|
||||
errorResponse.put("errorDetails", e.getMessage());
|
||||
return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body(errorResponse);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 查询异步处理结果
|
||||
*/
|
||||
@GetMapping("/process/async/{requestId}")
|
||||
public ResponseEntity<Map<String, Object>> getAsyncResult(@PathVariable String requestId) {
|
||||
ProcessResult result = reportGeneratorService.getAsyncResult(requestId);
|
||||
|
||||
Map<String, Object> response = new HashMap<>();
|
||||
if (result == null) {
|
||||
response.put("found", false);
|
||||
response.put("message", "未找到对应的处理结果");
|
||||
return ResponseEntity.ok(response);
|
||||
}
|
||||
|
||||
response.put("found", true);
|
||||
response.put("success", result.isSuccess());
|
||||
response.put("message", result.getMessage());
|
||||
response.put("requestId", result.getRequestId());
|
||||
|
||||
if (result.getEndTime() != null) {
|
||||
// 处理已完成
|
||||
response.put("completed", true);
|
||||
response.put("processingTimeMs", result.getProcessingTimeMs());
|
||||
|
||||
if (result.isSuccess() && result.getOutputBytes() != null) {
|
||||
String base64Content = java.util.Base64.getEncoder().encodeToString(result.getOutputBytes());
|
||||
response.put("fileContent", base64Content);
|
||||
}
|
||||
|
||||
if (result.getStats() != null) {
|
||||
response.put("stats", result.getStats());
|
||||
}
|
||||
} else {
|
||||
response.put("completed", false);
|
||||
}
|
||||
|
||||
return ResponseEntity.ok(response);
|
||||
}
|
||||
|
||||
/**
|
||||
* 批量处理模板
|
||||
*/
|
||||
@PostMapping("/process/batch")
|
||||
public ResponseEntity<Map<String, Object>> batchProcessTemplates(@RequestBody List<AdvancedProcessRequest> requests) {
|
||||
log.info("接收到批量模板处理请求，数量: {}", requests.size());
|
||||
|
||||
try {
|
||||
// 构建模板请求列表
|
||||
List<TemplateRequest> templateRequests = new java.util.ArrayList<>();
|
||||
for (AdvancedProcessRequest req : requests) {
|
||||
templateRequests.add(buildTemplateRequest(req));
|
||||
}
|
||||
|
||||
// 批量处理
|
||||
List<ProcessResult> results = reportGeneratorService.batchProcessTemplates(templateRequests);
|
||||
|
||||
// 统计结果
|
||||
long successCount = results.stream().mapToLong(r -> r.isSuccess() ? 1 : 0).sum();
|
||||
long failureCount = results.size() - successCount;
|
||||
|
||||
Map<String, Object> response = new HashMap<>();
|
||||
response.put("success", true);
|
||||
response.put("message", "批量处理完成");
|
||||
response.put("totalCount", results.size());
|
||||
response.put("successCount", successCount);
|
||||
response.put("failureCount", failureCount);
|
||||
List<Map<String, Object>> resultSummaries = new java.util.ArrayList<>();
|
||||
for (ProcessResult result : results) {
|
||||
resultSummaries.add(buildResultSummary(result));
|
||||
}
|
||||
response.put("results", resultSummaries);
|
||||
|
||||
return ResponseEntity.ok(response);
|
||||
|
||||
} catch (Exception e) {
|
||||
log.error("批量模板处理异常: {}", e.getMessage(), e);
|
||||
Map<String, Object> errorResponse = new HashMap<>();
|
||||
errorResponse.put("success", false);
|
||||
errorResponse.put("message", "批量处理异常");
|
||||
errorResponse.put("errorDetails", e.getMessage());
|
||||
return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body(errorResponse);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 验证模板
|
||||
*/
|
||||
@PostMapping("/validate")
|
||||
public ResponseEntity<Map<String, Object>> validateTemplate(@RequestBody AdvancedProcessRequest request) {
|
||||
try {
|
||||
TemplateRequest templateRequest = buildTemplateRequest(request);
|
||||
ProcessResult result = reportGeneratorService.validateTemplate(templateRequest);
|
||||
|
||||
Map<String, Object> response = new HashMap<>();
|
||||
response.put("valid", result.isSuccess());
|
||||
response.put("message", result.getMessage());
|
||||
|
||||
if (!result.isSuccess()) {
|
||||
response.put("errorCode", result.getErrorCode());
|
||||
response.put("errorDetails", result.getErrorDetails());
|
||||
}
|
||||
|
||||
return ResponseEntity.ok(response);
|
||||
|
||||
} catch (Exception e) {
|
||||
log.error("模板验证异常: {}", e.getMessage(), e);
|
||||
Map<String, Object> errorResponse = new HashMap<>();
|
||||
errorResponse.put("valid", false);
|
||||
errorResponse.put("message", "验证异常");
|
||||
errorResponse.put("errorDetails", e.getMessage());
|
||||
return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body(errorResponse);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 构建模板请求
|
||||
*/
|
||||
private TemplateRequest buildTemplateRequest(AdvancedProcessRequest request) {
|
||||
// 构建模板源
|
||||
TemplateSource templateSource;
|
||||
if (request.getTemplateFilePath() != null) {
|
||||
templateSource = TemplateSource.fromFile(request.getTemplateFilePath());
|
||||
} else if (request.getTemplateContent() != null) {
|
||||
byte[] content = java.util.Base64.getDecoder().decode(request.getTemplateContent());
|
||||
templateSource = TemplateSource.fromBytes(content, request.getTemplateName());
|
||||
} else {
|
||||
throw new IllegalArgumentException("模板源不能为空");
|
||||
}
|
||||
|
||||
// 构建处理选项
|
||||
ProcessOptions options = ProcessOptions.builder()
|
||||
.enablePlaceholder(request.isEnablePlaceholder())
|
||||
.enableBookmark(request.isEnableBookmark())
|
||||
.enableDynamicTable(request.isEnableDynamicTable())
|
||||
.enableAutoPage(request.isEnableAutoPage())
|
||||
.build();
|
||||
|
||||
// 构建输出目标
|
||||
TemplateRequest.OutputTarget outputTarget;
|
||||
if (request.getOutputFilePath() != null) {
|
||||
outputTarget = TemplateRequest.OutputTarget.toFile(request.getOutputFilePath());
|
||||
} else {
|
||||
outputTarget = TemplateRequest.OutputTarget.toBytes();
|
||||
}
|
||||
|
||||
return TemplateRequest.builder()
|
||||
.templateSource(templateSource)
|
||||
.data(request.getData())
|
||||
.options(options)
|
||||
.outputTarget(outputTarget)
|
||||
.requestId(request.getRequestId())
|
||||
.build();
|
||||
}
|
||||
|
||||
/**
|
||||
* 构建结果摘要
|
||||
*/
|
||||
private Map<String, Object> buildResultSummary(ProcessResult result) {
|
||||
Map<String, Object> summary = new HashMap<>();
|
||||
summary.put("requestId", result.getRequestId());
|
||||
summary.put("success", result.isSuccess());
|
||||
summary.put("message", result.getMessage());
|
||||
summary.put("processingTimeMs", result.getProcessingTimeMs());
|
||||
|
||||
if (!result.isSuccess()) {
|
||||
summary.put("errorCode", result.getErrorCode());
|
||||
}
|
||||
|
||||
return summary;
|
||||
}
|
||||
|
||||
/**
|
||||
* 高级处理请求模型
|
||||
*/
|
||||
public static class AdvancedProcessRequest {
|
||||
private String requestId;
|
||||
private String templateFilePath;
|
||||
private String templateContent; // base64编码
|
||||
private String templateName;
|
||||
private Map<String, Object> data;
|
||||
private String outputFilePath;
|
||||
private boolean enablePlaceholder = true;
|
||||
private boolean enableBookmark = true;
|
||||
private boolean enableDynamicTable = true;
|
||||
private boolean enableAutoPage = true;
|
||||
|
||||
// Getters and Setters
|
||||
public String getRequestId() { return requestId; }
|
||||
public void setRequestId(String requestId) { this.requestId = requestId; }
|
||||
|
||||
public String getTemplateFilePath() { return templateFilePath; }
|
||||
public void setTemplateFilePath(String templateFilePath) { this.templateFilePath = templateFilePath; }
|
||||
|
||||
public String getTemplateContent() { return templateContent; }
|
||||
public void setTemplateContent(String templateContent) { this.templateContent = templateContent; }
|
||||
|
||||
public String getTemplateName() { return templateName; }
|
||||
public void setTemplateName(String templateName) { this.templateName = templateName; }
|
||||
|
||||
public Map<String, Object> getData() { return data; }
|
||||
public void setData(Map<String, Object> data) { this.data = data; }
|
||||
|
||||
public String getOutputFilePath() { return outputFilePath; }
|
||||
public void setOutputFilePath(String outputFilePath) { this.outputFilePath = outputFilePath; }
|
||||
|
||||
public boolean isEnablePlaceholder() { return enablePlaceholder; }
|
||||
public void setEnablePlaceholder(boolean enablePlaceholder) { this.enablePlaceholder = enablePlaceholder; }
|
||||
|
||||
public boolean isEnableBookmark() { return enableBookmark; }
|
||||
public void setEnableBookmark(boolean enableBookmark) { this.enableBookmark = enableBookmark; }
|
||||
|
||||
public boolean isEnableDynamicTable() { return enableDynamicTable; }
|
||||
public void setEnableDynamicTable(boolean enableDynamicTable) { this.enableDynamicTable = enableDynamicTable; }
|
||||
|
||||
public boolean isEnableAutoPage() { return enableAutoPage; }
|
||||
public void setEnableAutoPage(boolean enableAutoPage) { this.enableAutoPage = enableAutoPage; }
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,63 @@
|
||||
package com.njcn.gather.tools.report.engine;
|
||||
|
||||
import com.njcn.gather.tools.report.model.TemplateRequest;
|
||||
import com.njcn.gather.tools.report.model.ProcessResult;
|
||||
import com.njcn.common.pojo.exception.BusinessException;
|
||||
import com.njcn.gather.tools.report.util.ReportExceptionUtil;
|
||||
|
||||
/**
|
||||
* 文档处理器接口
|
||||
* 定义了文档模板处理的核心方法
|
||||
*
|
||||
* @author hongawen
|
||||
*/
|
||||
public interface DocumentProcessor {
|
||||
|
||||
/**
|
||||
* 处理模板文档
|
||||
*
|
||||
* @param request 处理请求
|
||||
* @return 处理结果
|
||||
*/
|
||||
ProcessResult process(TemplateRequest request);
|
||||
|
||||
/**
|
||||
* 检查处理器是否支持指定的模板类型
|
||||
*
|
||||
* @param request 处理请求
|
||||
* @return true表示支持，false表示不支持
|
||||
*/
|
||||
boolean supports(TemplateRequest request);
|
||||
|
||||
/**
|
||||
* 获取处理器名称
|
||||
*
|
||||
* @return 处理器名称
|
||||
*/
|
||||
String getName();
|
||||
|
||||
/**
|
||||
* 获取支持的模板类型
|
||||
*
|
||||
* @return 支持的模板类型数组
|
||||
*/
|
||||
String[] getSupportedTypes();
|
||||
|
||||
/**
|
||||
* 验证请求参数
|
||||
*
|
||||
* @param request 处理请求
|
||||
* @throws BusinessException 参数无效时抛出
|
||||
*/
|
||||
default void validateRequest(TemplateRequest request) {
|
||||
if (request == null) {
|
||||
throw ReportExceptionUtil.validationError("处理请求不能为空");
|
||||
}
|
||||
if (!request.isValid()) {
|
||||
throw ReportExceptionUtil.validationError("处理请求参数无效");
|
||||
}
|
||||
if (!supports(request)) {
|
||||
throw ReportExceptionUtil.unsupportedOperation("不支持的模板类型或处理请求");
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,253 @@
|
||||
package com.njcn.gather.tools.report.engine;
|
||||
|
||||
import com.njcn.gather.tools.report.model.*;
|
||||
import com.njcn.gather.tools.report.util.WordDocumentUtil;
|
||||
import com.njcn.common.pojo.exception.BusinessException;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.apache.poi.xwpf.usermodel.XWPFDocument;
|
||||
import org.springframework.stereotype.Component;
|
||||
|
||||
import java.io.*;
|
||||
import java.nio.file.Files;
|
||||
import java.nio.file.Paths;
|
||||
import java.time.LocalDateTime;
|
||||
import java.util.HashMap;
|
||||
import java.util.Map;
|
||||
|
||||
/**
|
||||
* Word文档处理器实现
|
||||
* 基于Apache POI的Word文档模板处理
|
||||
*
|
||||
* @author hongawen
|
||||
*/
|
||||
@Slf4j
|
||||
@Component
|
||||
public class WordDocumentProcessor implements DocumentProcessor {
|
||||
|
||||
@Override
|
||||
public ProcessResult process(TemplateRequest request) {
|
||||
// 验证请求参数
|
||||
validateRequest(request);
|
||||
|
||||
ProcessResult.ProcessResultBuilder resultBuilder = ProcessResult.builder()
|
||||
.requestId(request.getRequestId())
|
||||
.startTime(LocalDateTime.now());
|
||||
|
||||
try {
|
||||
log.info("开始处理Word模板: {}", request.getTemplateSource().getDescription());
|
||||
|
||||
// 加载模板文档
|
||||
XWPFDocument document = loadTemplate(request.getTemplateSource());
|
||||
|
||||
// 统计模板信息
|
||||
WordDocumentUtil.DocumentStats templateStats = WordDocumentUtil.getDocumentStats(document);
|
||||
log.debug("模板统计信息: {}", templateStats);
|
||||
|
||||
// 处理文档内容
|
||||
ProcessResult.ProcessingStats stats = processDocumentContent(document, request);
|
||||
|
||||
// 保存处理后的文档
|
||||
saveDocument(document, request.getOutputTarget(), resultBuilder);
|
||||
|
||||
// 构建成功结果
|
||||
ProcessResult result = resultBuilder
|
||||
.success(true)
|
||||
.message("Word模板处理成功")
|
||||
.stats(stats)
|
||||
.endTime(LocalDateTime.now())
|
||||
.build();
|
||||
|
||||
result.calculateProcessingTime();
|
||||
log.info("Word模板处理完成，耗时: {}ms", result.getProcessingTimeMs());
|
||||
|
||||
return result;
|
||||
|
||||
} catch (Exception e) {
|
||||
log.error("Word模板处理失败: {}", e.getMessage(), e);
|
||||
return resultBuilder
|
||||
.success(false)
|
||||
.errorCode("WORD_PROCESSING_ERROR")
|
||||
.message("Word模板处理失败")
|
||||
.errorDetails(e.getMessage())
|
||||
.endTime(LocalDateTime.now())
|
||||
.build();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 加载模板文档
|
||||
*/
|
||||
private XWPFDocument loadTemplate(TemplateSource templateSource) throws IOException {
|
||||
if (templateSource.getFilePath() != null) {
|
||||
// 从文件路径加载
|
||||
return new XWPFDocument(new FileInputStream(templateSource.getFilePath()));
|
||||
} else if (templateSource.getInputStream() != null) {
|
||||
// 从输入流加载
|
||||
return new XWPFDocument(templateSource.getInputStream());
|
||||
} else if (templateSource.getContent() != null) {
|
||||
// 从字节数组加载
|
||||
return new XWPFDocument(new ByteArrayInputStream(templateSource.getContent()));
|
||||
} else {
|
||||
throw new IllegalArgumentException("无效的模板源");
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 处理文档内容
|
||||
*/
|
||||
private ProcessResult.ProcessingStats processDocumentContent(XWPFDocument document, TemplateRequest request) {
|
||||
ProcessOptions options = request.getEffectiveOptions();
|
||||
ProcessResult.ProcessingStats.ProcessingStatsBuilder statsBuilder = ProcessResult.ProcessingStats.builder();
|
||||
|
||||
// 1. 占位符替换
|
||||
if (options.isEnablePlaceholder()) {
|
||||
int placeholderCount = processPlaceholders(document, request.getData(), options);
|
||||
statsBuilder.placeholdersReplaced(placeholderCount);
|
||||
log.debug("已替换占位符: {}个", placeholderCount);
|
||||
}
|
||||
|
||||
// 2. 动态表格处理 (如果需要的话)
|
||||
if (options.isEnableDynamicTable()) {
|
||||
int tablesCount = processDynamicTables(document, request.getData());
|
||||
statsBuilder.tablesGenerated(tablesCount);
|
||||
log.debug("已处理动态表格: {}个", tablesCount);
|
||||
}
|
||||
|
||||
// 统计数据项数量
|
||||
statsBuilder.dataItemCount(request.getData().size());
|
||||
|
||||
return statsBuilder.build();
|
||||
}
|
||||
|
||||
/**
|
||||
* 处理占位符替换
|
||||
*/
|
||||
private int processPlaceholders(XWPFDocument document, Map<String, Object> data, ProcessOptions options) {
|
||||
// 将数据转换为字符串映射
|
||||
Map<String, String> placeholders = convertDataToStringMap(data, options);
|
||||
|
||||
if (placeholders.isEmpty()) {
|
||||
return 0;
|
||||
}
|
||||
|
||||
// 执行占位符替换
|
||||
WordDocumentUtil.replacePlaceholders(document, placeholders);
|
||||
|
||||
return placeholders.size();
|
||||
}
|
||||
|
||||
/**
|
||||
* 将数据对象转换为字符串映射
|
||||
*/
|
||||
private Map<String, String> convertDataToStringMap(Map<String, Object> data, ProcessOptions options) {
|
||||
Map<String, String> result = new HashMap<>();
|
||||
ProcessOptions.PlaceholderPattern pattern = options.getPlaceholderPattern();
|
||||
|
||||
for (Map.Entry<String, Object> entry : data.entrySet()) {
|
||||
String key = entry.getKey();
|
||||
Object value = entry.getValue();
|
||||
|
||||
// 构建完整的占位符格式
|
||||
String placeholder = pattern.buildPlaceholder(key);
|
||||
String stringValue = value != null ? value.toString() : "";
|
||||
|
||||
result.put(placeholder, stringValue);
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
/**
|
||||
* 处理动态表格（简单实现，可根据需要扩展）
|
||||
*/
|
||||
private int processDynamicTables(XWPFDocument document, Map<String, Object> data) {
|
||||
// 这里可以实现动态表格的逻辑
|
||||
// 目前返回0，表示没有处理动态表格
|
||||
return 0;
|
||||
}
|
||||
|
||||
/**
|
||||
* 保存处理后的文档
|
||||
*/
|
||||
private void saveDocument(XWPFDocument document, TemplateRequest.OutputTarget outputTarget,
|
||||
ProcessResult.ProcessResultBuilder resultBuilder) throws IOException {
|
||||
|
||||
if (outputTarget.getFilePath() != null) {
|
||||
// 保存到文件
|
||||
saveToFile(document, outputTarget.getFilePath());
|
||||
resultBuilder.outputFilePath(outputTarget.getFilePath());
|
||||
|
||||
// 获取文件大小
|
||||
try {
|
||||
long fileSize = Files.size(Paths.get(outputTarget.getFilePath()));
|
||||
resultBuilder.fileSize(fileSize);
|
||||
} catch (Exception e) {
|
||||
log.warn("无法获取输出文件大小: {}", e.getMessage());
|
||||
}
|
||||
|
||||
} else if (outputTarget.getOutputStream() != null) {
|
||||
// 保存到输出流
|
||||
saveToStream(document, outputTarget.getOutputStream());
|
||||
|
||||
} else if (outputTarget.isReturnBytes()) {
|
||||
// 返回字节数组
|
||||
byte[] bytes = saveToBytes(document);
|
||||
resultBuilder.outputBytes(bytes);
|
||||
resultBuilder.fileSize((long) bytes.length);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 保存文档到文件
|
||||
*/
|
||||
private void saveToFile(XWPFDocument document, String filePath) throws IOException {
|
||||
// 确保目录存在
|
||||
File file = new File(filePath);
|
||||
File parentDir = file.getParentFile();
|
||||
if (parentDir != null && !parentDir.exists()) {
|
||||
parentDir.mkdirs();
|
||||
}
|
||||
|
||||
try (FileOutputStream outputStream = new FileOutputStream(filePath)) {
|
||||
document.write(outputStream);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 保存文档到输出流
|
||||
*/
|
||||
private void saveToStream(XWPFDocument document, OutputStream outputStream) throws IOException {
|
||||
document.write(outputStream);
|
||||
outputStream.flush();
|
||||
}
|
||||
|
||||
/**
|
||||
* 保存文档到字节数组
|
||||
*/
|
||||
private byte[] saveToBytes(XWPFDocument document) throws IOException {
|
||||
try (ByteArrayOutputStream outputStream = new ByteArrayOutputStream()) {
|
||||
document.write(outputStream);
|
||||
return outputStream.toByteArray();
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public boolean supports(TemplateRequest request) {
|
||||
if (request == null || request.getTemplateSource() == null) {
|
||||
return false;
|
||||
}
|
||||
|
||||
TemplateType type = request.getTemplateSource().getType();
|
||||
return type == TemplateType.DOCX;
|
||||
}
|
||||
|
||||
@Override
|
||||
public String getName() {
|
||||
return "WordDocumentProcessor";
|
||||
}
|
||||
|
||||
@Override
|
||||
public String[] getSupportedTypes() {
|
||||
return new String[]{"DOCX"};
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,175 @@
|
||||
package com.njcn.gather.tools.report.model;
|
||||
|
||||
import lombok.Data;
|
||||
import lombok.Builder;
|
||||
import lombok.NoArgsConstructor;
|
||||
import lombok.AllArgsConstructor;
|
||||
|
||||
/**
|
||||
* 模板处理选项配置
|
||||
* 用于控制模板处理过程中启用的功能
|
||||
*
|
||||
* @author hongawen
|
||||
*/
|
||||
@Data
|
||||
@Builder
|
||||
@NoArgsConstructor
|
||||
@AllArgsConstructor
|
||||
public class ProcessOptions {
|
||||
|
||||
/**
|
||||
* 启用占位符替换功能
|
||||
* 默认：true
|
||||
*/
|
||||
@Builder.Default
|
||||
private boolean enablePlaceholder = true;
|
||||
|
||||
/**
|
||||
* 启用书签定位插入功能
|
||||
* 默认：true
|
||||
*/
|
||||
@Builder.Default
|
||||
private boolean enableBookmark = true;
|
||||
|
||||
/**
|
||||
* 启用动态表格生成功能
|
||||
* 默认：true
|
||||
*/
|
||||
@Builder.Default
|
||||
private boolean enableDynamicTable = true;
|
||||
|
||||
/**
|
||||
* 启用自动分页功能
|
||||
* 默认：true
|
||||
*/
|
||||
@Builder.Default
|
||||
private boolean enableAutoPage = true;
|
||||
|
||||
/**
|
||||
* 启用样式管理功能
|
||||
* 默认：true
|
||||
*/
|
||||
@Builder.Default
|
||||
private boolean enableStyleManagement = true;
|
||||
|
||||
/**
|
||||
* 占位符格式模式
|
||||
* 默认：DOLLAR_BRACE (${key})
|
||||
*/
|
||||
@Builder.Default
|
||||
private PlaceholderPattern placeholderPattern = PlaceholderPattern.DOLLAR_BRACE;
|
||||
|
||||
/**
|
||||
* 输出格式
|
||||
* 默认：DOCX
|
||||
*/
|
||||
@Builder.Default
|
||||
private String outputFormat = "DOCX";
|
||||
|
||||
/**
|
||||
* 是否压缩输出
|
||||
* 默认：false
|
||||
*/
|
||||
@Builder.Default
|
||||
private boolean compressOutput = false;
|
||||
|
||||
/**
|
||||
* 处理超时时间（毫秒）
|
||||
* 默认：30秒
|
||||
*/
|
||||
@Builder.Default
|
||||
private long timeoutMs = 30000L;
|
||||
|
||||
/**
|
||||
* 是否启用调试模式
|
||||
* 默认：false
|
||||
*/
|
||||
@Builder.Default
|
||||
private boolean debugMode = false;
|
||||
|
||||
/**
|
||||
* 自定义配置项
|
||||
*/
|
||||
private java.util.Map<String, Object> customOptions;
|
||||
|
||||
/**
|
||||
* 占位符模式枚举
|
||||
*/
|
||||
public enum PlaceholderPattern {
|
||||
/**
|
||||
* ${key} 格式
|
||||
*/
|
||||
DOLLAR_BRACE("\\$\\{([^}]+)\\}", "${", "}"),
|
||||
|
||||
/**
|
||||
* #{key} 格式
|
||||
*/
|
||||
HASH_BRACE("\\#\\{([^}]+)\\}", "#{", "}"),
|
||||
|
||||
/**
|
||||
* {{key}} 格式
|
||||
*/
|
||||
DOUBLE_BRACE("\\{\\{([^}]+)\\}\\}", "{{", "}}");
|
||||
|
||||
private final String regex;
|
||||
private final String prefix;
|
||||
private final String suffix;
|
||||
|
||||
PlaceholderPattern(String regex, String prefix, String suffix) {
|
||||
this.regex = regex;
|
||||
this.prefix = prefix;
|
||||
this.suffix = suffix;
|
||||
}
|
||||
|
||||
public String getRegex() {
|
||||
return regex;
|
||||
}
|
||||
|
||||
public String getPrefix() {
|
||||
return prefix;
|
||||
}
|
||||
|
||||
public String getSuffix() {
|
||||
return suffix;
|
||||
}
|
||||
|
||||
/**
|
||||
* 构造完整的占位符
|
||||
*/
|
||||
public String buildPlaceholder(String key) {
|
||||
return prefix + key + suffix;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建默认配置
|
||||
*/
|
||||
public static ProcessOptions defaultOptions() {
|
||||
return ProcessOptions.builder().build();
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建简单占位符替换配置
|
||||
*/
|
||||
public static ProcessOptions simplePlaceholder() {
|
||||
return ProcessOptions.builder()
|
||||
.enablePlaceholder(true)
|
||||
.enableBookmark(false)
|
||||
.enableDynamicTable(false)
|
||||
.enableAutoPage(false)
|
||||
.build();
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建高级功能配置
|
||||
*/
|
||||
public static ProcessOptions advancedFeatures() {
|
||||
return ProcessOptions.builder()
|
||||
.enablePlaceholder(true)
|
||||
.enableBookmark(true)
|
||||
.enableDynamicTable(true)
|
||||
.enableAutoPage(true)
|
||||
.enableStyleManagement(true)
|
||||
.build();
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,232 @@
|
||||
package com.njcn.gather.tools.report.model;
|
||||
|
||||
import lombok.Data;
|
||||
import lombok.Builder;
|
||||
import lombok.NoArgsConstructor;
|
||||
import lombok.AllArgsConstructor;
|
||||
|
||||
import java.time.LocalDateTime;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
|
||||
/**
|
||||
* 模板处理结果对象
|
||||
* 包含处理状态、结果数据、错误信息等
|
||||
*
|
||||
* @author hongawen
|
||||
*/
|
||||
@Data
|
||||
@Builder
|
||||
@NoArgsConstructor
|
||||
@AllArgsConstructor
|
||||
public class ProcessResult {
|
||||
|
||||
/**
|
||||
* 处理是否成功
|
||||
*/
|
||||
private boolean success;
|
||||
|
||||
/**
|
||||
* 结果消息
|
||||
*/
|
||||
private String message;
|
||||
|
||||
/**
|
||||
* 错误代码（失败时）
|
||||
*/
|
||||
private String errorCode;
|
||||
|
||||
/**
|
||||
* 详细错误信息（失败时）
|
||||
*/
|
||||
private String errorDetails;
|
||||
|
||||
/**
|
||||
* 生成的文件路径（文件输出时）
|
||||
*/
|
||||
private String outputFilePath;
|
||||
|
||||
/**
|
||||
* 生成的文件字节数组（字节输出时）
|
||||
*/
|
||||
private byte[] outputBytes;
|
||||
|
||||
/**
|
||||
* 输出文件大小（字节）
|
||||
*/
|
||||
private Long fileSize;
|
||||
|
||||
/**
|
||||
* 处理开始时间
|
||||
*/
|
||||
private LocalDateTime startTime;
|
||||
|
||||
/**
|
||||
* 处理结束时间
|
||||
*/
|
||||
private LocalDateTime endTime;
|
||||
|
||||
/**
|
||||
* 处理耗时（毫秒）
|
||||
*/
|
||||
private Long processingTimeMs;
|
||||
|
||||
/**
|
||||
* 请求标识
|
||||
*/
|
||||
private String requestId;
|
||||
|
||||
/**
|
||||
* 处理统计信息
|
||||
*/
|
||||
private ProcessingStats stats;
|
||||
|
||||
/**
|
||||
* 警告信息列表
|
||||
*/
|
||||
private List<String> warnings;
|
||||
|
||||
/**
|
||||
* 额外的元数据
|
||||
*/
|
||||
private Map<String, Object> metadata;
|
||||
|
||||
/**
|
||||
* 处理统计信息
|
||||
*/
|
||||
@Data
|
||||
@Builder
|
||||
@NoArgsConstructor
|
||||
@AllArgsConstructor
|
||||
public static class ProcessingStats {
|
||||
/**
|
||||
* 替换的占位符数量
|
||||
*/
|
||||
private int placeholdersReplaced;
|
||||
|
||||
/**
|
||||
* 处理的书签数量
|
||||
*/
|
||||
private int bookmarksProcessed;
|
||||
|
||||
/**
|
||||
* 生成的表格数量
|
||||
*/
|
||||
private int tablesGenerated;
|
||||
|
||||
/**
|
||||
* 处理的页面数量
|
||||
*/
|
||||
private int pagesProcessed;
|
||||
|
||||
/**
|
||||
* 模板文件大小（字节）
|
||||
*/
|
||||
private long templateFileSize;
|
||||
|
||||
/**
|
||||
* 输出文件大小（字节）
|
||||
*/
|
||||
private long outputFileSize;
|
||||
|
||||
/**
|
||||
* 数据项数量
|
||||
*/
|
||||
private int dataItemCount;
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建成功结果
|
||||
*/
|
||||
public static ProcessResult success(String message) {
|
||||
return ProcessResult.builder()
|
||||
.success(true)
|
||||
.message(message)
|
||||
.endTime(LocalDateTime.now())
|
||||
.build();
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建失败结果
|
||||
*/
|
||||
public static ProcessResult failure(String errorCode, String message) {
|
||||
return ProcessResult.builder()
|
||||
.success(false)
|
||||
.errorCode(errorCode)
|
||||
.message(message)
|
||||
.errorDetails(message)
|
||||
.endTime(LocalDateTime.now())
|
||||
.build();
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建失败结果（带详细错误信息）
|
||||
*/
|
||||
public static ProcessResult failure(String errorCode, String message, String errorDetails) {
|
||||
return ProcessResult.builder()
|
||||
.success(false)
|
||||
.errorCode(errorCode)
|
||||
.message(message)
|
||||
.errorDetails(errorDetails)
|
||||
.endTime(LocalDateTime.now())
|
||||
.build();
|
||||
}
|
||||
|
||||
/**
|
||||
* 计算处理耗时
|
||||
*/
|
||||
public void calculateProcessingTime() {
|
||||
if (startTime != null && endTime != null) {
|
||||
processingTimeMs = java.time.Duration.between(startTime, endTime).toMillis();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 添加警告信息
|
||||
*/
|
||||
public void addWarning(String warning) {
|
||||
if (warnings == null) {
|
||||
warnings = new java.util.ArrayList<>();
|
||||
}
|
||||
warnings.add(warning);
|
||||
}
|
||||
|
||||
/**
|
||||
* 设置元数据
|
||||
*/
|
||||
public void setMetadata(String key, Object value) {
|
||||
if (metadata == null) {
|
||||
metadata = new java.util.HashMap<>();
|
||||
}
|
||||
metadata.put(key, value);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取格式化的处理信息
|
||||
*/
|
||||
public String getFormattedSummary() {
|
||||
StringBuilder sb = new StringBuilder();
|
||||
sb.append("处理结果: ").append(success ? "成功" : "失败").append("\n");
|
||||
sb.append("消息: ").append(message).append("\n");
|
||||
|
||||
if (processingTimeMs != null) {
|
||||
sb.append("耗时: ").append(processingTimeMs).append("ms\n");
|
||||
}
|
||||
|
||||
if (stats != null) {
|
||||
sb.append("统计: 占位符").append(stats.placeholdersReplaced)
|
||||
.append("个, 书签").append(stats.bookmarksProcessed)
|
||||
.append("个, 表格").append(stats.tablesGenerated).append("个\n");
|
||||
}
|
||||
|
||||
if (fileSize != null) {
|
||||
sb.append("文件大小: ").append(fileSize).append("字节\n");
|
||||
}
|
||||
|
||||
if (warnings != null && !warnings.isEmpty()) {
|
||||
sb.append("警告: ").append(String.join(", ", warnings)).append("\n");
|
||||
}
|
||||
|
||||
return sb.toString();
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,58 @@
|
||||
package com.njcn.gather.tools.report.model;
|
||||
|
||||
import lombok.Getter;
|
||||
|
||||
/**
|
||||
* 报告生成模块响应枚举
|
||||
* 用于定义报告生成过程中的错误代码和消息
|
||||
*
|
||||
* @author hongawen
|
||||
*/
|
||||
@Getter
|
||||
public enum ReportResponseEnum {
|
||||
|
||||
/**
|
||||
* 报告生成模块异常响应码范围：
|
||||
* R01000 ~ R01099
|
||||
*/
|
||||
TEMPLATE_LOAD_ERROR("R01000", "模板文件加载失败"),
|
||||
TEMPLATE_PROCESS_ERROR("R01001", "模板处理失败"),
|
||||
FILE_SAVE_ERROR("R01002", "文件保存失败"),
|
||||
VALIDATION_ERROR("R01003", "参数验证失败"),
|
||||
UNSUPPORTED_OPERATION("R01004", "不支持的操作"),
|
||||
REPORT_GENERATION_ERROR("R01005", "报告生成失败"),
|
||||
ASYNC_PROCESSING_ERROR("R01006", "异步处理失败"),
|
||||
BATCH_PROCESSING_ERROR("R01007", "批量处理失败"),
|
||||
|
||||
// 模板相关错误
|
||||
TEMPLATE_NOT_FOUND("R01010", "模板文件不存在"),
|
||||
TEMPLATE_FORMAT_ERROR("R01011", "模板文件格式不正确"),
|
||||
TEMPLATE_CORRUPTED("R01012", "模板文件已损坏"),
|
||||
|
||||
// 数据相关错误
|
||||
DATA_EMPTY("R01020", "填充数据为空"),
|
||||
DATA_FORMAT_ERROR("R01021", "数据格式不正确"),
|
||||
PLACEHOLDER_NOT_FOUND("R01022", "占位符未找到对应数据"),
|
||||
|
||||
// 输出相关错误
|
||||
OUTPUT_PATH_INVALID("R01030", "输出路径无效"),
|
||||
OUTPUT_PERMISSION_DENIED("R01031", "输出路径权限不足"),
|
||||
DISK_SPACE_INSUFFICIENT("R01032", "磁盘空间不足"),
|
||||
|
||||
// 处理器相关错误
|
||||
NO_SUITABLE_PROCESSOR("R01040", "找不到合适的文档处理器"),
|
||||
PROCESSOR_INIT_ERROR("R01041", "文档处理器初始化失败"),
|
||||
|
||||
// 系统相关错误
|
||||
MEMORY_INSUFFICIENT("R01050", "内存不足"),
|
||||
PROCESSING_TIMEOUT("R01051", "处理超时"),
|
||||
SYSTEM_BUSY("R01052", "系统繁忙，请稍后再试");
|
||||
|
||||
private final String code;
|
||||
private final String message;
|
||||
|
||||
ReportResponseEnum(String code, String message) {
|
||||
this.code = code;
|
||||
this.message = message;
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,184 @@
|
||||
package com.njcn.gather.tools.report.model;
|
||||
|
||||
import lombok.Data;
|
||||
import lombok.Builder;
|
||||
import lombok.NoArgsConstructor;
|
||||
import lombok.AllArgsConstructor;
|
||||
|
||||
import javax.validation.constraints.NotNull;
|
||||
import java.io.OutputStream;
|
||||
import java.util.Map;
|
||||
|
||||
/**
|
||||
* 模板处理请求对象
|
||||
* 封装模板处理所需的所有参数
|
||||
*
|
||||
* @author hongawen
|
||||
*/
|
||||
@Data
|
||||
@Builder
|
||||
@NoArgsConstructor
|
||||
@AllArgsConstructor
|
||||
public class TemplateRequest {
|
||||
|
||||
/**
|
||||
* 模板源（必填）
|
||||
*/
|
||||
@NotNull(message = "模板源不能为空")
|
||||
private TemplateSource templateSource;
|
||||
|
||||
/**
|
||||
* 填充数据（必填）
|
||||
* key-value格式的数据，支持嵌套对象
|
||||
*/
|
||||
@NotNull(message = "填充数据不能为空")
|
||||
private Map<String, Object> data;
|
||||
|
||||
/**
|
||||
* 处理选项（可选）
|
||||
* 如果为空则使用默认配置
|
||||
*/
|
||||
private ProcessOptions options;
|
||||
|
||||
/**
|
||||
* 输出目标配置（必填）
|
||||
*/
|
||||
@NotNull(message = "输出目标不能为空")
|
||||
private OutputTarget outputTarget;
|
||||
|
||||
/**
|
||||
* 请求标识（可选）
|
||||
* 用于追踪和日志记录
|
||||
*/
|
||||
private String requestId;
|
||||
|
||||
/**
|
||||
* 用户标识（可选）
|
||||
* 用于权限控制和审计
|
||||
*/
|
||||
private String userId;
|
||||
|
||||
/**
|
||||
* 额外的上下文数据（可选）
|
||||
*/
|
||||
private Map<String, Object> contextData;
|
||||
|
||||
/**
|
||||
* 输出目标配置
|
||||
*/
|
||||
@Data
|
||||
@Builder
|
||||
@NoArgsConstructor
|
||||
@AllArgsConstructor
|
||||
public static class OutputTarget {
|
||||
/**
|
||||
* 输出文件路径
|
||||
*/
|
||||
private String filePath;
|
||||
|
||||
/**
|
||||
* 输出流
|
||||
*/
|
||||
private OutputStream outputStream;
|
||||
|
||||
/**
|
||||
* 输出文件名（当使用流输出时）
|
||||
*/
|
||||
private String fileName;
|
||||
|
||||
/**
|
||||
* 是否返回字节数组
|
||||
*/
|
||||
@Builder.Default
|
||||
private boolean returnBytes = false;
|
||||
|
||||
/**
|
||||
* 创建文件输出目标
|
||||
*/
|
||||
public static OutputTarget toFile(String filePath) {
|
||||
return OutputTarget.builder()
|
||||
.filePath(filePath)
|
||||
.build();
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建流输出目标
|
||||
*/
|
||||
public static OutputTarget toStream(OutputStream outputStream, String fileName) {
|
||||
return OutputTarget.builder()
|
||||
.outputStream(outputStream)
|
||||
.fileName(fileName)
|
||||
.build();
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建字节数组输出目标
|
||||
*/
|
||||
public static OutputTarget toBytes() {
|
||||
return OutputTarget.builder()
|
||||
.returnBytes(true)
|
||||
.build();
|
||||
}
|
||||
|
||||
/**
|
||||
* 验证输出目标是否有效
|
||||
*/
|
||||
public boolean isValid() {
|
||||
return filePath != null || outputStream != null || returnBytes;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 构建器便捷方法
|
||||
*/
|
||||
public static class TemplateRequestBuilder {
|
||||
|
||||
/**
|
||||
* 设置模板文件路径
|
||||
*/
|
||||
public TemplateRequestBuilder templatePath(String filePath) {
|
||||
this.templateSource = TemplateSource.fromFile(filePath);
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* 设置输出文件路径
|
||||
*/
|
||||
public TemplateRequestBuilder outputPath(String filePath) {
|
||||
this.outputTarget = OutputTarget.toFile(filePath);
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* 设置输出流
|
||||
*/
|
||||
public TemplateRequestBuilder outputStream(OutputStream outputStream, String fileName) {
|
||||
this.outputTarget = OutputTarget.toStream(outputStream, fileName);
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* 设置返回字节数组
|
||||
*/
|
||||
public TemplateRequestBuilder outputBytes() {
|
||||
this.outputTarget = OutputTarget.toBytes();
|
||||
return this;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 验证请求是否有效
|
||||
*/
|
||||
public boolean isValid() {
|
||||
return templateSource != null && templateSource.isValid() &&
|
||||
data != null && !data.isEmpty() &&
|
||||
outputTarget != null && outputTarget.isValid();
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取处理选项，如果为空则返回默认选项
|
||||
*/
|
||||
public ProcessOptions getEffectiveOptions() {
|
||||
return options != null ? options : ProcessOptions.defaultOptions();
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,114 @@
|
||||
package com.njcn.gather.tools.report.model;
|
||||
|
||||
import lombok.Data;
|
||||
import lombok.Builder;
|
||||
import lombok.NoArgsConstructor;
|
||||
import lombok.AllArgsConstructor;
|
||||
|
||||
import java.io.InputStream;
|
||||
|
||||
/**
|
||||
* 模板来源定义
|
||||
* 支持多种模板输入方式：文件路径、输入流、字节数组
|
||||
*
|
||||
* @author hongawen
|
||||
*/
|
||||
@Data
|
||||
@Builder
|
||||
@NoArgsConstructor
|
||||
@AllArgsConstructor
|
||||
public class TemplateSource {
|
||||
|
||||
/**
|
||||
* 模板文件路径
|
||||
*/
|
||||
private String filePath;
|
||||
|
||||
/**
|
||||
* 模板输入流
|
||||
*/
|
||||
private InputStream inputStream;
|
||||
|
||||
/**
|
||||
* 模板字节数组内容
|
||||
*/
|
||||
private byte[] content;
|
||||
|
||||
/**
|
||||
* 模板类型
|
||||
*/
|
||||
private TemplateType type;
|
||||
|
||||
/**
|
||||
* 模板名称（用于标识和缓存）
|
||||
*/
|
||||
private String name;
|
||||
|
||||
/**
|
||||
* 是否启用缓存
|
||||
*/
|
||||
private boolean enableCache = true;
|
||||
|
||||
/**
|
||||
* 从文件路径创建模板源
|
||||
*/
|
||||
public static TemplateSource fromFile(String filePath) {
|
||||
return TemplateSource.builder()
|
||||
.filePath(filePath)
|
||||
.type(TemplateType.DOCX)
|
||||
.name(extractFileName(filePath))
|
||||
.build();
|
||||
}
|
||||
|
||||
/**
|
||||
* 从输入流创建模板源
|
||||
*/
|
||||
public static TemplateSource fromStream(InputStream inputStream, String name) {
|
||||
return TemplateSource.builder()
|
||||
.inputStream(inputStream)
|
||||
.type(TemplateType.DOCX)
|
||||
.name(name)
|
||||
.build();
|
||||
}
|
||||
|
||||
/**
|
||||
* 从字节数组创建模板源
|
||||
*/
|
||||
public static TemplateSource fromBytes(byte[] content, String name) {
|
||||
return TemplateSource.builder()
|
||||
.content(content)
|
||||
.type(TemplateType.DOCX)
|
||||
.name(name)
|
||||
.build();
|
||||
}
|
||||
|
||||
/**
|
||||
* 从文件路径提取文件名
|
||||
*/
|
||||
private static String extractFileName(String filePath) {
|
||||
if (filePath == null) return null;
|
||||
int lastSeparator = Math.max(filePath.lastIndexOf('/'), filePath.lastIndexOf('\\'));
|
||||
return lastSeparator >= 0 ? filePath.substring(lastSeparator + 1) : filePath;
|
||||
}
|
||||
|
||||
/**
|
||||
* 验证模板源是否有效
|
||||
*/
|
||||
public boolean isValid() {
|
||||
return filePath != null || inputStream != null ||
|
||||
(content != null && content.length > 0);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取模板源描述（用于日志）
|
||||
*/
|
||||
public String getDescription() {
|
||||
if (filePath != null) {
|
||||
return "file://" + filePath;
|
||||
} else if (name != null) {
|
||||
return "stream://" + name;
|
||||
} else {
|
||||
return "bytes://unknown";
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,63 @@
|
||||
package com.njcn.gather.tools.report.model;
|
||||
|
||||
/**
|
||||
* 模板文档类型枚举
|
||||
*
|
||||
* @author hongawen
|
||||
*/
|
||||
public enum TemplateType {
|
||||
|
||||
/**
|
||||
* Word文档 (.docx)
|
||||
*/
|
||||
DOCX("docx", "application/vnd.openxmlformats-officedocument.wordprocessingml.document"),
|
||||
|
||||
/**
|
||||
* PDF文档 (.pdf) - 未来扩展
|
||||
*/
|
||||
PDF("pdf", "application/pdf"),
|
||||
|
||||
/**
|
||||
* Excel文档 (.xlsx) - 未来扩展
|
||||
*/
|
||||
XLSX("xlsx", "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
|
||||
|
||||
private final String extension;
|
||||
private final String mimeType;
|
||||
|
||||
TemplateType(String extension, String mimeType) {
|
||||
this.extension = extension;
|
||||
this.mimeType = mimeType;
|
||||
}
|
||||
|
||||
public String getExtension() {
|
||||
return extension;
|
||||
}
|
||||
|
||||
public String getMimeType() {
|
||||
return mimeType;
|
||||
}
|
||||
|
||||
/**
|
||||
* 根据文件扩展名获取模板类型
|
||||
*/
|
||||
public static TemplateType fromExtension(String fileName) {
|
||||
if (fileName == null) return DOCX;
|
||||
|
||||
String extension = getFileExtension(fileName);
|
||||
for (TemplateType type : values()) {
|
||||
if (type.extension.equalsIgnoreCase(extension)) {
|
||||
return type;
|
||||
}
|
||||
}
|
||||
return DOCX; // 默认返回DOCX
|
||||
}
|
||||
|
||||
/**
|
||||
* 从文件名提取扩展名
|
||||
*/
|
||||
private static String getFileExtension(String fileName) {
|
||||
int lastDot = fileName.lastIndexOf('.');
|
||||
return lastDot > 0 ? fileName.substring(lastDot + 1) : "";
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,52 @@
|
||||
package com.njcn.gather.tools.report.service;
|
||||
|
||||
import com.njcn.gather.tools.report.model.TemplateRequest;
|
||||
import com.njcn.gather.tools.report.model.ProcessResult;
|
||||
|
||||
/**
|
||||
* 报告生成服务接口
|
||||
*
|
||||
* @author hongawen
|
||||
*/
|
||||
public interface ReportGeneratorService {
|
||||
|
||||
/**
|
||||
* 处理模板并生成报告
|
||||
*
|
||||
* @param request 模板处理请求
|
||||
* @return 处理结果
|
||||
*/
|
||||
ProcessResult processTemplate(TemplateRequest request);
|
||||
|
||||
/**
|
||||
* 异步处理模板并生成报告
|
||||
*
|
||||
* @param request 模板处理请求
|
||||
* @return 请求ID，用于后续查询结果
|
||||
*/
|
||||
String processTemplateAsync(TemplateRequest request);
|
||||
|
||||
/**
|
||||
* 查询异步处理结果
|
||||
*
|
||||
* @param requestId 请求ID
|
||||
* @return 处理结果，如果还在处理中则返回null
|
||||
*/
|
||||
ProcessResult getAsyncResult(String requestId);
|
||||
|
||||
/**
|
||||
* 批量处理模板
|
||||
*
|
||||
* @param requests 模板处理请求列表
|
||||
* @return 处理结果列表
|
||||
*/
|
||||
java.util.List<ProcessResult> batchProcessTemplates(java.util.List<TemplateRequest> requests);
|
||||
|
||||
/**
|
||||
* 验证模板是否有效
|
||||
*
|
||||
* @param request 模板处理请求
|
||||
* @return 验证结果
|
||||
*/
|
||||
ProcessResult validateTemplate(TemplateRequest request);
|
||||
}
|
||||
@@ -0,0 +1,216 @@
|
||||
package com.njcn.gather.tools.report.service.impl;
|
||||
|
||||
import cn.hutool.core.util.IdUtil;
|
||||
import com.njcn.gather.tools.report.engine.DocumentProcessor;
|
||||
import com.njcn.gather.tools.report.model.TemplateRequest;
|
||||
import com.njcn.gather.tools.report.model.ProcessResult;
|
||||
import com.njcn.gather.tools.report.service.ReportGeneratorService;
|
||||
import com.njcn.common.pojo.exception.BusinessException;
|
||||
import lombok.RequiredArgsConstructor;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.stereotype.Service;
|
||||
|
||||
import java.time.LocalDateTime;
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.concurrent.CompletableFuture;
|
||||
import java.util.concurrent.ConcurrentHashMap;
|
||||
import java.util.concurrent.Executor;
|
||||
import java.util.concurrent.Executors;
|
||||
|
||||
/**
|
||||
* 报告生成服务实现
|
||||
*
|
||||
* @author hongawen
|
||||
*/
|
||||
@Slf4j
|
||||
@Service
|
||||
@RequiredArgsConstructor
|
||||
public class ReportGeneratorServiceImpl implements ReportGeneratorService {
|
||||
|
||||
private final List<DocumentProcessor> documentProcessors;
|
||||
|
||||
// 异步处理结果缓存
|
||||
private final Map<String, ProcessResult> asyncResults = new ConcurrentHashMap<>();
|
||||
|
||||
// 异步处理线程池
|
||||
private final Executor asyncExecutor = Executors.newFixedThreadPool(4);
|
||||
|
||||
@Override
|
||||
public ProcessResult processTemplate(TemplateRequest request) {
|
||||
log.info("开始处理模板请求: {}", request.getRequestId());
|
||||
|
||||
try {
|
||||
// 设置请求ID（如果没有设置的话）
|
||||
if (request.getRequestId() == null) {
|
||||
request.setRequestId(IdUtil.simpleUUID());
|
||||
}
|
||||
|
||||
// 查找合适的处理器
|
||||
DocumentProcessor processor = findProcessor(request);
|
||||
if (processor == null) {
|
||||
return ProcessResult.failure("NO_SUITABLE_PROCESSOR", "找不到合适的文档处理器");
|
||||
}
|
||||
|
||||
// 执行处理
|
||||
ProcessResult result = processor.process(request);
|
||||
log.info("模板处理完成: {}, 状态: {}", request.getRequestId(), result.isSuccess());
|
||||
|
||||
return result;
|
||||
|
||||
} catch (Exception e) {
|
||||
log.error("模板处理异常: {}", e.getMessage(), e);
|
||||
return ProcessResult.failure("PROCESSING_EXCEPTION", "模板处理异常", e.getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public String processTemplateAsync(TemplateRequest request) {
|
||||
String requestId = request.getRequestId();
|
||||
if (requestId == null) {
|
||||
requestId = IdUtil.simpleUUID();
|
||||
request.setRequestId(requestId);
|
||||
}
|
||||
|
||||
log.info("启动异步模板处理: {}", requestId);
|
||||
|
||||
// 创建处理中的结果
|
||||
ProcessResult processingResult = ProcessResult.builder()
|
||||
.requestId(requestId)
|
||||
.success(false)
|
||||
.message("处理中...")
|
||||
.startTime(LocalDateTime.now())
|
||||
.build();
|
||||
|
||||
asyncResults.put(requestId, processingResult);
|
||||
|
||||
// 异步执行处理
|
||||
final String finalRequestId = requestId;
|
||||
final TemplateRequest finalRequest = request;
|
||||
CompletableFuture.runAsync(new Runnable() {
|
||||
@Override
|
||||
public void run() {
|
||||
try {
|
||||
ProcessResult result = processTemplate(finalRequest);
|
||||
asyncResults.put(finalRequestId, result);
|
||||
} catch (Exception e) {
|
||||
ProcessResult errorResult = ProcessResult.failure("ASYNC_PROCESSING_ERROR",
|
||||
"异步处理异常", e.getMessage());
|
||||
errorResult.setRequestId(finalRequestId);
|
||||
asyncResults.put(finalRequestId, errorResult);
|
||||
}
|
||||
}
|
||||
}, asyncExecutor);
|
||||
|
||||
return requestId;
|
||||
}
|
||||
|
||||
@Override
|
||||
public ProcessResult getAsyncResult(String requestId) {
|
||||
return asyncResults.get(requestId);
|
||||
}
|
||||
|
||||
@Override
|
||||
public List<ProcessResult> batchProcessTemplates(List<TemplateRequest> requests) {
|
||||
log.info("开始批量处理模板，数量: {}", requests.size());
|
||||
|
||||
List<ProcessResult> results = new ArrayList<>();
|
||||
|
||||
for (TemplateRequest request : requests) {
|
||||
try {
|
||||
ProcessResult result = processTemplate(request);
|
||||
results.add(result);
|
||||
} catch (Exception e) {
|
||||
log.error("批量处理中的单个模板处理失败: {}", e.getMessage(), e);
|
||||
ProcessResult errorResult = ProcessResult.failure("BATCH_ITEM_ERROR",
|
||||
"批量处理中的模板处理失败", e.getMessage());
|
||||
if (request.getRequestId() != null) {
|
||||
errorResult.setRequestId(request.getRequestId());
|
||||
}
|
||||
results.add(errorResult);
|
||||
}
|
||||
}
|
||||
|
||||
long successCount = 0;
|
||||
long failureCount = 0;
|
||||
for (ProcessResult result : results) {
|
||||
if (result.isSuccess()) {
|
||||
successCount++;
|
||||
} else {
|
||||
failureCount++;
|
||||
}
|
||||
}
|
||||
log.info("批量处理完成，成功: {}, 失败: {}", successCount, failureCount);
|
||||
|
||||
return results;
|
||||
}
|
||||
|
||||
@Override
|
||||
public ProcessResult validateTemplate(TemplateRequest request) {
|
||||
log.debug("验证模板请求: {}", request.getRequestId());
|
||||
|
||||
try {
|
||||
// 基本参数验证
|
||||
if (request == null) {
|
||||
return ProcessResult.failure("VALIDATION_ERROR", "请求对象不能为空");
|
||||
}
|
||||
|
||||
if (!request.isValid()) {
|
||||
return ProcessResult.failure("VALIDATION_ERROR", "请求参数无效");
|
||||
}
|
||||
|
||||
// 查找处理器
|
||||
DocumentProcessor processor = findProcessor(request);
|
||||
if (processor == null) {
|
||||
return ProcessResult.failure("VALIDATION_ERROR", "找不到合适的文档处理器");
|
||||
}
|
||||
|
||||
// 处理器特定验证
|
||||
try {
|
||||
processor.validateRequest(request);
|
||||
return ProcessResult.success("模板验证通过");
|
||||
} catch (BusinessException e) {
|
||||
return ProcessResult.failure("VALIDATION_ERROR", "模板验证失败: " + e.getMessage());
|
||||
}
|
||||
|
||||
} catch (Exception e) {
|
||||
log.error("模板验证异常: {}", e.getMessage(), e);
|
||||
return ProcessResult.failure("VALIDATION_EXCEPTION", "模板验证异常", e.getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 查找合适的文档处理器
|
||||
*/
|
||||
private DocumentProcessor findProcessor(TemplateRequest request) {
|
||||
for (DocumentProcessor processor : documentProcessors) {
|
||||
if (processor.supports(request)) {
|
||||
log.debug("找到合适的处理器: {}", processor.getName());
|
||||
return processor;
|
||||
}
|
||||
}
|
||||
|
||||
log.warn("未找到合适的处理器，模板类型: {}",
|
||||
request.getTemplateSource() != null ? request.getTemplateSource().getType() : "unknown");
|
||||
return null;
|
||||
}
|
||||
|
||||
/**
|
||||
* 清理过期的异步结果
|
||||
* 可以通过定时任务调用
|
||||
*/
|
||||
public void cleanupExpiredAsyncResults() {
|
||||
LocalDateTime expireTime = LocalDateTime.now().minusHours(24);
|
||||
List<String> expiredKeys = new ArrayList<>();
|
||||
for (Map.Entry<String, ProcessResult> entry : asyncResults.entrySet()) {
|
||||
ProcessResult result = entry.getValue();
|
||||
if (result.getEndTime() != null && result.getEndTime().isBefore(expireTime)) {
|
||||
expiredKeys.add(entry.getKey());
|
||||
}
|
||||
}
|
||||
for (String key : expiredKeys) {
|
||||
asyncResults.remove(key);
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,603 @@
|
||||
package com.njcn.gather.tools.report.util;
|
||||
|
||||
import cn.hutool.core.collection.CollUtil;
|
||||
import cn.hutool.core.text.StrPool;
|
||||
import cn.hutool.core.util.StrUtil;
|
||||
import org.docx4j.XmlUtils;
|
||||
import org.docx4j.openpackaging.packages.WordprocessingMLPackage;
|
||||
import org.docx4j.openpackaging.parts.WordprocessingML.MainDocumentPart;
|
||||
import org.docx4j.wml.*;
|
||||
|
||||
import javax.xml.bind.JAXBElement;
|
||||
import java.math.BigInteger;
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
|
||||
/**
|
||||
* 基于docx4j的高级Word文档工具类
|
||||
* 从原有Docx4jUtil中抽取的通用功能，去除业务相关逻辑
|
||||
*
|
||||
* @author hongawen
|
||||
*/
|
||||
public class Docx4jAdvancedUtil {
|
||||
|
||||
/**
|
||||
* 创建标题段落
|
||||
*
|
||||
* @param factory 对象工厂
|
||||
* @param paragraph 段落容器
|
||||
* @param content 标题内容
|
||||
* @param fontSize 字体大小
|
||||
* @param isBold 是否加粗
|
||||
*/
|
||||
public static void createTitle(ObjectFactory factory, P paragraph, String content, int fontSize, boolean isBold) {
|
||||
R run = factory.createR();
|
||||
Text text = factory.createText();
|
||||
text.setValue(content);
|
||||
|
||||
// 创建运行属性
|
||||
RPr rPr = factory.createRPr();
|
||||
|
||||
// 设置字体
|
||||
RFonts fonts = factory.createRFonts();
|
||||
fonts.setAscii("Arial");
|
||||
fonts.setEastAsia("SimSun"); // 宋体
|
||||
rPr.setRFonts(fonts);
|
||||
|
||||
// 设置字号
|
||||
HpsMeasure size = new HpsMeasure();
|
||||
size.setVal(new BigInteger(String.valueOf(fontSize))); // 12号字=24
|
||||
rPr.setSz(size);
|
||||
rPr.setSzCs(size);
|
||||
|
||||
// 设置粗体
|
||||
if (isBold) {
|
||||
BooleanDefaultTrue b = new BooleanDefaultTrue();
|
||||
rPr.setB(b);
|
||||
}
|
||||
|
||||
run.setRPr(rPr);
|
||||
run.getContent().add(text);
|
||||
paragraph.getContent().add(run);
|
||||
}
|
||||
|
||||
/**
|
||||
* 提取文档中指定标题级别的内容
|
||||
*
|
||||
* @param allContent 文档中所有内容
|
||||
* @param headingLevel 标题级别 (如 "5" 表示 Heading 5)
|
||||
* @return 标题内容列表
|
||||
*/
|
||||
public static List<HeadingContent> extractHeadingContents(List<Object> allContent, String headingLevel) {
|
||||
List<HeadingContent> result = new ArrayList<>();
|
||||
boolean inHeadingSection = false;
|
||||
HeadingContent currentHeading = null;
|
||||
|
||||
for (Object obj : allContent) {
|
||||
if (obj instanceof P) {
|
||||
P paragraph = (P) obj;
|
||||
if (isHeadingLevel(paragraph, headingLevel)) {
|
||||
// 发现新的指定级别标题，保存前一个并创建新的
|
||||
if (currentHeading != null) {
|
||||
result.add(currentHeading);
|
||||
}
|
||||
currentHeading = new HeadingContent();
|
||||
currentHeading.setHeadingText(getTextFromP(paragraph));
|
||||
inHeadingSection = true;
|
||||
} else if (inHeadingSection) {
|
||||
// 当前内容属于标题的子内容
|
||||
currentHeading.addSubContent(paragraph);
|
||||
}
|
||||
} else if (obj instanceof JAXBElement && inHeadingSection) {
|
||||
// 表格属于当前标题的子内容
|
||||
JAXBElement<?> jaxbElement = (JAXBElement<?>) obj;
|
||||
if (jaxbElement.getValue() instanceof Tbl) {
|
||||
currentHeading.addSubContent(obj);
|
||||
}
|
||||
} else if (isHigherLevelHeading(obj, headingLevel)) {
|
||||
// 遇到更高级别的标题，结束当前标题的收集
|
||||
if (currentHeading != null) {
|
||||
result.add(currentHeading);
|
||||
currentHeading = null;
|
||||
}
|
||||
inHeadingSection = false;
|
||||
}
|
||||
}
|
||||
|
||||
// 添加最后一个标题
|
||||
if (currentHeading != null) {
|
||||
result.add(currentHeading);
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
/**
|
||||
* 判断段落是否为指定级别的标题
|
||||
*
|
||||
* @param paragraph 段落
|
||||
* @param headingLevel 标题级别
|
||||
* @return 是否匹配
|
||||
*/
|
||||
private static boolean isHeadingLevel(P paragraph, String headingLevel) {
|
||||
PPr ppr = paragraph.getPPr();
|
||||
if (ppr != null) {
|
||||
PPrBase.PStyle pStyle = ppr.getPStyle();
|
||||
if (pStyle != null && headingLevel.equals(pStyle.getVal())) {
|
||||
return true;
|
||||
}
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
/**
|
||||
* 判断是否为更高级别的标题
|
||||
*
|
||||
* @param obj 对象
|
||||
* @param currentLevel 当前级别
|
||||
* @return 是否为更高级别
|
||||
*/
|
||||
private static boolean isHigherLevelHeading(Object obj, String currentLevel) {
|
||||
if (obj instanceof P) {
|
||||
PPr ppr = ((P) obj).getPPr();
|
||||
if (ppr != null) {
|
||||
PPrBase.PStyle pStyle = ppr.getPStyle();
|
||||
if (pStyle != null) {
|
||||
String style = pStyle.getVal();
|
||||
if (style != null && style.matches("[1-9]")) {
|
||||
try {
|
||||
int current = Integer.parseInt(currentLevel);
|
||||
int found = Integer.parseInt(style);
|
||||
return found < current;
|
||||
} catch (NumberFormatException e) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
/**
|
||||
* 判断表格是否为横向布局
|
||||
*
|
||||
* @param obj 表格行对象
|
||||
* @return true表示横向，false表示纵向
|
||||
*/
|
||||
public static boolean judgeTableCross(Object obj) {
|
||||
if (!(obj instanceof Tr)) {
|
||||
return true;
|
||||
}
|
||||
|
||||
Tr row = (Tr) obj;
|
||||
List<Object> content = row.getContent();
|
||||
if (content.isEmpty()) {
|
||||
return true;
|
||||
}
|
||||
|
||||
// 取最后一个单元格，判断是否包含中文
|
||||
Object cellObject = content.get(content.size() - 1);
|
||||
if (cellObject instanceof JAXBElement) {
|
||||
@SuppressWarnings("unchecked")
|
||||
JAXBElement<Tc> cellElement = (JAXBElement<Tc>) cellObject;
|
||||
Tc cell = cellElement.getValue();
|
||||
String text = getTextFromCell(cell);
|
||||
|
||||
if (StrUtil.isBlank(text)) {
|
||||
return true;
|
||||
}
|
||||
|
||||
// 检查是否包含中文字符
|
||||
return containsChinese(text);
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* 读取单元格内的文本内容
|
||||
*
|
||||
* @param cell 单元格
|
||||
* @return 文本内容
|
||||
*/
|
||||
public static String getTextFromCell(Tc cell) {
|
||||
List<Object> cellContent = cell.getContent();
|
||||
StringBuilder cellText = new StringBuilder();
|
||||
|
||||
for (Object content : cellContent) {
|
||||
if (content instanceof P) {
|
||||
P paragraph = (P) content;
|
||||
cellText.append(getTextFromP(paragraph));
|
||||
}
|
||||
}
|
||||
|
||||
return cellText.toString();
|
||||
}
|
||||
|
||||
/**
|
||||
* 从段落中提取纯文本
|
||||
*
|
||||
* @param paragraph 段落
|
||||
* @return 段落内容
|
||||
*/
|
||||
public static String getTextFromP(P paragraph) {
|
||||
StringBuilder textContent = new StringBuilder();
|
||||
|
||||
for (Object runObj : paragraph.getContent()) {
|
||||
if (runObj instanceof R) {
|
||||
R run = (R) runObj;
|
||||
for (Object textObj : run.getContent()) {
|
||||
if (textObj instanceof Text) {
|
||||
textContent.append(((Text) textObj).getValue());
|
||||
} else if (textObj instanceof JAXBElement) {
|
||||
JAXBElement<?> jaxbElement = (JAXBElement<?>) textObj;
|
||||
if (jaxbElement.getValue() instanceof Text) {
|
||||
Text temp = (Text) jaxbElement.getValue();
|
||||
textContent.append(temp.getValue());
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return textContent.toString().trim();
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取段落的样式属性
|
||||
*
|
||||
* @param paragraph 段落
|
||||
* @return 运行属性
|
||||
*/
|
||||
public static RPr getTcPrFromParagraph(P paragraph) {
|
||||
List<Object> content = paragraph.getContent();
|
||||
RPr preservedRPr = null;
|
||||
|
||||
if (!content.isEmpty()) {
|
||||
Object firstObj = content.get(0);
|
||||
if (firstObj instanceof R) {
|
||||
preservedRPr = ((R) firstObj).getRPr();
|
||||
}
|
||||
}
|
||||
|
||||
return preservedRPr;
|
||||
}
|
||||
|
||||
/**
|
||||
* 向段落中添加内容
|
||||
*
|
||||
* @param factory 对象工厂
|
||||
* @param paragraph 段落
|
||||
* @param content 内容
|
||||
* @param rPr 运行属性
|
||||
* @param pPr 段落属性
|
||||
*/
|
||||
public static void addPContent(ObjectFactory factory, P paragraph, String content, RPr rPr, PPr pPr) {
|
||||
R run = factory.createR();
|
||||
Text text = factory.createText();
|
||||
text.setValue(content);
|
||||
run.setRPr(rPr);
|
||||
run.getContent().add(text);
|
||||
paragraph.getContent().add(run);
|
||||
paragraph.setPPr(pPr);
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建N个换行符
|
||||
*
|
||||
* @param factory 对象工厂
|
||||
* @param paragraph 段落
|
||||
* @param n 换行符数量
|
||||
*/
|
||||
public static void addBr(ObjectFactory factory, P paragraph, int n) {
|
||||
R run = factory.createR();
|
||||
for (int i = 0; i < n; i++) {
|
||||
Br br = factory.createBr();
|
||||
run.getContent().add(br);
|
||||
}
|
||||
paragraph.getContent().add(run);
|
||||
}
|
||||
|
||||
/**
|
||||
* 根据表格行获取需要填充的键列表
|
||||
*
|
||||
* @param row 表格行
|
||||
* @return 键列表
|
||||
*/
|
||||
public static List<String> getTableKeys(Tr row) {
|
||||
List<String> keys = new ArrayList<>();
|
||||
List<Object> content = row.getContent();
|
||||
|
||||
for (Object cellObject : content) {
|
||||
if (cellObject instanceof JAXBElement) {
|
||||
@SuppressWarnings("unchecked")
|
||||
JAXBElement<Tc> cellElement = (JAXBElement<Tc>) cellObject;
|
||||
Tc cell = cellElement.getValue();
|
||||
keys.add(getTextFromCell(cell));
|
||||
}
|
||||
}
|
||||
|
||||
return keys;
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建自定义表格行
|
||||
*
|
||||
* @param factory 对象工厂
|
||||
* @param valueMap 数据映射
|
||||
* @param tableKeys 表格键列表
|
||||
* @param trPr 行属性
|
||||
* @param tcPr 单元格属性
|
||||
* @param centerFlag 是否居中
|
||||
* @return 创建的表格行
|
||||
*/
|
||||
public static Tr createCustomRow(ObjectFactory factory, Map<String, String> valueMap,
|
||||
List<String> tableKeys, TrPr trPr, TcPr tcPr, boolean centerFlag) {
|
||||
Tr row = factory.createTr();
|
||||
|
||||
for (String tableKey : tableKeys) {
|
||||
Tc cell = factory.createTc();
|
||||
P paragraph = factory.createP();
|
||||
R run = factory.createR();
|
||||
|
||||
String value = valueMap.getOrDefault(tableKey, "");
|
||||
Text text = factory.createText();
|
||||
text.setValue(value);
|
||||
run.getContent().add(text);
|
||||
paragraph.getContent().add(run);
|
||||
|
||||
// 设置字体和样式
|
||||
RPr rPr = factory.createRPr();
|
||||
RFonts rFonts = factory.createRFonts();
|
||||
|
||||
if (containsChinese(value)) {
|
||||
rFonts.setEastAsia("宋体");
|
||||
rFonts.setAscii("宋体");
|
||||
rFonts.setHAnsi("宋体");
|
||||
} else {
|
||||
rFonts.setEastAsia("Arial");
|
||||
rFonts.setAscii("Arial");
|
||||
rFonts.setHAnsi("Arial");
|
||||
}
|
||||
rPr.setRFonts(rFonts);
|
||||
|
||||
// 设置段落居中
|
||||
if (centerFlag) {
|
||||
PPr pPr = factory.createPPr();
|
||||
Jc jc = factory.createJc();
|
||||
jc.setVal(JcEnumeration.CENTER);
|
||||
pPr.setJc(jc);
|
||||
paragraph.setPPr(pPr);
|
||||
}
|
||||
|
||||
// 设置特殊颜色（如不合格为红色）
|
||||
if ("不合格".equals(value)) {
|
||||
Color color = factory.createColor();
|
||||
color.setVal("FF0000"); // 红色
|
||||
rPr.setColor(color);
|
||||
}
|
||||
|
||||
// 设置字体大小
|
||||
HpsMeasure sz = factory.createHpsMeasure();
|
||||
sz.setVal(new BigInteger("20")); // 10号字体 = 20 half-points
|
||||
rPr.setSz(sz);
|
||||
|
||||
run.setRPr(rPr);
|
||||
cell.getContent().add(paragraph);
|
||||
cell.setTcPr(tcPr);
|
||||
row.getContent().add(cell);
|
||||
}
|
||||
|
||||
row.setTrPr(trPr);
|
||||
return row;
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建自定义表格行（使用列表数据）
|
||||
*
|
||||
* @param factory 对象工厂
|
||||
* @param cellValues 单元格值列表
|
||||
* @param ascFontStyle 西文字体
|
||||
* @param eastFontStyle 中文字体
|
||||
* @param size 字体大小
|
||||
* @param boldFlag 是否加粗
|
||||
* @param centerFlag 居中标志列表
|
||||
* @return 创建的表格行
|
||||
*/
|
||||
public static Tr createCustomRow(ObjectFactory factory, List<String> cellValues,
|
||||
String ascFontStyle, String eastFontStyle, Integer size,
|
||||
boolean boldFlag, List<Integer> centerFlag) {
|
||||
Tr row = factory.createTr();
|
||||
|
||||
for (int i = 0; i < cellValues.size(); i++) {
|
||||
String value = cellValues.get(i);
|
||||
Tc cell = factory.createTc();
|
||||
P paragraph = factory.createP();
|
||||
R run = factory.createR();
|
||||
|
||||
Text text = factory.createText();
|
||||
text.setValue(value);
|
||||
run.getContent().add(text);
|
||||
paragraph.getContent().add(run);
|
||||
|
||||
// 设置段落居中
|
||||
if (!centerFlag.contains(i)) {
|
||||
PPr pPr = factory.createPPr();
|
||||
Jc jc = factory.createJc();
|
||||
jc.setVal(JcEnumeration.CENTER);
|
||||
pPr.setJc(jc);
|
||||
paragraph.setPPr(pPr);
|
||||
}
|
||||
|
||||
// 设置字体和样式
|
||||
RPr rPr = factory.createRPr();
|
||||
|
||||
// 设置颜色
|
||||
if ("不合格".equals(value)) {
|
||||
Color color = factory.createColor();
|
||||
color.setVal("FF0000"); // 红色
|
||||
rPr.setColor(color);
|
||||
}
|
||||
|
||||
// 设置加粗
|
||||
if (boldFlag) {
|
||||
BooleanDefaultTrue bold = factory.createBooleanDefaultTrue();
|
||||
rPr.setB(bold);
|
||||
}
|
||||
|
||||
// 设置字体
|
||||
RFonts fonts = factory.createRFonts();
|
||||
fonts.setAscii(ascFontStyle); // 西文字体
|
||||
fonts.setEastAsia(eastFontStyle); // 中文字体
|
||||
rPr.setRFonts(fonts);
|
||||
|
||||
// 设置字号
|
||||
HpsMeasure fontSize = factory.createHpsMeasure();
|
||||
fontSize.setVal(BigInteger.valueOf(size));
|
||||
rPr.setSz(fontSize); // 西文字号
|
||||
rPr.setSzCs(fontSize); // 中文字号
|
||||
|
||||
run.setRPr(rPr);
|
||||
cell.getContent().add(paragraph);
|
||||
|
||||
// 设置单元格边距
|
||||
TcPr cellProperties = factory.createTcPr();
|
||||
TcMar mar = factory.createTcMar();
|
||||
|
||||
TblWidth top = factory.createTblWidth();
|
||||
top.setW(BigInteger.valueOf(100));
|
||||
mar.setTop(top);
|
||||
|
||||
TblWidth bottom = factory.createTblWidth();
|
||||
bottom.setW(BigInteger.valueOf(100));
|
||||
mar.setBottom(bottom);
|
||||
|
||||
cellProperties.setTcMar(mar);
|
||||
cell.setTcPr(cellProperties);
|
||||
row.getContent().add(cell);
|
||||
}
|
||||
|
||||
return row;
|
||||
}
|
||||
|
||||
/**
|
||||
* 深拷贝表格元素
|
||||
*
|
||||
* @param original 原始表格元素
|
||||
* @return 拷贝的表格元素
|
||||
* @throws Exception 拷贝失败时抛出异常
|
||||
*/
|
||||
public static JAXBElement<Tbl> deepCopyTbl(JAXBElement<Tbl> original) throws Exception {
|
||||
// 使用 docx4j 的 XmlUtils 进行深拷贝
|
||||
Tbl clonedTbl = (Tbl) XmlUtils.deepCopy(original.getValue());
|
||||
|
||||
// 重新包装为 JAXBElement
|
||||
return new JAXBElement<>(
|
||||
original.getName(),
|
||||
original.getDeclaredType(),
|
||||
original.getScope(),
|
||||
clonedTbl
|
||||
);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取表格样式属性
|
||||
*
|
||||
* @param factory 对象工厂
|
||||
* @return 表格属性
|
||||
*/
|
||||
public static TblPr getTblPr(ObjectFactory factory) {
|
||||
TblPr tblPr = factory.createTblPr();
|
||||
TblBorders borders = factory.createTblBorders();
|
||||
|
||||
// 定义边框样式（1磅黑色单实线）
|
||||
CTBorder border = new CTBorder();
|
||||
border.setVal(STBorder.SINGLE); // 实线类型
|
||||
border.setSz(BigInteger.valueOf(4)); // 1磅=4单位（1/8磅）
|
||||
border.setColor("000000"); // 黑色
|
||||
|
||||
// 应用边框到所有边
|
||||
borders.setTop(border);
|
||||
borders.setBottom(border);
|
||||
borders.setLeft(border);
|
||||
borders.setRight(border);
|
||||
borders.setInsideH(border); // 内部水平线
|
||||
borders.setInsideV(border); // 内部垂直线
|
||||
|
||||
tblPr.setTblBorders(borders);
|
||||
|
||||
// 设置表格宽度
|
||||
TblWidth tblWidth = factory.createTblWidth();
|
||||
tblWidth.setType("pct"); // 百分比类型
|
||||
tblWidth.setW(BigInteger.valueOf(5000)); // 96% = 4800/5000 (ISO标准)
|
||||
tblPr.setTblW(tblWidth);
|
||||
|
||||
return tblPr;
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建分页符段落
|
||||
*
|
||||
* @return 包含分页符的段落
|
||||
*/
|
||||
public static P getPageBreak() {
|
||||
try {
|
||||
ObjectFactory factory = new ObjectFactory();
|
||||
R run = factory.createR();
|
||||
Br br = factory.createBr();
|
||||
br.setType(STBrType.PAGE);
|
||||
run.getContent().add(br);
|
||||
|
||||
P pageBreakParagraph = factory.createP();
|
||||
pageBreakParagraph.getContent().add(run);
|
||||
return pageBreakParagraph;
|
||||
} catch (Exception e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
return null;
|
||||
}
|
||||
|
||||
/**
|
||||
* 判断字符串是否包含中文
|
||||
*
|
||||
* @param str 需要判断的字符串
|
||||
* @return 是否包含中文
|
||||
*/
|
||||
private static boolean containsChinese(String str) {
|
||||
if (str == null) {
|
||||
return false;
|
||||
}
|
||||
for (char c : str.toCharArray()) {
|
||||
if (Character.UnicodeScript.of(c) == Character.UnicodeScript.HAN) {
|
||||
return true;
|
||||
}
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
/**
|
||||
* 存储标题及其子内容的辅助类
|
||||
*/
|
||||
public static class HeadingContent {
|
||||
private String headingText;
|
||||
private List<Object> subContent = new ArrayList<>();
|
||||
|
||||
public void setHeadingText(String text) {
|
||||
this.headingText = text;
|
||||
}
|
||||
|
||||
public String getHeadingText() {
|
||||
return headingText;
|
||||
}
|
||||
|
||||
public void addSubContent(Object obj) {
|
||||
subContent.add(obj);
|
||||
}
|
||||
|
||||
public List<Object> getSubContent() {
|
||||
return subContent;
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,111 @@
|
||||
package com.njcn.gather.tools.report.util;
|
||||
|
||||
import com.njcn.common.pojo.exception.BusinessException;
|
||||
import com.njcn.gather.tools.report.model.ReportResponseEnum;
|
||||
|
||||
/**
|
||||
* 报告生成异常工具类
|
||||
* 基于现有的BusinessException创建报告生成相关异常
|
||||
*
|
||||
* @author hongawen
|
||||
*/
|
||||
public class ReportExceptionUtil {
|
||||
|
||||
/**
|
||||
* 模板加载异常
|
||||
*/
|
||||
public static BusinessException templateLoadError(String message) {
|
||||
return new BusinessException(ReportResponseEnum.TEMPLATE_LOAD_ERROR, message);
|
||||
}
|
||||
|
||||
/**
|
||||
* 模板处理异常
|
||||
*/
|
||||
public static BusinessException templateProcessError(String message) {
|
||||
return new BusinessException(ReportResponseEnum.TEMPLATE_PROCESS_ERROR, message);
|
||||
}
|
||||
|
||||
/**
|
||||
* 文件保存异常
|
||||
*/
|
||||
public static BusinessException fileSaveError(String message) {
|
||||
return new BusinessException(ReportResponseEnum.FILE_SAVE_ERROR, message);
|
||||
}
|
||||
|
||||
/**
|
||||
* 参数验证异常
|
||||
*/
|
||||
public static BusinessException validationError(String message) {
|
||||
return new BusinessException(ReportResponseEnum.VALIDATION_ERROR, message);
|
||||
}
|
||||
|
||||
/**
|
||||
* 不支持的操作异常
|
||||
*/
|
||||
public static BusinessException unsupportedOperation(String message) {
|
||||
return new BusinessException(ReportResponseEnum.UNSUPPORTED_OPERATION, message);
|
||||
}
|
||||
|
||||
/**
|
||||
* 通用报告生成异常
|
||||
*/
|
||||
public static BusinessException reportGenerationError(String message) {
|
||||
return new BusinessException(ReportResponseEnum.REPORT_GENERATION_ERROR, message);
|
||||
}
|
||||
|
||||
/**
|
||||
* 异步处理异常
|
||||
*/
|
||||
public static BusinessException asyncProcessingError(String message) {
|
||||
return new BusinessException(ReportResponseEnum.ASYNC_PROCESSING_ERROR, message);
|
||||
}
|
||||
|
||||
/**
|
||||
* 批量处理异常
|
||||
*/
|
||||
public static BusinessException batchProcessingError(String message) {
|
||||
return new BusinessException(ReportResponseEnum.BATCH_PROCESSING_ERROR, message);
|
||||
}
|
||||
|
||||
/**
|
||||
* 模板文件不存在异常
|
||||
*/
|
||||
public static BusinessException templateNotFound(String message) {
|
||||
return new BusinessException(ReportResponseEnum.TEMPLATE_NOT_FOUND, message);
|
||||
}
|
||||
|
||||
/**
|
||||
* 数据为空异常
|
||||
*/
|
||||
public static BusinessException dataEmpty(String message) {
|
||||
return new BusinessException(ReportResponseEnum.DATA_EMPTY, message);
|
||||
}
|
||||
|
||||
/**
|
||||
* 找不到合适的处理器异常
|
||||
*/
|
||||
public static BusinessException noSuitableProcessor(String message) {
|
||||
return new BusinessException(ReportResponseEnum.NO_SUITABLE_PROCESSOR, message);
|
||||
}
|
||||
|
||||
/**
|
||||
* 处理超时异常
|
||||
*/
|
||||
public static BusinessException processingTimeout(String message) {
|
||||
return new BusinessException(ReportResponseEnum.PROCESSING_TIMEOUT, message);
|
||||
}
|
||||
|
||||
/**
|
||||
* 根据异常类型和消息创建BusinessException
|
||||
*/
|
||||
public static BusinessException create(ReportResponseEnum responseEnum, String customMessage) {
|
||||
return new BusinessException(responseEnum, customMessage);
|
||||
}
|
||||
|
||||
/**
|
||||
* 根据异常类型创建BusinessException（使用默认消息）
|
||||
*/
|
||||
public static BusinessException create(ReportResponseEnum responseEnum) {
|
||||
return new BusinessException(responseEnum);
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,332 @@
|
||||
package com.njcn.gather.tools.report.util;
|
||||
|
||||
import org.apache.poi.xwpf.usermodel.*;
|
||||
import org.apache.xmlbeans.XmlCursor;
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
|
||||
/**
|
||||
* Word文档工具类
|
||||
* 基于Apache POI的通用Word文档操作工具
|
||||
* 从原有WordUtil中抽取的通用功能
|
||||
*
|
||||
* @author hongawen
|
||||
*/
|
||||
public class WordDocumentUtil {
|
||||
|
||||
/**
|
||||
* 替换文档中的占位符
|
||||
*
|
||||
* @param document 文档对象
|
||||
* @param placeholders 占位符键值对
|
||||
*/
|
||||
public static void replacePlaceholders(XWPFDocument document, Map<String, String> placeholders) {
|
||||
if (document == null || placeholders == null || placeholders.isEmpty()) {
|
||||
return;
|
||||
}
|
||||
|
||||
replacePlaceholdersInParagraphs(document, placeholders);
|
||||
replacePlaceholdersInTables(document, placeholders);
|
||||
}
|
||||
|
||||
/**
|
||||
* 替换段落中的占位符
|
||||
*
|
||||
* @param document 文档对象
|
||||
* @param placeholders 占位符键值对
|
||||
*/
|
||||
public static void replacePlaceholdersInParagraphs(XWPFDocument document, Map<String, String> placeholders) {
|
||||
for (XWPFParagraph paragraph : document.getParagraphs()) {
|
||||
replacePlaceholdersInParagraph(paragraph, placeholders);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 替换单个段落中的占位符
|
||||
*
|
||||
* @param paragraph 段落对象
|
||||
* @param placeholders 占位符键值对
|
||||
*/
|
||||
public static void replacePlaceholdersInParagraph(XWPFParagraph paragraph, Map<String, String> placeholders) {
|
||||
List<XWPFRun> runs = paragraph.getRuns();
|
||||
if (runs != null) {
|
||||
for (XWPFRun run : runs) {
|
||||
String text = run.getText(0);
|
||||
if (text != null) {
|
||||
for (Map.Entry<String, String> entry : placeholders.entrySet()) {
|
||||
text = text.replace(entry.getKey(), entry.getValue());
|
||||
}
|
||||
run.setText(text, 0);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 替换表格中的占位符
|
||||
*
|
||||
* @param document 文档对象
|
||||
* @param placeholders 占位符键值对
|
||||
*/
|
||||
public static void replacePlaceholdersInTables(XWPFDocument document, Map<String, String> placeholders) {
|
||||
for (XWPFTable table : document.getTables()) {
|
||||
replacePlaceholdersInTable(table, placeholders);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 替换单个表格中的占位符
|
||||
*
|
||||
* @param table 表格对象
|
||||
* @param placeholders 占位符键值对
|
||||
*/
|
||||
public static void replacePlaceholdersInTable(XWPFTable table, Map<String, String> placeholders) {
|
||||
for (XWPFTableRow row : table.getRows()) {
|
||||
for (XWPFTableCell cell : row.getTableCells()) {
|
||||
for (XWPFParagraph paragraph : cell.getParagraphs()) {
|
||||
replacePlaceholdersInParagraph(paragraph, placeholders);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 将源文档的内容追加到目标文档中
|
||||
*
|
||||
* @param target 目标文档
|
||||
* @param source 源文档
|
||||
*/
|
||||
public static void appendDocument(XWPFDocument target, XWPFDocument source) {
|
||||
if (target == null || source == null) {
|
||||
return;
|
||||
}
|
||||
|
||||
// 在追加内容之前，插入分页符
|
||||
insertPageBreak(target);
|
||||
|
||||
// 遍历源文档的所有块（段落、表格等）
|
||||
source.getBodyElements().forEach(bodyElement -> {
|
||||
switch (bodyElement.getElementType()) {
|
||||
case PARAGRAPH:
|
||||
// 处理段落
|
||||
XWPFParagraph sourceParagraph = (XWPFParagraph) bodyElement;
|
||||
XWPFParagraph newParagraph = target.createParagraph();
|
||||
copyParagraphContent(sourceParagraph, newParagraph);
|
||||
break;
|
||||
case TABLE:
|
||||
// 处理表格
|
||||
XWPFTable sourceTable = (XWPFTable) bodyElement;
|
||||
XWPFTable newTable = target.createTable();
|
||||
copyTableContent(sourceTable, newTable);
|
||||
break;
|
||||
default:
|
||||
// 其他类型的内容处理
|
||||
break;
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* 复制段落内容
|
||||
*
|
||||
* @param source 源段落
|
||||
* @param target 目标段落
|
||||
*/
|
||||
private static void copyParagraphContent(XWPFParagraph source, XWPFParagraph target) {
|
||||
try {
|
||||
// 简单的内容复制，保持样式
|
||||
target.getCTP().set(source.getCTP());
|
||||
} catch (Exception e) {
|
||||
// 如果复制失败，采用文本复制方式
|
||||
target.createRun().setText(source.getText());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 复制表格内容
|
||||
*
|
||||
* @param source 源表格
|
||||
* @param target 目标表格
|
||||
*/
|
||||
private static void copyTableContent(XWPFTable source, XWPFTable target) {
|
||||
try {
|
||||
// 简单的内容复制，保持样式
|
||||
target.getCTTbl().set(source.getCTTbl());
|
||||
} catch (Exception e) {
|
||||
// 如果复制失败，采用逐行复制方式
|
||||
copyTableRowByRow(source, target);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 逐行复制表格内容
|
||||
*
|
||||
* @param source 源表格
|
||||
* @param target 目标表格
|
||||
*/
|
||||
private static void copyTableRowByRow(XWPFTable source, XWPFTable target) {
|
||||
// 先清空目标表格的默认行
|
||||
if (target.getRows().size() > 0) {
|
||||
target.removeRow(0);
|
||||
}
|
||||
|
||||
// 复制每一行
|
||||
for (XWPFTableRow sourceRow : source.getRows()) {
|
||||
XWPFTableRow targetRow = target.createRow();
|
||||
|
||||
// 确保目标行有足够的单元格
|
||||
int cellsNeeded = sourceRow.getTableCells().size();
|
||||
while (targetRow.getTableCells().size() < cellsNeeded) {
|
||||
targetRow.createCell();
|
||||
}
|
||||
|
||||
// 复制每个单元格的内容
|
||||
for (int i = 0; i < cellsNeeded; i++) {
|
||||
XWPFTableCell sourceCell = sourceRow.getTableCells().get(i);
|
||||
XWPFTableCell targetCell = targetRow.getTableCells().get(i);
|
||||
|
||||
// 复制单元格文本内容
|
||||
StringBuilder cellText = new StringBuilder();
|
||||
for (XWPFParagraph para : sourceCell.getParagraphs()) {
|
||||
if (cellText.length() > 0) {
|
||||
cellText.append("\n");
|
||||
}
|
||||
cellText.append(para.getText());
|
||||
}
|
||||
|
||||
if (cellText.length() > 0) {
|
||||
targetCell.getParagraphs().get(0).createRun().setText(cellText.toString());
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 在文档中插入分页符
|
||||
*
|
||||
* @param document 文档对象
|
||||
*/
|
||||
public static void insertPageBreak(XWPFDocument document) {
|
||||
XWPFParagraph paragraph = document.createParagraph();
|
||||
XWPFRun run = paragraph.createRun();
|
||||
run.addBreak(BreakType.PAGE);
|
||||
}
|
||||
|
||||
/**
|
||||
* 查找文档中指定样式的段落
|
||||
*
|
||||
* @param document 文档对象
|
||||
* @param styleId 样式ID
|
||||
* @return 匹配的段落列表
|
||||
*/
|
||||
public static List<XWPFParagraph> findParagraphsByStyle(XWPFDocument document, String styleId) {
|
||||
List<XWPFParagraph> result = new ArrayList<>();
|
||||
for (XWPFParagraph paragraph : document.getParagraphs()) {
|
||||
String style = paragraph.getStyle();
|
||||
if (styleId.equals(style)) {
|
||||
result.add(paragraph);
|
||||
}
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取段落在文档中的位置索引
|
||||
*
|
||||
* @param document 文档对象
|
||||
* @param paragraph 段落对象
|
||||
* @return 位置索引，找不到返回-1
|
||||
*/
|
||||
public static int getParagraphPosition(XWPFDocument document, XWPFParagraph paragraph) {
|
||||
List<IBodyElement> bodyElements = document.getBodyElements();
|
||||
for (int i = 0; i < bodyElements.size(); i++) {
|
||||
if (bodyElements.get(i) instanceof XWPFParagraph &&
|
||||
bodyElements.get(i).equals(paragraph)) {
|
||||
return i;
|
||||
}
|
||||
}
|
||||
return -1;
|
||||
}
|
||||
|
||||
/**
|
||||
* 在指定位置插入段落
|
||||
*
|
||||
* @param document 文档对象
|
||||
* @param position 插入位置
|
||||
* @param text 段落文本
|
||||
* @return 创建的段落对象
|
||||
*/
|
||||
public static XWPFParagraph insertParagraphAt(XWPFDocument document, int position, String text) {
|
||||
try {
|
||||
XWPFParagraph paragraph = document.insertNewParagraph(getCursorAtPosition(document, position));
|
||||
if (text != null && !text.isEmpty()) {
|
||||
paragraph.createRun().setText(text);
|
||||
}
|
||||
return paragraph;
|
||||
} catch (Exception e) {
|
||||
// 如果插入失败，则在末尾添加
|
||||
XWPFParagraph paragraph = document.createParagraph();
|
||||
if (text != null && !text.isEmpty()) {
|
||||
paragraph.createRun().setText(text);
|
||||
}
|
||||
return paragraph;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取指定位置的游标
|
||||
*
|
||||
* @param document 文档对象
|
||||
* @param position 位置
|
||||
* @return XML游标
|
||||
*/
|
||||
private static XmlCursor getCursorAtPosition(XWPFDocument document, int position) {
|
||||
List<IBodyElement> bodyElements = document.getBodyElements();
|
||||
if (position >= 0 && position < bodyElements.size()) {
|
||||
IBodyElement element = bodyElements.get(position);
|
||||
if (element instanceof XWPFParagraph) {
|
||||
return ((XWPFParagraph) element).getCTP().newCursor();
|
||||
}
|
||||
}
|
||||
// 默认返回文档末尾的游标
|
||||
return document.getDocument().getBody().newCursor();
|
||||
}
|
||||
|
||||
/**
|
||||
* 统计文档信息
|
||||
*
|
||||
* @param document 文档对象
|
||||
* @return 文档统计信息
|
||||
*/
|
||||
public static DocumentStats getDocumentStats(XWPFDocument document) {
|
||||
DocumentStats stats = new DocumentStats();
|
||||
|
||||
stats.paragraphCount = document.getParagraphs().size();
|
||||
stats.tableCount = document.getTables().size();
|
||||
|
||||
// 统计文本长度
|
||||
int textLength = 0;
|
||||
for (XWPFParagraph paragraph : document.getParagraphs()) {
|
||||
textLength += paragraph.getText().length();
|
||||
}
|
||||
stats.textLength = textLength;
|
||||
|
||||
return stats;
|
||||
}
|
||||
|
||||
/**
|
||||
* 文档统计信息
|
||||
*/
|
||||
public static class DocumentStats {
|
||||
public int paragraphCount;
|
||||
public int tableCount;
|
||||
public int textLength;
|
||||
|
||||
@Override
|
||||
public String toString() {
|
||||
return String.format("段落: %d, 表格: %d, 文本长度: %d",
|
||||
paragraphCount, tableCount, textLength);
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,103 @@
|
||||
package com.njcn.gather.tools.report;
|
||||
|
||||
import com.njcn.gather.tools.report.model.*;
|
||||
import com.njcn.gather.tools.report.engine.WordDocumentProcessor;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.junit.jupiter.api.Test;
|
||||
import org.springframework.boot.test.context.SpringBootTest;
|
||||
|
||||
import java.util.HashMap;
|
||||
import java.util.Map;
|
||||
|
||||
/**
|
||||
* 报告生成工具测试类
|
||||
*
|
||||
* @author hongawen
|
||||
*/
|
||||
@Slf4j
|
||||
public class ReportGeneratorTest {
|
||||
|
||||
@Test
|
||||
public void testWordDocumentProcessor() {
|
||||
// 创建测试数据
|
||||
Map<String, Object> data = new HashMap<>();
|
||||
data.put("name", "张三");
|
||||
data.put("date", "2024-01-01");
|
||||
data.put("amount", "1000.00");
|
||||
data.put("company", "灿能公司");
|
||||
|
||||
// 创建处理选项
|
||||
ProcessOptions options = ProcessOptions.simplePlaceholder();
|
||||
|
||||
// 这里需要真实的模板文件进行测试
|
||||
// 由于测试环境可能没有模板文件，这个测试主要验证代码结构
|
||||
log.info("报告生成工具基础结构测试通过");
|
||||
|
||||
// 测试模型创建
|
||||
TemplateSource templateSource = TemplateSource.builder()
|
||||
.filePath("test-template.docx")
|
||||
.type(TemplateType.DOCX)
|
||||
.name("测试模板")
|
||||
.build();
|
||||
|
||||
TemplateRequest.OutputTarget outputTarget = TemplateRequest.OutputTarget.toBytes();
|
||||
|
||||
TemplateRequest request = TemplateRequest.builder()
|
||||
.templateSource(templateSource)
|
||||
.data(data)
|
||||
.options(options)
|
||||
.outputTarget(outputTarget)
|
||||
.requestId("test-001")
|
||||
.build();
|
||||
|
||||
// 验证请求有效性
|
||||
assert request.isValid() : "请求应该是有效的";
|
||||
assert request.getEffectiveOptions() != null : "应该有有效的处理选项";
|
||||
|
||||
log.info("模型验证测试通过");
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testProcessOptions() {
|
||||
// 测试默认选项
|
||||
ProcessOptions defaultOptions = ProcessOptions.defaultOptions();
|
||||
assert defaultOptions.isEnablePlaceholder() : "默认应启用占位符替换";
|
||||
|
||||
// 测试简单占位符选项
|
||||
ProcessOptions simpleOptions = ProcessOptions.simplePlaceholder();
|
||||
assert simpleOptions.isEnablePlaceholder() : "简单选项应启用占位符替换";
|
||||
assert !simpleOptions.isEnableBookmark() : "简单选项应禁用书签处理";
|
||||
|
||||
// 测试高级功能选项
|
||||
ProcessOptions advancedOptions = ProcessOptions.advancedFeatures();
|
||||
assert advancedOptions.isEnablePlaceholder() : "高级选项应启用占位符替换";
|
||||
assert advancedOptions.isEnableBookmark() : "高级选项应启用书签处理";
|
||||
assert advancedOptions.isEnableDynamicTable() : "高级选项应启用动态表格";
|
||||
|
||||
log.info("处理选项测试通过");
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testPlaceholderPattern() {
|
||||
ProcessOptions.PlaceholderPattern pattern = ProcessOptions.PlaceholderPattern.DOLLAR_BRACE;
|
||||
|
||||
String placeholder = pattern.buildPlaceholder("name");
|
||||
assert "${name}".equals(placeholder) : "占位符格式应该正确";
|
||||
|
||||
log.info("占位符模式测试通过");
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testTemplateType() {
|
||||
TemplateType type = TemplateType.fromExtension("test.docx");
|
||||
assert type == TemplateType.DOCX : "应该识别为DOCX类型";
|
||||
|
||||
type = TemplateType.fromExtension("test.pdf");
|
||||
assert type == TemplateType.PDF : "应该识别为PDF类型";
|
||||
|
||||
type = TemplateType.fromExtension("unknown.xyz");
|
||||
assert type == TemplateType.DOCX : "未知类型应该默认为DOCX";
|
||||
|
||||
log.info("模板类型测试通过");
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user