-
Notifications
You must be signed in to change notification settings - Fork 6
/
jobs.go
56 lines (48 loc) · 1.23 KB
/
jobs.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
/*
Copyright 2021 Adevinta
*/
package store
import (
"strings"
"github.com/adevinta/errors"
"github.com/adevinta/vulcan-api/pkg/api"
"github.com/jinzhu/gorm"
)
// FindJob retrieves a Job by its ID
func (db vulcanitoStore) FindJob(jobID string) (*api.Job, error) {
if jobID == "" {
return nil, db.logError(errors.Validation(`ID is empty`))
}
job := &api.Job{ID: jobID}
res := db.Conn.Find(job)
if res.Error != nil {
if strings.HasPrefix(res.Error.Error(), `pq: invalid input syntax for type uuid`) {
return nil, db.logError(errors.Validation(`ID is malformed`))
}
if !db.NotFoundError(res.Error) {
return nil, db.logError(errors.Database(res.Error))
}
}
if res.RowsAffected == 0 {
return nil, db.logError(errors.NotFound("Job does not exist"))
}
return job, nil
}
func (db vulcanitoStore) createJobTx(tx *gorm.DB, job api.Job) (*api.Job, error) {
res := tx.Create(&job)
err := res.Error
if err != nil {
return nil, db.logError(errors.Create(err))
}
tx.First(&job)
return &job, nil
}
func (db vulcanitoStore) UpdateJob(job api.Job) (*api.Job, error) {
res := db.Conn.Model(&job).Update(&job)
err := res.Error
if err != nil {
return nil, db.logError(errors.Update(err))
}
db.Conn.First(&job)
return &job, nil
}