60 lines
1.1 KiB
Go
60 lines
1.1 KiB
Go
package internal
|
|
|
|
import (
|
|
"github.com/Selly-Modules/mongodb"
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
"go.mongodb.org/mongo-driver/mongo/options"
|
|
)
|
|
|
|
type CommonQuery struct {
|
|
Page int64
|
|
Limit int64
|
|
Keyword string
|
|
RoleID string
|
|
Status string
|
|
Sort interface{}
|
|
}
|
|
|
|
// AssignKeyword ...
|
|
func (q *CommonQuery) AssignKeyword(cond bson.M) {
|
|
if q.Keyword != "" {
|
|
cond["searchString"] = mongodb.GenerateQuerySearchString(q.Keyword)
|
|
}
|
|
}
|
|
|
|
// AssignRoleID ...
|
|
func (q *CommonQuery) AssignRoleID(cond bson.M) {
|
|
if q.RoleID != "" {
|
|
if id, isValid := mongodb.NewIDFromString(q.RoleID); isValid {
|
|
cond["roleId"] = id
|
|
}
|
|
}
|
|
}
|
|
|
|
// AssignStatus ...
|
|
func (q *CommonQuery) AssignStatus(cond bson.M) {
|
|
if q.Status != "" {
|
|
cond["status"] = q.Status
|
|
}
|
|
}
|
|
|
|
// GetFindOptionsUsingPage ...
|
|
func (q *CommonQuery) GetFindOptionsUsingPage() *options.FindOptions {
|
|
opts := options.Find()
|
|
if q.Limit > 0 {
|
|
opts.SetLimit(q.Limit).SetSkip(q.Limit * q.Page)
|
|
}
|
|
if q.Sort != nil {
|
|
opts.SetSort(q.Sort)
|
|
}
|
|
return opts
|
|
}
|
|
|
|
// SetDefaultLimit ...
|
|
func (q *CommonQuery) SetDefaultLimit() {
|
|
if q.Limit <= 0 || q.Limit > 20 {
|
|
q.Limit = 20
|
|
}
|
|
}
|
|
|