-
Notifications
You must be signed in to change notification settings - Fork 2
/
task.go
116 lines (90 loc) · 3.44 KB
/
task.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
package service
import (
"context"
"github.com/limes-cloud/kratosx"
"github.com/limes-cloud/kratosx/pkg/util"
"google.golang.org/protobuf/types/known/emptypb"
"github.com/limes-cloud/cron/api/errors"
v1 "github.com/limes-cloud/cron/api/server/v1"
"github.com/limes-cloud/cron/internal/server/biz"
)
func (s *Service) AllTaskGroup(ctx context.Context, _ *emptypb.Empty) (*v1.AllTaskGroupReply, error) {
list, err := s.task.AllTaskGroup(kratosx.MustContext(ctx))
if err != nil {
return nil, err
}
reply := v1.AllTaskGroupReply{}
if err := util.Transform(list, &reply.List); err != nil {
return nil, errors.Transform()
}
return &reply, nil
}
func (s *Service) AddTaskGroup(ctx context.Context, in *v1.AddTaskGroupRequest) (*v1.AddTaskGroupReply, error) {
wk := biz.TaskGroup{}
if err := util.Transform(in, &wk); err != nil {
return nil, errors.TransformFormat(err.Error())
}
id, err := s.task.AddTaskGroup(kratosx.MustContext(ctx), &wk)
if err != nil {
return nil, err
}
return &v1.AddTaskGroupReply{Id: id}, nil
}
func (s *Service) UpdateTaskGroup(ctx context.Context, in *v1.UpdateTaskGroupRequest) (*emptypb.Empty, error) {
wk := biz.TaskGroup{}
if err := util.Transform(in, &wk); err != nil {
return nil, errors.TransformFormat(err.Error())
}
return nil, s.task.UpdateTaskGroup(kratosx.MustContext(ctx), &wk)
}
func (s *Service) DeleteTaskGroup(ctx context.Context, in *v1.DeleteTaskGroupRequest) (*emptypb.Empty, error) {
return nil, s.task.DeleteTaskGroup(kratosx.MustContext(ctx), in.Id)
}
func (s *Service) PageTask(ctx context.Context, in *v1.PageTaskRequest) (*v1.PageTaskReply, error) {
var req biz.PageTaskRequest
if err := util.Transform(in, &req); err != nil {
return nil, errors.Transform()
}
list, total, err := s.task.PageTask(kratosx.MustContext(ctx), &req)
if err != nil {
return nil, err
}
reply := v1.PageTaskReply{Total: total}
if err := util.Transform(list, &reply.List); err != nil {
return nil, errors.Transform()
}
return &reply, nil
}
func (s *Service) AddTask(ctx context.Context, in *v1.AddTaskRequest) (*v1.AddTaskReply, error) {
wk := biz.Task{}
if err := util.Transform(in, &wk); err != nil {
return nil, errors.TransformFormat(err.Error())
}
id, err := s.task.AddTask(kratosx.MustContext(ctx), &wk)
if err != nil {
return nil, err
}
return &v1.AddTaskReply{Id: id}, nil
}
func (s *Service) UpdateTask(ctx context.Context, in *v1.UpdateTaskRequest) (*emptypb.Empty, error) {
wk := biz.Task{}
if err := util.Transform(in, &wk); err != nil {
return nil, errors.TransformFormat(err.Error())
}
return nil, s.task.UpdateTask(kratosx.MustContext(ctx), &wk)
}
func (s *Service) EnableTask(ctx context.Context, in *v1.EnableTaskRequest) (*emptypb.Empty, error) {
return nil, s.task.EnableTask(kratosx.MustContext(ctx), in.Id)
}
func (s *Service) DisableTask(ctx context.Context, in *v1.DisableTaskRequest) (*emptypb.Empty, error) {
return nil, s.task.DisableTask(kratosx.MustContext(ctx), in.Id)
}
func (s *Service) CancelExecTask(ctx context.Context, in *v1.CancelExecTaskRequest) (*emptypb.Empty, error) {
return nil, s.task.CancelTask(kratosx.MustContext(ctx), in.Uuid)
}
func (s *Service) ExecTask(ctx context.Context, in *v1.ExecTaskRequest) (*emptypb.Empty, error) {
return nil, s.task.ExecTask(kratosx.MustContext(ctx), in.Id)
}
func (s *Service) DeleteTask(ctx context.Context, in *v1.DeleteTaskRequest) (*emptypb.Empty, error) {
return nil, s.task.DeleteTask(kratosx.MustContext(ctx), in.Id)
}