⚡ Optimize batch notification speed

pkg/internal/grpc/notifier.go

@@ -3,7 +3,9 @@ package grpc
 import (
 	"context"
 	"fmt"
+	"git.solsynth.dev/hydrogen/passport/pkg/internal/database"
 	jsoniter "github.com/json-iterator/go"
+	"github.com/samber/lo"
 
 	"git.solsynth.dev/hydrogen/dealer/pkg/proto"
 	"git.solsynth.dev/hydrogen/passport/pkg/internal/models"
@@ -45,3 +47,82 @@ func (v *Server) NotifyUser(_ context.Context, in *proto.NotifyUserRequest) (*pr
 		IsSuccess: true,
 	}, nil
 }
+
+func (v *Server) NotifyUserBatch(_ context.Context, in *proto.NotifyUserBatchRequest) (*proto.NotifyResponse, error) {
+	var err error
+	var users []models.Account
+	if users, err = services.GetAccountList(lo.Map(in.GetUserId(), func(item uint64, index int) uint {
+		return uint(item)
+	})); err != nil {
+		return nil, fmt.Errorf("unable to get account: %v", err)
+	}
+
+	var metadata map[string]any
+	_ = jsoniter.Unmarshal(in.GetNotify().GetMetadata(), &metadata)
+
+	var notifications []models.Notification
+	for _, user := range users {
+		notification := models.Notification{
+			Topic:       in.GetNotify().GetTopic(),
+			Title:       in.GetNotify().GetTitle(),
+			Subtitle:    in.GetNotify().Subtitle,
+			Body:        in.GetNotify().GetBody(),
+			Metadata:    metadata,
+			IsRealtime:  in.GetNotify().GetIsRealtime(),
+			IsForcePush: in.GetNotify().GetIsForcePush(),
+			AccountID:   user.ID,
+		}
+
+		notifications = append(notifications, notification)
+	}
+
+	if in.GetNotify().GetIsRealtime() {
+		services.PushNotificationBatch(notifications)
+	} else {
+		if err := services.NewNotificationBatch(notifications); err != nil {
+			return nil, err
+		}
+	}
+
+	return &proto.NotifyResponse{
+		IsSuccess: true,
+	}, nil
+}
+
+func (v *Server) NotifyAllUser(_ context.Context, in *proto.NotifyRequest) (*proto.NotifyResponse, error) {
+	var users []models.Account
+	if err := database.C.Find(&users).Error; err != nil {
+		return nil, fmt.Errorf("unable to get account: %v", err)
+	}
+
+	var metadata map[string]any
+	_ = jsoniter.Unmarshal(in.GetMetadata(), &metadata)
+
+	var notifications []models.Notification
+	for _, user := range users {
+		notification := models.Notification{
+			Topic:       in.GetTopic(),
+			Title:       in.GetTitle(),
+			Subtitle:    in.Subtitle,
+			Body:        in.GetBody(),
+			Metadata:    metadata,
+			IsRealtime:  in.GetIsRealtime(),
+			IsForcePush: in.GetIsForcePush(),
+			AccountID:   user.ID,
+		}
+
+		notifications = append(notifications, notification)
+	}
+
+	if in.GetIsRealtime() {
+		services.PushNotificationBatch(notifications)
+	} else {
+		if err := services.NewNotificationBatch(notifications); err != nil {
+			return nil, err
+		}
+	}
+
+	return &proto.NotifyResponse{
+		IsSuccess: true,
+	}, nil
+}