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

refactor: Update main.go to process Civial Aviation Authority Telegram messages and add CLI flag for Nats server address and topic

Browse Source
This commit is contained in:
windyboy
2024-08-05 10:58:20 +08:00
parent c54bcbc0a8
commit 3c917c93af
3 changed files with 49 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/handlers/nats_handler.go → internal/handlers/message_handler.go
+5 -5
View File
@@ -15,17 +15,17 @@ import (
nc "github.com/nats-io/nats.go"
)
type NatsHandler struct {
type MessageHandler struct {
mu sync.Mutex
config *config.Config
hasuraRepo *repository.HasuraRepository
}
func NewNatsHandler(config *config.Config) *NatsHandler {
return &NatsHandler{config: config, hasuraRepo: repository.NewHasuraRepo(config.Hasura.Endpoint, config.Hasura.Secret)}
func NewNatsHandler(config *config.Config) *MessageHandler {
return &MessageHandler{config: config, hasuraRepo: repository.NewHasuraRepo(config.Hasura.Endpoint, config.Hasura.Secret)}
}
func (n *NatsHandler) HandleMessage(msg *message.Message) error {
func (n *MessageHandler) HandleMessage(msg *message.Message) error {
n.mu.Lock()
defer n.mu.Unlock()
log := utils.GetSugaredLogger()
@@ -45,7 +45,7 @@ func (n *NatsHandler) HandleMessage(msg *message.Message) error {
return nil
}
func (n *NatsHandler) SaveMessage(parsed *domain.ParsedMessage, uuid string) {
func (n *MessageHandler) SaveMessage(parsed *domain.ParsedMessage, uuid string) {
logger := watermill.NewStdLogger(false, false)
if parsed != nil {
parsed.Uuid = uuid