Hydrogen/Dealer
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

⚡ Directly use multiple goroutines to speed up delivery speed

Browse Source
This commit is contained in:
LittleSheep
2024-07-23 18:27:43 +08:00
parent 7541292f68
commit fbfb340058
4 changed files with 38 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
pkg/internal/grpc/postman.go
View File

@ -4,37 +4,50 @@ import (
"context"
"git.solsynth.dev/hydrogen/dealer/pkg/internal/services"
"git.solsynth.dev/hydrogen/dealer/pkg/proto"
"sync"
)
func (v *Server) DeliverNotification(ctx context.Context, request *proto.DeliverNotificationRequest) (*proto.DeliverResponse, error) {
services.PublishDeliveryTask(request)
services.DealDeliveryTask(request)
return &proto.DeliverResponse{}, nil
}
func (v *Server) DeliverNotificationBatch(ctx context.Context, request *proto.DeliverNotificationBatchRequest) (*proto.DeliverResponse, error) {
var wg sync.WaitGroup
for idx, provider := range request.GetProviders() {
token := request.GetDeviceTokens()[idx]
services.PublishDeliveryTask(&proto.DeliverNotificationRequest{
Provider: provider,
DeviceToken: token,
Notify: request.GetNotify(),
})
provider := provider
go func() {
wg.Add(1)
services.DealDeliveryTask(&proto.DeliverNotificationRequest{
Provider: provider,
DeviceToken: token,
Notify: request.GetNotify(),
})
wg.Done()
}()
}
return &proto.DeliverResponse{}, nil
}
func (v *Server) DeliverEmail(ctx context.Context, request *proto.DeliverEmailRequest) (*proto.DeliverResponse, error) {
services.PublishDeliveryTask(request)
services.DealDeliveryTask(request)
return &proto.DeliverResponse{}, nil
}
func (v *Server) DeliverEmailBatch(ctx context.Context, request *proto.DeliverEmailBatchRequest) (*proto.DeliverResponse, error) {
var wg sync.WaitGroup
for _, to := range request.GetTo() {
services.PublishDeliveryTask(&proto.DeliverEmailRequest{
To: to,
Email: request.GetEmail(),
})
to := to
go func() {
wg.Add(1)
services.DealDeliveryTask(&proto.DeliverEmailRequest{
To: to,
Email: request.GetEmail(),
})
wg.Done()
}()
}
return &proto.DeliverResponse{}, nil

37
pkg/internal/services/postman.go
View File

@ -17,29 +17,20 @@ import (
"time"
)
var deliveryTaskQueue = make(chan any, 256)
func PublishDeliveryTask(task any) {
deliveryTaskQueue <- task
}
func ConsumeDeliveryTasks() {
for {
task := <-deliveryTaskQueue
switch tk := task.(type) {
case *proto.DeliverEmailRequest:
if tk.GetEmail().HtmlBody != nil {
_ = SendMailHTML(tk.GetTo(), tk.GetEmail().GetSubject(), tk.GetEmail().GetHtmlBody())
} else {
_ = SendMail(tk.GetTo(), tk.GetEmail().GetSubject(), tk.GetEmail().GetTextBody())
}
case *proto.DeliverNotificationRequest:
switch tk.GetProvider() {
case "firebase":
_ = PushFirebaseNotify(tk.GetDeviceToken(), tk.GetNotify())
case "apple":
_ = PushAppleNotify(tk.GetDeviceToken(), tk.GetNotify())
}
func DealDeliveryTask(task any) {
switch tk := task.(type) {
case *proto.DeliverEmailRequest:
if tk.GetEmail().HtmlBody != nil {
_ = SendMailHTML(tk.GetTo(), tk.GetEmail().GetSubject(), tk.GetEmail().GetHtmlBody())
} else {
_ = SendMail(tk.GetTo(), tk.GetEmail().GetSubject(), tk.GetEmail().GetTextBody())
}
case *proto.DeliverNotificationRequest:
switch tk.GetProvider() {
case "firebase":
_ = PushFirebaseNotify(tk.GetDeviceToken(), tk.GetNotify())
case "apple":
_ = PushAppleNotify(tk.GetDeviceToken(), tk.GetNotify())
}
}
}