Fixes race condition with token rate limit INCR ?

AutoMuteUs-Portable · Nov 29, 2020 · 69503d2 · 69503d2
1 parent 35a9399
commit 69503d2
Showing 1 changed file with 3 additions and 12 deletions.
diff --git a/galactus/server.go b/galactus/server.go
@@ -171,20 +171,11 @@ func (tokenProvider *TokenProvider) getAnySession(guildID string, limit int) (*d
 }
 
 func (tokenProvider *TokenProvider) IncrGuildTokenComboLock(guildID, hashToken string) {
-	v := int64(0)
-	vStr, err := tokenProvider.client.Get(context.Background(), guildTokenLock(guildID, hashToken)).Result()
-	if err == redis.Nil {
-		v = 0
-	} else {
-		v, _ = strconv.ParseInt(vStr, 10, 64)
-	}
-	v++
-
-	//5 second TTL
-	err = tokenProvider.client.Set(context.Background(), guildTokenLock(guildID, hashToken), v, time.Second*5).Err()
+	err := tokenProvider.client.Incr(context.Background(), guildTokenLock(guildID, hashToken)).Err()
 	if err != nil {
-		log.Println(err)
+		log.Println()
 	}
+	tokenProvider.client.Expire(context.Background(), guildTokenLock(guildID, hashToken), time.Second*5)
 }
 
 func (tokenProvider *TokenProvider) CanUseGuildTokenCombo(guildID, hashToken string) bool {