Optimize batch notification speed

This commit is contained in:
2024-07-17 14:04:55 +08:00
parent 7436d4b2cc
commit 282a0891d0
6 changed files with 165 additions and 58 deletions

View File

@ -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{

View File

@ -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()
}()
}
}