windy/go-caatsm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

✨ Add configuration to interfaces.

Browse Source
This commit is contained in:
windyboy
2024-08-14 11:48:18 +08:00
parent 6b44c2d7a0
commit a5fa4955f4
5 changed files with 99 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/nats/handler.go
+28 -29
View File
@@ -3,62 +3,61 @@ package nats
import (
"caatsm/internal/config"
"caatsm/internal/domain"
"caatsm/internal/iface"
"caatsm/internal/parsers"
"caatsm/internal/repository"
"caatsm/pkg/utils"
"fmt"
"sync"
"github.com/ThreeDotsLabs/watermill"
"github.com/ThreeDotsLabs/watermill/message"
)
type MessageHandler struct {
mu sync.Mutex
config *config.Config
hasuraRepo *repository.HasuraRepository
repository iface.MessageRepository
publisher iface.MessagePublisher
}
func New(config *config.Config) *MessageHandler {
func NewHandler(config *config.Config, publisher iface.MessagePublisher, repository iface.MessageRepository) *MessageHandler {
return &MessageHandler{
config: config,
hasuraRepo: repository.New(config.Hasura.Endpoint, config.Hasura.Secret),
repository: repository,
publisher: publisher,
}
}
func (handler *MessageHandler) HandleMessage(msg *message.Message) error {
func (handler *MessageHandler) HandleMessage(msg []byte, uuid []byte) error {
handler.mu.Lock()
defer handler.mu.Unlock()
log := utils.GetSugaredLogger()
if msg.Payload == nil {
if msg == nil {
log.Error("empty message")
return fmt.Errorf("empty message")
}
payload := string(msg.Payload)
payload := string(msg)
var parsed *domain.ParsedMessage
if parsed = parsers.Parse(payload); !parsed.Parsed {
log.Infof("not parsed: [%s] : {%s} \n", msg.UUID, msg.Payload)
log.Infof("not parsed: [%s] : {%s} \n", uuid, payload)
} else {
log.Infof("parsed [%s]: %v\n", msg.UUID, parsed.ToString())
log.Infof("parsed [%s]: %v\n", uuid, parsed.ToString())
}
handler.SaveMessage(parsed, msg.UUID)
handler.Publish(parsed)
handler.repository.CreateNew(parsed, uuid)
handler.publisher.Publish(parsed)
return nil
}
func (n *MessageHandler) SaveMessage(parsed *domain.ParsedMessage, uuid string) {
logger := watermill.NewStdLogger(false, false)
if parsed != nil {
parsed.Uuid = uuid
if err := n.hasuraRepo.CreateNew(parsed); err != nil {
logger.Error("error inserting message", err, map[string]interface{}{"message": parsed})
}
logger.Info("message inserted", map[string]interface{}{"message": parsed.Uuid})
}
}
// func (n *MessageHandler) SaveMessage(parsed *domain.ParsedMessage, uuid string) {
// logger := watermill.NewStdLogger(false, false)
// if parsed != nil {
// parsed.Uuid = uuid
// if err := n.hasuraRepo.CreateNew(parsed); err != nil {
// logger.Error("error inserting message", err, map[string]interface{}{"message": parsed})
// }
// logger.Info("message inserted", map[string]interface{}{"message": parsed.Uuid})
// }
// }
func (n *MessageHandler) Publish(parsed *domain.ParsedMessage) {
if err := Publish(n.config, parsed); err != nil {
utils.GetSugaredLogger().Error("error publishing message", err, map[string]interface{}{"message": parsed})
}
}
// func (n *MessageHandler) Publish(parsed *domain.ParsedMessage) {
// if err := Publish(n.config, parsed); err != nil {
// utils.GetSugaredLogger().Error("error publishing message", err, map[string]interface{}{"message": parsed})
// }
// }