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

fix skipped accounts and locks

Browse source
This commit is contained in:
Andre Medeiros 2021-07-08 20:26:01 -04:00
parent c1f1d2a309
commit aad2434d83
1 changed files with 12 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
cmd/apollo-scheduler/main.go
View file

@ -150,17 +150,27 @@ func enqueueAccounts(ctx context.Context, logger *logrus.Logger, pool *pgxpool.P
}).Debug("enqueueing account batch") }).Debug("enqueueing account batch")
enqueued := 0 enqueued := 0
skipped := 0
for _, id := range ids { for _, id := range ids {
payload := fmt.Sprintf("%d", id) payload := fmt.Sprintf("%d", id)
if redisConn.HGet(ctx, "locks:accounts", payload).Val() != "" { if _, err := redisConn.HGet(ctx, "locks:accounts", payload).Result(); err != redis.Nil {
skipped++
continue continue
} }
if err := redisConn.HSet(ctx, "locks:accounts", payload, true).Err(); err != nil {
logger.WithFields(logrus.Fields{
"err": err,
}).Error("failed to lock account")
}
enqueued++ enqueued++
_ = queue.Publish(payload) _ = queue.Publish(payload)
} }
logger.WithFields(logrus.Fields{ logger.WithFields(logrus.Fields{
"count": enqueued, "count": enqueued,
"skipped": skipped,
}).Info("done enqueueing account batch") }).Info("done enqueueing account batch")
} }