Repository navigation
Expand file tree
/
Copy pathgorm_heartbeat_repository.go
More file actions
112 lines (90 loc) · 3.59 KB
/
Copy pathgorm_heartbeat_repository.go
File metadata and controls
112 lines (90 loc) · 3.59 KB
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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
package repositories
import (
"context"
"fmt"
"github.com/pkg/errors"
"github.com/NdoleStudio/httpsms/pkg/entities"
"github.com/NdoleStudio/httpsms/pkg/telemetry"
"github.com/palantir/stacktrace"
"gorm.io/gorm"
)
// gormHeartbeatRepository is responsible for persisting entities.Heartbeat
type gormHeartbeatRepository struct {
logger telemetry.Logger
tracer telemetry.Tracer
db *gorm.DB
}
// NewGormHeartbeatRepository creates the GORM version of the HeartbeatRepository
func NewGormHeartbeatRepository(
logger telemetry.Logger,
tracer telemetry.Tracer,
db *gorm.DB,
) HeartbeatRepository {
return &gormHeartbeatRepository{
logger: logger.WithService(fmt.Sprintf("%T", &gormHeartbeatRepository{})),
tracer: tracer,
db: db,
}
}
func (repository *gormHeartbeatRepository) DeleteAllForUser(ctx context.Context, userID entities.UserID) error {
ctx, span := repository.tracer.Start(ctx)
defer span.End()
err := repository.db.WithContext(ctx).Where("user_id = ?", userID).Delete(&entities.Heartbeat{}).Error
if err != nil {
msg := fmt.Sprintf("cannot delete all [%T] for user with ID [%s]", &entities.Heartbeat{}, userID)
return repository.tracer.WrapErrorSpan(span, stacktrace.Propagate(err, msg))
}
return nil
}
func (repository *gormHeartbeatRepository) Last(ctx context.Context, userID entities.UserID, owner string) (*entities.Heartbeat, error) {
ctx, span := repository.tracer.Start(ctx)
defer span.End()
ctx, cancel := context.WithTimeout(ctx, dbOperationDuration)
defer cancel()
heartbeat := new(entities.Heartbeat)
err := repository.db.WithContext(ctx).
Where("user_id = ?", userID).
Where("owner = ?", owner).
Order("timestamp DESC").
First(&heartbeat).Error
if errors.Is(err, gorm.ErrRecordNotFound) {
msg := fmt.Sprintf("heartbeat with userID [%s] and owner [%s] does not exist", userID, owner)
return nil, repository.tracer.WrapErrorSpan(span, stacktrace.PropagateWithCode(err, ErrCodeNotFound, msg))
}
if err != nil {
msg := fmt.Sprintf("cannot load heartbeat with userID [%s] and owner [%s]", userID, owner)
return nil, repository.tracer.WrapErrorSpan(span, stacktrace.Propagate(err, msg))
}
return heartbeat, nil
}
// Index entities.Message between 2 parties
func (repository *gormHeartbeatRepository) Index(ctx context.Context, userID entities.UserID, owner string, params IndexParams) (*[]entities.Heartbeat, error) {
ctx, span := repository.tracer.Start(ctx)
defer span.End()
ctx, cancel := context.WithTimeout(ctx, dbOperationDuration)
defer cancel()
query := repository.db.WithContext(ctx).Where("user_id = ?", userID).Where("owner = ?", owner)
if len(params.Query) > 0 {
queryPattern := "%" + params.Query + "%"
query.Where("version LIKE ?", queryPattern)
}
heartbeats := new([]entities.Heartbeat)
err := query.Order("timestamp DESC").Limit(params.Limit).Offset(params.Skip).Find(&heartbeats).Error
if err != nil {
msg := fmt.Sprintf("cannot fetch heartbeats with owner [%s] and params [%+#v]", owner, params)
return nil, repository.tracer.WrapErrorSpan(span, stacktrace.Propagate(err, msg))
}
return heartbeats, nil
}
// Store a new entities.Message
func (repository *gormHeartbeatRepository) Store(ctx context.Context, heartbeat *entities.Heartbeat) error {
ctx, span := repository.tracer.Start(ctx)
defer span.End()
ctx, cancel := context.WithTimeout(ctx, dbOperationDuration)
defer cancel()
if err := repository.db.WithContext(ctx).Create(heartbeat).Error; err != nil {
msg := fmt.Sprintf("cannot save heartbeat with ID [%s]", heartbeat.ID)
return repository.tracer.WrapErrorSpan(span, stacktrace.Propagate(err, msg))
}
return nil
}