forked from NdoleStudio/httpsms
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathheartbeat_listener.go
More file actions
151 lines (123 loc) · 5.49 KB
/
Copy pathheartbeat_listener.go
File metadata and controls
151 lines (123 loc) · 5.49 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
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
package listeners
import (
"context"
"fmt"
cloudevents "github.com/cloudevents/sdk-go/v2"
"github.com/davecgh/go-spew/spew"
"github.com/palantir/stacktrace"
"github.com/NdoleStudio/httpsms/pkg/events"
"github.com/NdoleStudio/httpsms/pkg/services"
"github.com/NdoleStudio/httpsms/pkg/telemetry"
)
// HeartbeatListener handles cloud events which need to register entities.Heartbeat
type HeartbeatListener struct {
logger telemetry.Logger
tracer telemetry.Tracer
service *services.HeartbeatService
}
// NewHeartbeatListener creates a new instance of HeartbeatListener
func NewHeartbeatListener(
logger telemetry.Logger,
tracer telemetry.Tracer,
service *services.HeartbeatService,
) (l *HeartbeatListener, routes map[string]events.EventListener) {
l = &HeartbeatListener{
logger: logger.WithService(fmt.Sprintf("%T", l)),
tracer: tracer,
service: service,
}
return l, map[string]events.EventListener{
events.EventTypePhoneUpdated: l.onPhoneUpdated,
events.EventTypePhoneDeleted: l.onPhoneDeleted,
events.EventTypePhoneHeartbeatCheck: l.onPhoneHeartbeatCheck,
events.EventTypePhoneHeartbeatOffline: l.onPhoneHeartbeatOffline,
events.UserAccountDeleted: l.onUserAccountDeleted,
}
}
// onPhoneUpdated handles the events.EventTypePhoneUpdated event
func (listener *HeartbeatListener) onPhoneUpdated(ctx context.Context, event cloudevents.Event) error {
ctx, span := listener.tracer.Start(ctx)
defer span.End()
var payload events.PhoneUpdatedPayload
if err := event.DataAs(&payload); err != nil {
msg := fmt.Sprintf("cannot decode [%s] into [%T]", event.Data(), payload)
return listener.tracer.WrapErrorSpan(span, stacktrace.Propagate(err, msg))
}
storeParams := &services.HeartbeatMonitorStoreParams{
Owner: payload.Owner,
PhoneID: payload.PhoneID,
UserID: payload.UserID,
Source: event.Source(),
}
if _, err := listener.service.StoreMonitor(ctx, storeParams); err != nil {
msg := fmt.Sprintf("cannot store heartbeat monitor with params [%s] for event with ID [%s]", spew.Sdump(storeParams), event.ID())
return listener.tracer.WrapErrorSpan(span, stacktrace.Propagate(err, msg))
}
return nil
}
// onPhoneDeleted handles the events.EventTypePhoneDeleted event
func (listener *HeartbeatListener) onPhoneDeleted(ctx context.Context, event cloudevents.Event) error {
ctx, span := listener.tracer.Start(ctx)
defer span.End()
var payload events.PhoneDeletedPayload
if err := event.DataAs(&payload); err != nil {
msg := fmt.Sprintf("cannot decode [%s] into [%T]", event.Data(), payload)
return listener.tracer.WrapErrorSpan(span, stacktrace.Propagate(err, msg))
}
if err := listener.service.DeleteMonitor(ctx, payload.UserID, payload.Owner); err != nil {
msg := fmt.Sprintf("cannot delete heartbeat monitor with userID [%s] and owner [%s] for event with ID [%s]", payload.UserID, payload.Owner, event.ID())
return listener.tracer.WrapErrorSpan(span, stacktrace.Propagate(err, msg))
}
return nil
}
// onPhoneHeartbeatCheck handles the events.EventTypePhoneHeartbeatCheck event
func (listener *HeartbeatListener) onPhoneHeartbeatCheck(ctx context.Context, event cloudevents.Event) error {
ctx, span := listener.tracer.Start(ctx)
defer span.End()
var payload events.PhoneHeartbeatCheckPayload
if err := event.DataAs(&payload); err != nil {
msg := fmt.Sprintf("cannot decode [%s] into [%T]", event.Data(), payload)
return listener.tracer.WrapErrorSpan(span, stacktrace.Propagate(err, msg))
}
monitorParams := &services.HeartbeatMonitorParams{
Owner: payload.Owner,
PhoneID: payload.PhoneID,
MonitorID: payload.MonitorID,
UserID: payload.UserID,
Source: event.Source(),
}
if err := listener.service.Monitor(ctx, monitorParams); err != nil {
msg := fmt.Sprintf("cannot monitor heartbeats for userID [%s] and owner [%s] for event with ID [%s]", payload.UserID, payload.Owner, event.ID())
return listener.tracer.WrapErrorSpan(span, stacktrace.Propagate(err, msg))
}
return nil
}
// onPhoneDeleted handles the events.EventTypePhoneDeleted event
func (listener *HeartbeatListener) onPhoneHeartbeatOffline(ctx context.Context, event cloudevents.Event) error {
ctx, span := listener.tracer.Start(ctx)
defer span.End()
var payload events.PhoneHeartbeatOfflinePayload
if err := event.DataAs(&payload); err != nil {
msg := fmt.Sprintf("cannot decode [%s] into [%T]", event.Data(), payload)
return listener.tracer.WrapErrorSpan(span, stacktrace.Propagate(err, msg))
}
if err := listener.service.UpdatePhoneOnline(ctx, payload.UserID, payload.MonitorID, false); err != nil {
msg := fmt.Sprintf("cannot delete heartbeat monitor with userID [%s] and owner [%s] for event with ID [%s]", payload.UserID, payload.Owner, event.ID())
return listener.tracer.WrapErrorSpan(span, stacktrace.Propagate(err, msg))
}
return nil
}
func (listener *HeartbeatListener) onUserAccountDeleted(ctx context.Context, event cloudevents.Event) error {
ctx, span := listener.tracer.Start(ctx)
defer span.End()
var payload events.UserAccountDeletedPayload
if err := event.DataAs(&payload); err != nil {
msg := fmt.Sprintf("cannot decode [%s] into [%T]", event.Data(), payload)
return listener.tracer.WrapErrorSpan(span, stacktrace.Propagate(err, msg))
}
if err := listener.service.DeleteAllForUser(ctx, payload.UserID); err != nil {
msg := fmt.Sprintf("cannot delete [entities.Heartbeat] for user [%s] on [%s] event with ID [%s]", payload.UserID, event.Type(), event.ID())
return listener.tracer.WrapErrorSpan(span, stacktrace.Propagate(err, msg))
}
return nil
}