33 lines
739 B
Go
33 lines
739 B
Go
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
|
|
}
|