This repository has been archived by the owner on Jun 27, 2023. It is now read-only.
/
server_pg_repo.go
89 lines (74 loc) · 1.8 KB
/
server_pg_repo.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
package repository
import (
"context"
"github.com/tribalwarshelp/dcbot/model"
"github.com/tribalwarshelp/dcbot/server"
"github.com/go-pg/pg/v10"
"github.com/go-pg/pg/v10/orm"
"github.com/pkg/errors"
)
type PGRepository struct {
*pg.DB
}
var _ server.Repository = &PGRepository{}
func NewPgRepository(db *pg.DB) (*PGRepository, error) {
if err := db.Model((*model.Server)(nil)).CreateTable(&orm.CreateTableOptions{
IfNotExists: true,
}); err != nil {
return nil, errors.Wrap(err, "couldn't create the 'servers' table")
}
return &PGRepository{db}, nil
}
func (repo *PGRepository) Store(ctx context.Context, server *model.Server) error {
if _, err := repo.
Model(server).
Where("id = ?id").
Returning("*").
Context(ctx).
Relation("Groups").
SelectOrInsert(); err != nil {
return err
}
return nil
}
func (repo *PGRepository) Update(ctx context.Context, server *model.Server) error {
if _, err := repo.
Model(server).
WherePK().
Returning("*").
Context(ctx).
Update(); err != nil {
return err
}
return nil
}
func (repo *PGRepository) Fetch(ctx context.Context, f *model.ServerFilter) ([]*model.Server, int, error) {
var err error
var data []*model.Server
query := repo.Model(&data).Context(ctx).Relation("Groups")
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) Delete(ctx context.Context, f *model.ServerFilter) ([]*model.Server, error) {
var data []*model.Server
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
}