f

internal/infrastructure/http/handlers/pdfg_handler.go

@@ -0,0 +1,93 @@
+package handlers
+
+import (
+	"fmt"
+	"net/http"
+	"time"
+
+	"github.com/gin-gonic/gin"
+	"go.uber.org/zap"
+
+	"tyapi-server/internal/shared/interfaces"
+	"tyapi-server/internal/shared/pdf"
+)
+
+// PDFGHandler PDFG处理器
+type PDFGHandler struct {
+	cacheManager   *pdf.PDFCacheManager
+	responseBuilder interfaces.ResponseBuilder
+	logger         *zap.Logger
+}
+
+// NewPDFGHandler 创建PDFG处理器
+func NewPDFGHandler(
+	cacheManager *pdf.PDFCacheManager,
+	responseBuilder interfaces.ResponseBuilder,
+	logger *zap.Logger,
+) *PDFGHandler {
+	return &PDFGHandler{
+		cacheManager:    cacheManager,
+		responseBuilder: responseBuilder,
+		logger:          logger,
+	}
+}
+
+// DownloadPDF 下载PDF文件
+// GET /api/v1/pdfg/download?name=xxx&id_card=xxx
+func (h *PDFGHandler) DownloadPDF(c *gin.Context) {
+	name := c.Query("name")
+	idCard := c.Query("id_card")
+
+	if name == "" || idCard == "" {
+		h.responseBuilder.BadRequest(c, "姓名和身份证号不能为空")
+		return
+	}
+
+	// 从缓存获取PDF
+	pdfBytes, hit, createdAt, err := h.cacheManager.Get(name, idCard)
+	if err != nil {
+		h.logger.Error("获取PDF缓存失败",
+			zap.String("name", name),
+			zap.String("id_card", idCard),
+			zap.Error(err),
+		)
+		h.responseBuilder.InternalError(c, "获取PDF文件失败")
+		return
+	}
+
+	if !hit {
+		h.logger.Warn("PDF文件不存在或已过期",
+			zap.String("name", name),
+			zap.String("id_card", idCard),
+		)
+		h.responseBuilder.NotFound(c, "PDF文件不存在或已过期，请重新生成")
+		return
+	}
+
+	// 检查是否过期（从文件生成时间开始算24小时）
+	expiresAt := createdAt.Add(24 * time.Hour)
+	if time.Now().After(expiresAt) {
+		h.logger.Warn("PDF文件已过期",
+			zap.String("name", name),
+			zap.String("id_card", idCard),
+			zap.Time("expires_at", expiresAt),
+		)
+		h.responseBuilder.NotFound(c, "PDF文件已过期，请重新生成")
+		return
+	}
+
+	// 设置响应头
+	c.Header("Content-Type", "application/pdf")
+	c.Header("Content-Disposition", `attachment; filename="大数据租赁风险报告.pdf"`)
+	c.Header("Content-Length", fmt.Sprintf("%d", len(pdfBytes)))
+
+	// 发送PDF文件
+	c.Data(http.StatusOK, "application/pdf", pdfBytes)
+
+	h.logger.Info("PDF文件下载成功",
+		zap.String("name", name),
+		zap.String("id_card", idCard),
+		zap.Int("file_size", len(pdfBytes)),
+	)
+}
+