-
-
Notifications
You must be signed in to change notification settings - Fork 10
/
inbox_middleware.go
39 lines (31 loc) · 892 Bytes
/
inbox_middleware.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
package tm
import (
"context"
"fmt"
"github.com/stackus/errors"
"github.com/LordMoMA/Intelli-Mall/internal/am"
)
type ErrDuplicateMessage string
type InboxStore interface {
Save(ctx context.Context, msg am.IncomingMessage) error
}
func InboxHandler(store InboxStore) am.MessageHandlerMiddleware {
return func(next am.MessageHandler) am.MessageHandler {
return am.MessageHandlerFunc(func(ctx context.Context, msg am.IncomingMessage) error {
// try to insert the message
err := store.Save(ctx, msg)
if err != nil {
var errDupe ErrDuplicateMessage
if errors.As(err, &errDupe) {
// duplicate message; return without an error to let the message Ack
return nil
}
return err
}
return next.HandleMessage(ctx, msg)
})
}
}
func (e ErrDuplicateMessage) Error() string {
return fmt.Sprintf("duplicate message id encountered: %s", string(e))
}