forked from NdoleStudio/httpsms
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathbilling_listener.go
More file actions
96 lines (78 loc) · 3.42 KB
/
Copy pathbilling_listener.go
File metadata and controls
96 lines (78 loc) · 3.42 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
package listeners
import (
"context"
"fmt"
"github.com/davecgh/go-spew/spew"
"github.com/NdoleStudio/httpsms/pkg/events"
"github.com/NdoleStudio/httpsms/pkg/services"
"github.com/NdoleStudio/httpsms/pkg/telemetry"
cloudevents "github.com/cloudevents/sdk-go/v2"
"github.com/palantir/stacktrace"
)
// BillingListener handles cloud events which affect billing
type BillingListener struct {
logger telemetry.Logger
tracer telemetry.Tracer
service *services.BillingService
}
// NewBillingListener creates a new instance of UserListener
func NewBillingListener(
logger telemetry.Logger,
tracer telemetry.Tracer,
service *services.BillingService,
) (l *BillingListener, routes map[string]events.EventListener) {
l = &BillingListener{
logger: logger.WithService(fmt.Sprintf("%T", l)),
tracer: tracer,
service: service,
}
return l, map[string]events.EventListener{
events.EventTypeMessageAPISent: l.OnMessageAPISent,
events.UserAccountDeleted: l.onUserAccountDeleted,
events.EventTypeMessagePhoneReceived: l.OnMessagePhoneReceived,
}
}
// OnMessageAPISent handles the events.EventTypeMessageAPISent event
func (listener *BillingListener) OnMessageAPISent(ctx context.Context, event cloudevents.Event) error {
ctx, span := listener.tracer.Start(ctx)
defer span.End()
var payload events.MessageAPISentPayload
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.RegisterSentMessage(ctx, payload.MessageID, payload.RequestReceivedAt, payload.UserID); err != nil {
msg := fmt.Sprintf("cannot register sent message for event [%s] for event with ID [%s]", spew.Sdump(payload), event.ID())
return listener.tracer.WrapErrorSpan(span, stacktrace.Propagate(err, msg))
}
return nil
}
// OnMessagePhoneReceived handles the events.EventTypeMessagePhoneReceived event
func (listener *BillingListener) OnMessagePhoneReceived(ctx context.Context, event cloudevents.Event) error {
ctx, span := listener.tracer.Start(ctx)
defer span.End()
var payload events.MessagePhoneReceivedPayload
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.RegisterReceivedMessage(ctx, payload.MessageID, payload.Timestamp, payload.UserID); err != nil {
msg := fmt.Sprintf("cannot register received message for event [%s] for event with ID [%s]", spew.Sdump(payload), event.ID())
return listener.tracer.WrapErrorSpan(span, stacktrace.Propagate(err, msg))
}
return nil
}
func (listener *BillingListener) 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.BillingUsage] 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
}