YogeLiu/doc_ai_backed
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: query by task-no

Browse Source
This commit is contained in:
yoge
2025-12-26 16:28:49 +08:00
parent 930d782f18
commit d0c0d2cbc3
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
internal/model/task/request.go
View File

@@ -36,6 +36,6 @@ type TaskListResponse struct {
} }
type ExportTaskRequest struct { type ExportTaskRequest struct {
TaskID int64 `json:"task_id" binding:"required"` TaskNo string `json:"task_no" binding:"required"`
Type string `json:"type" binding:"required,oneof=pdf docx"` Type string `json:"type" binding:"required,oneof=pdf docx"`
} }

4
internal/service/task.go
View File

@@ -118,7 +118,7 @@ func (svc *TaskService) GetTaskList(ctx context.Context, req *task.TaskListReque
} }
func (svc *TaskService) ExportTask(ctx context.Context, req *task.ExportTaskRequest) ([]byte, string, error) { func (svc *TaskService) ExportTask(ctx context.Context, req *task.ExportTaskRequest) ([]byte, string, error) {
recognitionTask, err := svc.recognitionTaskDao.GetTaskByID(dao.DB.WithContext(ctx), req.TaskID) recognitionTask, err := svc.recognitionTaskDao.GetByTaskNo(dao.DB.WithContext(ctx), req.TaskNo)
if err != nil { if err != nil {
log.Error(ctx, "func", "ExportTask", "msg", "get task by task id failed", "error", err) log.Error(ctx, "func", "ExportTask", "msg", "get task by task id failed", "error", err)
return nil, "", err return nil, "", err
@@ -134,7 +134,7 @@ func (svc *TaskService) ExportTask(ctx context.Context, req *task.ExportTaskRequ
return nil, "", errors.New("task not finished") return nil, "", errors.New("task not finished")
} }
recognitionResult, err := svc.recognitionResultDao.GetByTaskID(dao.DB.WithContext(ctx), req.TaskID) recognitionResult, err := svc.recognitionResultDao.GetByTaskID(dao.DB.WithContext(ctx), recognitionTask.ID)
if err != nil { if err != nil {
log.Error(ctx, "func", "ExportTask", "msg", "get recognition result by task id failed", "error", err) log.Error(ctx, "func", "ExportTask", "msg", "get recognition result by task id failed", "error", err)
return nil, "", err return nil, "", err