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
|
package handler
import (
"context"
"github.com/Baitinq/fs-tracer-backend/lib"
"github.com/jmoiron/sqlx"
)
//go:generate mockgen -source=$GOFILE -package=$GOPACKAGE -destination=mock_$GOFILE
type DB interface {
GetLatestFileByPath(ctx context.Context, path string, user_id string) (*lib.File, error)
GetUserIDByAPIKey(ctx context.Context, apiKey string) (string, error)
}
type DBImpl struct {
db *sqlx.DB
}
var _ DB = (*DBImpl)(nil)
func NewDB(db *sqlx.DB) DB {
return &DBImpl{db: db}
}
func (db DBImpl) GetLatestFileByPath(ctx context.Context, path string, user_id string) (*lib.File, error) {
var file lib.File
err := db.db.GetContext(ctx, &file, `
SELECT * FROM public.file
WHERE
user_id = $1
AND absolute_path = $2
ORDER BY timestamp DESC
LIMIT 1
`, user_id, path)
if err != nil {
return nil, err
}
return &file, nil
}
// TODO: Add test
func (db DBImpl) GetUserIDByAPIKey(ctx context.Context, apiKey string) (string, error) {
if len(apiKey) != 44 {
return "", nil
}
var userID string
err := db.db.GetContext(ctx, &userID, `
SELECT user_id FROM public.api_key
WHERE api_key = $1
LIMIT 1
`, apiKey)
if err != nil {
return "", err
}
return userID, nil
}
|
