This repository has been archived by the owner on Jun 27, 2023. It is now read-only.
/
pg_repository.go
103 lines (86 loc) · 2.3 KB
/
pg_repository.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
package repository
import (
"context"
"github.com/tribalwarshelp/dcbot/model"
"github.com/tribalwarshelp/dcbot/observation"
"github.com/go-pg/pg/v10"
"github.com/go-pg/pg/v10/orm"
"github.com/pkg/errors"
)
type PGRepository struct {
*pg.DB
}
var _ observation.Repository = &PGRepository{}
func NewPgRepository(db *pg.DB) (*PGRepository, error) {
if err := db.Model((*model.Observation)(nil)).CreateTable(&orm.CreateTableOptions{
IfNotExists: true,
FKConstraints: true,
}); err != nil {
return nil, errors.Wrap(err, "couldn't create the 'observations' table")
}
return &PGRepository{db}, nil
}
func (repo *PGRepository) Store(ctx context.Context, observation *model.Observation) error {
if _, err := repo.Model(observation).Returning("*").Context(ctx).Insert(); err != nil {
return err
}
return nil
}
func (repo *PGRepository) StoreMany(ctx context.Context, observations []*model.Observation) error {
if _, err := repo.Model(&observations).Returning("*").Context(ctx).Insert(); err != nil {
return err
}
return nil
}
func (repo *PGRepository) Update(ctx context.Context, observation *model.Observation) error {
if _, err := repo.
Model(observation).
WherePK().
Returning("*").
Context(ctx).
UpdateNotZero(); err != nil {
return err
}
return nil
}
func (repo *PGRepository) Fetch(ctx context.Context, f *model.ObservationFilter) ([]*model.Observation, int, error) {
var err error
var data []*model.Observation
query := repo.Model(&data).Context(ctx)
if f != nil {
query = query.
Apply(f.Apply).
Limit(f.Limit).
Offset(f.Offset)
}
total, err := query.SelectAndCount()
if err != nil && err != pg.ErrNoRows {
return nil, 0, err
}
return data, total, nil
}
func (repo *PGRepository) FetchServers(ctx context.Context) ([]string, error) {
var res []string
err := repo.
Model(&model.Observation{}).
Column("server").
Context(ctx).
Group("server").
Order("server ASC").
Select(&res)
return res, err
}
func (repo *PGRepository) Delete(ctx context.Context, f *model.ObservationFilter) ([]*model.Observation, error) {
var data []*model.Observation
query := repo.Model(&data).Context(ctx)
if f != nil {
query = query.Apply(f.Apply)
}
_, err := query.
Returning("*").
Delete()
if err != nil && err != pg.ErrNoRows {
return nil, err
}
return data, nil
}