feat(api): adjust timeout for get_message to 1s

This commit is contained in:
QuantumGhost
2026-02-05 15:22:09 +08:00
parent f21782a9a3
commit d898bcff90
2 changed files with 3 additions and 3 deletions

View File

@ -64,7 +64,7 @@ class _RedisSubscription(RedisSubscriptionBase):
def _get_message(self) -> dict | None:
assert self._pubsub is not None
return self._pubsub.get_message(ignore_subscribe_messages=True, timeout=0.1)
return self._pubsub.get_message(ignore_subscribe_messages=True, timeout=1)
def _get_message_type(self) -> str:
return "message"

View File

@ -73,11 +73,11 @@ class _RedisShardedSubscription(RedisSubscriptionBase):
node = self._client.get_node_from_key(self._topic)
return self._pubsub.get_sharded_message(
ignore_subscribe_messages=False,
timeout=0.1,
timeout=1,
target_node=node,
)
else:
return self._pubsub.get_sharded_message(ignore_subscribe_messages=False, timeout=0.1) # type: ignore[attr-defined]
return self._pubsub.get_sharded_message(ignore_subscribe_messages=False, timeout=1) # type: ignore[attr-defined]
def _get_message_type(self) -> str:
return "smessage"