⚡ Optimize batch notification speed
This commit is contained in:
@ -25,6 +25,15 @@ func GetAccount(id uint) (models.Account, error) {
|
||||
return account, nil
|
||||
}
|
||||
|
||||
func GetAccountList(id []uint) ([]models.Account, error) {
|
||||
var accounts []models.Account
|
||||
if err := database.C.Where("id IN ?", id).Find(&accounts).Error; err != nil {
|
||||
return accounts, err
|
||||
}
|
||||
|
||||
return accounts, nil
|
||||
}
|
||||
|
||||
func GetAccountWithName(alias string) (models.Account, error) {
|
||||
var account models.Account
|
||||
if err := database.C.Where(models.Account{
|
||||
|
@ -6,6 +6,7 @@ import (
|
||||
"git.solsynth.dev/hydrogen/dealer/pkg/proto"
|
||||
"git.solsynth.dev/hydrogen/passport/pkg/internal/gap"
|
||||
"reflect"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"firebase.google.com/go/messaging"
|
||||
@ -60,6 +61,15 @@ func NewNotification(notification models.Notification) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func NewNotificationBatch(notifications []models.Notification) error {
|
||||
if err := database.C.CreateInBatches(notifications, 1000).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
PushNotificationBatch(notifications)
|
||||
return nil
|
||||
}
|
||||
|
||||
// PushNotification will push the notification whatever it exists record in the
|
||||
// database Recommend pushing another goroutine when you need to push a lot of
|
||||
// notifications And just use a block statement when you just push one
|
||||
@ -153,3 +163,15 @@ func PushNotification(notification models.Notification) error {
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func PushNotificationBatch(notifications []models.Notification) {
|
||||
var wg sync.WaitGroup
|
||||
for _, notification := range notifications {
|
||||
wg.Add(1)
|
||||
item := notification
|
||||
go func() {
|
||||
_ = PushNotification(item)
|
||||
wg.Done()
|
||||
}()
|
||||
}
|
||||
}
|
||||
|
Reference in New Issue
Block a user