70 lines
1.5 KiB
Go
70 lines
1.5 KiB
Go
package membership_repo
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/XingfenD/yoresee_doc/internal/model"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type QueryUserGroupOperation struct {
|
|
repo *MembershipRepository
|
|
tx *gorm.DB
|
|
keyword *string
|
|
page int
|
|
pageSize int
|
|
}
|
|
|
|
func (r *MembershipRepository) QueryUserGroup() *QueryUserGroupOperation {
|
|
return &QueryUserGroupOperation{
|
|
repo: r,
|
|
}
|
|
}
|
|
|
|
func (op *QueryUserGroupOperation) WithTx(tx *gorm.DB) *QueryUserGroupOperation {
|
|
op.tx = tx
|
|
return op
|
|
}
|
|
|
|
func (op *QueryUserGroupOperation) WithKeyword(keyword *string) *QueryUserGroupOperation {
|
|
op.keyword = keyword
|
|
return op
|
|
}
|
|
|
|
func (op *QueryUserGroupOperation) WithPagination(page, pageSize int) *QueryUserGroupOperation {
|
|
op.page = page
|
|
op.pageSize = pageSize
|
|
return op
|
|
}
|
|
|
|
func (op *QueryUserGroupOperation) ExecWithTotal() ([]model.UserGroupMeta, int64, error) {
|
|
if op.tx == nil {
|
|
op.tx = op.repo.db
|
|
}
|
|
|
|
query := op.tx.Model(&model.UserGroupMeta{})
|
|
if op.keyword != nil {
|
|
trimmed := strings.TrimSpace(*op.keyword)
|
|
if trimmed != "" {
|
|
like := "%" + trimmed + "%"
|
|
query = query.Where("name ILIKE ? OR description ILIKE ?", like, like)
|
|
}
|
|
}
|
|
|
|
var total int64
|
|
if err := query.Count(&total).Error; err != nil {
|
|
return nil, 0, err
|
|
}
|
|
|
|
if op.page > 0 && op.pageSize > 0 {
|
|
offset := (op.page - 1) * op.pageSize
|
|
query = query.Offset(offset).Limit(op.pageSize)
|
|
}
|
|
|
|
var userGroups []model.UserGroupMeta
|
|
if err := query.Order("id DESC").Find(&userGroups).Error; err != nil {
|
|
return nil, 0, err
|
|
}
|
|
return userGroups, total, nil
|
|
}
|