mirrors/apollo-backend
2
0
Fork 0
mirror of https://github.com/christianselig/apollo-backend synced 2024-11-22 19:57:43 +00:00
Code Issues Projects Releases Packages Wiki Activity

try refreshing token on mailbox error

Browse source
This commit is contained in:
Andre Medeiros 2021-07-13 13:58:52 -04:00
parent 9a37a05bf0
commit d2909b8003
1 changed files with 25 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

40
internal/worker/notifications.go
View file

@ -186,6 +186,24 @@ func (nc *notificationsConsumer) Consume(delivery rmq.Delivery) {
nc.statsd.Histogram("apollo.queue.delay", latency, []string{}, rate) nc.statsd.Histogram("apollo.queue.delay", latency, []string{}, rate)
} }
if err = nc.db.BeginFunc(ctx, func(tx pgx.Tx) error {
stmt := `
UPDATE accounts
SET last_checked_at = $1
WHERE id = $2`
_, err := tx.Exec(ctx, stmt, now, account.ID)
return err
}); err != nil {
nc.logger.WithFields(logrus.Fields{
"accountID": id,
"err": err,
}).Error("failed to update last_checked_at for account")
return
}
retried := false
refreshToken:
rac := nc.reddit.NewAuthenticatedClient(account.RefreshToken, account.AccessToken) rac := nc.reddit.NewAuthenticatedClient(account.RefreshToken, account.AccessToken)
if account.ExpiresAt < int64(now) { if account.ExpiresAt < int64(now) {
nc.logger.WithFields(logrus.Fields{ nc.logger.WithFields(logrus.Fields{
@ -226,6 +244,13 @@ func (nc *notificationsConsumer) Consume(delivery rmq.Delivery) {
msgs, err := rac.MessageInbox(account.LastMessageID) msgs, err := rac.MessageInbox(account.LastMessageID)
if err != nil { if err != nil {
// Check if maybe we need to refresh the token.
if !retried {
retried = true
account.ExpiresAt = 0
goto refreshToken
}
nc.logger.WithFields(logrus.Fields{ nc.logger.WithFields(logrus.Fields{
"accountID": id, "accountID": id,
"err": err, "err": err,
@ -238,21 +263,6 @@ func (nc *notificationsConsumer) Consume(delivery rmq.Delivery) {
"count": len(msgs.MessageListing.Messages), "count": len(msgs.MessageListing.Messages),
}).Debug("fetched messages") }).Debug("fetched messages")
if err = nc.db.BeginFunc(ctx, func(tx pgx.Tx) error {
stmt := `
UPDATE accounts
SET last_checked_at = $1
WHERE id = $2`
_, err := tx.Exec(ctx, stmt, now, account.ID)
return err
}); err != nil {
nc.logger.WithFields(logrus.Fields{
"accountID": id,
"err": err,
}).Error("failed to update last_checked_at for account")
return
}
if len(msgs.MessageListing.Messages) == 0 { if len(msgs.MessageListing.Messages) == 0 {
nc.logger.WithFields(logrus.Fields{ nc.logger.WithFields(logrus.Fields{
"accountID": id, "accountID": id,