-
Notifications
You must be signed in to change notification settings - Fork 0
/
nutritionists.go
68 lines (58 loc) · 1.68 KB
/
nutritionists.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
package store
import (
"context"
sq "github.com/Masterminds/squirrel"
"github.com/google/uuid"
"github.com/l-orlov/slim-fairy/bot/internal/model"
)
// CreateNutritionist creates model.Nutritionist
func (s *Storage) CreateNutritionist(ctx context.Context, record *model.Nutritionist) error {
query := psql().
Insert(record.DbTable()).
SetMap(nutritionistAttrs(record)).
Suffix("RETURNING " + asteriskNutritionists)
err := Getx(ctx, s.pool, record, query)
if err != nil {
return dbError(err)
}
return nil
}
// UpdateNutritionist updates model.Nutritionist
func (s *Storage) UpdateNutritionist(ctx context.Context, record *model.Nutritionist) error {
query := psql().
Update(record.DbTable()).
Where(sq.Eq{"id": record.ID}).
SetMap(nutritionistAttrs(record)).
Suffix("RETURNING " + asteriskNutritionists)
err := Getx(ctx, s.pool, record, query)
if err != nil {
return dbError(err)
}
return nil
}
// GetNutritionistByID gets model.Nutritionist by id
func (s *Storage) GetNutritionistByID(ctx context.Context, id uuid.UUID) (*model.Nutritionist, error) {
record := &model.Nutritionist{}
query := psql().
Select(asteriskNutritionists).
From(record.DbTable()).
Where(sq.Eq{"id": id}).
Limit(1)
err := Getx(ctx, s.pool, record, query)
if err != nil {
return nil, dbError(err)
}
return record, nil
}
func nutritionistAttrs(record *model.Nutritionist) map[string]interface{} {
return map[string]interface{}{
"name": record.Name,
"email": record.Email,
"phone": record.Phone,
"telegram_id": record.TelegramID,
"age": record.Age,
"gender": record.Gender,
"info": record.Info,
"created_by": record.CreatedBy,
}
}