-
Notifications
You must be signed in to change notification settings - Fork 3
/
keys.go
43 lines (34 loc) · 834 Bytes
/
keys.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
package repository
import (
"database/sql"
"errors"
"github.com/labstack/gommon/log"
"github.com/lalizita/streaming-key-server-manager/internal/model"
)
var (
QueryErr = errors.New("Error query key")
)
type KeysRepository interface {
FindStreamKey(name, key string) (*model.Keys, error)
}
type keysRepository struct {
*sql.DB
}
func NewKeysRepository(db *sql.DB) KeysRepository {
return &keysRepository{
db,
}
}
func (r *keysRepository) FindStreamKey(name, key string) (*model.Keys, error) {
keys := &model.Keys{}
row := r.QueryRow(`SELECT * FROM "Lives" WHERE "name"=$1 AND "stream_key"=$2`, name, key)
err := row.Scan(&keys.Name, &keys.Key)
if err != nil {
log.Error(err.Error())
if errors.Is(err, sql.ErrNoRows) {
return &model.Keys{}, nil
}
return &model.Keys{}, QueryErr
}
return keys, nil
}