yoge
ac078a16bc
- Downgrade go directive in go.mod from 1.23.0 back to 1.20 to match Docker builder image (golang:1.20-alpine); re-run go mod tidy with go1.20 (via gvm) to keep go.sum consistent - GetPDFTask now verifies callerUserID matches task.UserID to prevent cross-user data exposure of PDF page content Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
96 lines
2.9 KiB
Go
96 lines
2.9 KiB
Go
package pdf
|
|
|
|
import (
|
|
"net/http"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
pdfmodel "gitea.com/texpixel/document_ai/internal/model/pdf"
|
|
"gitea.com/texpixel/document_ai/internal/service"
|
|
"gitea.com/texpixel/document_ai/pkg/common"
|
|
"gitea.com/texpixel/document_ai/pkg/constant"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
type PDFEndpoint struct {
|
|
pdfService *service.PDFRecognitionService
|
|
}
|
|
|
|
func NewPDFEndpoint() *PDFEndpoint {
|
|
return &PDFEndpoint{
|
|
pdfService: service.NewPDFRecognitionService(),
|
|
}
|
|
}
|
|
|
|
// CreateTask godoc
|
|
// @Summary Create a PDF recognition task
|
|
// @Description Create a new PDF recognition task (max 10 pages processed)
|
|
// @Tags PDF
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param request body pdfmodel.CreatePDFRecognitionRequest true "Create PDF task request"
|
|
// @Success 200 {object} common.Response{data=pdfmodel.CreatePDFTaskResponse}
|
|
// @Failure 400 {object} common.Response
|
|
// @Failure 500 {object} common.Response
|
|
// @Router /v1/pdf/recognition [post]
|
|
func (e *PDFEndpoint) CreateTask(c *gin.Context) {
|
|
var req pdfmodel.CreatePDFRecognitionRequest
|
|
if err := c.BindJSON(&req); err != nil {
|
|
c.JSON(http.StatusOK, common.ErrorResponse(c, common.CodeParamError, "参数错误"))
|
|
return
|
|
}
|
|
req.UserID = c.GetInt64(constant.ContextUserID)
|
|
|
|
if strings.ToLower(filepath.Ext(req.FileName)) != ".pdf" {
|
|
c.JSON(http.StatusOK, common.ErrorResponse(c, common.CodeParamError, "仅支持PDF文件"))
|
|
return
|
|
}
|
|
|
|
task, err := e.pdfService.CreatePDFTask(c, &req)
|
|
if err != nil {
|
|
if bizErr, ok := err.(*common.BusinessError); ok {
|
|
c.JSON(http.StatusOK, common.ErrorResponse(c, int(bizErr.Code), bizErr.Message))
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, common.ErrorResponse(c, common.CodeSystemError, "创建任务失败"))
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, common.SuccessResponse(c, &pdfmodel.CreatePDFTaskResponse{
|
|
TaskNo: task.TaskUUID,
|
|
Status: int(task.Status),
|
|
}))
|
|
}
|
|
|
|
// GetTaskStatus godoc
|
|
// @Summary Get PDF recognition task status and results
|
|
// @Description Poll task status; pages field populated when status=2 (completed)
|
|
// @Tags PDF
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param task_no path string true "Task No"
|
|
// @Success 200 {object} common.Response{data=pdfmodel.GetPDFTaskResponse}
|
|
// @Failure 404 {object} common.Response
|
|
// @Failure 500 {object} common.Response
|
|
// @Router /v1/pdf/recognition/{task_no} [get]
|
|
func (e *PDFEndpoint) GetTaskStatus(c *gin.Context) {
|
|
var req pdfmodel.GetPDFTaskRequest
|
|
if err := c.ShouldBindUri(&req); err != nil {
|
|
c.JSON(http.StatusOK, common.ErrorResponse(c, common.CodeParamError, "参数错误"))
|
|
return
|
|
}
|
|
|
|
resp, err := e.pdfService.GetPDFTask(c, req.TaskNo, c.GetInt64(constant.ContextUserID))
|
|
if err != nil {
|
|
if bizErr, ok := err.(*common.BusinessError); ok {
|
|
c.JSON(http.StatusOK, common.ErrorResponse(c, int(bizErr.Code), bizErr.Message))
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, common.ErrorResponse(c, common.CodeSystemError, "查询任务失败"))
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, common.SuccessResponse(c, resp))
|
|
}
|