✨ Batch push websocket to improve performance

pkg/internal/grpc/stream.go

@@ -3,8 +3,10 @@ package grpc
 import (
 	"context"
 	"fmt"
+
 	"git.solsynth.dev/hydrogen/dealer/pkg/internal/services"
 	"git.solsynth.dev/hydrogen/dealer/pkg/proto"
+	"github.com/samber/lo"
 )
 
 func (v *Server) CountStreamConnection(ctx context.Context, request *proto.CountConnectionRequest) (*proto.CountConnectionResponse, error) {
@@ -34,3 +36,29 @@ func (v *Server) PushStream(ctx context.Context, request *proto.PushStreamReques
 		FailedCount:   int64(cnt - success),
 	}, nil
 }
+
+func (v *Server) PushStreamBatch(ctx context.Context, request *proto.PushStreamBatchRequest) (*proto.PushStreamResponse, error) {
+	cnt, success, errs := services.WebsocketPushBatch(
+		lo.Map(request.GetUserId(), func(item uint64, idx int) uint {
+			return uint(item)
+		},
+		), request.GetBody(),
+	)
+	if len(errs) > 0 {
+		// Partial fail
+		return &proto.PushStreamResponse{
+			IsAllSuccess:  false,
+			AffectedCount: int64(success),
+			FailedCount:   int64(cnt - success),
+		}, nil
+	} else if cnt > 0 && success == 0 {
+		// All fail
+		return nil, fmt.Errorf("all push request failed: %v", errs)
+	}
+
+	return &proto.PushStreamResponse{
+		IsAllSuccess:  true,
+		AffectedCount: int64(success),
+		FailedCount:   int64(cnt - success),
+	}, nil
+}