-
Notifications
You must be signed in to change notification settings - Fork 8
/
UserModel.go
41 lines (34 loc) 路 1.04 KB
/
UserModel.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
package pg
import (
"github.com/quan-to/chevron/pkg/models"
)
// AddUser adds a user in the database if not exists
func (h *PostgreSQLDBDriver) AddUser(um models.User) (string, error) {
h.log.Debug("AddUser(%s)", um.Username)
tx, err := h.conn.Beginx()
if err != nil {
return "", err
}
defer func() { h.rollbackIfErrorCommitIfNot(err, tx) }()
return h.addUser(tx, um)
}
// GetUser fetchs a user from the database by it's username
func (h *PostgreSQLDBDriver) GetUser(username string) (um *models.User, err error) {
h.log.Debug("GetUser(%s)", username)
tx, err := h.conn.Beginx()
if err != nil {
return nil, err
}
defer func() { h.rollbackIfErrorCommitIfNot(err, tx) }()
return h.getUser(tx, username)
}
// UpdateUser updates user fingerprint, password and / or fullname by it's ID
func (h *PostgreSQLDBDriver) UpdateUser(um models.User) error {
h.log.Debug("UpdateUser(%s)", um.Username)
tx, err := h.conn.Beginx()
if err != nil {
return err
}
defer func() { h.rollbackIfErrorCommitIfNot(err, tx) }()
return h.updateUser(tx, um)
}