migrate from github
This commit is contained in:
@@ -0,0 +1,74 @@
|
||||
package notification_repo
|
||||
|
||||
import (
|
||||
"github.com/XingfenD/yoresee_doc/internal/model"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
type NotificationListOperation struct {
|
||||
repo *NotificationRepository
|
||||
receiverID int64
|
||||
status *string
|
||||
page int
|
||||
pageSize int
|
||||
tx *gorm.DB
|
||||
}
|
||||
|
||||
func (r *NotificationRepository) List(receiverID int64) *NotificationListOperation {
|
||||
return &NotificationListOperation{
|
||||
repo: r,
|
||||
receiverID: receiverID,
|
||||
}
|
||||
}
|
||||
|
||||
func (op *NotificationListOperation) WithTx(tx *gorm.DB) *NotificationListOperation {
|
||||
op.tx = tx
|
||||
return op
|
||||
}
|
||||
|
||||
func (op *NotificationListOperation) WithStatus(status *string) *NotificationListOperation {
|
||||
op.status = status
|
||||
return op
|
||||
}
|
||||
|
||||
func (op *NotificationListOperation) WithPagination(page, pageSize int) *NotificationListOperation {
|
||||
op.page = page
|
||||
op.pageSize = pageSize
|
||||
return op
|
||||
}
|
||||
|
||||
func (op *NotificationListOperation) ExecWithTotal() ([]model.Notification, int64, error) {
|
||||
db := op.repo.db
|
||||
if op.tx != nil {
|
||||
db = op.tx
|
||||
}
|
||||
|
||||
query := db.Model(&model.Notification{}).Where("receiver_id = ?", op.receiverID)
|
||||
if op.status != nil && *op.status != "" {
|
||||
query = query.Where("status = ?", *op.status)
|
||||
}
|
||||
|
||||
var total int64
|
||||
if err := query.Count(&total).Error; err != nil {
|
||||
return nil, 0, err
|
||||
}
|
||||
|
||||
page := op.page
|
||||
pageSize := op.pageSize
|
||||
if page <= 0 {
|
||||
page = 1
|
||||
}
|
||||
if pageSize <= 0 {
|
||||
pageSize = 10
|
||||
}
|
||||
if pageSize > 100 {
|
||||
pageSize = 100
|
||||
}
|
||||
offset := (page - 1) * pageSize
|
||||
|
||||
var list []model.Notification
|
||||
if err := query.Order("created_at DESC").Offset(offset).Limit(pageSize).Find(&list).Error; err != nil {
|
||||
return nil, 0, err
|
||||
}
|
||||
return list, total, nil
|
||||
}
|
||||
Reference in New Issue
Block a user