This commit is contained in:
2026-01-29 16:34:37 +08:00
parent f50e11a052
commit 5bff33547c
3 changed files with 51 additions and 32 deletions

View File

@@ -2,6 +2,8 @@ package pdfg
import (
"context"
"crypto/rand"
"encoding/hex"
"encoding/json"
"errors"
"fmt"
@@ -94,9 +96,6 @@ func ProcessPDFG01GZRequest(ctx context.Context, params []byte, deps *processors
pdfGenService = pdfgen.NewPDFGenService(defaultCfg, zapLogger)
}
// 计算缓存键(内部使用,不直接暴露给用户)
cacheKey := cacheManager.GetCacheKey(paramsDto.Name, paramsDto.IDCard)
// 直接生成PDF不检查缓存每次都重新生成
zapLogger.Info("开始生成PDF",
zap.String("name", paramsDto.Name),
@@ -153,15 +152,15 @@ func ProcessPDFG01GZRequest(ctx context.Context, params []byte, deps *processors
return nil, errors.Join(processors.ErrSystem, fmt.Errorf("生成PDF失败: %w", err))
}
// 保存到缓存(基于姓名+身份证
if err := cacheManager.Set(paramsDto.Name, paramsDto.IDCard, pdfResp.PDFBytes); err != nil {
// 生成报告ID每次请求都生成唯一的ID
reportID := generateReportID()
// 保存到缓存基于报告ID文件名包含时间戳确保唯一性
if err := cacheManager.SetByReportID(reportID, pdfResp.PDFBytes); err != nil {
zapLogger.Warn("保存PDF到缓存失败", zap.Error(err))
// 不影响返回结果,只记录警告
}
// 生成报告ID与内部缓存键解耦对外只暴露带前缀的ID
reportID := generateReportID(cacheKey)
// 生成下载链接基于报告ID
downloadURL := generateDownloadURL(apiDomain, reportID)
expiresAt := time.Now().Add(cacheTTL)
@@ -473,10 +472,18 @@ func generateReportNumber() string {
}
// generateReportID 生成对外可见的报告ID
// 结构{report_number}-{cacheKey},前缀变化以避免用户轻易看出是否命中缓存
func generateReportID(cacheKey string) string {
// 每次请求都生成唯一的ID格式{report_number}-{随机字符串}
// 注意不再包含cacheKey因为每次请求都会重新生成不需要通过ID定位缓存文件
func generateReportID() string {
reportNumber := generateReportNumber()
return fmt.Sprintf("%s-%s", reportNumber, cacheKey)
// 生成8字节随机字符串确保每次请求ID都不同
randomBytes := make([]byte, 8)
if _, err := rand.Read(randomBytes); err != nil {
// 如果随机数生成失败,使用纳秒时间戳作为后备
randomBytes = []byte(fmt.Sprintf("%d", time.Now().UnixNano()))
}
randomStr := hex.EncodeToString(randomBytes)
return fmt.Sprintf("%s-%s", reportNumber, randomStr)
}
// generateDownloadURL 生成下载链接基于报告ID/缓存键)

View File

@@ -36,25 +36,18 @@ func NewPDFGHandler(
// DownloadPDF 下载PDF文件
// GET /api/v1/pdfg/download?id=报告ID
func (h *PDFGHandler) DownloadPDF(c *gin.Context) {
rawReportID := c.Query("id")
reportID := c.Query("id")
if rawReportID == "" {
if reportID == "" {
h.responseBuilder.BadRequest(c, "报告ID不能为空")
return
}
// 从对外报告ID中解析内部缓存键(使用最后一个'-'后的部分作为缓存键)
cacheKey := rawReportID
if idx := strings.LastIndex(rawReportID, "-"); idx > 0 && idx < len(rawReportID)-1 {
cacheKey = rawReportID[idx+1:]
}
// 从缓存获取PDF
pdfBytes, hit, createdAt, err := h.cacheManager.GetByCacheKey(cacheKey)
// 通过报告ID获取PDF文件
pdfBytes, hit, createdAt, err := h.cacheManager.GetByReportID(reportID)
if err != nil {
h.logger.Error("获取PDF缓存失败",
zap.String("report_id", rawReportID),
zap.String("cache_key", cacheKey),
zap.String("report_id", reportID),
zap.Error(err),
)
h.responseBuilder.InternalError(c, "获取PDF文件失败")
@@ -63,8 +56,7 @@ func (h *PDFGHandler) DownloadPDF(c *gin.Context) {
if !hit {
h.logger.Warn("PDF文件不存在或已过期",
zap.String("report_id", rawReportID),
zap.String("cache_key", cacheKey),
zap.String("report_id", reportID),
)
h.responseBuilder.NotFound(c, "PDF文件不存在或已过期请重新生成")
return
@@ -74,8 +66,7 @@ func (h *PDFGHandler) DownloadPDF(c *gin.Context) {
expiresAt := createdAt.Add(24 * time.Hour)
if time.Now().After(expiresAt) {
h.logger.Warn("PDF文件已过期",
zap.String("report_id", rawReportID),
zap.String("cache_key", cacheKey),
zap.String("report_id", reportID),
zap.Time("expires_at", expiresAt),
)
h.responseBuilder.NotFound(c, "PDF文件已过期请重新生成")
@@ -84,11 +75,11 @@ func (h *PDFGHandler) DownloadPDF(c *gin.Context) {
// 设置响应头
c.Header("Content-Type", "application/pdf")
// 使用报告ID前缀作为下载文件名的一部分,避免泄露内部缓存键
// 使用报告ID前缀作为下载文件名的一部分
filename := "大数据租赁风险报告.pdf"
if idx := strings.LastIndex(rawReportID, "-"); idx > 0 {
if idx := strings.LastIndex(reportID, "-"); idx > 0 {
// 使用前缀(报告编号部分)作为文件名的一部分
prefix := rawReportID[:idx]
prefix := reportID[:idx]
filename = fmt.Sprintf("大数据租赁风险报告_%s.pdf", prefix)
}
c.Header("Content-Disposition", fmt.Sprintf(`attachment; filename="%s"`, filename))
@@ -98,8 +89,7 @@ func (h *PDFGHandler) DownloadPDF(c *gin.Context) {
c.Data(http.StatusOK, "application/pdf", pdfBytes)
h.logger.Info("PDF文件下载成功",
zap.String("report_id", rawReportID),
zap.String("cache_key", cacheKey),
zap.String("report_id", reportID),
zap.Int("file_size", len(pdfBytes)),
)
}

View File

@@ -78,6 +78,14 @@ func (m *PDFCacheManager) GetCacheKeyByProduct(productID, version string) string
return hex.EncodeToString(hash[:])
}
// GetCacheKeyByReportID 生成缓存键基于报告ID
// 文件名格式MD5(report_id).pdf
// report_id 本身已经包含时间戳和随机数,所以 MD5 后就是唯一的
func (m *PDFCacheManager) GetCacheKeyByReportID(reportID string) string {
hash := md5.Sum([]byte(reportID))
return hex.EncodeToString(hash[:])
}
// GetCachePath 获取缓存文件路径
func (m *PDFCacheManager) GetCachePath(cacheKey string) string {
return filepath.Join(m.cacheDir, fmt.Sprintf("%s.pdf", cacheKey))
@@ -105,6 +113,20 @@ func (m *PDFCacheManager) GetByCacheKey(cacheKey string) ([]byte, bool, time.Tim
return m.getByKey(cacheKey, "", "")
}
// SetByReportID 将PDF文件保存到缓存基于报告ID
func (m *PDFCacheManager) SetByReportID(reportID string, pdfBytes []byte) error {
cacheKey := m.GetCacheKeyByReportID(reportID)
return m.setByKey(cacheKey, pdfBytes, reportID, "")
}
// GetByReportID 从缓存获取PDF文件基于报告ID
// 直接通过 report_id 的 MD5 计算文件名,无需遍历
// 返回PDF字节流、是否命中缓存、文件创建时间、错误
func (m *PDFCacheManager) GetByReportID(reportID string) ([]byte, bool, time.Time, error) {
cacheKey := m.GetCacheKeyByReportID(reportID)
return m.getByKey(cacheKey, reportID, "")
}
// getByKey 内部方法:根据缓存键获取文件
func (m *PDFCacheManager) getByKey(cacheKey string, key1, key2 string) ([]byte, bool, time.Time, error) {
cachePath := m.GetCachePath(cacheKey)