Skip to content

Commit 03dde17

Browse files
committed
Add PkgFetchMulti and PkgQuerySQL helpers to pkg DB layer
1 parent d5ab1b4 commit 03dde17

File tree

1 file changed

+25
-8
lines changed

1 file changed

+25
-8
lines changed

db/pkg.go

Lines changed: 25 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -20,18 +20,35 @@ func PkgFetch(pkg_id uuid.UUID) (*Pkg, error) {
2020
return &pkg, nil
2121
}
2222

23+
func PkgFetchMulti(pkg_ids []uuid.UUID) ([]Pkg, error) {
24+
var pkgs []Pkg
25+
if err := DB.Where("id IN ?", pkg_ids).Find(&pkgs).Error; err != nil {
26+
return nil, err
27+
}
28+
return pkgs, nil
29+
}
30+
31+
// GORM helper for querying packages with arbitrary conditions
32+
func PkgQuerySQL(query interface{}, args ...interface{}) ([]Pkg, error) {
33+
var pkgs []Pkg
34+
if err := DB.Where(query, args...).Find(&pkgs).Error; err != nil {
35+
return nil, err
36+
}
37+
return pkgs, nil
38+
}
39+
2340
func SubmitPkg(pkg *Pkg) (*Pkg, error) {
24-
if err := DB.Create(pkg).Error; err != nil {
25-
return nil, err
26-
}
27-
return pkg, nil
41+
if err := DB.Create(pkg).Error; err != nil {
42+
return nil, err
43+
}
44+
return pkg, nil
2845
}
2946

3047
func (p *Pkg) Update() error {
31-
if err := DB.Save(p).Error; err != nil {
32-
return err
33-
}
34-
return nil
48+
if err := DB.Save(p).Error; err != nil {
49+
return err
50+
}
51+
return nil
3552
}
3653

3754
func (p *Pkg) Delete() error {

0 commit comments

Comments
 (0)