56 lines
1.5 KiB
Go
56 lines
1.5 KiB
Go
package grpc
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
|
|
"git.solsynth.dev/hydrogen/dealer/pkg/internal/services"
|
|
"git.solsynth.dev/hydrogen/dealer/pkg/proto"
|
|
)
|
|
|
|
func (v *Server) DeliverNotification(ctx context.Context, request *proto.DeliverNotificationRequest) (*proto.DeliverResponse, error) {
|
|
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]
|
|
provider := provider
|
|
wg.Add(1)
|
|
go func() {
|
|
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.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() {
|
|
to := to
|
|
wg.Add(1)
|
|
go func() {
|
|
services.DealDeliveryTask(&proto.DeliverEmailRequest{
|
|
To: to,
|
|
Email: request.GetEmail(),
|
|
})
|
|
wg.Done()
|
|
}()
|
|
}
|
|
|
|
return &proto.DeliverResponse{}, nil
|
|
}
|