Revert "ref(redis): Do not use redis-blaster for snowflake IDs generation (#103967)"

This reverts commit 278ba81.

Co-authored-by: markstory <[email protected]>

Author: getsentry-bot

src/sentry/conf/server.py

@@ -195,7 +195,6 @@ def env(
 SENTRY_HYBRIDCLOUD_DELETIONS_REDIS_CLUSTER = "default"
 SENTRY_SESSION_STORE_REDIS_CLUSTER = "default"
 SENTRY_AUTH_IDPMIGRATION_REDIS_CLUSTER = "default"
-SENTRY_SNOWFLAKE_REDIS_CLUSTER = "default"
 
 # Hosts that are allowed to use system token authentication.
 # http://en.wikipedia.org/wiki/Reserved_IP_addresses

src/sentry/utils/snowflake.py

@@ -8,13 +8,12 @@
 from django.conf import settings
 from django.db import IntegrityError, router, transaction
 from django.db.models import Model
+from redis.client import StrictRedis
 from rest_framework import status
 from rest_framework.exceptions import APIException
-from sentry_redis_tools.clients import RedisCluster, StrictRedis
 
 from sentry.db.postgres.transactions import enforce_constraints
 from sentry.types.region import RegionContextError, get_local_region
-from sentry.utils import redis
 
 if TYPE_CHECKING:
     from sentry.db.models.base import Model as BaseModel
@@ -140,16 +139,14 @@ def generate_snowflake_id(redis_key: str) -> int:
     return snowflake_id
 
 
-def get_redis_cluster() -> RedisCluster[str] | StrictRedis[str]:
-    return redis.redis_clusters.get(settings.SENTRY_SNOWFLAKE_REDIS_CLUSTER)
+def get_redis_cluster(redis_key: str) -> StrictRedis[str]:
+    from sentry.utils import redis
 
-
-def get_timestamp_redis_key(redis_key: str, timestamp: int) -> str:
-    return f"snowflakeid:{redis_key}:{str(timestamp)}"
+    return redis.clusters.get("default").get_local_client_for_key(redis_key)
 
 
 def get_sequence_value_from_redis(redis_key: str, starting_timestamp: int) -> tuple[int, int]:
-    cluster = get_redis_cluster()
+    cluster = get_redis_cluster(redis_key)
 
     # this is the amount we want to lookback for previous timestamps
     # the below is more of a safety net if starting_timestamp is ever
@@ -159,16 +156,14 @@ def get_sequence_value_from_redis(redis_key: str, starting_timestamp: int) -> tu
     for i in range(time_range):
         timestamp = starting_timestamp - i
 
-        timestamp_redis_key = get_timestamp_redis_key(redis_key, timestamp)
-
         # We are decreasing the value by 1 each time since the incr operation in redis
         # initializes the counter at 1. For our region sequences, we want the value to
         # be from 0-15 and not 1-16
-        sequence_value = cluster.incr(timestamp_redis_key)
+        sequence_value = cluster.incr(str(timestamp))
         sequence_value -= 1
 
         if sequence_value == 0:
-            cluster.expire(timestamp_redis_key, int(_TTL.total_seconds()))
+            cluster.expire(str(timestamp), int(_TTL.total_seconds()))
 
         if sequence_value < MAX_AVAILABLE_REGION_SEQUENCES:
             return timestamp, sequence_value

tests/sentry/utils/test_snowflake.py

@@ -20,7 +20,6 @@
     SnowflakeBitSegment,
     generate_snowflake_id,
     get_redis_cluster,
-    get_timestamp_redis_key,
     uses_snowflake_id,
 )
 
@@ -65,16 +64,14 @@ def test_generate_correct_ids_with_region_sequence(self) -> None:
 
     @freeze_time(CURRENT_TIME)
     def test_out_of_region_sequences(self) -> None:
-        cluster = get_redis_cluster()
+        cluster = get_redis_cluster("test_redis_key")
         current_timestamp = int(datetime.now().timestamp() - settings.SENTRY_SNOWFLAKE_EPOCH_START)
-        redis_key = "test_redis_key"
-
         for i in range(int(_TTL.total_seconds())):
             timestamp = current_timestamp - i
-            cluster.set(get_timestamp_redis_key(redis_key, timestamp), 16)
+            cluster.set(str(timestamp), 16)
 
         with pytest.raises(Exception) as context:
-            generate_snowflake_id(redis_key)
+            generate_snowflake_id("test_redis_key")
 
         assert str(context.value) == "No available ID"