331 lines
11 KiB
Markdown
331 lines
11 KiB
Markdown
# AI 服务重构总结
|
||
|
||
---
|
||
title: AI调用统一封装到SmartHandleService
|
||
author: AI Assistant
|
||
date: 2026-02-10
|
||
status: final
|
||
category: 重构
|
||
---
|
||
|
||
## 概述
|
||
|
||
本次重构将项目中所有分散的 AI 调用统一封装到 `SmartHandleService` 中，实现了AI功能的集中管理和统一维护。
|
||
|
||
## 重构前的问题
|
||
|
||
在重构前，项目中有多个地方直接依赖 `IOpenAiService` 进行 AI 调用：
|
||
|
||
1. **EmailParseServicesBase** (`Service/EmailServices/EmailParse/IEmailParseServices.cs`)
|
||
- 邮件解析AI兜底逻辑
|
||
- 包含大量重复的类型判断代码
|
||
|
||
2. **CategoryIconGenerationJob** (`Service/Jobs/CategoryIconGenerationJob.cs`)
|
||
- 定时任务批量生成分类图标
|
||
- 包含复杂的Prompt构建逻辑
|
||
|
||
3. **TransactionCategoryApplication** (`Application/TransactionCategoryApplication.cs`)
|
||
- 手动触发单个图标生成
|
||
- 包含SVG提取和验证逻辑
|
||
|
||
4. **BudgetService** (`Service/Budget/BudgetService.cs`)
|
||
- 生成预算执行报告
|
||
- 包含复杂的数据格式化逻辑
|
||
|
||
### 存在的问题
|
||
|
||
- **代码重复**：多处存在相似的AI调用代码
|
||
- **职责分散**：AI相关逻辑散落在不同层级
|
||
- **难以维护**：修改AI调用逻辑需要改动多个文件
|
||
- **测试困难**：需要在多个测试类中Mock `IOpenAiService`
|
||
|
||
## 重构方案
|
||
|
||
### 1. 扩展 SmartHandleService 接口
|
||
|
||
在 `Service/AI/SmartHandleService.cs` 中新增以下方法：
|
||
|
||
```csharp
|
||
public interface ISmartHandleService
|
||
{
|
||
// 原有方法
|
||
Task SmartClassifyAsync(long[] transactionIds, Action<(string type, string data)> chunkAction);
|
||
Task AnalyzeBillAsync(string userInput, Action<string> chunkAction);
|
||
Task<TransactionParseResult?> ParseOneLineBillAsync(string text);
|
||
|
||
// 新增方法
|
||
|
||
/// <summary>
|
||
/// 从邮件正文中使用AI提取交易记录（AI兜底方案）
|
||
/// </summary>
|
||
Task<(string card, string reason, decimal amount, decimal balance, TransactionType type, DateTime? occurredAt)[]?>
|
||
ParseEmailByAiAsync(string emailBody);
|
||
|
||
/// <summary>
|
||
/// 为分类生成多个SVG图标（定时任务使用）
|
||
/// </summary>
|
||
Task<List<string>?> GenerateCategoryIconsAsync(string categoryName, TransactionType categoryType, int iconCount = 5);
|
||
|
||
/// <summary>
|
||
/// 为分类生成单个SVG图标（手动触发使用）
|
||
/// </summary>
|
||
Task<string?> GenerateSingleCategoryIconAsync(string categoryName, TransactionType categoryType);
|
||
|
||
/// <summary>
|
||
/// 生成预算执行报告（HTML格式）
|
||
/// </summary>
|
||
Task<string?> GenerateBudgetReportAsync(string promptWithData, int year, int month);
|
||
}
|
||
```
|
||
|
||
### 2. 实现新增方法
|
||
|
||
#### ParseEmailByAiAsync - 邮件解析
|
||
|
||
将 `EmailParseServicesBase.ParseByAiAsync` 方法迁移到 `SmartHandleService`：
|
||
|
||
- 保留完整的JSON解析逻辑
|
||
- 保留markdown代码块清理逻辑
|
||
- 保留单个对象兼容处理
|
||
- 将类型判断逻辑保留在基类中供子类使用
|
||
|
||
#### GenerateCategoryIconsAsync - 批量图标生成
|
||
|
||
将 `CategoryIconGenerationJob.GenerateIconsForCategoryAsync` 的核心逻辑迁移：
|
||
|
||
- 封装5种不同风格的图标生成Prompt
|
||
- 处理JSON数组解析和验证
|
||
- 统一的错误处理和日志记录
|
||
|
||
#### GenerateSingleCategoryIconAsync - 单个图标生成
|
||
|
||
将 `TransactionCategoryApplication.GenerateIconAsync` 的核心逻辑迁移：
|
||
|
||
- 简化的极简风格Prompt
|
||
- SVG标签提取逻辑
|
||
- 统一的返回格式
|
||
|
||
#### GenerateBudgetReportAsync - 预算报告生成
|
||
|
||
将 `BudgetService` 中的报告生成逻辑迁移：
|
||
|
||
- 接收完整的Prompt（包含数据和格式要求）
|
||
- 统一的HTML格式要求
|
||
- 统一的错误处理
|
||
|
||
### 3. 更新调用方
|
||
|
||
#### EmailParseServicesBase
|
||
|
||
```csharp
|
||
// 修改前
|
||
public abstract class EmailParseServicesBase(
|
||
ILogger<EmailParseServicesBase> logger,
|
||
IOpenAiService openAiService
|
||
) : IEmailParseServices
|
||
|
||
// 修改后
|
||
public abstract class EmailParseServicesBase(
|
||
ILogger<EmailParseServicesBase> logger,
|
||
ISmartHandleService smartHandleService
|
||
) : IEmailParseServices
|
||
```
|
||
|
||
调用改为：
|
||
```csharp
|
||
result = await smartHandleService.ParseEmailByAiAsync(emailContent) ?? [];
|
||
```
|
||
|
||
#### CategoryIconGenerationJob
|
||
|
||
```csharp
|
||
// 修改前
|
||
public class CategoryIconGenerationJob(
|
||
ITransactionCategoryRepository categoryRepository,
|
||
IOpenAiService openAiService,
|
||
ILogger<CategoryIconGenerationJob> logger) : IJob
|
||
|
||
// 修改后
|
||
public class CategoryIconGenerationJob(
|
||
ITransactionCategoryRepository categoryRepository,
|
||
ISmartHandleService smartHandleService,
|
||
ILogger<CategoryIconGenerationJob> logger) : IJob
|
||
```
|
||
|
||
调用简化为：
|
||
```csharp
|
||
var icons = await smartHandleService.GenerateCategoryIconsAsync(category.Name, category.Type, iconCount: 5);
|
||
```
|
||
|
||
#### TransactionCategoryApplication
|
||
|
||
```csharp
|
||
// 修改前
|
||
public class TransactionCategoryApplication(
|
||
...
|
||
IOpenAiService openAiService,
|
||
...) : ITransactionCategoryApplication
|
||
|
||
// 修改后
|
||
public class TransactionCategoryApplication(
|
||
...
|
||
ISmartHandleService smartHandleService,
|
||
...) : ITransactionCategoryApplication
|
||
```
|
||
|
||
调用简化为：
|
||
```csharp
|
||
var svg = await smartHandleService.GenerateSingleCategoryIconAsync(category.Name, category.Type);
|
||
```
|
||
|
||
#### BudgetService
|
||
|
||
```csharp
|
||
// 修改前
|
||
public class BudgetService(
|
||
...
|
||
IOpenAiService openAiService,
|
||
...) : IBudgetService
|
||
|
||
// 修改后
|
||
public class BudgetService(
|
||
...
|
||
ISmartHandleService smartHandleService,
|
||
...) : IBudgetService
|
||
```
|
||
|
||
调用简化为：
|
||
```csharp
|
||
var htmlReport = await smartHandleService.GenerateBudgetReportAsync(dataPrompt, year, month);
|
||
```
|
||
|
||
### 4. 更新测试代码
|
||
|
||
更新所有测试类，将 `IOpenAiService` 的Mock改为 `ISmartHandleService`：
|
||
|
||
- `BudgetStatsTest.cs`
|
||
- `TransactionCategoryApplicationTest.cs`
|
||
|
||
## 重构收益
|
||
|
||
### 1. 代码集中管理
|
||
|
||
- 所有AI调用逻辑集中在 `SmartHandleService` 中
|
||
- 便于统一修改Prompt策略
|
||
- 便于添加新的AI功能
|
||
|
||
### 2. 职责清晰
|
||
|
||
- `OpenAiService`：底层AI API调用（同步/流式）
|
||
- `SmartHandleService`：业务级AI功能封装
|
||
- 业务服务层：专注业务逻辑，无需关心AI调用细节
|
||
|
||
### 3. 易于测试
|
||
|
||
- 只需Mock `ISmartHandleService`
|
||
- 测试更简洁，Mock对象更少
|
||
- 可以为 `SmartHandleService` 编写独立的单元测试
|
||
|
||
### 4. 易于扩展
|
||
|
||
- 新增AI功能只需在 `SmartHandleService` 中添加方法
|
||
- 不影响现有业务代码
|
||
- 符合开闭原则（对扩展开放，对修改封闭）
|
||
|
||
### 5. 代码复用
|
||
|
||
- 消除了多处重复的AI调用代码
|
||
- 统一的错误处理和日志记录
|
||
- 统一的返回格式和验证逻辑
|
||
|
||
## 测试结果
|
||
|
||
重构后运行全部211个单元测试，全部通过：
|
||
|
||
```
|
||
已通过! - 失败: 0，通过: 211，已跳过: 0，总计: 211，持续时间: 22 s
|
||
```
|
||
|
||
## 受影响的文件清单
|
||
|
||
### 新增/修改的文件
|
||
|
||
1. **Service/AI/SmartHandleService.cs**
|
||
- 新增4个方法接口定义
|
||
- 新增4个方法实现
|
||
- 新增辅助方法：`ParseEmailSingleRecord`, `DetermineTransactionType`
|
||
|
||
### 修改的业务服务
|
||
|
||
2. **Service/EmailServices/EmailParse/IEmailParseServices.cs**
|
||
- 构造函数参数改为 `ISmartHandleService`
|
||
- 移除 `ParseByAiAsync` 方法，改为调用 `smartHandleService.ParseEmailByAiAsync`
|
||
- 保留 `DetermineTransactionType` 为protected方法供子类使用
|
||
|
||
3. **Service/EmailServices/EmailParse/EmailParseFormCcsvc.cs**
|
||
- 构造函数参数改为 `ISmartHandleService`
|
||
|
||
4. **Service/EmailServices/EmailParse/EmailParseForm95555.cs**
|
||
- 构造函数参数改为 `ISmartHandleService`
|
||
|
||
5. **Service/Jobs/CategoryIconGenerationJob.cs**
|
||
- 构造函数参数改为 `ISmartHandleService`
|
||
- 简化 `GenerateIconsForCategoryAsync` 方法
|
||
|
||
6. **Application/TransactionCategoryApplication.cs**
|
||
- 构造函数参数改为 `ISmartHandleService`
|
||
- 简化 `GenerateIconAsync` 方法
|
||
|
||
7. **Service/Budget/BudgetService.cs**
|
||
- 构造函数参数改为 `ISmartHandleService`
|
||
- 简化报告生成调用
|
||
|
||
### 修改的测试文件
|
||
|
||
8. **WebApi.Test/Budget/BudgetStatsTest.cs**
|
||
- Mock对象改为 `ISmartHandleService`
|
||
|
||
9. **WebApi.Test/Application/TransactionCategoryApplicationTest.cs**
|
||
- Mock对象改为 `ISmartHandleService`
|
||
|
||
## 架构图
|
||
|
||
```
|
||
┌─────────────────────────────────────────────────────────────┐
|
||
│ 业务层 (Business Layer) │
|
||
│ ┌──────────────┐ ┌──────────────┐ ┌──────────────┐ │
|
||
│ │ BudgetService│ │CategoryApp │ │EmailParse │ │
|
||
│ │ │ │ │ │Services │ │
|
||
│ └──────┬───────┘ └──────┬───────┘ └──────┬───────┘ │
|
||
│ │ │ │ │
|
||
│ └─────────────────┼─────────────────┘ │
|
||
│ │ │
|
||
└───────────────────────────┼────────────────────────────────┘
|
||
│ 依赖
|
||
┌───────────────────────────┼────────────────────────────────┐
|
||
│ AI 服务层 (AI Service Layer) │
|
||
│ │ │
|
||
│ ┌───────▼────────┐ │
|
||
│ │SmartHandleService│ ◄── 统一封装AI调用 │
|
||
│ │ (业务级AI功能) │ │
|
||
│ └───────┬────────┘ │
|
||
│ │ │
|
||
│ ┌───────▼────────┐ │
|
||
│ │ OpenAiService │ ◄── 底层API调用 │
|
||
│ │ (HTTP Client) │ │
|
||
│ └────────────────┘ │
|
||
└─────────────────────────────────────────────────────────────┘
|
||
```
|
||
|
||
## 后续建议
|
||
|
||
1. **添加单元测试**：为 `SmartHandleService` 的新方法添加独立的单元测试
|
||
2. **监控日志**：关注AI调用的性能和错误日志
|
||
3. **Prompt优化**：基于实际使用反馈持续优化Prompt
|
||
4. **缓存机制**：考虑为图标生成等场景添加缓存
|
||
5. **重试机制**：考虑为AI调用添加重试逻辑
|
||
|
||
## 总结
|
||
|
||
本次重构成功将项目中所有AI调用统一封装到 `SmartHandleService`，实现了代码的集中管理和职责分离。重构后的代码更易维护、更易测试、更易扩展，符合单一职责原则和依赖倒置原则。所有测试用例全部通过，证明重构没有引入功能回归问题。
|