sesamy-go/measurementprotocol/v2/messagehandler.go
Kevin Franklin Kim d80faf60d8
initial commit
2024-03-04 14:53:55 +01:00

30 lines
718 B
Go

package v2
import (
"encoding/json"
"github.com/ThreeDotsLabs/watermill/message"
"github.com/pkg/errors"
)
func MessageEventHandler(eventHandler func(event *Event, msg *message.Message) error) func(msg *message.Message) ([]*message.Message, error) {
return func(msg *message.Message) ([]*message.Message, error) {
var event *Event
if err := json.Unmarshal(msg.Payload, &event); err != nil {
return nil, errors.Wrap(err, "failed to unmarshal event")
}
if err := eventHandler(event, msg); err != nil {
return nil, err
}
b, err := json.Marshal(event)
if err != nil {
return nil, errors.Wrap(err, "failed to marshal event")
}
msg.Payload = b
return []*message.Message{msg}, nil
}
}