youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-04-30 15:38:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(api): Prevent messages from being dropped.

Browse Source 
This is cuased by an issue in upstream code:

If Sharded PubSub is used with Cluster, the `
ClusterPubSub.get_sharded_message` will return `None` regardless of
message['type'].
This commit is contained in:
QuantumGhost
2026-01-26 00:32:08 +08:00
parent 33a830cbc9
commit 98d9dc620b
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
api/libs/broadcast_channel/redis/sharded_channel.py
View File

@ -59,7 +59,14 @@ class _RedisShardedSubscription(RedisSubscriptionBase):
def _get_message(self) -> dict | None:
assert self._pubsub is not None
return self._pubsub.get_sharded_message(ignore_subscribe_messages=True, timeout=0.1) # type: ignore[attr-defined]
# NOTE(QuantumGhost): this is an issue in
# upstream code. If Sharded PubSub is used with Cluster, the
# `ClusterPubSub.get_sharded_message` will return `None` regardless of
# message['type'].
#
# Since we have already filtered at the caller's site, we can safely set
# `ignore_subscribe_messages=False`.
return self._pubsub.get_sharded_message(ignore_subscribe_messages=False, timeout=0.1) # type: ignore[attr-defined]
def _get_message_type(self) -> str:
return "smessage"