✨ Disconnect websocket auto cleaning up subscribed channels

2025-03-01 18:02:55 +08:00
parent e3a4988ccf
commit eea9b85745
3 changed files with 31 additions and 6 deletions
--- a/pkg/internal/grpc/stream.go
+++ b/pkg/internal/grpc/stream.go
@ -76,7 +76,7 @@ func (v *Server) PushStream(_ context.Context, request *proto.PushStreamRequest)
 		action := strings.Split(in.Action, ".")[1]
 		switch action {
 		case "subscribe":
-			services.SubscribeChannel(uint(request.GetUserId()), data.ChannelID)
+			services.SubscribeChannel(uint(request.GetUserId()), data.ChannelID, request.GetClientId())
 		case "unsubscribe":
 			services.UnsubscribeChannel(uint(request.GetUserId()), data.ChannelID)
 		case "unsubscribeAll":