62 lines
2.0 KiB
Go
62 lines
2.0 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
pb "datarush/pkg/api/task"
|
|
|
|
"google.golang.org/protobuf/types/known/emptypb"
|
|
)
|
|
|
|
type TaskRepository interface {
|
|
CreateTask(ctx context.Context, t *pb.Task) (*pb.Task, error)
|
|
GetTask(ctx context.Context, id string) (*pb.Task, error)
|
|
EditTask(ctx context.Context, t *pb.Task) (*pb.Task, error)
|
|
DeleteTask(ctx context.Context, id string) error
|
|
ListCompetitionTasks(ctx context.Context, competitionID string) ([]*pb.Task, error)
|
|
GetTaskAttachments(ctx context.Context, taskID string, showPrivate bool) ([]*pb.TaskAttachment, error)
|
|
}
|
|
|
|
type TaskService struct {
|
|
repo TaskRepository
|
|
}
|
|
|
|
func NewTaskService(repo TaskRepository) *TaskService {
|
|
return &TaskService{repo: repo}
|
|
}
|
|
|
|
func (s *TaskService) CreateTask(ctx context.Context, req *pb.Task) (*pb.Task, error) {
|
|
return s.repo.CreateTask(ctx, req)
|
|
}
|
|
|
|
func (s *TaskService) GetTask(ctx context.Context, req *pb.GetTaskRequest) (*pb.Task, error) {
|
|
return s.repo.GetTask(ctx, req.TaskId)
|
|
}
|
|
|
|
func (s *TaskService) EditTask(ctx context.Context, req *pb.Task) (*pb.Task, error) {
|
|
return s.repo.EditTask(ctx, req)
|
|
}
|
|
|
|
func (s *TaskService) DeleteTask(ctx context.Context, req *pb.DeleteTaskRequest) (*emptypb.Empty, error) {
|
|
err := s.repo.DeleteTask(ctx, req.TaskId)
|
|
return &emptypb.Empty{}, err
|
|
}
|
|
|
|
func (s *TaskService) ListCompetitionTasks(ctx context.Context, req *pb.ListCompetitionTasksRequest) (*pb.ListCompetitionTasksResponse, error) {
|
|
tasks, err := s.repo.ListCompetitionTasks(ctx, req.CompetitionId)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &pb.ListCompetitionTasksResponse{Tasks: tasks}, nil
|
|
}
|
|
|
|
func (s *TaskService) GetTaskAttachments(ctx context.Context, req *pb.GetTaskAttachmentsRequest) (*pb.GetTaskAttachmentsResponse, error) {
|
|
showPrivate := false
|
|
if req.ShowPrivate != nil {
|
|
showPrivate = *req.ShowPrivate
|
|
}
|
|
attachments, err := s.repo.GetTaskAttachments(ctx, req.TaskId, showPrivate)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &pb.GetTaskAttachmentsResponse{Attachments: attachments}, nil
|
|
} |