✨ Websocket stream control count connection for specific user

2024-07-14 23:49:34 +08:00
parent 8f996bb697
commit 7812a06cb1
5 changed files with 228 additions and 36 deletions
--- a/pkg/internal/grpc/stream.go
+++ b/pkg/internal/grpc/stream.go
@ -7,6 +7,13 @@ import (
 	"git.solsynth.dev/hydrogen/dealer/pkg/proto"
 )

+func (v *Server) CountStreamConnection(ctx context.Context, request *proto.CountConnectionRequest) (*proto.CountConnectionResponse, error) {
+	out := services.ClientCount(uint(request.GetUserId()))
+	return &proto.CountConnectionResponse{
+		Count: int64(out),
+	}, nil
+}
+
 func (v *Server) PushStream(ctx context.Context, request *proto.PushStreamRequest) (*proto.PushStreamResponse, error) {
 	cnt, success, errs := services.WebsocketPush(uint(request.GetUserId()), request.GetBody())
 	if len(errs) > 0 {