36 lines
732 B
Go
36 lines
732 B
Go
package notification_repo
|
|
|
|
import (
|
|
"github.com/XingfenD/yoresee_doc/internal/model"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type NotificationCreateBatchOperation struct {
|
|
repo *NotificationRepository
|
|
items []model.Notification
|
|
tx *gorm.DB
|
|
}
|
|
|
|
func (r *NotificationRepository) CreateBatch(items []model.Notification) *NotificationCreateBatchOperation {
|
|
return &NotificationCreateBatchOperation{
|
|
repo: r,
|
|
items: items,
|
|
}
|
|
}
|
|
|
|
func (op *NotificationCreateBatchOperation) WithTx(tx *gorm.DB) *NotificationCreateBatchOperation {
|
|
op.tx = tx
|
|
return op
|
|
}
|
|
|
|
func (op *NotificationCreateBatchOperation) Exec() error {
|
|
if len(op.items) == 0 {
|
|
return nil
|
|
}
|
|
db := op.repo.db
|
|
if op.tx != nil {
|
|
db = op.tx
|
|
}
|
|
return db.Create(&op.items).Error
|
|
}
|