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

change locking around

Browse source
This commit is contained in:
Andre Medeiros 2021-07-08 21:44:06 -04:00
parent 83fb4cdcc7
commit 3d8492e757
2 changed files with 21 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -158,13 +158,22 @@ func enqueueAccounts(ctx context.Context, logger *logrus.Logger, pool *pgxpool.P
skipped := 0
for _, id := range ids {
payload := fmt.Sprintf("%d", id)
if _, err := redisConn.HGet(ctx, "locks:accounts", payload).Result(); err != redis.Nil {
lockKey := fmt.Sprintf("locks:accounts:%s", payload)
_, err := redisConn.Get(ctx, lockKey).Result()
if err == nil || err == redis.Nil {
skipped++
continue
} else {
logger.WithFields(logrus.Fields{
"lockKey": lockKey,
"err": err,
}).Error("failed to check for account lock")
}
if err := redisConn.HSet(ctx, "locks:accounts", payload, true).Err(); err != nil {
if err := redisConn.SetEX(ctx, lockKey, true, 60*time.Second).Err(); err != nil {
logger.WithFields(logrus.Fields{
"lockKey": lockKey,
"err": err,
}).Error("failed to lock account")
}

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

@ -185,7 +185,13 @@ func (c *Consumer) Consume(delivery rmq.Delivery) {
ctx := context.Background()
defer func() {
c.redis.HDel(ctx, "locks:accounts", delivery.Payload()).Err()
lockKey := fmt.Sprintf("locks:accounts:%s", delivery.Payload())
if err := c.redis.Del(ctx, lockKey).Err(); err != nil {
c.logger.WithFields(logrus.Fields{
"lockKey": lockKey,
"err": err,
}).Error("failed to remove lock")
}
}()
c.logger.WithFields(logrus.Fields{
@ -207,6 +213,7 @@ func (c *Consumer) Consume(delivery rmq.Delivery) {
stmt := `SELECT
id,
username,
access_token,
refresh_token,
expires_at,
@ -217,6 +224,7 @@ func (c *Consumer) Consume(delivery rmq.Delivery) {
account := &data.Account{}
if err := c.pool.QueryRow(ctx, stmt, id).Scan(
&account.ID,
&account.Username,
&account.AccessToken,
&account.RefreshToken,
&account.ExpiresAt,