mirrors/apollo-backend
2
0
Fork 0
mirror of https://github.com/christianselig/apollo-backend synced 2024-12-26 00:05:28 +00:00
Code Issues Projects Releases Packages Wiki Activity

actually remember last message id so we don't duplicate notifications

Browse source
This commit is contained in:
Andre Medeiros 2021-07-08 23:29:53 -04:00
parent 3ce858927b
commit b9373832eb
1 changed files with 19 additions and 21 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

40
cmd/apollo-worker-notifications/main.go
View file

@ -327,30 +327,28 @@ func (c *Consumer) Consume(delivery rmq.Delivery) {
// Set latest message we alerted on
latestMsg := msgs.MessageListing.Messages[0]
if err = c.pool.BeginFunc(ctx, func(tx pgx.Tx) error {
stmt := `
UPDATE accounts
SET last_message_id = $1
WHERE id = $2`
_, err := tx.Exec(ctx, stmt, latestMsg.FullName(), account.ID)
return err
}); err != nil {
c.logger.WithFields(logrus.Fields{
"accountID": id,
"err": err,
}).Error("failed to update last_message_id for account")
delivery.Reject()
return
}
// Let's populate this with the latest message so we don't flood users with stuff
if account.LastMessageID == "" {
if err = c.pool.BeginFunc(ctx, func(tx pgx.Tx) error {
stmt := `
UPDATE accounts
SET last_message_id = $1
WHERE id = $2`
_, err := tx.Exec(ctx, stmt, latestMsg.FullName(), account.ID)
return err
}); err != nil {
c.logger.WithFields(logrus.Fields{
"accountID": id,
"err": err,
}).Error("failed to update last_message_id for account")
delivery.Reject()
} else {
c.logger.WithFields(logrus.Fields{
"accountID": delivery.Payload(),
}).Debug("populating first message ID to prevent spamming")
delivery.Ack()
}
c.logger.WithFields(logrus.Fields{
"accountID": delivery.Payload(),
}).Debug("populating first message ID to prevent spamming")
delivery.Ack()
return
}