redis/go-redis
1
0
Fork 0
mirror of https://github.com/redis/go-redis.git synced 2025-07-28 06:42:00 +03:00
Code Activity

Merge pull request #2177 from j178/sharded-pubsub

Browse Source 
feat(pubsub): support sharded pub/sub
This commit is contained in:
Vladimir Mihailenco
2022-10-05 10:21:19 +03:00
committed by GitHub
parent 965ecdeb07 baa48a4415
commit fda4feef7a
8 changed files with 233 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
ring.go
View File

@ -560,6 +560,19 @@ func (c *Ring) PSubscribe(ctx context.Context, channels ...string) *PubSub {
return shard.Client.PSubscribe(ctx, channels...)
}
// SSubscribe Subscribes the client to the specified shard channels.
func (c *Ring) SSubscribe(ctx context.Context, channels ...string) *PubSub {
if len(channels) == 0 {
panic("at least one channel is required")
}
shard, err := c.shards.GetByKey(channels[0])
if err != nil {
// TODO: return PubSub with sticky error
panic(err)
}
return shard.Client.SSubscribe(ctx, channels...)
}
// ForEachShard concurrently calls the fn on each live shard in the ring.
// It returns the first error if any.
func (c *Ring) ForEachShard(