migrate from github
This commit is contained in:
@@ -0,0 +1,69 @@
|
||||
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
|
||||
}
|
||||
Reference in New Issue
Block a user