forked from content-services/content-sources-backend
-
Notifications
You must be signed in to change notification settings - Fork 0
/
task_info.go
87 lines (71 loc) · 2.37 KB
/
task_info.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
package dao
import (
"time"
"github.com/content-services/content-sources-backend/pkg/api"
ce "github.com/content-services/content-sources-backend/pkg/errors"
"github.com/content-services/content-sources-backend/pkg/models"
"gorm.io/gorm"
)
type taskInfoDaoImpl struct {
db *gorm.DB
}
func GetTaskInfoDao(db *gorm.DB) TaskInfoDao {
return taskInfoDaoImpl{
db: db,
}
}
func (t taskInfoDaoImpl) Fetch(orgId string, id string) (api.TaskInfoResponse, error) {
taskInfo := models.TaskInfo{}
result := t.db.Where("id = ? AND org_id = ?", id, orgId).First(&taskInfo)
taskInfoResponse := api.TaskInfoResponse{}
if result.Error != nil {
if result.Error == gorm.ErrRecordNotFound {
return taskInfoResponse, &ce.DaoError{NotFound: true, Message: "Could not find task with UUID " + id}
} else {
return taskInfoResponse, result.Error
}
}
taskInfoModelToApiFields(&taskInfo, &taskInfoResponse)
return taskInfoResponse, nil
}
func (t taskInfoDaoImpl) List(
orgId string,
pageData api.PaginationData,
statusFilter string,
) (api.TaskInfoCollectionResponse, int64, error) {
var totalTasks int64
tasks := make([]models.TaskInfo, 0)
filteredDB := t.db.Where("org_id = ?", orgId)
if statusFilter != "" {
filteredDB = filteredDB.Where("status = ?", statusFilter)
}
filteredDB.Find(&tasks).Count(&totalTasks)
// Most recently queued (created) first
filteredDB.Order("queued_at DESC").Offset(pageData.Offset).Limit(pageData.Limit).Find(&tasks)
if filteredDB.Error != nil {
return api.TaskInfoCollectionResponse{}, totalTasks, filteredDB.Error
}
taskResponses := convertTaskInfoToResponses(tasks)
return api.TaskInfoCollectionResponse{Data: taskResponses}, totalTasks, nil
}
func taskInfoModelToApiFields(taskInfo *models.TaskInfo, apiTaskInfo *api.TaskInfoResponse) {
apiTaskInfo.UUID = taskInfo.Id.String()
apiTaskInfo.OrgId = taskInfo.OrgId
apiTaskInfo.Status = taskInfo.Status
if taskInfo.Error != nil {
apiTaskInfo.Error = *taskInfo.Error
}
if taskInfo.Queued != nil {
apiTaskInfo.CreatedAt = taskInfo.Queued.Format(time.RFC3339)
}
if taskInfo.Finished != nil {
apiTaskInfo.EndedAt = taskInfo.Finished.Format(time.RFC3339)
}
}
func convertTaskInfoToResponses(taskInfo []models.TaskInfo) []api.TaskInfoResponse {
tasks := make([]api.TaskInfoResponse, len(taskInfo))
for i := range taskInfo {
taskInfoModelToApiFields(&taskInfo[i], &tasks[i])
}
return tasks
}