generated from broadinstitute/golang-project-template
-
Notifications
You must be signed in to change notification settings - Fork 0
/
builds.go
97 lines (79 loc) · 2.87 KB
/
builds.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
package v1models
// v1models.go contains the type for modeling build entities in sherlocks database
// and methods for interacting with the persistence layer. It should only contain
// logic related to interacting with build entities in sherlock's db
import (
"errors"
"fmt"
"strings"
"time"
"gorm.io/gorm"
)
const duplicateVersionStringErrorCheck string = `duplicate key value violates unique constraint "builds_version_string_key" (SQLSTATE 23505)`
// ErrBuildNotFound is returned when a specific build look up fails
var (
ErrBuildNotFound error = gorm.ErrRecordNotFound
ErrDuplicateVersionString error = errors.New("field version_string for builds must be unique")
// ErrBadCreateRequest is an error type used when a create servie request fails validation checks
ErrBadCreateRequest error = errors.New("error invalid create build request")
)
type buildStore struct {
*gorm.DB
}
// Build is the structure used to represent a build entity in sherlock's db persistence layer
type Build struct {
ID int
VersionString string `gorm:"not null;default:null"`
CommitSha string
BuildURL string
BuiltAt time.Time `gorm:"autoCreateTime"`
CreatedAt time.Time
UpdatedAt time.Time
ServiceID int
Service Service `gorm:"foreignKey:ServiceID;references:ID"`
}
type BuildStore interface {
ListAll() ([]Build, error)
CreateNew(Build) (Build, error)
GetByID(int) (Build, error)
GetByVersionString(string) (Build, error)
}
func NewBuildStore(dbConn *gorm.DB) buildStore {
return buildStore{dbConn}
}
func (db buildStore) ListAll() ([]Build, error) {
builds := make([]Build, 0)
if err := db.Preload("Service").Find(&builds).Error; err != nil {
return []Build{}, fmt.Errorf("error listing builds: %v", err)
}
return builds, nil
}
func (db buildStore) CreateNew(newBuild Build) (Build, error) {
err := db.Create(&newBuild).Error
if err != nil {
// check for error due to duplicate VersionString field
if strings.Contains(err.Error(), duplicateVersionStringErrorCheck) {
return Build{}, ErrDuplicateVersionString
}
return Build{}, ErrBadCreateRequest
}
// retrieve the same build record back from DB so it can be returned with associations updated
// gorm will not update the associations in the input struct when performing create operations,
// even though those associations will be modeled properly in the db
err = db.Preload("Service").First(&newBuild, newBuild.ID).Error
return newBuild, err
}
func (db buildStore) GetByID(id int) (Build, error) {
build := Build{}
if err := db.Preload("Service").First(&build, id).Error; err != nil {
return Build{}, err
}
return build, nil
}
func (db buildStore) GetByVersionString(versionString string) (Build, error) {
build := Build{}
if err := db.Preload("Service").First(&build, &Build{VersionString: versionString}).Error; err != nil {
return Build{}, err
}
return build, nil
}