1

kb/controller/parse_task_controller.go

@@ -0,0 +1,32 @@
+package controller
+
+import (
+	"context"
+
+	"rag-local/kb/model/dto"
+	"rag-local/kb/service"
+)
+
+type parseTask struct{}
+
+var ParseTask = &parseTask{}
+
+func (c *parseTask) List(ctx context.Context, req *dto.ListParseTaskReq) (*dto.ListParseTaskRes, error) {
+	list, total, err := service.ParseTaskService.List(ctx, req.Page, req.PageSize)
+	if err != nil {
+		return nil, err
+	}
+	return &dto.ListParseTaskRes{
+		List:     list,
+		Total:    total,
+		Page:     req.Page,
+		PageSize: req.PageSize,
+	}, nil
+}
+
+func (c *parseTask) Retry(ctx context.Context, req *dto.RetryParseTaskReq) (*dto.RetryParseTaskRes, error) {
+	if err := service.ParseTaskService.Retry(ctx, req.Id); err != nil {
+		return nil, err
+	}
+	return &dto.RetryParseTaskRes{}, nil
+}